feat: windows agent update (#790)

agent/update/update_windows.go

@@ -1,9 +1,141 @@
 package update
 
 import (
+	"archive/zip"
+	"bufio"
 	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
 )
 
+func download(file string) (string, error) {
+	fname := filepath.Base(file)
+	log.Println("downloading file:", file, "save to:", fname)
+	res, err := http.Get(file)
+	if err != nil {
+		return fname, fmt.Errorf("cannot download file from %s", file)
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode != 200 {
+		return fname, fmt.Errorf("download  %s error response: %s", file, res.Status)
+	}
+	f, err := os.OpenFile(fname, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0755)
+	if err != nil {
+		return fname, err
+	}
+	defer f.Close()
+	bufWriter := bufio.NewWriter(f)
+
+	_, err = io.Copy(bufWriter, res.Body)
+	if err != nil {
+		return fname, err
+	}
+	bufWriter.Flush()
+	return fname, nil
+}
+
 func Update(tar string) error {
-	return fmt.Errorf("linux support only")
+	// download
+	fname, err := download(tar)
+	if err != nil {
+		return err
+	}
+	nv, err := UnTar("./", fname)
+	if err != nil {
+		return err
+	}
+
+	// old version
+	ov, err := os.Executable()
+	if err != nil {
+		return err
+	}
+	fm, err := os.Stat(ov)
+	if err != nil {
+		return err
+	}
+	fi, err := os.Stat(nv)
+	if err != nil {
+		return err
+	}
+	if fi.Mode().IsDir() {
+		return fmt.Errorf("%s is directory", nv)
+	}
+	log.Printf("I! replace old version:%s with new version:%s", ov, "./"+nv)
+
+	// replace
+	err = os.Rename(nv, ov)
+	if err != nil {
+		return err
+	}
+	err = os.RemoveAll("./" + filepath.Dir(nv))
+	if err != nil {
+		log.Println("E! clean dir:", "./"+filepath.Dir(nv), "error:", err)
+	} else {
+		log.Println("I! clean dir:", "./"+filepath.Dir(nv), "success")
+	}
+	err = os.Remove("./" + fname)
+	if err != nil {
+		log.Println("E! clean file:", "./"+fname, "error:", err)
+	} else {
+		log.Println("I! clean file:", "./"+fname, "success")
+	}
+	return os.Chmod(ov, fm.Mode().Perm())
+}
+
+func UnTar(dst, src string) (target string, err error) {
+	fr, err := zip.OpenReader(src)
+	if err != nil {
+		return
+	}
+	defer fr.Close()
+
+	err = os.MkdirAll(dst, os.ModePerm)
+	if err != nil {
+		return "", err
+	}
+
+	// 遍历 ZIP 文件中的每个文件
+	for _, file := range fr.File {
+		// 构建文件解压后的路径
+		destPath := filepath.Join(dst, file.Name)
+
+		// 如果是目录，创建相应的目录
+		if file.FileInfo().IsDir() {
+			err := os.MkdirAll(destPath, os.ModePerm)
+			if err != nil {
+				return "", err
+			}
+		}
+
+		// 打开 ZIP 文件中的每个文件
+		srcFile, err := file.Open()
+		if err != nil {
+			return "", err
+		}
+		defer srcFile.Close()
+
+		// 创建目标文件
+		dest, err := os.Create(destPath)
+		if err != nil {
+			return "", err
+		}
+		defer dest.Close()
+
+		// 将 ZIP 文件中的内容复制到目标文件
+		_, err = io.Copy(dest, srcFile)
+		if err != nil {
+			return "", err
+		}
+		if strings.HasSuffix(destPath, "categraf") {
+			target = destPath
+		}
+	}
+
+	return target, nil
 }