pawan/
init.rs

1//! Project skeleton templates for `pawan init`.
2//!
3//! Generates scaffolding files for new Rust projects that pass `cargo check` immediately.
4
5use std::path::Path;
6
7/// A project skeleton: a named collection of (relative path, content) pairs.
8pub struct ProjectSkeleton {
9    pub name: String,
10    pub files: Vec<(String, String)>,
11}
12
13impl ProjectSkeleton {
14    /// Write all skeleton files under `root`, creating intermediate directories as needed.
15    pub fn write_to(&self, root: &Path) -> std::io::Result<()> {
16        for (rel_path, content) in &self.files {
17            let full = root.join(rel_path);
18            if let Some(parent) = full.parent() {
19                std::fs::create_dir_all(parent)?;
20            }
21            std::fs::write(&full, content)?;
22        }
23        Ok(())
24    }
25}
26
27/// Generate a simple Rust binary project skeleton.
28pub fn rust_binary_skeleton(name: &str) -> ProjectSkeleton {
29    let cargo_toml = format!(
30        r#"[package]
31name = "{name}"
32version = "0.1.0"
33edition = "2021"
34rust-version = "1.75"
35"#
36    );
37    let main_rs = "fn main() {\n    println!(\"Hello, world!\");\n}\n".to_string();
38
39    ProjectSkeleton {
40        name: name.to_string(),
41        files: vec![
42            ("Cargo.toml".into(), cargo_toml),
43            ("src/main.rs".into(), main_rs),
44        ],
45    }
46}
47
48/// Generate a Rust library project skeleton.
49pub fn rust_library_skeleton(name: &str) -> ProjectSkeleton {
50    let cargo_toml = format!(
51        r#"[package]
52name = "{name}"
53version = "0.1.0"
54edition = "2021"
55rust-version = "1.75"
56"#
57    );
58    let lib_rs = format!(
59        "/// Add two numbers.\npub fn add(a: i32, b: i32) -> i32 {{\n    a + b\n}}\n\n#[cfg(test)]\nmod tests {{\n    use super::*;\n\n    #[test]\n    fn it_works() {{\n        assert_eq!(add(2, 2), 4);\n    }}\n}}\n"
60    );
61
62    ProjectSkeleton {
63        name: name.to_string(),
64        files: vec![
65            ("Cargo.toml".into(), cargo_toml),
66            ("src/lib.rs".into(), lib_rs),
67        ],
68    }
69}
70
71/// Generate a Pawan agent project skeleton (binary with pawan dep + pawan.toml config).
72pub fn pawan_agent_skeleton(name: &str) -> ProjectSkeleton {
73    let cargo_toml = format!(
74        r#"[package]
75name = "{name}"
76version = "0.1.0"
77edition = "2021"
78rust-version = "1.75"
79
80[dependencies]
81pawan = {{ git = "https://github.com/dirmacs/pawan.git" }}
82"#
83    );
84    let main_rs = format!(
85        r#"fn main() {{
86    println!("Pawan agent: {name}");
87}}
88"#
89    );
90    let pawan_toml = format!(
91        r#"[agent]
92name = "{name}"
93model = "qwen/qwen3.5-122b-a10b"
94
95[provider]
96name = "nvidia"
97api_url = "https://integrate.api.nvidia.com/v1"
98"#
99    );
100
101    ProjectSkeleton {
102        name: name.to_string(),
103        files: vec![
104            ("Cargo.toml".into(), cargo_toml),
105            ("src/main.rs".into(), main_rs),
106            ("pawan.toml".into(), pawan_toml),
107        ],
108    }
109}
110
111#[cfg(test)]
112mod tests {
113    use super::*;
114    use std::fs;
115
116    #[test]
117    fn binary_skeleton_files() {
118        let sk = rust_binary_skeleton("myapp");
119        assert_eq!(sk.name, "myapp");
120        let paths: Vec<&str> = sk.files.iter().map(|(p, _)| p.as_str()).collect();
121        assert_eq!(paths, vec!["Cargo.toml", "src/main.rs"]);
122    }
123
124    #[test]
125    fn library_skeleton_files() {
126        let sk = rust_library_skeleton("mylib");
127        assert_eq!(sk.name, "mylib");
128        let paths: Vec<&str> = sk.files.iter().map(|(p, _)| p.as_str()).collect();
129        assert_eq!(paths, vec!["Cargo.toml", "src/lib.rs"]);
130    }
131
132    #[test]
133    fn agent_skeleton_files() {
134        let sk = pawan_agent_skeleton("myagent");
135        assert_eq!(sk.name, "myagent");
136        let paths: Vec<&str> = sk.files.iter().map(|(p, _)| p.as_str()).collect();
137        assert_eq!(paths, vec!["Cargo.toml", "src/main.rs", "pawan.toml"]);
138    }
139
140    #[test]
141    fn write_to_creates_files() {
142        let dir = tempfile::tempdir().unwrap();
143        let sk = rust_binary_skeleton("testproj");
144        sk.write_to(dir.path()).unwrap();
145
146        let cargo = fs::read_to_string(dir.path().join("Cargo.toml")).unwrap();
147        assert!(cargo.contains("name = \"testproj\""));
148        assert!(cargo.contains("edition = \"2021\""));
149        assert!(cargo.contains("rust-version = \"1.75\""));
150
151        let main = fs::read_to_string(dir.path().join("src/main.rs")).unwrap();
152        assert!(main.contains("fn main()"));
153    }
154
155    #[test]
156    fn cargo_toml_is_valid_toml() {
157        for sk in [
158            rust_binary_skeleton("a"),
159            rust_library_skeleton("b"),
160            pawan_agent_skeleton("c"),
161        ] {
162            let cargo_content = &sk.files.iter().find(|(p, _)| p == "Cargo.toml").unwrap().1;
163            let parsed: Result<toml::Value, _> = toml::from_str(cargo_content);
164            assert!(parsed.is_ok(), "Invalid TOML in {} skeleton", sk.name);
165        }
166    }
167
168    #[test]
169    fn pawan_agent_skeleton_includes_pawan_toml_in_file_list() {
170        let sk = pawan_agent_skeleton("demo");
171        let has_pawan_toml = sk.files.iter().any(|(p, _)| p == "pawan.toml");
172        assert!(has_pawan_toml, "pawan_agent_skeleton must include pawan.toml");
173    }
174
175    #[test]
176    fn write_to_creates_nested_directories() {
177        let dir = tempfile::tempdir().unwrap();
178        let sk = rust_binary_skeleton("nested");
179        sk.write_to(dir.path()).unwrap();
180        // src/ directory must have been created for src/main.rs
181        assert!(dir.path().join("src").is_dir(), "src/ directory not created");
182        assert!(dir.path().join("src/main.rs").is_file());
183    }
184
185    #[test]
186    fn skeleton_names_set_correctly() {
187        assert_eq!(rust_binary_skeleton("alpha").name, "alpha");
188        assert_eq!(rust_library_skeleton("beta").name, "beta");
189        assert_eq!(pawan_agent_skeleton("gamma").name, "gamma");
190    }
191
192    #[test]
193    fn generated_cargo_toml_contains_edition_and_rust_version() {
194        for (sk, expected_name) in [
195            (rust_binary_skeleton("x"), "x"),
196            (rust_library_skeleton("y"), "y"),
197            (pawan_agent_skeleton("z"), "z"),
198        ] {
199            let cargo = &sk.files.iter().find(|(p, _)| p == "Cargo.toml").unwrap().1;
200            assert!(cargo.contains("edition = \"2021\""), "{expected_name} missing edition");
201            assert!(cargo.contains("rust-version = \"1.75\""), "{expected_name} missing rust-version");
202        }
203    }
204
205    #[test]
206    fn agent_skeleton_has_pawan_toml() {
207        let dir = tempfile::tempdir().unwrap();
208        let sk = pawan_agent_skeleton("agent1");
209        sk.write_to(dir.path()).unwrap();
210
211        let pawan_cfg = fs::read_to_string(dir.path().join("pawan.toml")).unwrap();
212        assert!(pawan_cfg.contains("name = \"agent1\""));
213        assert!(pawan_cfg.contains("model ="));
214    }
215}
pawan/init.rs

pawan/
init.rs
