[waiter-palypoo] ip

src/main/java/HinaBot.java

@@ -1,7 +1,7 @@
 import java.util.Scanner;
 
 public class HinaBot {
-    protected static String[] taskList = new String[100];
+    protected static Task[] taskList = new Task[100];
     protected static int taskCount = 0;
     public static void main(String[] args) {
         String line;
@@ -26,21 +26,49 @@ public static void handleCommand(String command) {
         else if (command.equalsIgnoreCase("list")) {
             listTasks();
         }
+        else if (command.split(" ")[0].equalsIgnoreCase("mark")) {
+            int taskIndex = Integer.parseInt(command.split(" ")[1]);
+            markTask(taskIndex);
+        }
+        else if (command.split(" ")[0].equalsIgnoreCase("unmark")) {
+            int taskIndex = Integer.parseInt(command.split(" ")[1]);
+            unmarkTask(taskIndex);
+        }
         else {
             addTask(command);
         }
     }
 
-    public static void addTask(String task) {
-        taskList[taskCount] = task;
+    public static void addTask(String description) {
+        Task newTask = new Task(description);
+        taskList[taskCount] = newTask;
         taskCount++;
-        System.out.println("added: " + task);
+        System.out.println("added: " + description);
     }
 
     public static void listTasks() {
         for (int i = 0; i < taskCount; i++) {
             System.out.print(i + 1);
-            System.out.println(". " + taskList[i]);
+            System.out.print(". ");
+            if (taskList[i].isDone()) {
+                System.out.print("[X] ");
+            }
+            else {
+                System.out.print("[ ] ");
+            }
+            System.out.println(taskList[i].getDescription());
         }
     }
+
+    public static void markTask(int taskIndex) {
+        taskList[taskIndex - 1].setDone(true);
+        System.out.println("Roger that! This task is marked as done: ");
+        System.out.println("[X] " + taskList[taskIndex - 1].getDescription());
+    }
+
+    public static void unmarkTask(int taskIndex) {
+        taskList[taskIndex - 1].setDone(false);
+        System.out.println("Roger that! This task is marked as done: ");
+        System.out.println("[ ] " + taskList[taskIndex - 1].getDescription());
+    }
 }

src/main/java/Task.java

@@ -0,0 +1,24 @@
+public class Task {
+    private String description;
+    private boolean isDone;
+    public boolean isDone() {
+        return isDone;
+    }
+
+    public void setDone(boolean done) {
+        isDone = done;
+    }
+
+    public Task(String description) {
+        this.description = description;
+        isDone = false;
+    }
+
+    public String getDescription() {
+        return description;
+    }
+
+    public void setDescription(String description) {
+        this.description = description;
+    }
+}