📝 Apply project and task edits

src/commands/edit.rs

@@ -1,5 +1,6 @@
 use crate::api::client::ApiClient;
-use crate::models;
+use crate::constants::DEFAULT_ENTITY_ID;
+use crate::models::{self, TimeEntry};
 use crate::parcel::Parcel;
 use crate::picker;
 use colored::Colorize;
@@ -18,12 +19,38 @@ impl EditCommand {
         match entities.running_time_entry() {
             None => println!("{}", "No time entry is running at the moment".yellow()),
             Some(running_time_entry) => {
-                let updated_time_entry = running_time_entry
-                    .launch_in_editor()
-                    .inspect_err(|e| {
-                        eprintln!("{}", e.to_string().red());
-                    })
-                    .unwrap();
+                let updated_time_entry = {
+                    let workspace_id = running_time_entry.workspace_id;
+
+                    let new_entry = running_time_entry
+                        .launch_in_editor()
+                        .inspect_err(|e| {
+                            eprintln!("{}", e.to_string().red());
+                        })
+                        .unwrap();
+
+                    let project = new_entry.project.and_then(|project| {
+                        if project.id == DEFAULT_ENTITY_ID {
+                            entities.project_for_name(workspace_id, &project.name)
+                        } else {
+                            Some(project)
+                        }
+                    });
+
+                    let task = new_entry.task.and_then(|task| {
+                        if task.id == DEFAULT_ENTITY_ID {
+                            entities.task_for_name(workspace_id, &task.name)
+                        } else {
+                            Some(task)
+                        }
+                    });
+
+                    TimeEntry {
+                        project,
+                        task,
+                        ..new_entry
+                    }
+                };
 
                 let updated_entry_id = api_client
                     .update_time_entry(updated_time_entry.clone())