User registration

Author: lilstiffy

src/main/java/org/pwss/Start.java

@@ -15,7 +15,7 @@ public static void main(String[] args) {
             // Create the main frame
             JFrame frame = new JFrame("Scan Integrity Scanner");
             frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
-            frame.setSize(250, 250);
+            frame.setSize(350, 250);
             frame.setLocationRelativeTo(null); // center on screen
 
             // Create presenter factory

src/main/java/org/pwss/model/service/AuthService.java

@@ -40,7 +40,7 @@ public boolean userExists() throws UserExistsLookupException, ExecutionException
 
         return switch (response.statusCode()) {
             case 200 -> Boolean.parseBoolean(response.body());
-            case 404 -> throw new UserExistsLookupException("User existence check failed: No user found");
+            case 404 -> false;
             case 500 -> throw new UserExistsLookupException("User existence check failed: Server error");
             default ->
                     throw new UserExistsLookupException("User existence check failed: Unexpected status code " + response.statusCode());

src/main/java/org/pwss/model/service/network/Endpoint.java

@@ -14,7 +14,7 @@ public enum Endpoint {
     /**
      * Endpoint for creating a user.
      */
-    CREATE_USER(HTTP_Method.POST, A.BASE_URL + A.USER + "create-user"),
+    CREATE_USER(HTTP_Method.POST, A.BASE_URL + A.USER + "create"),
     /**
      * Endpoint for user login.
      */

src/main/java/org/pwss/navigation/NavigationHandler.java

@@ -47,7 +47,7 @@ public NavigationHandler(JFrame frame, PresenterFactory factory) {
      */
     public void navigateTo(Screen screen) {
         BasePresenter<?> presenter = presenters.computeIfAbsent(screen, factory::createPresenter);
-        JPanel view = presenter.getView();
+        JPanel view = presenter.getScreen();
 
         frame.getContentPane().removeAll();
         frame.getContentPane().add(view);

src/main/java/org/pwss/presenter/BasePresenter.java

@@ -1,41 +1,42 @@
 package org.pwss.presenter;
 
-import javax.swing.JPanel;
+import org.pwss.view.screen.BaseScreen;
+
 
 /**
  * An abstract base class for presenters in the application.
- * Provides a common structure for managing a `JFrame` view and initializing event listeners.
+ * Provides a common structure for managing a `BaseScreen` view and initializing event listeners.
  */
-public abstract class BasePresenter<View extends JPanel> {
+public abstract class BasePresenter<Screen extends BaseScreen> {
     /**
      * The view instance managed by this presenter.
      * Represents the UI component associated with this presenter.
      */
-    protected View view;
+    protected Screen screen;
 
     /**
      * Constructs a `BasePresenter` with the specified view.
      * Initializes the view and sets up event listeners.
      *
-     * @param view The `View` instance to be managed by this presenter.
+     * @param screen The `Screen` instance to be managed by this presenter.
      */
-    public BasePresenter(View view) {
-        this.view = view;
+    public BasePresenter(Screen screen) {
+        this.screen = screen;
         initListeners();
     }
 
     /**
-     * Abstract method to initialize event listeners for the view.
+     * Abstract method to initialize event listeners for the Screen.
      * Subclasses must provide an implementation for this method.
      */
     protected abstract void initListeners();
 
     /**
      * Retrieves the view managed by this presenter.
      *
-     * @return The `View` instance representing the managed view.
+     * @return The `Screen` instance representing the managed view.
      */
-    public View getView() {
-        return view;
+    public Screen getScreen() {
+        return screen;
     }
 }

src/main/java/org/pwss/presenter/HomePresenter.java

@@ -7,34 +7,34 @@
 import org.pwss.exception.scan.StartScanAllException;
 import org.pwss.exception.scan.StopScanException;
 import org.pwss.model.service.ScanService;
-import org.pwss.view.screen.HomeView;
+import org.pwss.view.screen.HomeScreen;
 
-public class HomePresenter extends BasePresenter<HomeView> {
+public class HomePresenter extends BasePresenter<HomeScreen> {
     private final ScanService scanService;
 
-    public HomePresenter(HomeView view, ScanService scanService) {
+    public HomePresenter(HomeScreen view, ScanService scanService) {
         super(view);
         this.scanService = scanService;
     }
 
     @Override
     protected void initListeners() {
-        view.getStartButton().addActionListener(e -> performStartScan());
-        view.getStopButton().addActionListener(e -> performStopScan());
+        getScreen().getStartButton().addActionListener(e -> performStartScan());
+        getScreen().getStopButton().addActionListener(e -> performStopScan());
     }
 
     private void performStartScan() {
         try {
             boolean startScanSuccess = scanService.startScan();
             SwingUtilities.invokeLater(() -> {
                 if (startScanSuccess) {
-                    view.showSuccess("Scan started successfully!");
+                    getScreen().showSuccess("Scan started successfully!");
                 } else {
-                    view.showError("Failed to start scan.");
+                    getScreen().showError("Failed to start scan.");
                 }
             });
         } catch (ExecutionException | InterruptedException | StartScanAllException e) {
-            SwingUtilities.invokeLater(() -> view.showError("Error starting scan: " + e.getMessage()));
+            SwingUtilities.invokeLater(() -> screen.showError("Error starting scan: " + e.getMessage()));
         }
     }
 
@@ -43,13 +43,13 @@ private void performStopScan() {
             boolean stopScanSuccess = scanService.stopScan();
             SwingUtilities.invokeLater(() -> {
                 if (stopScanSuccess) {
-                    view.showSuccess("Scan stopped successfully!");
+                    getScreen().showSuccess("Scan stopped successfully!");
                 } else {
-                    view.showError("Failed to stop scan.");
+                    getScreen().showError("Failed to stop scan.");
                 }
             });
         } catch (ExecutionException | InterruptedException | StopScanException e) {
-            SwingUtilities.invokeLater(() -> view.showError("Error stopping scan: " + e.getMessage()));
+            SwingUtilities.invokeLater(() -> getScreen().showError("Error stopping scan: " + e.getMessage()));
         }
     }
 }

src/main/java/org/pwss/presenter/LoginPresenter.java

@@ -3,52 +3,183 @@
 import javax.swing.SwingUtilities;
 
 import org.pwss.exception.user.LoginException;
+import org.pwss.exception.user.UserExistsLookupException;
 import org.pwss.model.service.AuthService;
 import org.pwss.navigation.NavigationEvents;
 import org.pwss.navigation.Screen;
-import org.pwss.view.screen.LoginView;
+import org.pwss.view.screen.LoginScreen;
 
 import com.fasterxml.jackson.core.JsonProcessingException;
 
-public class LoginPresenter extends BasePresenter<LoginView> {
+import java.util.concurrent.ExecutionException;
+
+public class LoginPresenter extends BasePresenter<LoginScreen> {
+    /**
+     * Indicates whether the application is in create user mode (i.e., no users exist yet).
+     */
+    private final boolean createUserMode;
+
     private final AuthService authService;
 
-    public LoginPresenter(LoginView view, AuthService authService) {
+    /**
+     * Constructs a LoginPresenter with the specified view and authentication service.
+     *
+     * @param view        The LoginView instance to be managed by this presenter.
+     * @param authService The AuthService instance for handling authentication operations.
+     */
+    public LoginPresenter(LoginScreen view, AuthService authService) {
         super(view);
         this.authService = authService;
+        this.createUserMode = checkUserExists();
     }
 
     @Override
     protected void initListeners() {
-        getView().getLoginButton().addActionListener(e -> performLogin());
-        getView().getCancelButton().addActionListener(e -> System.exit(0));
+        getScreen().getProceedButton().addActionListener(e -> onProceedButtonClick());
+        getScreen().getCancelButton().addActionListener(e -> System.exit(0));
+    }
+
+    /**
+     * Checks if any users exist in the system.
+     * If no users exist, updates the view to prompt for user creation.
+     *
+     * @return true if no users exist (create user mode), false otherwise.
+     */
+    private boolean checkUserExists() {
+        try {
+            if (!authService.userExists()) {
+                // Notify user that no users exist and they need to create one
+                getScreen().showInfo("No user found.\nCreate one by entering a username and password.");
+                // Update message label to indicate user creation
+                getScreen().setMessage("Create a user for the first login");
+                // Change button text to "Register"
+                getScreen().getProceedButton().setText("Register");
+                return true;
+            } else {
+                // Update message label to indicate normal login
+                getScreen().setMessage("Login with your username and password.");
+                // Change button text to "Login"
+                getScreen().getProceedButton().setText("Login");
+                return false;
+            }
+        } catch (UserExistsLookupException | ExecutionException e) {
+            SwingUtilities.invokeLater(() ->
+                    screen.showError("Error checking user existence: " + e.getMessage()));
+        } catch (InterruptedException e) {
+            Thread.currentThread().interrupt();
+            SwingUtilities.invokeLater(() ->
+                    screen.showError("Operation interrupted: " + e.getMessage()));
+        } catch (Exception e) {
+            SwingUtilities.invokeLater(() ->
+                    screen.showError("An unexpected error occurred: " + e.getMessage()));
+        }
+        throw new IllegalStateException("Failed to determine user existence due to an error.");
+    }
+
+    /**
+     * Handles the proceed button click event.
+     * Validates input and either creates a new user or performs login based on the mode.
+     */
+    private void onProceedButtonClick() {
+        if (!validateInput()) {
+            return; // Input validation failed, do not proceed
+        }
+
+        if (createUserMode) {
+            // Handle user creation logic here and then login
+            createUserAndLogin();
+        } else {
+            // Handle normal login logic here
+            performLogin();
+        }
+    }
+
+    /**
+     * Validates the input fields for username and password.
+     *
+     * @return true if both fields are non-empty, false otherwise.
+     */
+    private boolean validateInput() {
+        String username = getScreen().getUsername();
+        String password = getScreen().getPassword();
+
+        if (username == null || username.trim().isEmpty()) {
+            getScreen().showError("Username cannot be empty.");
+            return false;
+        }
+
+        if (password == null || password.trim().isEmpty()) {
+            getScreen().showError("Password cannot be empty.");
+            return false;
+        }
+
+        return true;
+    }
+
+    /**
+     * Creates a new user and then performs login if creation is successful.
+     */
+    private void createUserAndLogin() {
+        if (createUser()) {
+            performLogin();
+        }
+    }
+
+    /**
+     * Creates a new user with the provided username and password.
+     *
+     * @return true if user creation is successful, false otherwise.
+     */
+    private boolean createUser() {
+        String username = getScreen().getUsername();
+        String password = getScreen().getPassword();
+
+        try {
+            return authService.createUser(username, password);
+        } catch (JsonProcessingException e) {
+            SwingUtilities.invokeLater(() ->
+                    screen.showError("Error preparing user creation request: " + e.getMessage()));
+            return false;
+        } catch (Exception e) {
+            SwingUtilities.invokeLater(() ->
+                    screen.showError("An unexpected error occurred: " + e.getMessage()));
+            return false;
+        }
     }
 
+    /**
+     * Performs the login operation using the provided username and password.
+     * Displays success or error messages based on the outcome.
+     */
     private void performLogin() {
-        String username = view.getUsername();
-        String password = view.getPassword();
+        String username = getScreen().getUsername();
+        String password = getScreen().getPassword();
 
         try {
             boolean loginSuccess = authService.login(username, password);
 
             SwingUtilities.invokeLater(() -> {
                 if (loginSuccess) {
-                    view.showSuccess("Login successful!");
+                    if (createUserMode) {
+                        screen.showInfo("User created and logged in successfully!");
+                    } else {
+                        screen.showInfo("Logged in successfully!");
+                    }
                     NavigationEvents.navigateTo(Screen.HOME);
                 } else {
-                    view.showError("Invalid username or password.");
+                    screen.showError("Invalid username or password.");
                 }
             });
 
         } catch (JsonProcessingException e) {
             SwingUtilities.invokeLater(() ->
-                    view.showError("Error preparing login request: " + e.getMessage()));
+                    screen.showError("Error preparing login request: " + e.getMessage()));
         } catch (LoginException e) {
             SwingUtilities.invokeLater(() ->
-                    view.showError("Login request failed: " + e.getMessage()));
+                    screen.showError("Login request failed: " + e.getMessage()));
         } catch (Exception e) {
             SwingUtilities.invokeLater(() ->
-                    view.showError("An unexpected error occurred: " + e.getMessage()));
+                    screen.showError("An unexpected error occurred: " + e.getMessage()));
         }
     }
 }

src/main/java/org/pwss/presenter/factory/AppPresenterFactory.java

@@ -6,8 +6,8 @@
 import org.pwss.presenter.BasePresenter;
 import org.pwss.presenter.LoginPresenter;
 import org.pwss.presenter.HomePresenter;
-import org.pwss.view.screen.LoginView;
-import org.pwss.view.screen.HomeView;
+import org.pwss.view.screen.HomeScreen;
+import org.pwss.view.screen.LoginScreen;
 
 /**
  * A factory class responsible for creating presenters for different screens in the application.
@@ -25,9 +25,9 @@ public class AppPresenterFactory implements PresenterFactory {
     public BasePresenter<?> createPresenter(Screen screen) {
         switch (screen) {
             case LOGIN:
-                return new LoginPresenter(new LoginView(), new AuthService());
+                return new LoginPresenter(new LoginScreen(), new AuthService());
             case HOME:
-                return new HomePresenter(new HomeView(), new ScanService());
+                return new HomePresenter(new HomeScreen(), new ScanService());
             default:
                 throw new IllegalArgumentException("Unknown screen: " + screen);
         }