mdvault_core/domain/behaviors/
task.rs

1//! Task note type behavior.
2//!
3//! Tasks have:
4//! - ID generated from project counter (TST-001) or inbox (INB-001)
5//! - Project selector prompt
6//! - Logging to daily note
7//! - Output path: Projects/{project}/Tasks/{id}.md or Inbox/{id}.md
8
9use std::path::PathBuf;
10use std::sync::Arc;
11
12use crate::types::TypeDefinition;
13
14use super::super::context::{CreationContext, FieldPrompt, PromptContext, PromptType};
15use super::super::traits::{
16    DomainError, DomainResult, NoteBehavior, NoteIdentity, NoteLifecycle, NotePrompts,
17};
18
19/// Behavior implementation for task notes.
20pub struct TaskBehavior {
21    typedef: Option<Arc<TypeDefinition>>,
22}
23
24impl TaskBehavior {
25    /// Create a new TaskBehavior, optionally wrapping a Lua typedef override.
26    pub fn new(typedef: Option<Arc<TypeDefinition>>) -> Self {
27        Self { typedef }
28    }
29}
30
31impl NoteIdentity for TaskBehavior {
32    fn generate_id(&self, ctx: &CreationContext) -> DomainResult<Option<String>> {
33        // ID generation is handled in before_create after project is known
34        // Return existing ID if already set
35        if let Some(ref id) = ctx.core_metadata.task_id {
36            return Ok(Some(id.clone()));
37        }
38        Ok(None)
39    }
40
41    fn output_path(&self, ctx: &CreationContext) -> DomainResult<PathBuf> {
42        let task_id = ctx
43            .core_metadata
44            .task_id
45            .as_ref()
46            .ok_or_else(|| DomainError::PathResolution("task-id not set".into()))?;
47
48        let project = ctx.get_var("project").unwrap_or("inbox");
49
50        // Check Lua typedef for output template first
51        if let Some(ref td) = self.typedef
52            && let Some(ref _output) = td.output
53        {
54            // TODO: render_output_path(output, ctx)
55        }
56
57        // Default path
58        if project == "inbox" {
59            Ok(ctx.config.vault_root.join(format!("Inbox/{}.md", task_id)))
60        } else {
61            Ok(ctx
62                .config
63                .vault_root
64                .join(format!("Projects/{}/Tasks/{}.md", project, task_id)))
65        }
66    }
67
68    fn core_fields(&self) -> Vec<&'static str> {
69        vec!["type", "title", "task-id", "project"]
70    }
71}
72
73impl NoteLifecycle for TaskBehavior {
74    fn before_create(&self, ctx: &mut CreationContext) -> DomainResult<()> {
75        let project = ctx
76            .get_var("project")
77            .map(|s| s.to_string())
78            .unwrap_or_else(|| "inbox".into());
79
80        // Generate task ID based on project
81        let task_id = if project == "inbox" {
82            generate_inbox_task_id(&ctx.config.vault_root)?
83        } else {
84            // Get project counter and generate ID
85            let (project_id, counter) = get_project_info(ctx.config, &project)?;
86            format!("{}-{:03}", project_id, counter + 1)
87        };
88
89        // Set core metadata
90        ctx.core_metadata.task_id = Some(task_id.clone());
91        ctx.core_metadata.project =
92            if project == "inbox" { None } else { Some(project.clone()) };
93        ctx.set_var("task-id", &task_id);
94        if project != "inbox" {
95            ctx.set_var("project", &project);
96        }
97
98        Ok(())
99    }
100
101    fn after_create(&self, ctx: &CreationContext, _content: &str) -> DomainResult<()> {
102        let project = ctx.get_var("project").unwrap_or("inbox");
103
104        // Increment project counter if not inbox
105        if project != "inbox" {
106            increment_project_counter(ctx.config, project)?;
107        }
108
109        // TODO: Log to daily note
110        // TODO: Run Lua on_create hook if defined
111        // TODO: Reindex vault
112
113        Ok(())
114    }
115}
116
117impl NotePrompts for TaskBehavior {
118    fn type_prompts(&self, ctx: &PromptContext) -> Vec<FieldPrompt> {
119        let mut prompts = vec![];
120
121        // Project selector (if not provided)
122        if !ctx.provided_vars.contains_key("project") && !ctx.batch_mode {
123            prompts.push(FieldPrompt {
124                field_name: "project".into(),
125                prompt_text: "Select project for this task".into(),
126                prompt_type: PromptType::ProjectSelector,
127                required: false, // Can be inbox
128                default_value: Some("inbox".into()),
129            });
130        }
131
132        prompts
133    }
134}
135
136impl NoteBehavior for TaskBehavior {
137    fn type_name(&self) -> &'static str {
138        "task"
139    }
140}
141
142// --- Helper functions (to be moved/refactored) ---
143
144use crate::config::types::ResolvedConfig;
145use std::fs;
146
147/// Generate an inbox task ID by scanning the Inbox directory.
148fn generate_inbox_task_id(vault_root: &std::path::Path) -> DomainResult<String> {
149    let inbox_dir = vault_root.join("Inbox");
150
151    let mut max_num = 0u32;
152
153    if inbox_dir.exists() {
154        for entry in fs::read_dir(&inbox_dir).map_err(DomainError::Io)? {
155            let entry = entry.map_err(DomainError::Io)?;
156            let name = entry.file_name();
157            let name_str = name.to_string_lossy();
158
159            // Parse INB-XXX.md pattern
160            if let Some(stem) = name_str.strip_suffix(".md")
161                && let Some(num_str) = stem.strip_prefix("INB-")
162                && let Ok(num) = num_str.parse::<u32>()
163            {
164                max_num = max_num.max(num);
165            }
166        }
167    }
168
169    Ok(format!("INB-{:03}", max_num + 1))
170}
171
172/// Get project info (project-id and task_counter) from project file.
173fn get_project_info(
174    config: &ResolvedConfig,
175    project: &str,
176) -> DomainResult<(String, u32)> {
177    let project_file = find_project_file(config, project)?;
178
179    let content = fs::read_to_string(&project_file).map_err(DomainError::Io)?;
180
181    // Parse frontmatter
182    let parsed = crate::frontmatter::parse(&content).map_err(|e| {
183        DomainError::Other(format!("Failed to parse project frontmatter: {}", e))
184    })?;
185
186    let fields = parsed.frontmatter.map(|fm| fm.fields).unwrap_or_default();
187
188    let project_id = fields
189        .get("project-id")
190        .and_then(|v| v.as_str())
191        .map(|s| s.to_string())
192        .unwrap_or_else(|| project.to_uppercase());
193
194    let counter = fields
195        .get("task_counter")
196        .and_then(|v| v.as_u64())
197        .map(|n| n as u32)
198        .unwrap_or(0);
199
200    Ok((project_id, counter))
201}
202
203/// Find the project file by project name/ID.
204fn find_project_file(config: &ResolvedConfig, project: &str) -> DomainResult<PathBuf> {
205    // Try common patterns
206    let patterns = [
207        format!("Projects/{}/{}.md", project, project),
208        format!("Projects/{}.md", project),
209        format!("projects/{}/{}.md", project.to_lowercase(), project.to_lowercase()),
210    ];
211
212    for pattern in &patterns {
213        let path = config.vault_root.join(pattern);
214        if path.exists() {
215            return Ok(path);
216        }
217    }
218
219    Err(DomainError::Other(format!("Project file not found for: {}", project)))
220}
221
222/// Increment the task_counter in a project file.
223fn increment_project_counter(config: &ResolvedConfig, project: &str) -> DomainResult<()> {
224    let project_file = find_project_file(config, project)?;
225
226    let content = fs::read_to_string(&project_file).map_err(DomainError::Io)?;
227
228    // Parse frontmatter
229    let parsed = crate::frontmatter::parse(&content).map_err(|e| {
230        DomainError::Other(format!("Failed to parse project frontmatter: {}", e))
231    })?;
232
233    let mut fields = parsed.frontmatter.map(|fm| fm.fields).unwrap_or_default();
234
235    let current = fields
236        .get("task_counter")
237        .and_then(|v| v.as_u64())
238        .map(|n| n as u32)
239        .unwrap_or(0);
240
241    fields.insert(
242        "task_counter".to_string(),
243        serde_yaml::Value::Number((current + 1).into()),
244    );
245
246    // Rebuild content with updated frontmatter
247    let yaml = serde_yaml::to_string(&fields).map_err(|e| {
248        DomainError::Other(format!("Failed to serialize frontmatter: {}", e))
249    })?;
250
251    let new_content = format!("---\n{}---\n{}", yaml, parsed.body);
252    fs::write(&project_file, new_content).map_err(DomainError::Io)?;
253
254    Ok(())
255}
mdvault_core/domain/behaviors/task.rs

mdvault_core/domain/behaviors/
task.rs
