初始化项目

2025-03-27 16:09:20 +08:00
parent e09a32d1e8
commit fc2d585489
709 changed files with 516391 additions and 0 deletions
--- a/agent-wdd/cmd/Zsh.go
+++ b/agent-wdd/cmd/Zsh.go
@@ -0,0 +1,152 @@
+package cmd
+
+import (
+	"agent-wdd/config"
+	"agent-wdd/log"
+	"agent-wdd/op"
+	"agent-wdd/utils"
+
+	"github.com/spf13/cobra"
+)
+
+func addZshSubcommands(cmd *cobra.Command) {
+	cmd.Run = func(cmd *cobra.Command, args []string) {
+		log.Info("安装zsh")
+
+		// 初始化安装命令
+		utils.CreateFolder("/root/wdd")
+
+		if config.CanConnectInternet() <= 1 {
+			log.Error("无法连接互联网，无法安装zsh")
+			return
+		}
+
+		// 判定config
+		if config.ConfigCache.Agent.Network.Public.IPv4 == "" {
+			network := config.ConfigCache.Agent.Network
+			network.Gather()
+			network.SaveConfig()
+		}
+
+		// 下载并安装zsh和git
+		ok := op.AgentPackOperator.Install([]string{"git", "zsh"})
+		if !ok {
+			log.Error("安装git和zsh失败, 无法继续后面的的安装！")
+			return
+		}
+
+		// 清理残留
+		utils.RemoveFolderComplete("/root/.oh-my-zsh")
+		utils.RemoveFolderComplete("/root/wdd/zsh-install.sh")
+
+		ohMyZshInstallUrl := ""
+
+		if config.ConfigCache.Agent.Network.Internet == 7 {
+			ohMyZshInstallUrl = "https://gitea.107421.xyz/zeaslity/ohmyzsh/raw/branch/master/tools/install.sh"
+		} else {
+			ohMyZshInstallUrl = "https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh"
+		}
+
+		downloadOk, resultLog := utils.DownloadFile(ohMyZshInstallUrl, "/root/wdd/zsh-install.sh")
+		if !downloadOk {
+			log.Error("下载zsh安装脚本失败: %v", resultLog)
+			return
+		}
+
+		op.SingleLineCommandExecutor([]string{
+			"chmod",
+			"+x",
+			"/root/wdd/zsh-install.sh",
+		})
+
+		installZshCommand := []string{
+			"sh",
+			"-c",
+			"/root/wdd/zsh-install.sh",
+		}
+
+		if config.ConfigCache.Agent.Network.Internet == 7 {
+			op.SingleLineCommandExecutor([]string{
+				"export",
+				"REMOTE=https://gitea.107421.xyz/zeaslity/ohmyzsh.git",
+			})
+		}
+
+		// 执行 oh-my-zsh 安装命令
+		op.RealTimeCommandExecutor(installZshCommand)
+
+		log.Info("安装zsh完成, 开始安装插件!")
+
+		pluginsHardCodeUrl := [][]string{
+			{
+				"https://gitee.com/wangl-cc/zsh-autosuggestions.git",
+				"/root/.oh-my-zsh/plugins/zsh-autosuggestions",
+			},
+			{
+				"https://gitee.com/xiaoqqya/zsh-syntax-highlighting.git",
+				"/root/.oh-my-zsh/plugins/zsh-syntax-highlighting",
+			},
+		}
+
+		pluginsHardCodeOutsideUrl := [][]string{
+			{
+				"https://github.com/zsh-users/zsh-autosuggestions",
+				"/root/.oh-my-zsh/plugins/zsh-autosuggestions",
+			},
+			{
+				"https://github.com/zsh-users/zsh-syntax-highlighting.git",
+				"/root/.oh-my-zsh/plugins/zsh-syntax-highlighting",
+			},
+		}
+
+		pluginsListUrl := []string{
+			"https://gitea.107421.xyz/zeaslity/ohmyzsh/src/branch/master/plugins/command-not-found/command-not-found.plugin.zsh",
+			"https://gitea.107421.xyz/zeaslity/ohmyzsh/src/branch/master/plugins/autojump/autojump.plugin.zsh",
+			"https://gitea.107421.xyz/zeaslity/ohmyzsh/src/branch/master/plugins/themes/themes.plugin.zsh",
+		}
+
+		// 下载插件
+		if config.ConfigCache.Agent.Network.Internet == 7 {
+			// 执行硬编码命令, 安装插件
+			for _, command := range pluginsHardCodeUrl {
+				op.SingleLineCommandExecutor([]string{"git", "clone", command[0], command[1]})
+			}
+
+		} else {
+			// 执行硬编码命令, 安装插件 国外
+			for _, command := range pluginsHardCodeOutsideUrl {
+				op.SingleLineCommandExecutor([]string{"git", "clone", command[0], command[1]})
+			}
+		}
+		// 下载插件
+		for _, url := range pluginsListUrl {
+			downloadOk, resultLog := utils.DownloadFile(url, "/root/.oh-my-zsh/plugins/")
+			if !downloadOk {
+				log.Error("下载插件失败: %v", resultLog)
+			}
+		}
+
+		// 修改ZSH配置
+		utils.FindAndReplaceContentInFile("robbyrussell", "agnoster", "/root/.zshrc")
+		utils.FindAndReplaceContentInFile("# DISABLE_AUTO_UPDATE=\"true\"", "DISABLE_AUTO_UPDATE=\"true\"", "/root/.zshrc")
+		utils.FindAndReplaceContentInFile("plugins=(git)", "plugins=(git zsh-autosuggestions zsh-syntax-highlighting command-not-found z themes)", "/root/.zshrc")
+
+		//
+		op.SingleLineCommandExecutor([]string{
+			"source",
+			"/root/.zshrc",
+		})
+
+		op.SingleLineCommandExecutor([]string{
+			"chsh",
+			"-s",
+			"/bin/zsh",
+		})
+
+		op.SingleLineCommandExecutor([]string{
+			"zsh",
+		})
+
+		log.Info("Zsh安装完成")
+	}
+}