refactor for managing upcoming and current

projects!

pentest_tool/src/project_controls.rs

@@ -7,6 +7,7 @@ use std::process;
 use std::thread;
 use std::time::Duration;
 use std::str::FromStr;
+use crate::project_controls;
 use crate::Project;
 use crate::box_controls::make_box;
 
@@ -53,13 +54,13 @@ pub fn save_projects(projects: &Vec<Project>, config_path: &PathBuf){
         let mut _outline = String::new();
         if project.active{
             if active_set == false{
-                _outline = format!("{}yes:{}\n", default, project.boxname);
+                _outline = format!("{}yes:{}:{}\n", default, project.boxname, project.stage);
                 active_set = true;
             }
 
         }
         else{
-            _outline = format!("{}no:{}\n", default, project.boxname);
+            _outline = format!("{}no:{}:{}\n", default, project.boxname, project.stage);
         }
         save_file.write_all(_outline.as_bytes()).expect("error writing outline");
     }
@@ -71,6 +72,7 @@ pub fn new_project(projects: &mut Vec<Project>, project_dir: &PathBuf, notes_dir
     let mut existing_folders = String::new();
     let mut customer_name = String::new();
     let mut project_name = String::new();
+    let mut project_stage = String::new();
     println!("do you have an existing notes and folder structure to copy over?\ny/n");
     std::io::stdin().read_line(&mut existing_folders).unwrap();
     if existing_folders.contains("y") || existing_folders.contains("Y"){
@@ -182,6 +184,25 @@ pub fn new_project(projects: &mut Vec<Project>, project_dir: &PathBuf, notes_dir
         fs::create_dir_all(&new_project_dir).expect("error creating new files folder");
         fs::create_dir_all(&new_note_dir).expect("error creating new notes folder");
     }
+    loop{
+        let mut stage_response = String::new();
+        println!("what stage is this project in?");
+        print!("
+1.) current
+2.) upcoming
+");
+        let stage_result = stdin().read_line(&mut stage_response);
+        if stage_result.is_err(){
+            println!("we need input here dummy, try again...");
+        }
+        else{
+            match &stage_response.trim_end(){
+                &"1" => {project_stage = "current".to_owned(); break;},
+                &"2" => {project_stage = "upcoming".to_owned(); break;},
+                _ => println!("unknown option, try again...")
+            }
+        }
+    }
     thread::sleep(Duration::from_secs(2));
     let box_name = format!("atarchbox_{}", customer_name);
     let new_project = Project{customer: customer_name.trim_end().to_owned(), 
@@ -191,6 +212,7 @@ pub fn new_project(projects: &mut Vec<Project>, project_dir: &PathBuf, notes_dir
                                     active: false,
                                     id: new_id,
                                     boxname: box_name,
+                                    stage: project_stage.to_owned()
                                       };
     make_box(&new_project, &tools_dir, &boxtemplate, true);
     projects.push(new_project);
@@ -281,11 +303,51 @@ pub fn get_projects(config_path: &PathBuf) -> Vec<Project>{
                         active = true;
                     }
                 }
-                let new_project = Project{customer: customer, project_name: project, files_folder: project_folder, notes_folder: notes_folder, active: active, id: first, boxname: boxname};
+                let project_stage = settings[6].to_owned();
+                let new_project = Project{customer: customer, project_name: project, files_folder: project_folder, notes_folder: notes_folder, active: active, id: first, boxname: boxname, stage: project_stage};
                 println!("{} {} LOADED!", &new_project.customer, &new_project.project_name);
                 projects.push(new_project);
             }
         }
     }
     return projects
+}
+
+pub fn print_upcoming_projects(projects: &Vec<Project>){
+    for project in projects{
+        if project.stage.contains("upcoming"){
+            println!("{}:{}", project.customer, project.project_name);
+        }
+    }
+}
+
+pub fn promote_project(projects: &mut Vec<Project>, config_path: &PathBuf){
+    let working_projects = projects.clone();
+    for project in &working_projects{
+        if project.stage.contains("upcoming"){
+            println!("{}.) {}:{}", project.id, project.customer, project.project_name);
+        }
+    }
+    println!("which project to promote?");
+    let mut selection = String::new();
+    let result = stdin().read_line(&mut selection);
+    if result.is_err(){
+        println!("we need input here dummy try again....");
+    } 
+    result.unwrap();
+    let promote_id: i32 = selection.trim_end().parse().unwrap(); 
+    let mut projects_to_save = Vec::new();
+    for project in &working_projects{
+        if project.id == promote_id{
+            let mut promoted_project = project.clone();
+            promoted_project.stage = "current".to_owned();
+            projects_to_save.push(promoted_project);
+        }
+        else{
+            projects_to_save.push(project.clone());
+        }
+    }
+    projects.clear();
+    projects.append(&mut projects_to_save);
+    save_projects(&projects_to_save, config_path);
 }