Lab

README.md

@@ -1,3 +1,4 @@
+[![Review Assignment Due Date](https://classroom.github.com/assets/deadline-readme-button-22041afd0340ce965d47ae6ef1cefeee28c7c493a6346c4f15d667ab976d596c.svg)](https://classroom.github.com/a/339Lr3BJ)
 ### How the tests work (and Docker requirement)
 
 This project ships with an end‑to‑end CLI integration test suite that uses Testcontainers to spin up a temporary MySQL database. 

src/main/java/com/example/Main.java

@@ -1,9 +1,8 @@
 package com.example;
 
-import java.sql.Connection;
-import java.sql.DriverManager;
-import java.sql.SQLException;
+import java.sql.*;
 import java.util.Arrays;
+import java.util.Scanner;
 
 public class Main {
 
@@ -14,8 +13,11 @@ static void main(String[] args) {
         new Main().run();
     }
 
+    /**
+     * Runs the main application with login and menu.
+     */
     public void run() {
-        // Resolve DB settings with precedence: System properties -> Environment variables
+        // Resolve database configuration
         String jdbcUrl = resolveConfig("APP_JDBC_URL", "APP_JDBC_URL");
         String dbUser = resolveConfig("APP_DB_USER", "APP_DB_USER");
         String dbPass = resolveConfig("APP_DB_PASS", "APP_DB_PASS");
@@ -26,11 +28,276 @@ public void run() {
                             "as system properties (-Dkey=value) or environment variables.");
         }
 
+        // Test connection
         try (Connection connection = DriverManager.getConnection(jdbcUrl, dbUser, dbPass)) {
         } catch (SQLException e) {
             throw new RuntimeException(e);
         }
-        //Todo: Starting point for your code
+
+        Scanner scanner = new Scanner(System.in);
+
+        // Login flow
+        System.out.print("Username: ");
+        String username = scanner.nextLine().trim();
+
+        System.out.print("Password: ");
+        String password = scanner.nextLine().trim();
+
+        if (!validateLogin(jdbcUrl, dbUser, dbPass, username, password)) {
+            System.out.println("Invalid username or password.");
+            System.out.println("0) Exit");
+            scanner.nextLine();
+            scanner.close();
+            return;
+        }
+
+        System.out.println("Login successful!");
+
+        // Menu loop
+        boolean running = true;
+        while (running) {
+            System.out.println("\n1) List moon missions");
+            System.out.println("2) Get a moon mission by mission_id");
+            System.out.println("3) Count missions for a given year");
+            System.out.println("4) Create an account");
+            System.out.println("5) Update an account password");
+            System.out.println("6) Delete an account");
+            System.out.println("0) Exit");
+            System.out.print("Choose an option: ");
+
+            String choice = scanner.nextLine().trim();
+
+            switch (choice) {
+                case "1":
+                    listMoonMissions(jdbcUrl, dbUser, dbPass);
+                    break;
+                case "2":
+                    getMissionById(scanner, jdbcUrl, dbUser, dbPass);
+                    break;
+                case "3":
+                    countMissionsByYear(scanner, jdbcUrl, dbUser, dbPass);
+                    break;
+                case "4":
+                    createAccount(scanner, jdbcUrl, dbUser, dbPass);
+                    break;
+                case "5":
+                    updateAccountPassword(scanner, jdbcUrl, dbUser, dbPass);
+                    break;
+                case "6":
+                    deleteAccount(scanner, jdbcUrl, dbUser, dbPass);
+                    break;
+                case "0":
+                    running = false;
+                    break;
+                default:
+                    System.out.println("Invalid option");
+            }
+        }
+
+        scanner.close();
+    }
+
+    /**
+     * Lists all spacecraft from moon_mission table.
+     */
+    private void listMoonMissions(String jdbcUrl, String dbUser, String dbPass) {
+        String sql = "SELECT spacecraft FROM moon_mission ORDER BY mission_id";
+
+        try (Connection conn = DriverManager.getConnection(jdbcUrl, dbUser, dbPass);
+             PreparedStatement stmt = conn.prepareStatement(sql);
+             ResultSet rs = stmt.executeQuery()) {
+
+            while (rs.next()) {
+                System.out.println(rs.getString("spacecraft"));
+            }
+
+        } catch (SQLException e) {
+            System.err.println("Error listing missions: " + e.getMessage());
+        }
+    }
+
+    /**
+     * Gets a specific mission by ID and prints details.
+     */
+    private void getMissionById(Scanner scanner, String jdbcUrl, String dbUser, String dbPass) {
+        System.out.print("Enter mission_id: ");
+        String input = scanner.nextLine().trim();
+
+        try {
+            long missionId = Long.parseLong(input);
+            String sql = "SELECT mission_id, spacecraft, launch_date FROM moon_mission WHERE mission_id = ?";
+
+            try (Connection conn = DriverManager.getConnection(jdbcUrl, dbUser, dbPass);
+                 PreparedStatement stmt = conn.prepareStatement(sql)) {
+
+                stmt.setLong(1, missionId);
+
+                try (ResultSet rs = stmt.executeQuery()) {
+                    if (rs.next()) {
+                        System.out.println("Mission ID: " + rs.getLong("mission_id"));
+                        System.out.println("Spacecraft: " + rs.getString("spacecraft"));
+                        System.out.println("Launch Date: " + rs.getDate("launch_date"));
+                    }
+                }
+            }
+        } catch (NumberFormatException e) {
+            System.out.println("Invalid mission ID format");
+        } catch (SQLException e) {
+            System.err.println("Error: " + e.getMessage());
+        }
+    }
+
+    /**
+     * Counts how many missions were launched in a specific year.
+     */
+    private void countMissionsByYear(Scanner scanner, String jdbcUrl, String dbUser, String dbPass) {
+        System.out.print("Enter year: ");
+        String input = scanner.nextLine().trim();
+
+        try {
+            int year = Integer.parseInt(input);
+            String sql = "SELECT COUNT(*) as count FROM moon_mission WHERE YEAR(launch_date) = ?";
+
+            try (Connection conn = DriverManager.getConnection(jdbcUrl, dbUser, dbPass);
+                 PreparedStatement stmt = conn.prepareStatement(sql)) {
+
+                stmt.setInt(1, year);
+
+                try (ResultSet rs = stmt.executeQuery()) {
+                    if (rs.next()) {
+                        int count = rs.getInt("count");
+                        System.out.println("Number of missions in " + year + ": " + count);
+                    }
+                }
+            }
+        } catch (NumberFormatException e) {
+            System.out.println("Invalid year format");
+        } catch (SQLException e) {
+            System.err.println("Error: " + e.getMessage());
+        }
+    }
+
+    /**
+     * Creates a new account in the database.
+     */
+    private void createAccount(Scanner scanner, String jdbcUrl, String dbUser, String dbPass) {
+        System.out.print("First name: ");
+        String firstName = scanner.nextLine().trim();
+
+        System.out.print("Last name: ");
+        String lastName = scanner.nextLine().trim();
+
+        System.out.print("SSN: ");
+        String ssn = scanner.nextLine().trim();
+
+        System.out.print("Password: ");
+        String password = scanner.nextLine().trim();
+
+        String username = firstName.substring(0, Math.min(3, firstName.length())) +
+                lastName.substring(0, Math.min(3, lastName.length()));
+
+        String sql = "INSERT INTO account (name, first_name, last_name, ssn, password) VALUES (?, ?, ?, ?, ?)";
+
+        try (Connection conn = DriverManager.getConnection(jdbcUrl, dbUser, dbPass);
+             PreparedStatement stmt = conn.prepareStatement(sql)) {
+
+            stmt.setString(1, username);
+            stmt.setString(2, firstName);
+            stmt.setString(3, lastName);
+            stmt.setString(4, ssn);
+            stmt.setString(5, password);
+
+            int rowsAffected = stmt.executeUpdate();
+            if (rowsAffected > 0) {
+                System.out.println("Account created successfully");
+            }
+
+        } catch (SQLException e) {
+            System.err.println("Error: " + e.getMessage());
+        }
+    }
+
+    /**
+     * Updates password for an existing account.
+     */
+    private void updateAccountPassword(Scanner scanner, String jdbcUrl, String dbUser, String dbPass) {
+        System.out.print("Enter user_id: ");
+        String input = scanner.nextLine().trim();
+
+        System.out.print("Enter new password: ");
+        String newPassword = scanner.nextLine().trim();
+
+        try {
+            long userId = Long.parseLong(input);
+            String sql = "UPDATE account SET password = ? WHERE user_id = ?";
+
+            try (Connection conn = DriverManager.getConnection(jdbcUrl, dbUser, dbPass);
+                 PreparedStatement stmt = conn.prepareStatement(sql)) {
+
+                stmt.setString(1, newPassword);
+                stmt.setLong(2, userId);
+
+                int rowsAffected = stmt.executeUpdate();
+                if (rowsAffected > 0) {
+                    System.out.println("Password updated successfully");
+                }
+            }
+        } catch (NumberFormatException e) {
+            System.out.println("Invalid user ID format");
+        } catch (SQLException e) {
+            System.err.println("Error: " + e.getMessage());
+        }
+    }
+
+    /**
+     * Deletes an account from the database.
+     */
+    private void deleteAccount(Scanner scanner, String jdbcUrl, String dbUser, String dbPass) {
+        System.out.print("Enter user_id to delete: ");
+        String input = scanner.nextLine().trim();
+
+        try {
+            long userId = Long.parseLong(input);
+            String sql = "DELETE FROM account WHERE user_id = ?";
+
+            try (Connection conn = DriverManager.getConnection(jdbcUrl, dbUser, dbPass);
+                 PreparedStatement stmt = conn.prepareStatement(sql)) {
+
+                stmt.setLong(1, userId);
+
+                int rowsAffected = stmt.executeUpdate();
+                if (rowsAffected > 0) {
+                    System.out.println("Account deleted successfully");
+                }
+            }
+        } catch (NumberFormatException e) {
+            System.out.println("Invalid user ID format");
+        } catch (SQLException e) {
+            System.err.println("Error: " + e.getMessage());
+        }
+    }
+
+    /**
+     * Checks if username and password are correct.
+     */
+    private boolean validateLogin(String jdbcUrl, String dbUser, String dbPass,
+                                  String username, String password) {
+        String sql = "SELECT user_id FROM account WHERE name = ? AND password = ?";
+
+        try (Connection conn = DriverManager.getConnection(jdbcUrl, dbUser, dbPass);
+             PreparedStatement stmt = conn.prepareStatement(sql)) {
+
+            stmt.setString(1, username);
+            stmt.setString(2, password);
+
+            try (ResultSet rs = stmt.executeQuery()) {
+                return rs.next();
+            }
+
+        } catch (SQLException e) {
+            System.err.println("Database error: " + e.getMessage());
+            return false;
+        }
     }
 
     /**
@@ -41,11 +308,11 @@ public void run() {
      * @return {@code true} if the application is in development mode; {@code false} otherwise
      */
     private static boolean isDevMode(String[] args) {
-        if (Boolean.getBoolean("devMode"))  //Add VM option -DdevMode=true
+        if (Boolean.getBoolean("devMode"))
             return true;
-        if ("true".equalsIgnoreCase(System.getenv("DEV_MODE")))  //Environment variable DEV_MODE=true
+        if ("true".equalsIgnoreCase(System.getenv("DEV_MODE")))
             return true;
-        return Arrays.asList(args).contains("--dev"); //Argument --dev
+        return Arrays.asList(args).contains("--dev");
     }
 
     /**
@@ -59,4 +326,4 @@ private static String resolveConfig(String propertyKey, String envKey) {
         }
         return (v == null || v.trim().isEmpty()) ? null : v.trim();
     }
-}
+}