feat: 新增 pkg.install/remove/search/info 包管理命令

2026-11-03 22:05:00 +08:00
parent d05096e29a
commit 0e89ed3248
5 changed files with 572 additions and 56 deletions
@@ -32,62 +32,72 @@ type BuiltinRequest struct {
 type Builtin func(ctx context.Context, req BuiltinRequest) error

 var builtins = map[string]Builtin{
-	"archive.extract":    runArchiveExtract,
-	"disk.usage":         runDiskUsage,
-	"diagnose":           runDiagnose,
-	"docker.df":          runDockerDF,
-	"docker.exec":        runDockerExec,
-	"docker.load":        runDockerLoad,
-	"docker.logs":        runDockerLogs,
-	"docker.restart":     runDockerRestart,
-	"docker.rm":          runDockerRM,
+	"archive.create":   runArchiveCreate,
+	"archive.extract":  runArchiveExtract,
+	"archive.list":     runArchiveList,
+	"disk.check":       runDiskCheck,
+	"disk.inode":       runDiskInode,
+	"disk.usage":       runDiskUsage,
+	"diagnose":         runDiagnose,
+	"docker.df":        runDockerDF,
+	"docker.exec":      runDockerExec,
+	"docker.load":      runDockerLoad,
+	"docker.logs":      runDockerLogs,
+	"docker.restart":   runDockerRestart,
+	"docker.rm":        runDockerRM,
 	"docker.rm.preview":  runDockerPreviewRM,
-	"docker.rmi":         runDockerRMI,
+	"docker.rmi":       runDockerRMI,
 	"docker.rmi.preview": runDockerPreviewRMI,
-	"docker.save":        runDockerSave,
-	"docker.stats":       runDockerStats,
-	"docker.stop":        runDockerStop,
-	"file.empty":         runFileEmpty,
-	"file.findbig":       runFileFindBig,
-	"file.findname":      runFileFindName,
-	"file.findrecent":    runFileFindRecent,
-	"fw.allow":           runFwAllow,
-	"fw.deny":            runFwDeny,
-	"fw.status":          runFwStatus,
-	"git.retime":         runGitRetime,
+	"docker.save":      runDockerSave,
+	"docker.stats":     runDockerStats,
+	"docker.stop":      runDockerStop,
+	"file.empty":       runFileEmpty,
+	"file.findbig":     runFileFindBig,
+	"file.findname":    runFileFindName,
+	"file.findrecent":  runFileFindRecent,
+	"fw.allow":         runFwAllow,
+	"fw.deny":          runFwDeny,
+	"fw.status":        runFwStatus,
+	"git.retime":       runGitRetime,
 	"git.retime.preview": runGitRetimePreview,
-	"java.run":           runJavaRun,
-	"java.run.preview":   runJavaPreview,
-	"log.last":           runLogLast,
-	"log.list":           runLogList,
-	"log.search":         runLogSearch,
-	"net.conn":           runNetConn,
-	"net.curl":           runNetCurl,
-	"net.dns":            runNetDNS,
-	"net.iface":          runNetIface,
-	"net.ping":           runNetPing,
-	"net.pids":           runNetPIDs,
-	"net.ports":          runNetPorts,
-	"net.speed":          runNetSpeed,
-	"net.trace":          runNetTrace,
-	"pkg.list":           runPkgList,
-	"proc.killname":      runProcKillName,
-	"proc.list":          runProcList,
-	"proc.nohup":         runNohup,
-	"proc.topcpu":        runProcTopCPU,
-	"proc.topmem":        runProcTopMem,
-	"proc.tree":          runProcTree,
-	"proc.wait":          runProcWait,
-	"svc.disable":        runSvcDisable,
-	"svc.enable":         runSvcEnable,
-	"svc.list":           runSvcList,
-	"svc.logs":           runSvcLogs,
-	"svc.status":         runSvcStatus,
-	"sysinfo":            runSysInfo,
-	"user.grant":         runUserGrant,
-	"user.info":          runUserInfo,
-	"user.list":          runUserList,
-	"user.revoke":        runUserRevoke,
+	"git.time":         runGitTime,
+	"java.run":         runJavaRun,
+	"java.run.preview": runJavaPreview,
+	"log.last":         runLogLast,
+	"log.list":         runLogList,
+	"log.search":       runLogSearch,
+	"net.conn":         runNetConn,
+	"net.curl":         runNetCurl,
+	"net.dns":          runNetDNS,
+	"net.http":         runNetHTTP,
+	"net.iface":        runNetIface,
+	"net.ping":         runNetPing,
+	"net.pids":         runNetPIDs,
+	"net.ports":        runNetPorts,
+	"net.speed":        runNetSpeed,
+	"net.trace":        runNetTrace,
+	"pkg.info":         runPkgInfo,
+	"pkg.install":      runPkgInstall,
+	"pkg.list":         runPkgList,
+	"pkg.remove":       runPkgRemove,
+	"pkg.search":       runPkgSearch,
+	"proc.killname":    runProcKillName,
+	"proc.list":        runProcList,
+	"proc.nohup":       runNohup,
+	"proc.topcpu":      runProcTopCPU,
+	"proc.topmem":      runProcTopMem,
+	"proc.tree":        runProcTree,
+	"proc.wait":        runProcWait,
+	"svc.disable":      runSvcDisable,
+	"svc.enable":       runSvcEnable,
+	"svc.list":         runSvcList,
+	"svc.logs":         runSvcLogs,
+	"svc.status":       runSvcStatus,
+	"sysinfo":          runSysInfo,
+	"user.grant":       runUserGrant,
+	"user.info":        runUserInfo,
+	"user.list":        runUserList,
+	"user.revoke":      runUserRevoke,
 }

 // runDiagnose bridges the registry builtin to the diagnose package.
@@ -0,0 +1,288 @@
+package executor
+
+import (
+	"context"
+	"fmt"
+	"os/exec"
+	"runtime"
+	"strings"
+
+	"gitea.odboy.cn/odboy/kenaito-kopsc/internal/platform"
+)
+
+// runPkgInstall implements `kopsc pkg install <package>`: install a package.
+func runPkgInstall(ctx context.Context, req BuiltinRequest) error {
+	opts, err := parsePkgManageArgs(req.Args, "install")
+	if err != nil {
+		return err
+	}
+	if len(opts.packages) == 0 {
+		return fmt.Errorf("缺少包名；用法: kopsc pkg install <包名> [包名...]")
+	}
+
+	switch runtime.GOOS {
+	case "linux":
+		return pkgInstallLinux(ctx, req, opts.packages)
+	case "darwin":
+		args := append([]string{"install"}, opts.packages...)
+		cmd := exec.CommandContext(ctx, "brew", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	case "windows":
+		args := append([]string{"install"}, opts.packages...)
+		cmd := exec.CommandContext(ctx, "winget", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	default:
+		return fmt.Errorf("不支持的操作系统: %s", runtime.GOOS)
+	}
+}
+
+func pkgInstallLinux(ctx context.Context, req BuiltinRequest, packages []string) error {
+	info := platform.System().Detect()
+
+	// Debian/Ubuntu: apt-get install
+	if distroLikeContains(info, "debian", "ubuntu") {
+		args := append([]string{"install"}, packages...)
+		cmd := exec.CommandContext(ctx, "apt-get", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	// CentOS/RHEL/Fedora: dnf install
+	if distroLikeContains(info, "rhel", "fedora") || info.Distro == "centos" {
+		args := append([]string{"install"}, packages...)
+		cmd := exec.CommandContext(ctx, "dnf", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	// Alpine: apk add
+	if info.Distro == "alpine" {
+		args := append([]string{"add"}, packages...)
+		cmd := exec.CommandContext(ctx, "apk", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	return fmt.Errorf("不支持的 Linux 发行版: %s", info.Distro)
+}
+
+// runPkgRemove implements `kopsc pkg remove <package>`: remove a package.
+func runPkgRemove(ctx context.Context, req BuiltinRequest) error {
+	opts, err := parsePkgManageArgs(req.Args, "remove")
+	if err != nil {
+		return err
+	}
+	if len(opts.packages) == 0 {
+		return fmt.Errorf("缺少包名；用法: kopsc pkg remove <包名> [包名...]")
+	}
+
+	switch runtime.GOOS {
+	case "linux":
+		return pkgRemoveLinux(ctx, req, opts.packages)
+	case "darwin":
+		args := append([]string{"uninstall"}, opts.packages...)
+		cmd := exec.CommandContext(ctx, "brew", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	case "windows":
+		args := append([]string{"uninstall"}, opts.packages...)
+		cmd := exec.CommandContext(ctx, "winget", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	default:
+		return fmt.Errorf("不支持的操作系统: %s", runtime.GOOS)
+	}
+}
+
+func pkgRemoveLinux(ctx context.Context, req BuiltinRequest, packages []string) error {
+	info := platform.System().Detect()
+
+	// Debian/Ubuntu: apt-get remove
+	if distroLikeContains(info, "debian", "ubuntu") {
+		args := append([]string{"remove"}, packages...)
+		cmd := exec.CommandContext(ctx, "apt-get", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	// CentOS/RHEL/Fedora: dnf remove
+	if distroLikeContains(info, "rhel", "fedora") || info.Distro == "centos" {
+		args := append([]string{"remove"}, packages...)
+		cmd := exec.CommandContext(ctx, "dnf", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	// Alpine: apk del
+	if info.Distro == "alpine" {
+		args := append([]string{"del"}, packages...)
+		cmd := exec.CommandContext(ctx, "apk", args...)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	return fmt.Errorf("不支持的 Linux 发行版: %s", info.Distro)
+}
+
+// runPkgSearch implements `kopsc pkg search <keyword>`: search for packages.
+func runPkgSearch(ctx context.Context, req BuiltinRequest) error {
+	opts, err := parsePkgManageArgs(req.Args, "search")
+	if err != nil {
+		return err
+	}
+	if len(opts.packages) == 0 {
+		return fmt.Errorf("缺少搜索关键字；用法: kopsc pkg search <关键字>")
+	}
+	keyword := opts.packages[0]
+
+	switch runtime.GOOS {
+	case "linux":
+		return pkgSearchLinux(ctx, req, keyword)
+	case "darwin":
+		cmd := exec.CommandContext(ctx, "brew", "search", keyword)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	case "windows":
+		cmd := exec.CommandContext(ctx, "winget", "search", keyword)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	default:
+		return fmt.Errorf("不支持的操作系统: %s", runtime.GOOS)
+	}
+}
+
+func pkgSearchLinux(ctx context.Context, req BuiltinRequest, keyword string) error {
+	info := platform.System().Detect()
+
+	// Debian/Ubuntu: apt-cache search
+	if distroLikeContains(info, "debian", "ubuntu") {
+		cmd := exec.CommandContext(ctx, "apt-cache", "search", keyword)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	// CentOS/RHEL/Fedora: dnf search
+	if distroLikeContains(info, "rhel", "fedora") || info.Distro == "centos" {
+		cmd := exec.CommandContext(ctx, "dnf", "search", keyword)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	// Alpine: apk search
+	if info.Distro == "alpine" {
+		cmd := exec.CommandContext(ctx, "apk", "search", keyword)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	return fmt.Errorf("不支持的 Linux 发行版: %s", info.Distro)
+}
+
+// runPkgInfo implements `kopsc pkg info <package>`: show package details.
+func runPkgInfo(ctx context.Context, req BuiltinRequest) error {
+	opts, err := parsePkgManageArgs(req.Args, "info")
+	if err != nil {
+		return err
+	}
+	if len(opts.packages) == 0 {
+		return fmt.Errorf("缺少包名；用法: kopsc pkg info <包名>")
+	}
+	pkg := opts.packages[0]
+
+	switch runtime.GOOS {
+	case "linux":
+		return pkgInfoLinux(ctx, req, pkg)
+	case "darwin":
+		cmd := exec.CommandContext(ctx, "brew", "info", pkg)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	case "windows":
+		cmd := exec.CommandContext(ctx, "winget", "show", pkg)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	default:
+		return fmt.Errorf("不支持的操作系统: %s", runtime.GOOS)
+	}
+}
+
+func pkgInfoLinux(ctx context.Context, req BuiltinRequest, pkg string) error {
+	info := platform.System().Detect()
+
+	// Debian/Ubuntu: apt show (fallback to dpkg -s)
+	if distroLikeContains(info, "debian", "ubuntu") {
+		cmd := exec.CommandContext(ctx, "apt", "show", pkg)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		if err := cmd.Run(); err != nil {
+			// Fallback to dpkg -s
+			cmd = exec.CommandContext(ctx, "dpkg", "-s", pkg)
+			cmd.Stdout = req.Stdout
+			cmd.Stderr = req.Stderr
+			return cmd.Run()
+		}
+		return nil
+	}
+
+	// CentOS/RHEL/Fedora: rpm -qi
+	if distroLikeContains(info, "rhel", "fedora") || info.Distro == "centos" {
+		cmd := exec.CommandContext(ctx, "rpm", "-qi", pkg)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	// Alpine: apk info
+	if info.Distro == "alpine" {
+		cmd := exec.CommandContext(ctx, "apk", "info", "-a", pkg)
+		cmd.Stdout = req.Stdout
+		cmd.Stderr = req.Stderr
+		return cmd.Run()
+	}
+
+	return fmt.Errorf("不支持的 Linux 发行版: %s", info.Distro)
+}
+
+type pkgManageOptions struct {
+	packages []string
+}
+
+func parsePkgManageArgs(args []string, action string) (pkgManageOptions, error) {
+	var opts pkgManageOptions
+	var positional []string
+
+	for i := 0; i < len(args); i++ {
+		arg := args[i]
+		switch {
+		case arg == "--":
+			positional = append(positional, args[i+1:]...)
+			i = len(args)
+		case strings.HasPrefix(arg, "-") && arg != "-":
+			// Ignore unknown flags, pass through
+			positional = append(positional, arg)
+		default:
+			positional = append(positional, arg)
+		}
+	}
+
+	opts.packages = positional
+	return opts, nil
+}