tsk-ai 0.10.6

tsk-tsk: keeping your agents out of trouble with sandboxed coding agent automation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use crate::assets::frontmatter::parse_frontmatter;
use crate::assets::{find_template, list_all_templates};
use crate::commands::Command;
use crate::context::AppContext;
use crate::display::print_columns;
use crate::repo_utils::find_repository_root;
use async_trait::async_trait;
use std::error::Error;
use std::path::Path;

pub struct TemplateListCommand;

#[async_trait]
impl Command for TemplateListCommand {
    async fn execute(&self, ctx: &AppContext) -> Result<(), Box<dyn Error>> {
        let project_root = find_repository_root(Path::new(".")).ok();

        // List all available templates
        let templates = list_all_templates(project_root.as_deref(), &ctx.tsk_env());

        if templates.is_empty() {
            println!("No templates available");
            return Ok(());
        }

        // Determine source and description for each template
        let mut rows = Vec::new();
        for template in &templates {
            let source = determine_template_source(template, project_root.as_deref(), ctx)?;
            let description = find_template(template, project_root.as_deref(), &ctx.tsk_env())
                .ok()
                .and_then(|content| parse_frontmatter(&content).description)
                .unwrap_or_default();
            rows.push(vec![template.to_string(), source, description]);
        }

        println!("Available Templates:\n");
        print_columns(&["Name", "Source", "Description"], &rows);

        // Print additional information
        println!("\nTemplate locations (in priority order):");
        if let Some(root) = &project_root {
            println!("  1. Project: {}/.tsk/templates/", root.display());
        }
        println!(
            "  2. User: {}/templates/",
            ctx.tsk_env().config_dir().display()
        );
        println!("  3. Built-in: Embedded in tsk binary");

        Ok(())
    }
}

fn determine_template_source(
    template_name: &str,
    project_root: Option<&Path>,
    ctx: &AppContext,
) -> Result<String, Box<dyn Error>> {
    // Check project level first
    if let Some(root) = project_root {
        let project_template = root
            .join(".tsk")
            .join("templates")
            .join(format!("{template_name}.md"));
        if project_template.exists() {
            return Ok("Project".to_string());
        }
    }

    // Check user level
    let user_template = ctx
        .tsk_env()
        .config_dir()
        .join("templates")
        .join(format!("{template_name}.md"));
    if user_template.exists() {
        return Ok("User".to_string());
    }

    // Must be built-in
    Ok("Built-in".to_string())
}