From 58e4775738d9dcb625777b11715a84892667c39d Mon Sep 17 00:00:00 2001
From: Patrick Lipka <patrick.lipka@emea.nec.com>
Date: Mon, 13 Dec 2021 10:22:49 +0100
Subject: [PATCH] new task / switch project commands added

---
 src/project.cpp |  3 ---
 src/track.cpp   |  1 +
 src/ui.cpp      | 62 +++++++++++++++++++++++++++++++++++++++++++++++--
 src/ui.h        |  2 ++
 4 files changed, 63 insertions(+), 5 deletions(-)

diff --git a/src/project.cpp b/src/project.cpp
index f233e5f..0ddf22d 100644
--- a/src/project.cpp
+++ b/src/project.cpp
@@ -31,8 +31,6 @@ Task* Project::find_task_by_name(std::string task_name){
     for (int i=0; i<num_tasks; i++){
         if(tasks[i].name == task_name){
             return &(tasks[i]);
-        }else{
-            std::cout << "Task " << task_name << " not found in project " << name << std::endl;;
         }
     }
     return NULL;
@@ -44,7 +42,6 @@ int Project::find_task_id_by_name(std::string task_name){
             return i;
        }
     }
-    std::cout << "Task " << task_name << " not found in project " << name << std::endl;
     return -1;
 }
 
diff --git a/src/track.cpp b/src/track.cpp
index aa12510..aaab810 100644
--- a/src/track.cpp
+++ b/src/track.cpp
@@ -23,6 +23,7 @@ void track(Project *proj){
     time(&start);
     tracking = 1;
     std::cout << "Started tracking of task " << proj->name << "/" << proj->active_task->name << " at " << ctime(&start) << std::endl;
+    std::cout << "Press CTRL-C to stop tracking" << std::endl;
     sigint = 0;
 
     // will be interrupted by signal handler in case of SIGINT (CTRL-C)
diff --git a/src/ui.cpp b/src/ui.cpp
index aeec4c6..ed60770 100644
--- a/src/ui.cpp
+++ b/src/ui.cpp
@@ -96,10 +96,22 @@ void parse_input(std::string input, ProjectList *proj_list){
             
     }else if (command == "np"){
         if (command_end == std::string::npos){
-            std::cout << "new peoject: Please specify project name." << std::endl;
+            std::cout << "new project: Please specify project name." << std::endl;
         }else{
             command_np(argument,proj_list);
         }
+    }else if (command == "nt"){
+        if (command_end == std::string::npos){
+            std::cout << "new task: Please specify task name." << std::endl;
+        }else{
+            command_nt(argument, proj_list);
+        }
+    }else if (command == "sp"){
+        if (command_end == std::string::npos){
+            std::cout << "switch project: Please specify project name." << std::endl;
+        }else{
+            command_sp(argument, proj_list);
+        }
     }
 }
 
@@ -146,19 +158,25 @@ void command_start(std::string input, ProjectList *proj_list){
         std::string proj_name = str.substr(0,place_of_slash);
         std::string task_name = str.substr(place_of_slash+1);
         Project *proj = proj_list->find_project_by_name(underscore_to_space(proj_name));
+        if (proj == NULL){
+            std::cout << "Project " << underscore_to_space(proj_name) << " does not exist." << std::endl;
+            return;
+        }
         int id=proj->find_task_id_by_name(underscore_to_space(task_name));
         if(id >= 0){
             std::cout << "id=" << id << std::endl;
             proj->set_active_task(id);
             std::cout << "now active: "<< proj->active_task->name << std::endl;
             track(proj);
+        }else{
+            std::cout << "Task " << underscore_to_space(str) << " does not exist." << std::endl;
         }
     }
 }
 
 // create new project
 void command_np(std::string name, ProjectList *proj_list){
-    if (proj_list->find_project_id_by_name(name) == -1){
+    if (proj_list->find_project_id_by_name(underscore_to_space(name)) == -1){
         Project proj(name); 
         std::string auto_complete_str = space_to_underscore(name);
         autocomplete_names.push_back(auto_complete_str);
@@ -167,3 +185,43 @@ void command_np(std::string name, ProjectList *proj_list){
         std::cout << "A project with the name " << name << " already exists." << std::endl;
     }
 }
+
+// create new task
+void command_nt(std::string input, ProjectList *proj_list){
+    // change active task and start tracking
+    size_t place_of_slash = input.find("/");
+    std::string proj_name = input.substr(0,place_of_slash);
+    std::string task_name = input.substr(place_of_slash+1);
+    Project *proj = proj_list->find_project_by_name(underscore_to_space(proj_name));
+    if (place_of_slash == std::string::npos){
+        proj = proj_list->active_project;
+    }
+    if (proj == NULL){
+    std::cout << "Project " << underscore_to_space(proj_name) << " does not exist." << std::endl;
+    return;
+    }
+    int id = proj->find_task_id_by_name(underscore_to_space(task_name));
+    if (id < 0){
+        Task task(task_name);
+        proj->add_task(task);
+        std::string auto_complete_str = space_to_underscore(proj->name+"/"+task_name);
+        autocomplete_names.push_back(auto_complete_str);
+    }else{
+        std::string str = proj->name+"/"+task_name;
+        std::cout << "Task " << str << " already exists." << std::endl;
+    }
+}
+
+// switch project
+void command_sp(std::string name, ProjectList *proj_list){
+    int id = proj_list->find_project_id_by_name(underscore_to_space(name));
+    if (id >= 0){
+        proj_list->set_active_project(id);
+        std::cout << "Switched to project " << underscore_to_space(name) << std::endl;
+    }else{
+        std::cout << "Project " << underscore_to_space(name) << " does not exist." << std::endl;
+    }
+}
+
+
+
diff --git a/src/ui.h b/src/ui.h
index c71168a..550b74b 100644
--- a/src/ui.h
+++ b/src/ui.h
@@ -20,4 +20,6 @@ void parse_input(std::string input, ProjectList *proj_list);
 void command_ls(std::string input, ProjectList *proj_list);
 void command_start(std::string input, ProjectList *proj_list);
 void command_np(std::string name, ProjectList *proj_list);
+void command_nt(std::string input, ProjectList *proj_list);
+void command_sp(std::string name, ProjectList *proj_list);
 #endif