Dev

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 {
 
@@ -27,12 +26,174 @@ public void run() {
         }
 
         try (Connection connection = DriverManager.getConnection(jdbcUrl, dbUser, dbPass)) {
+            Scanner scanner = new Scanner(System.in);
+            boolean loggedIn = false;
+            while (!loggedIn) {
+                System.out.print("Username: ");
+                String username = scanner.nextLine();
+                System.out.print("Password: ");
+                String password = scanner.nextLine();
+
+                if (isValidLogin(connection,username,password)) {
+                    loggedIn = true;
+                    System.out.println("Logged in as " + username);
+                } else  {
+                    System.out.println("Invalid username or password");
+                    System.out.println("0) Exit");
+                    System.out.println("Press any other key to try again");
+
+                    String choice = scanner.nextLine();
+                    if (choice.equals("0")) {
+                        return;
+                    }
+                }
+            }
+            while (true) {
+                printMenu();
+                String option = scanner.nextLine();
+
+                    switch (option) {
+                        case "1":
+                            listMoonMissions(connection);
+                            break;
+                        case "2":
+                            System.out.print("Enter mission_id: ");
+                            String idStr = scanner.nextLine();
+                            getMoonMissionById(connection, idStr);
+                            break;
+                        case "3":
+                            System.out.print("Enter year: ");
+                            String yearStr = scanner.nextLine();
+                            countMissionsByYear(connection, yearStr);
+                            break;
+                        case "4":
+                            createAccount(connection, scanner);
+                            break;
+                        case "5":
+                            updateAccountPassword(connection, scanner);
+                            break;
+                        case "6":
+                            deleteAccount(connection, scanner);
+                            break;
+                        case "0":
+                            return;
+                        default:
+                            System.out.println("Invalid option, try again.");
+                }
+
+            }
         } catch (SQLException e) {
             throw new RuntimeException(e);
         }
-        //Todo: Starting point for your code
+
+    }
+
+    private void printMenu() {
+        System.out.println("\n--- Moon Mission Menu ---");
+        System.out.println("1. List moon missions");
+        System.out.println("2. Get moon mission by id");
+        System.out.println("3. Count missions by 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("Choice: ");
+    }
+
+    private void listMoonMissions(Connection conn) throws SQLException {
+        String query = "SELECT * FROM moon_mission";
+        try (PreparedStatement stmt = conn.prepareStatement(query);
+             ResultSet rs = stmt.executeQuery()) {
+
+            while (rs.next()) {
+                System.out.println(rs.getString("spacecraft"));
+            }
+        }
+    }
+
+    private void getMoonMissionById(Connection conn, String id) throws SQLException {
+        String query = "SELECT * FROM moon_mission WHERE mission_id = ?";
+        try (PreparedStatement stmt = conn.prepareStatement(query)) {
+            stmt.setString(1, id);
+            try (ResultSet rs = stmt.executeQuery()) {
+                if (rs.next()) {
+                    String dateStr = rs.getString("launch_date");
+                    String year = (dateStr != null && dateStr.length() >= 4) ? dateStr.substring(0, 4) : "N/A";
+
+                    System.out.println("Mission: " + rs.getString("spacecraft") +
+                            ", Year: " + year);
+                } else {
+                    System.out.println("Mission not found.");
+                }
+            }
+        }
+    }
+
+    private void countMissionsByYear(Connection conn, String year) throws SQLException {
+        String query = "SELECT COUNT(*) FROM moon_mission WHERE YEAR(launch_date) = ?";
+        try (PreparedStatement stmt = conn.prepareStatement(query)) {
+            stmt.setString(1, year);
+            try (ResultSet rs = stmt.executeQuery()) {
+                if (rs.next()) {
+                    int count = rs.getInt(1);
+                    System.out.println("Missions in " + year + ": " + count);
+                }
+            }
+        }
     }
 
+    private void createAccount(Connection conn, Scanner sc) throws SQLException {
+        System.out.print("First name: ");
+        String first = sc.nextLine();
+        System.out.print("Last name: ");
+        String last = sc.nextLine();
+        System.out.print("SSN: ");
+        String ssn = sc.nextLine();
+        System.out.print("Password: ");
+        String pass = sc.nextLine();
+
+        String query = "INSERT INTO account (first_name, last_name, ssn, password) VALUES (?, ?, ?, ?)";
+        try (PreparedStatement stmt = conn.prepareStatement(query)) {
+            stmt.setString(1, first);
+            stmt.setString(2, last);
+            stmt.setString(3, ssn);
+            stmt.setString(4, pass);
+            stmt.executeUpdate();
+            System.out.println("Account created.");
+        }
+    }
+
+    private void updateAccountPassword(Connection conn, Scanner sc) throws SQLException {
+        System.out.print("Enter user_id: ");
+        String id = sc.nextLine();
+        System.out.print("Enter new password: ");
+        String pass = sc.nextLine();
+
+        String query = "UPDATE account SET password = ? WHERE user_id = ?";
+        try (PreparedStatement stmt = conn.prepareStatement(query)) {
+            stmt.setString(1, pass);
+            stmt.setString(2, id);
+            int rows = stmt.executeUpdate();
+            if (rows > 0) System.out.println("Password updated.");
+            else System.out.println("User not found.");
+        }
+    }
+    private void deleteAccount(Connection conn, Scanner sc) throws SQLException {
+        System.out.print("Enter user_id: ");
+        String id = sc.nextLine();
+
+        String query = "DELETE FROM account WHERE user_id = ?";
+        try (PreparedStatement stmt = conn.prepareStatement(query)) {
+            stmt.setString(1, id);
+            int rows = stmt.executeUpdate();
+            if (rows > 0) System.out.println("Account deleted.");
+            else System.out.println("User not found.");
+        }
+    }
+
+
+
+
     /**
      * Determines if the application is running in development mode based on system properties,
      * environment variables, or command-line arguments.
@@ -59,4 +220,16 @@ private static String resolveConfig(String propertyKey, String envKey) {
         }
         return (v == null || v.trim().isEmpty()) ? null : v.trim();
     }
+    private boolean isValidLogin(Connection conn, String user, String pass) throws SQLException {
+        String query = "SELECT * FROM account WHERE name = ? AND password = ?";
+
+        try (PreparedStatement stmt = conn.prepareStatement(query)) {
+            stmt.setString(1, user);
+            stmt.setString(2, pass);
+
+            try (ResultSet rs = stmt.executeQuery()) {
+                return rs.next();
+            }
+        }
+    }
 }