actually add skill

Amolith created 1 week ago

jj hard

Change summary

src/cli.rs       |  7 +++++++
src/cmd/mod.rs   |  2 ++
src/cmd/skill.rs | 34 ++++++++++++++++++++++++++++++++++
3 files changed, 43 insertions(+)

Detailed changes

src/cli.rs

@@ -141,6 +141,13 @@ pub enum Command {
         /// Path to JSONL file (- for stdin)
         file: String,
     },
+
+    /// Install the agent skill file (SKILL.md)
+    Skill {
+        /// Skills directory (writes managing-tasks-with-td/SKILL.md inside)
+        #[arg(long)]
+        dir: Option<String>,
+    },
 }
 
 #[derive(Subcommand)]

src/cmd/mod.rs

@@ -11,6 +11,7 @@ mod ready;
 mod reopen;
 mod search;
 mod show;
+mod skill;
 mod stats;
 mod update;
 
@@ -128,5 +129,6 @@ pub fn dispatch(cli: &Cli) -> Result<()> {
             let root = require_root()?;
             import::run(&root, file)
         }
+        Command::Skill { dir } => skill::run(dir.as_deref()),
     }
 }

src/cmd/skill.rs

@@ -0,0 +1,34 @@
+use anyhow::{bail, Result};
+use std::path::PathBuf;
+
+const SKILL_CONTENT: &str = include_str!("../../SKILL.md");
+const SKILL_DIR: &str = "managing-tasks-with-td";
+const SKILL_FILE: &str = "SKILL.md";
+
+fn default_skills_dir() -> Result<PathBuf> {
+    if let Ok(xdg) = std::env::var("XDG_CONFIG_HOME") {
+        Ok(PathBuf::from(xdg).join("agents/skills"))
+    } else if let Ok(home) = std::env::var("HOME") {
+        Ok(PathBuf::from(home).join(".config/agents/skills"))
+    } else {
+        bail!("neither $XDG_CONFIG_HOME nor $HOME is set");
+    }
+}
+
+pub fn run(dir: Option<&str>) -> Result<()> {
+    let skills_dir = match dir {
+        Some(d) => PathBuf::from(d),
+        None => default_skills_dir()?,
+    };
+
+    let dest_dir = skills_dir.join(SKILL_DIR);
+    std::fs::create_dir_all(&dest_dir)?;
+
+    let dest = dest_dir.join(SKILL_FILE);
+    std::fs::write(&dest, SKILL_CONTENT)?;
+
+    let c = crate::color::stderr_theme();
+    eprintln!("{}info:{} wrote {}", c.blue, c.reset, dest.display());
+
+    Ok(())
+}