大量的更新

agent-wdd/op/PackageOperator.go

@@ -4,7 +4,12 @@ import (
 	"agent-wdd/config"
 	"agent-wdd/log"
 	"agent-wdd/utils"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
 	"strings"
+	"time"
 )
 
 type PackageOperator struct {
@@ -36,12 +41,19 @@ var (
 			"yum", "remove", "-y",
 		},
 		upgradePrefix: []string{},
+		initCommand: []string{
+			"yum", "makecache",
+		},
 	}
 )
 
-func (op *PackageOperator) Install(tools []string) {
+func (op *PackageOperator) Install(tools []string) bool {
 	// 判定本机的包管理Operator
-	generatePackageOperator()
+	ok := generatePackageOperator()
+	if !ok {
+		log.Error("PackageOperator init failed! 无法执行安装操作 %s", tools)
+		return false
+	}
 
 	// install seperately
 	for _, tool := range tools {
@@ -51,14 +63,19 @@ func (op *PackageOperator) Install(tools []string) {
 			utils.BeautifulPrint(result)
 		}
 	}
+
+	return true
 }
 
-// PackageInit 初始化包管理器, 如果本机是ubuntu或者debian, 则使用apt, 否则使用yum
-func (op *PackageOperator) PackageInit() {
+// PackageInit 初始化包管理器, 同样会判定主机是否可以联网，如果本机是ubuntu或者debian, 则使用apt, 否则使用yum
+func (op *PackageOperator) PackageInit() bool {
 	log.Info("PackageInit !")
 
 	// 判定本机的包管理Operator
-	generatePackageOperator()
+	ok := generatePackageOperator()
+	if !ok {
+		return false
+	}
 
 	// package init
 	os := config.ConfigCache.Agent.OS
@@ -72,6 +89,8 @@ func (op *PackageOperator) PackageInit() {
 		}
 	}
 
+	return true
+
 }
 
 func (op *PackageOperator) Remove(tools []string) {
@@ -89,13 +108,20 @@ func (op *PackageOperator) Remove(tools []string) {
 
 }
 
-func generatePackageOperator() {
+func generatePackageOperator() bool {
 
 	// cache return
 	if AgentPackOperator.initCommand != nil {
-		return
+		return true
 	}
-	// 检查本机是否存在Os的信息
+
+	// 检查是否可以连接互联网
+	if config.CanConnectInternet() <= 1 {
+		log.Error("服务器无法连接互联网，无法初始化包管理器")
+		return false
+	}
+
+	// 检查本机是否存在OS的信息
 	os := config.ConfigCache.Agent.OS
 	if os.Hostname == "" {
 		os.Gather()
@@ -108,4 +134,64 @@ func generatePackageOperator() {
 	} else {
 		AgentPackOperator = yumPackageOperator
 	}
+
+	return true
+}
+
+// GetLatestGithubReleaseVersion 获取GitHub仓库的最新版本
+func GetLatestGithubReleaseVersion(repoOwner, repoName string) (string, error) {
+	// 检查是否可以连接互联网
+	if config.CanConnectInternet() <= 1 {
+		return "", fmt.Errorf("服务器无法连接互联网，无法获取GitHub最新版本")
+	}
+
+	// 设置超时时间为10秒的HTTP客户端
+	client := &http.Client{
+		Timeout: 10 * time.Second,
+	}
+
+	// 构建GitHub API URL
+	apiURL := fmt.Sprintf("https://api.github.com/repos/%s/%s/releases/latest", repoOwner, repoName)
+
+	// 发送GET请求
+	resp, err := client.Get(apiURL)
+	if err != nil {
+		return "", fmt.Errorf("请求GitHub API失败: %v", err)
+	}
+	defer resp.Body.Close()
+
+	// 检查响应状态码
+	if resp.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("GitHub API返回非200状态码: %d", resp.StatusCode)
+	}
+
+	// 读取响应体
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", fmt.Errorf("读取响应内容失败: %v", err)
+	}
+
+	// 解析JSON响应
+	var release struct {
+		TagName string `json:"tag_name"`
+	}
+
+	if err := json.Unmarshal(body, &release); err != nil {
+		return "", fmt.Errorf("解析JSON响应失败: %v", err)
+	}
+
+	return release.TagName, nil
+}
+
+// CommandExists 判定命令是否存在
+func CommandExists(command string) bool {
+	ok, _ := SingleLineCommandExecutor([]string{"command", "-v", command})
+	return ok
+}
+
+func CommandExistsByPath(command string) bool {
+
+	// 查询 /usr/bin /usr/local/bin中是否有可执行文件
+	ok, _ := SingleLineCommandExecutor([]string{"find", "/usr/bin", "/usr/local/bin", "-name", command})
+	return ok
 }