public class UiStepDefs {
    @LocalPort private int port;
    @Autowired private UserService userService;
    private WebDriver webDriver;

    @Before
    public void setupWebDriver() {
        WebDriverManager.chromedriver().setup();
        final ChromeOptions options = new ChromeOptions();
        options.addArguments("headless");
        webDriver = new ChromeDriver(options);
    }

    @After
    public void closeDriver() {
        webDriver.quit();
    }

    @Given("a user {string} with password {string} exists")
    public void createUser(final String username, final String password) {
        userService.addUser(username, password);
    }

    @Given("I am currently not logged in")
    public void notLoggedIn() {
        webDriver.manage().deleteAllCookies();
        webDriver.get("http://localhost:" + port);
    }

    @When("logging in with username {string} and password {string}")
    public void login(final String username, final String password) {
        webDriver.findElement(By.id("username")).sendKeys(username);
        webDriver.findElement(By.id("password")).sendKeys(password);
        webDriver.findElement(By.id("login")).click();
    }

    @Then("login MUST be successful")
    public void assertLoginSuccess() {
        assertNotNull(webDriver.manage().getCookieNamed("JWT"));
    }
}

class UiStepDefs(
    @LocalPort private val port = 0,
    @Autowired private val userService: UserService? = null,
) {
    private var webDriver: WebDriver? = null

    @Before
    fun setupWebDriver() {
        WebDriverManager.chromedriver().setup()
        webDriver = ChromeDriver(ChromeOptions().apply {
            addArguments("headless")
        })
    }

    @After
    fun closeDriver() {
        webDriver.quit()
    }

    @Given("a user {string} with password {string} exists")
    fun createUser(username: String?, password: String?) {
        userService.addUser(username, password)
    }

    @Given("I am currently not logged in")
    fun notLoggedIn() {
        webDriver.manage().deleteAllCookies()
        webDriver.get("http://localhost:$port")
    }

    @When("logging in with username {string} and password {string}")
    fun login(username: String?, password: String?) {
        webDriver.findElement(By.id("username")).sendKeys(username)
        webDriver.findElement(By.id("password")).sendKeys(password)
        webDriver.findElement(By.id("login")).click()
    }

    @Then("login MUST be successful")
    fun assertLoginSuccess() {
        assertNotNull(webDriver.manage().getCookieNamed("JWT"))
    }
}

public class ApiStepDefs {
    @Autowired private TestRestTemplate testRestTemplate;
    private Response response;

    @Given("a user {string} with password {string} exists")
    public void createUser(final String username, final String password) {
        userService.addUser(username, password);
    }

    @Given("I am currently not logged in")
    public void notLoggedIn() {
        // Nothing to do
    }

    @When("logging in with username {string} and password {string}")
    public void login(final String username, final String password) {
        response = testRestTemplate.postForEntity("/session", null, Object.class);
    }

    @Then("login MUST be successful")
    public void assertLoginSuccess() {
        assertEquals(HttpStatus.OK, response.getStatusCode());
    }
}

class ApiStepDefs(
    @Autowired private val testRestTemplate: TestRestTemplate,
) {
    private var response: Response? = null

    @Given("a user {string} with password {string} exists")
    fun createUser(username: String, password: String) {
        userService.addUser(username, password)
    }

    @Given("I am currently not logged in")
    fun notLoggedIn() {
        // Nothing to do
    }

    @When("logging in with username {string} and password {string}")
    fun login(username: String, password: String) {
        response = testRestTemplate.postForEntity<Any>("/session", null)
    }

    @Then("login MUST be successful")
    fun assertLoginSuccess() =
        assertEquals(HttpStatus.OK, response.getStatusCode())
}

public class UnitStepDefs {
    private UserRepository userRepository = Mockito.mock(UserRepository.class);
    private SessionRepository sessionRepository = Mockito.mock(SessionRepository.class);

    private SessionService sessionService = new SessionService(userRepository);
    private Optional<Session> session;

    @Given("a user {string} with password {string} exists")
    public void createUser(final String username, final String password) {
        Mockito.when(userRepository.findByUsername(username))
                .thenReturn(new User(username, Bcrypt.encrypt(password)));
    }

    @Given("I am currently not logged in")
    public void notLoggedIn() {
        // Nothing to do
    }

    @When("logging in with username {string} and password {string}")
    public void login(final String username, final String password) {
        session = sessionService.login(username, password);
    }

    @Then("login MUST be successful")
    public void assertLoginSuccess() {
        assertTrue(session.isPresent());
    }
}

class UnitStepDefs {
    private val userRepository = mock<UserRepository>()
    private val sessionRepository = mock<SessionRepository>
    private val sessionService = SessionService(userRepository)

    private var session: Optional<Session>? = null

    @Given("a user {string} with password {string} exists")
    fun createUser(username: String, password: String) {
        `when`(userRepository.findByUsername(username))
            .thenReturn(User(username, Bcrypt.encrypt(password)))
    }

    @Given("I am currently not logged in")
    fun notLoggedIn() {
        // Nothing to do
    }

    @When("logging in with username {string} and password {string}")
    fun login(username: String, password: String) {
        session = sessionService.login(username, password)
    }

    @Then("login MUST be successful")
    fun assertLoginSuccess() = assertTrue(session.isPresent())
}