feat: ✨ add use command

Author: unreadcode

src/cmd/use.go

@@ -0,0 +1,54 @@
+//Package cmd
+/*
+Copyright © 2024 UnreadCode <i@unreadcode.com>
+
+*/
+
+package cmd
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+	"pvm/utils"
+	"regexp"
+)
+
+// useCmd represents the use command
+var useCmd = &cobra.Command{
+	Use:   "use",
+	Short: "Switch to the specified PHP version",
+	Long: `Quickly switch between different PHP versions.
+usage example:
+	pvm use 8.0
+`,
+
+	Run: useRun,
+}
+
+func init() {
+	rootCmd.AddCommand(useCmd)
+}
+
+func useRun(cmd *cobra.Command, args []string) {
+	if len(args) == 0 {
+		utils.PrintMsg("Please specify a PHP version", "Warning", 1)
+	}
+	version := args[0]
+	// 是一个有效的版本号
+	if !regexp.MustCompile(`^\d+\.\d+$`).MatchString(version) {
+		utils.PrintMsg("Invalid PHP version number.", "Error", 1)
+	}
+	// 是否已经在使用
+	if currentVersion := utils.GetCurrentPhpVersion(); currentVersion == version {
+		utils.PrintMsg(fmt.Sprintf("Already using PHP v%s", version), "Info", 0)
+	}
+	// 是否已经安装
+	if !utils.IsInstalled(version) {
+		utils.PrintMsg(fmt.Sprintf("PHP v%s is not installed.", version), "Error", 1)
+	}
+	// 切换到指定版本
+	if err := utils.SwitchToVersion(version); err != nil {
+		utils.PrintMsg(err.Error(), "Error", 1)
+	}
+	utils.PrintMsg(fmt.Sprintf("Switched to PHP v%s", version), "Success", 0)
+}

src/utils/config.go

@@ -10,6 +10,8 @@ const (
 
 	PVM_ROOT = "PVM_ROOT"
 
+	PHP_PATH = "PHP_PATH"
+
 	RELEASES = "https://windows.php.net/downloads/releases/releases.json"
 
 	DOWNLOAD = "https://windows.php.net/downloads/releases/"

src/utils/utils.go

@@ -30,6 +30,7 @@ type Zip struct {
 }
 
 var PvmRoot = os.Getenv(PVM_ROOT)
+var PhpPath = os.Getenv(PHP_PATH)
 
 var MsgTypeMap = map[string]color.Attribute{
 	"Error":   color.FgRed,
@@ -314,6 +315,38 @@ func UninstallPhpVersion(version string) error {
 	return nil
 }
 
+// SwitchToVersion 切换到指定版本
+func SwitchToVersion(version string) error {
+	targetDir := filepath.Join(PvmRoot, fmt.Sprintf("v%s", version))
+	if hasPermission() {
+		os.Remove(PhpPath)
+		if err := os.Symlink(targetDir, PhpPath); err != nil {
+			return err
+		}
+		return nil
+	} else {
+		uacRun("pvm", "use", version)
+	}
+	return nil
+}
+
+// 是否具有管理员权限
+func hasPermission() bool {
+	if _, err := os.Open("\\\\.\\PHYSICALDRIVE0"); err != nil {
+		return false
+	}
+	return true
+}
+
+// 请求管理员权限运行程序
+func uacRun(command ...string) {
+	//This executable file comes from https://jianqiezhushou.com/
+	//Original author https://jpassing.com/2007/12/08/launch-elevated-processes-from-the-command-line/
+	cmd := exec.Command("elevate.exe", "-k")
+	cmd.Args = append(cmd.Args, command...)
+	cmd.Run()
+}
+
 // 下载composer installer
 func downloadComposer(installer string) error {
 	resp, err := http.Get(COMPOSER)