feat: add support for basic vyper project using forge init --vyper

Author: startup-dreamer

crates/forge/src/cmd/init.rs

@@ -37,13 +37,17 @@ pub struct InitArgs {
     #[arg(long, conflicts_with = "template")]
     pub vscode: bool,
 
+    /// Initialize a Vyper project template
+    #[arg(long, conflicts_with = "template")]
+    pub vyper: bool,
+
     #[command(flatten)]
     pub install: DependencyInstallOpts,
 }
 
 impl InitArgs {
     pub fn run(self) -> Result<()> {
-        let Self { root, template, branch, install, offline, force, vscode } = self;
+        let Self { root, template, branch, install, offline, force, vscode, vyper } = self;
         let DependencyInstallOpts { shallow, no_git, commit } = install;
 
         // create the root dir if it does not exist
@@ -53,6 +57,13 @@ impl InitArgs {
         let root = dunce::canonicalize(root)?;
         let git = Git::new(&root).shallow(shallow);
 
+        // If vyper flag is set, use the Vyper template
+        let template = if vyper {
+            Some("https://github.com/Patronum-Labs/foundry-vyper".to_string())
+        } else {
+            template
+        };
+
         // if a template is provided, then this command initializes a git repo,
         // fetches the template repo, and resets the git history to the head of the fetched
         // repo with no other history

crates/forge/tests/cli/cmd.rs

@@ -281,6 +281,29 @@ Warning: Target directory is not empty, but `--force` was specified
     let _config: BasicConfig = parse_with_profile(&s).unwrap().unwrap().1;
 });
 
+// checks that init works with vyper flag
+forgetest!(can_init_vyper_project, |prj, cmd| {
+    prj.wipe();
+
+    cmd.args(["init", "--vyper"]).arg(prj.root()).assert_success().stdout_eq(str![[r#"
+Initializing [..] from https://github.com/Patronum-Labs/foundry-vyper...
+    Initialized forge project
+
+"#]]);
+
+    // Check that the Vyper template repository was cloned correctly
+    assert!(prj.root().join(".git").exists());
+    assert!(prj.root().join("foundry.toml").exists());
+    assert!(prj.root().join("lib/forge-std").exists());
+    assert!(prj.root().join(".git/modules").exists());
+    assert!(prj.root().join("src").exists());
+    assert!(prj.root().join("test").exists());
+    assert!(prj.root().join("src").read_dir().unwrap().any(|entry| {
+        let path = entry.unwrap().path();
+        path.to_string_lossy().ends_with(".vy") || path.to_string_lossy().ends_with(".vpy")
+    }));
+});
+
 // Checks that a forge project fails to initialise if dir is already git repo and dirty
 forgetest!(can_detect_dirty_git_status_on_init, |prj, cmd| {
     prj.wipe();