feat(pm): add link and unlink commands

Author: fengmk2

crates/vite_install/src/commands/link.rs

@@ -0,0 +1,189 @@
+use std::{collections::HashMap, process::ExitStatus};
+
+use vite_error::Error;
+use vite_path::AbsolutePath;
+
+use crate::package_manager::{
+    PackageManager, PackageManagerType, ResolveCommandResult, format_path_env, run_command,
+};
+
+/// Options for the link command.
+#[derive(Debug, Default)]
+pub struct LinkCommandOptions<'a> {
+    pub package: Option<&'a str>,
+    pub pass_through_args: Option<&'a [String]>,
+}
+
+impl PackageManager {
+    /// Run the link command with the package manager.
+    /// Return the exit status of the command.
+    #[must_use]
+    pub async fn run_link_command(
+        &self,
+        options: &LinkCommandOptions<'_>,
+        cwd: impl AsRef<AbsolutePath>,
+    ) -> Result<ExitStatus, Error> {
+        let resolve_command = self.resolve_link_command(options);
+        run_command(&resolve_command.bin_path, &resolve_command.args, &resolve_command.envs, cwd)
+            .await
+    }
+
+    /// Resolve the link command.
+    #[must_use]
+    pub fn resolve_link_command(&self, options: &LinkCommandOptions) -> ResolveCommandResult {
+        let bin_name: String;
+        let envs = HashMap::from([("PATH".to_string(), format_path_env(self.get_bin_prefix()))]);
+        let mut args: Vec<String> = Vec::new();
+
+        match self.client {
+            PackageManagerType::Pnpm => {
+                bin_name = "pnpm".into();
+                args.push("link".into());
+
+                // pnpm link behavior:
+                // - pnpm link --global: register current package globally
+                // - pnpm link --global <pkg>: link global package to current
+                // - pnpm link <dir>: link local directory
+
+                // For registering current package or linking global package
+                // Add --global unless it's a directory path (starts with . or /)
+                if let Some(package) = options.package {
+                    if !package.starts_with('.') && !package.starts_with('/') {
+                        args.push("--global".into());
+                    }
+                } else {
+                    // No package specified, register current package globally
+                    args.push("--global".into());
+                }
+            }
+            PackageManagerType::Yarn => {
+                bin_name = "yarn".into();
+                args.push("link".into());
+            }
+            PackageManagerType::Npm => {
+                bin_name = "npm".into();
+                args.push("link".into());
+            }
+        }
+
+        // Add package/directory if specified
+        if let Some(package) = options.package {
+            args.push(package.to_string());
+        }
+
+        // Add pass-through args
+        if let Some(pass_through_args) = options.pass_through_args {
+            args.extend_from_slice(pass_through_args);
+        }
+
+        ResolveCommandResult { bin_path: bin_name, args, envs }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use tempfile::{TempDir, tempdir};
+    use vite_path::AbsolutePathBuf;
+    use vite_str::Str;
+
+    use super::*;
+
+    fn create_temp_dir() -> TempDir {
+        tempdir().expect("Failed to create temp directory")
+    }
+
+    fn create_mock_package_manager(pm_type: PackageManagerType, version: &str) -> PackageManager {
+        let temp_dir = create_temp_dir();
+        let temp_dir_path = AbsolutePathBuf::new(temp_dir.path().to_path_buf()).unwrap();
+        let install_dir = temp_dir_path.join("install");
+
+        PackageManager {
+            client: pm_type,
+            package_name: pm_type.to_string().into(),
+            version: Str::from(version),
+            hash: None,
+            bin_name: pm_type.to_string().into(),
+            workspace_root: temp_dir_path.clone(),
+            install_dir,
+        }
+    }
+
+    #[test]
+    fn test_pnpm_link_no_package() {
+        let pm = create_mock_package_manager(PackageManagerType::Pnpm, "10.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions { ..Default::default() });
+        assert_eq!(result.bin_path, "pnpm");
+        assert_eq!(result.args, vec!["link", "--global"]);
+    }
+
+    #[test]
+    fn test_pnpm_link_package() {
+        let pm = create_mock_package_manager(PackageManagerType::Pnpm, "10.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions {
+            package: Some("react"),
+            ..Default::default()
+        });
+        assert_eq!(result.bin_path, "pnpm");
+        assert_eq!(result.args, vec!["link", "--global", "react"]);
+    }
+
+    #[test]
+    fn test_pnpm_link_directory() {
+        let pm = create_mock_package_manager(PackageManagerType::Pnpm, "10.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions {
+            package: Some("./packages/utils"),
+            ..Default::default()
+        });
+        assert_eq!(result.bin_path, "pnpm");
+        assert_eq!(result.args, vec!["link", "./packages/utils"]);
+    }
+
+    #[test]
+    fn test_pnpm_link_absolute_directory() {
+        let pm = create_mock_package_manager(PackageManagerType::Pnpm, "10.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions {
+            package: Some("/absolute/path/to/package"),
+            ..Default::default()
+        });
+        assert_eq!(result.bin_path, "pnpm");
+        assert_eq!(result.args, vec!["link", "/absolute/path/to/package"]);
+    }
+
+    #[test]
+    fn test_yarn_link_basic() {
+        let pm = create_mock_package_manager(PackageManagerType::Yarn, "4.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions { ..Default::default() });
+        assert_eq!(result.bin_path, "yarn");
+        assert_eq!(result.args, vec!["link"]);
+    }
+
+    #[test]
+    fn test_yarn_link_package() {
+        let pm = create_mock_package_manager(PackageManagerType::Yarn, "4.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions {
+            package: Some("react"),
+            ..Default::default()
+        });
+        assert_eq!(result.bin_path, "yarn");
+        assert_eq!(result.args, vec!["link", "react"]);
+    }
+
+    #[test]
+    fn test_npm_link_basic() {
+        let pm = create_mock_package_manager(PackageManagerType::Npm, "11.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions { ..Default::default() });
+        assert_eq!(result.bin_path, "npm");
+        assert_eq!(result.args, vec!["link"]);
+    }
+
+    #[test]
+    fn test_npm_link_package() {
+        let pm = create_mock_package_manager(PackageManagerType::Npm, "11.0.0");
+        let result = pm.resolve_link_command(&LinkCommandOptions {
+            package: Some("react"),
+            ..Default::default()
+        });
+        assert_eq!(result.bin_path, "npm");
+        assert_eq!(result.args, vec!["link", "react"]);
+    }
+}

