UPDATE

Author: Nipuna-Lakruwan

Beginner/Project1_ToDoList/src/Main.java

@@ -3,58 +3,52 @@
 public class Main {
     public static void main(String[] args) {
         Scanner scanner = new Scanner(System.in);
-        TodoListManager todoListManager = new TodoListManager();
-        String command = "";
-
-        System.out.println("Welcome to the To-Do List Application!");
+        TodoListManager<String> toDoListManager = new TodoListManager<>();  // Managing String tasks
+        String command;
 
         do {
-            try {
-                System.out.println("\nEnter a command (add/view/remove/complete/save/load/exit):");
-                command = scanner.nextLine();
+            System.out.println("\nEnter a command (add/view/remove/complete/save/load/exit):");
+            command = scanner.nextLine();
 
-                switch (command) {
-                    case "add":
-                        System.out.println("Enter the task:");
-                        String task = scanner.nextLine();
-                        todoListManager.addTask(task);
-                        break;
-                    case "view":
-                        todoListManager.viewTasks();
-                        break;
-                    case "remove":
-                        System.out.println("Enter the task number to remove:");
-                        int taskNumber = scanner.nextInt();
-                        scanner.nextLine();
-                        todoListManager.removeTask(taskNumber);
-                        break;
-                    case "complete":
-                        System.out.println("Enter the task number to complete:");
-                        int completeTaskNumber = scanner.nextInt();
-                        scanner.nextLine();
-                        todoListManager.completeTask(completeTaskNumber);
-                        break;
-                    case "save":
-                        System.out.println("Enter the filename to save tasks:");
-                        String saveFilename = scanner.nextLine();
-                        todoListManager.saveTasksToFile(saveFilename);
-                        break;
-                    case "load":
-                        System.out.println("Enter the filename to load tasks:");
-                        String loadFilename = scanner.nextLine();
-                        todoListManager.loadTasksFromFile(loadFilename);
-                        break;
-                    case "exit":
-                        System.out.println("Exiting the application");
-                        break;
-                    default:
-                        System.out.println("Invalid command");
-                }
-            } catch (Exception e) {
-                System.out.println("An error occurred: " + e.getMessage());
-                scanner.nextLine();
+            switch (command) {
+                case "add":
+                    System.out.println("Enter the task:");
+                    String task = scanner.nextLine();
+                    toDoListManager.addTask(task);
+                    break;
+                case "view":
+                    toDoListManager.viewTasks();
+                    break;
+                case "remove":
+                    System.out.println("Enter the task number to remove:");
+                    int taskNumber = scanner.nextInt();
+                    scanner.nextLine(); // Consume newline
+                    toDoListManager.removeTask(taskNumber);
+                    break;
+                case "complete":
+                    System.out.println("Enter the task number to complete:");
+                    int completeTaskNumber = scanner.nextInt();
+                    scanner.nextLine(); // Consume newline
+                    toDoListManager.completeTask(completeTaskNumber);
+                    break;
+                case "save":
+                    System.out.println("Enter the filename to save tasks:");
+                    String saveFilename = scanner.nextLine();
+                    toDoListManager.saveTasksToFile(saveFilename);
+                    break;
+                case "load":
+                    System.out.println("Enter the filename to load tasks:");
+                    String loadFilename = scanner.nextLine();
+                    toDoListManager.loadTasksFromFile(loadFilename);
+                    break;
+                case "exit":
+                    System.out.println("Exiting the application.");
+                    break;
+                default:
+                    System.out.println("Invalid command.");
             }
         } while (!command.equals("exit"));
 
+        scanner.close();
     }
 }

Beginner/Project1_ToDoList/src/TodoList.java

@@ -1,76 +1,66 @@
-import java.io.*;
 import java.util.ArrayList;
+import java.io.*;
 
-/**
- * ToDoListManager is responsible for managing the to-do list tasks,
- * including adding, removing, completing, and persisting tasks.
- */
-
-public class TodoListManager {
-    private ArrayList<String> tasks;
+public class TodoListManager<T> {
+    private ArrayList<T> tasks;
 
     public TodoListManager() {
         tasks = new ArrayList<>();
     }
 
-    // Add task to the list
-    public void addTask(String task) {
+    public void addTask(T task) {
         tasks.add(task);
         System.out.println("Task added: " + task);
     }
 
-    // View all tasks
-    public void viewTasks(){
-        System.out.println("Your To-Do- List:");
-        for (int i = 0; i < tasks.size(); i++) {
-            System.out.println((i + 1) + ". " + tasks.get(i));
+    public void viewTasks() {
+        if (tasks.isEmpty()) {
+            System.out.println("No tasks in the list.");
+        } else {
+            for (int i = 0; i < tasks.size(); i++) {
+                System.out.println((i + 1) + ". " + tasks.get(i));
+            }
         }
     }
 
-    // Remove a task by index
     public void removeTask(int index) {
         if (index >= 1 && index <= tasks.size()) {
-            String removedTask = tasks.remove(index - 1);
-            System.out.println("Task removed: " + removedTask);
+            System.out.println("Task removed: " + tasks.get(index - 1));
+            tasks.remove(index - 1);
         } else {
             System.out.println("Invalid task number.");
         }
     }
 
-    // Complete a task
     public void completeTask(int index) {
         if (index >= 1 && index <= tasks.size()) {
             System.out.println("Task completed: " + tasks.get(index - 1));
-            tasks.remove(index - 1); // Remove the task after completion
+            tasks.remove(index - 1);
         } else {
             System.out.println("Invalid task number.");
         }
     }
 
-    // Save tasks to a file
     public void saveTasksToFile(String filename) {
         try (FileWriter writer = new FileWriter(filename)) {
-            for (String task : tasks) {
-                writer.write(task + "\n");
+            for (T task : tasks) {
+                writer.write(task.toString() + "\n");
             }
             System.out.println("Tasks saved to " + filename);
         } catch (IOException e) {
             System.out.println("Error saving tasks: " + e.getMessage());
         }
     }
 
-    // Load tasks from a file
     public void loadTasksFromFile(String filename) {
         try (BufferedReader reader = new BufferedReader(new FileReader(filename))) {
             String task;
             while ((task = reader.readLine()) != null) {
-                tasks.add(task);
+                // This will need casting depending on the task type
+                System.out.println("Task loaded: " + task);
             }
-            System.out.println("Tasks loaded from " + filename);
         } catch (IOException e) {
             System.out.println("Error loading tasks: " + e.getMessage());
         }
     }
-}
-
-
+}