jany 0.1.0

jev x any command: turn loosely ordered words into a command line, defined per command by a schema
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

//! `/jany-register` スキルの配布と `jany --register <name>` の雛形。
//! スキル本体は `skill/jany-register/` をバイナリに埋め込み、`jany --init` のたびに
//! `~/.agents/skills/jany-register/` へ書く(Claude Code / Codex のどちらからも読める場所)。

use crate::error::JanyError;
use std::path::{Path, PathBuf};

/// 配布するファイル。examples は examples/ の find・curl そのもの。
const FILES: &[(&str, &str)] = &[
    ("SKILL.md", include_str!("../skill/jany-register/SKILL.md")),
    ("reference.md", include_str!("../skill/jany-register/reference.md")),
    ("template/schema.toml", include_str!("../skill/jany-register/template/schema.toml")),
    ("template/assemble.sh", include_str!("../skill/jany-register/template/assemble.sh")),
    ("template/cases.toml", include_str!("../skill/jany-register/template/cases.toml")),
    ("examples/find/schema.toml", include_str!("../examples/find/schema.toml")),
    ("examples/find/assemble.sh", include_str!("../examples/find/assemble.sh")),
    ("examples/find/cases.toml", include_str!("../examples/find/cases.toml")),
    ("examples/curl/schema.toml", include_str!("../examples/curl/schema.toml")),
    ("examples/curl/assemble.sh", include_str!("../examples/curl/assemble.sh")),
    ("examples/curl/cases.toml", include_str!("../examples/curl/cases.toml")),
];

/// 組み込みのコマンド定義。`jany --init` が `~/.config/jany/cmd/<name>/` にまだ無いものだけ置く。
/// 中身は examples/ の原本そのもの(examples と同じ)。
const COMMANDS: &[(&str, &[(&str, &str)])] = &[
    ("find", &[
        ("schema.toml", include_str!("../examples/find/schema.toml")),
        ("assemble.sh", include_str!("../examples/find/assemble.sh")),
        ("cases.toml", include_str!("../examples/find/cases.toml")),
    ]),
    ("curl", &[
        ("schema.toml", include_str!("../examples/curl/schema.toml")),
        ("assemble.sh", include_str!("../examples/curl/assemble.sh")),
        ("cases.toml", include_str!("../examples/curl/cases.toml")),
    ]),
    ("docker/run", &[
        ("schema.toml", include_str!("../examples/docker/run/schema.toml")),
        ("assemble.sh", include_str!("../examples/docker/run/assemble.sh")),
        ("cases.toml", include_str!("../examples/docker/run/cases.toml")),
    ]),
];

/// まだ無い定義だけ置く。既に schema.toml があるディレクトリは(古くても)触らない。
/// 戻り値は置いたディレクトリ。
pub fn install_commands(cmd_dir: &Path) -> Result<Vec<String>, JanyError> {
    let mut placed = Vec::new();
    for (name, files) in COMMANDS {
        let dir = cmd_dir.join(name);
        if dir.join("schema.toml").exists() {
            continue;
        }
        std::fs::create_dir_all(&dir)?;
        for (rel, body) in *files {
            let p = dir.join(rel);
            std::fs::write(&p, body)?;
            set_executable(&p, rel.ends_with(".sh"))?;
        }
        placed.push(dir.display().to_string());
    }
    Ok(placed)
}

/// `JANY_SKILL_DIR`、無ければ `~/.agents/skills/jany-register`。
pub fn skill_dir() -> Option<PathBuf> {
    if let Some(d) = std::env::var_os("JANY_SKILL_DIR") {
        return Some(PathBuf::from(d));
    }
    let home = std::env::var_os("HOME")?;
    Some(Path::new(&home).join(".agents").join("skills").join("jany-register"))
}

/// 中身が違うファイルだけ書き直す。戻り値は書いたパスと作ったリンク。
pub fn install() -> Result<Vec<String>, JanyError> {
    let dir = skill_dir().ok_or_else(|| JanyError::Config("cannot determine skill dir (HOME unset)".into()))?;
    let mut changed = Vec::new();
    for (rel, body) in FILES {
        let p = dir.join(rel);
        if std::fs::read_to_string(&p).map(|cur| cur == *body).unwrap_or(false) {
            continue;
        }
        if let Some(parent) = p.parent() {
            std::fs::create_dir_all(parent)?;
        }
        std::fs::write(&p, body)?;
        set_executable(&p, rel.ends_with(".sh"))?;
        changed.push(p.display().to_string());
    }
    // Claude Code / Codex がそれぞれのディレクトリしか見ない場合に備えてリンクを置く。
    // ディレクトリが既にあって、その名前がまだ無いときだけ。
    if let Some(home) = std::env::var_os("HOME") {
        for tool in [".claude", ".codex"] {
            let skills = Path::new(&home).join(tool).join("skills");
            let link = skills.join("jany-register");
            if skills.is_dir() && std::fs::symlink_metadata(&link).is_err() {
                #[cfg(unix)]
                if std::os::unix::fs::symlink(&dir, &link).is_ok() {
                    changed.push(format!("{} -> {}", link.display(), dir.display()));
                }
            }
        }
    }
    Ok(changed)
}

fn set_executable(p: &Path, on: bool) -> std::io::Result<()> {
    #[cfg(unix)]
    if on {
        use std::os::unix::fs::PermissionsExt;
        let mut perm = std::fs::metadata(p)?.permissions();
        perm.set_mode(perm.mode() | 0o111);
        std::fs::set_permissions(p, perm)?;
    }
    Ok(())
}

/// `jany --register <name> [sub…]`: 雛形 3 ファイルを置く。既にあれば触らない。
pub fn register(cmd_dir: &Path, names: &[String]) -> Result<i32, JanyError> {
    if names.is_empty() {
        return Err(JanyError::Usage("jany --register <name> [sub …]  e.g. jany --register docker run".into()));
    }
    for n in names {
        if n.is_empty() || n.contains('/') || n.starts_with('.') || n.starts_with('-') {
            return Err(JanyError::Usage(format!("bad command name `{n}`: use plain words (docker run)")));
        }
    }
    let name = names.join(" ");
    // argv の先頭は語ごとに分ける("docker run" → "docker", "run")。
    let argv0 = names.iter().map(|n| format!("{n:?}")).collect::<Vec<_>>().join(", ");
    let dir = names.iter().fold(cmd_dir.to_path_buf(), |d, n| d.join(n));
    if dir.join("schema.toml").exists() {
        return Err(JanyError::Usage(format!("{} already exists; edit it or remove it first", dir.join("schema.toml").display())));
    }
    std::fs::create_dir_all(&dir)?;
    for rel in ["schema.toml", "assemble.sh", "cases.toml"] {
        let body = FILES.iter().find(|(r, _)| *r == format!("template/{rel}")).map(|(_, b)| *b).expect("template embedded");
        let p = dir.join(rel);
        std::fs::write(&p, body.replace("__ARGV0__", &argv0).replace("__NAME__", &name))?;
        set_executable(&p, rel.ends_with(".sh"))?;
        eprintln!("wrote {}", p.display());
    }
    let skill = skill_dir().map(|d| d.display().to_string()).unwrap_or_else(|| "~/.agents/skills/jany-register".into());
    eprintln!(
        "\nnext: fill them in with the skill (Claude Code / Codex): /jany-register {name}\n      reference: {skill}/reference.md\n      then run:  jany --test {name}"
    );
    Ok(0)
}