crates/vite_install/src/commands/mod.rs

@@ -1,6 +1,7 @@
 pub mod add;
 pub mod dedupe;
 mod install;
+pub mod link;
 pub mod outdated;
 pub mod remove;
 pub mod update;

packages/cli/binding/src/cli.rs

@@ -24,6 +24,7 @@ use crate::commands::{
     fmt::{FmtConfig, fmt},
     install::InstallCommand,
     lib_cmd::lib,
+    link::LinkCommand,
     lint::{LintConfig, lint},
     outdated::OutdatedCommand,
     remove::RemoveCommand,
@@ -401,6 +402,17 @@ pub enum Commands {
         #[arg(last = true, allow_hyphen_values = true)]
         pass_through_args: Option<Vec<String>>,
     },
+    /// Link packages for local development
+    #[command(disable_help_flag = true, alias = "ln")]
+    Link {
+        /// Package name or directory to link
+        /// If empty, registers current package globally
+        package: Option<String>,
+
+        /// Arguments to pass to package manager
+        #[arg(allow_hyphen_values = true, trailing_var_arg = true)]
+        args: Vec<String>,
+    },
 }
 
 impl Commands {
@@ -414,6 +426,7 @@ impl Commands {
                 | Commands::Dedupe { .. }
                 | Commands::Outdated { .. }
                 | Commands::Why { .. }
+                | Commands::Link { .. }
         )
     }
 }
@@ -875,6 +888,10 @@ pub async fn main<
                 .await?;
             return Ok(exit_status);
         }
+        Commands::Link { package, args } => {
+            let exit_status = LinkCommand::new(cwd).execute(package.as_deref(), Some(args)).await?;
+            return Ok(exit_status);
+        }
         Commands::Why {
             packages,
             json,

packages/cli/binding/src/commands/link.rs

@@ -0,0 +1,49 @@
+use std::process::ExitStatus;
+
+use vite_install::{commands::link::LinkCommandOptions, package_manager::PackageManager};
+use vite_path::AbsolutePathBuf;
+
+use crate::Error;
+
+/// Link command for local package development.
+///
+/// This command automatically detects the package manager and translates
+/// the link command to the appropriate package manager-specific syntax.
+pub struct LinkCommand {
+    cwd: AbsolutePathBuf,
+}
+
+impl LinkCommand {
+    pub fn new(cwd: AbsolutePathBuf) -> Self {
+        Self { cwd }
+    }
+
+    pub async fn execute(
+        self,
+        package: Option<&str>,
+        pass_through_args: Option<&[String]>,
+    ) -> Result<ExitStatus, Error> {
+        // Detect package manager
+        let package_manager = PackageManager::builder(&self.cwd).build().await?;
+
+        let link_command_options = LinkCommandOptions { package, pass_through_args };
+        package_manager.run_link_command(&link_command_options, &self.cwd).await
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_link_command_new() {
+        let workspace_root = if cfg!(windows) {
+            AbsolutePathBuf::new("C:\\test".into()).unwrap()
+        } else {
+            AbsolutePathBuf::new("/test".into()).unwrap()
+        };
+
+        let cmd = LinkCommand::new(workspace_root.clone());
+        assert_eq!(cmd.cwd, workspace_root);
+    }
+}

packages/cli/binding/src/commands/mod.rs

@@ -4,6 +4,7 @@ pub(crate) mod doc;
 pub(crate) mod fmt;
 pub(crate) mod install;
 pub(crate) mod lib_cmd;
+pub(crate) mod link;
 pub(crate) mod lint;
 pub(crate) mod outdated;
 pub(crate) mod remove;