From 98aa5b7de715d43a6b501e062fa6597a66cfc2bf Mon Sep 17 00:00:00 2001 From: wh_zcy Date: Fri, 12 Aug 2022 18:47:23 +0800 Subject: [PATCH] upgrade --- cmd/upgrade/build.sh | 0 cmd/upgrade/build_linux.bat | 5 ++ cmd/upgrade/main.go | 167 +++++++++++++++++++++++++++++++++++ cmd/upgrade/readme.md | 0 cmd/upgrade/tools/gz.go | 125 ++++++++++++++++++++++++++ cmd/upgrade/tools/zip.go | 54 +++++++++++ cmd/upgrade/views/index.html | 69 +++++++++++++++ 7 files changed, 420 insertions(+) create mode 100644 cmd/upgrade/build.sh create mode 100644 cmd/upgrade/build_linux.bat create mode 100644 cmd/upgrade/main.go create mode 100644 cmd/upgrade/readme.md create mode 100644 cmd/upgrade/tools/gz.go create mode 100644 cmd/upgrade/tools/zip.go create mode 100644 cmd/upgrade/views/index.html diff --git a/cmd/upgrade/build.sh b/cmd/upgrade/build.sh new file mode 100644 index 000000000..e69de29bb diff --git a/cmd/upgrade/build_linux.bat b/cmd/upgrade/build_linux.bat new file mode 100644 index 000000000..3d2bf438e --- /dev/null +++ b/cmd/upgrade/build_linux.bat @@ -0,0 +1,5 @@ +SET CGO_ENABLED=0 +SET GOOS=linux +SET GOARCH=amd64 + +go build -o update main.go \ No newline at end of file diff --git a/cmd/upgrade/main.go b/cmd/upgrade/main.go new file mode 100644 index 000000000..83e5ff798 --- /dev/null +++ b/cmd/upgrade/main.go @@ -0,0 +1,167 @@ +package main + +import ( + "embed" + "flag" + "fmt" + "go_dreamfactory/cmd/upgrade/tools" + "html/template" + "io" + "io/ioutil" + "log" + "net/http" + "os" + "path" + "strconv" + "strings" + "time" + + "github.com/gin-gonic/gin" +) + +// func main() { +// http.HandleFunc("/download", FileDownload) +// http.HandleFunc("/upload", FileUpload) + +// log.Print("http server start") +// if err := http.ListenAndServe(":8910", nil); err != nil { +// log.Fatal(err) +// } +// } + +func FileUpload(w http.ResponseWriter, r *http.Request) { + //获取文件流,第三个返回值是错误对象 + file, header, _ := r.FormFile("file") + //读取文件流为[]byte + b, _ := ioutil.ReadAll(file) + //把文件保存到指定位置 + ioutil.WriteFile("/opt/upgrade", b, 0777) + //输出上传时文件名 + fmt.Println("上传文件名:", header.Filename) +} + +func FileDownload(w http.ResponseWriter, r *http.Request) { + filename := "E:\\projects\\workspace\\go_dreamfactory\\cmd\\v2\\RobotGUI.exe" + + file, _ := os.Open(filename) + defer file.Close() + + fileHeader := make([]byte, 512) + file.Read(fileHeader) + + fileStat, _ := file.Stat() + + w.Header().Set("Content-Disposition", "attachment; filename="+filename) + w.Header().Set("Content-Type", http.DetectContentType(fileHeader)) + w.Header().Set("Content-Length", strconv.FormatInt(fileStat.Size(), 10)) + + file.Seek(0, 0) + io.Copy(w, file) + + return +} + +//go:embed views/* +var f embed.FS + +var version = "0.0.1" + +func main() { + + host := flag.String("host", "10.0.0.9", "Host") + port := flag.String("port", "8080", "Port") + uploadsDir := flag.String("uploadsDir", "./uploads", "上传文件存储地址") + wwwDir := flag.String("wwwDir", "./www", "www服务地址，即解压地址") + + flag.Parse() + + gin.SetMode(gin.ReleaseMode) + r := gin.Default() + r.Static("/prd", "./www") + tmpl := template.Must(template.New("").ParseFS(f, "views/*")) + r.SetHTMLTemplate(tmpl) + + r.GET("/version", func(c *gin.Context) { + c.JSON(http.StatusOK, gin.H{ + "version": version, + }) + }) + + r.GET("/", func(c *gin.Context) { + fmt.Println(tmpl.DefinedTemplates()) + c.HTML(http.StatusOK, "index.html", nil) + }) + + r.POST("/upload", func(c *gin.Context) { + f, err := c.FormFile("file") + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{ + "error": err.Error(), + }) + } + + //从上传的文件名获取到版本号 + s := strings.SplitN(f.Filename, "-", 2) + if len(s) != 2 { + c.JSON(http.StatusBadRequest, gin.H{ + "error": "上传的文件名中没有包含版本号 eg. update.zip-0.0.1", + }) + return + } + version = s[1] //设置版本号 + + dst := path.Join(*uploadsDir, f.Filename) + err = c.SaveUploadedFile(f, dst) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{ + "error": err.Error(), + }) + } + _, err = tools.Unzip(dst, *wwwDir) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{ + "error": err.Error(), + }) + } + + c.JSON(http.StatusOK, gin.H{ + "msg": "upload success", + "url": strings.Join([]string{"http://" + *host + ":" + *port, "/prd/", strings.Split(f.Filename, ".")[0], "/index.html"}, ""), + }) + }) + + r.GET("/dirs", func(c *gin.Context) { + dir, err := ioutil.ReadDir(*wwwDir) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{ + "error": err.Error(), + }) + } + + type Item struct { + Name string `json:"name"` + Url string `json:"url"` + DateTime string `json:"dateTime"` + } + + var dirs []Item + for _, f := range dir { + // if f.IsDir() && len(f.Name()) > 0 && f.Name() != "__MACOSX" { + dirs = append(dirs, Item{ + Name: f.Name(), + Url: strings.Join([]string{"http://" + *host + ":" + *port, "/prd/", f.Name()}, ""), + DateTime: f.ModTime().Format(time.RFC3339), + }) + // } + } + + c.JSON(http.StatusOK, gin.H{ + "dirs": dirs, + }) + }) + + err := r.Run("0.0.0.0:" + *port) + if err != nil { + log.Fatal(err) + } +} diff --git a/cmd/upgrade/readme.md b/cmd/upgrade/readme.md new file mode 100644 index 000000000..e69de29bb diff --git a/cmd/upgrade/tools/gz.go b/cmd/upgrade/tools/gz.go new file mode 100644 index 000000000..85ab1fcbc --- /dev/null +++ b/cmd/upgrade/tools/gz.go @@ -0,0 +1,125 @@ +package tools + +import ( + "archive/tar" + "compress/gzip" + "fmt" + "io" + "os" + "path/filepath" + "strings" +) + +func UnGz(tarFile, dest string) error { + srcFile, err := os.Open(tarFile) + if err != nil { + return err + } + defer srcFile.Close() + gr, err := gzip.NewReader(srcFile) + if err != nil { + return err + } + defer gr.Close() + tr := tar.NewReader(gr) + for { + hdr, err := tr.Next() + if err != nil { + if err == io.EOF { + break + } else { + return err + } + } + filename := dest + hdr.Name + file, err := createFile(filename) + if err != nil { + return err + } + io.Copy(file, tr) + } + return nil +} + +func createFile(name string) (*os.File, error) { + err := os.MkdirAll(string([]rune(name)[0:strings.LastIndex(name, "/")]), 0755) + if err != nil { + return nil, err + } + return os.Create(name) +} + +func UnGzip(tarName, xpath string) (err error) { + tarFile, err := os.Open(tarName) + if err != nil { + return err + } + defer func() { + err = tarFile.Close() + }() + + os.Mkdir(xpath, 0755) + + absPath, err := filepath.Abs(xpath) + if err != nil { + return err + } + + tr := tar.NewReader(tarFile) + if strings.HasSuffix(tarName, ".gz") || strings.HasSuffix(tarName, ".gzip") { + gz, err := gzip.NewReader(tarFile) + if err != nil { + return err + } + defer gz.Close() + tr = tar.NewReader(gz) + } + + // untar each segment + for { + hdr, err := tr.Next() + if err == io.EOF { + break + } + if err != nil { + return err + } + + // determine proper file path info + finfo := hdr.FileInfo() + fileName := hdr.Name + if filepath.IsAbs(fileName) { + fmt.Printf("removing / prefix from %s\n", fileName) + fileName, err = filepath.Rel("/", fileName) + if err != nil { + return err + } + } + absFileName := filepath.Join(absPath, fileName) + + if finfo.Mode().IsDir() { + if err := os.MkdirAll(absFileName, 0755); err != nil { + return err + } + continue + } + + // create new file with original file mode + file, err := os.OpenFile(absFileName, os.O_RDWR|os.O_CREATE|os.O_TRUNC, finfo.Mode().Perm()) + if err != nil { + return err + } + fmt.Printf("x %s\n", absFileName) + n, cpErr := io.Copy(file, tr) + if closeErr := file.Close(); closeErr != nil { // close file immediately + return err + } + if cpErr != nil { + return cpErr + } + if n != finfo.Size() { + return fmt.Errorf("unexpected bytes written: wrote %d, want %d", n, finfo.Size()) + } + } + return nil +} diff --git a/cmd/upgrade/tools/zip.go b/cmd/upgrade/tools/zip.go new file mode 100644 index 000000000..d75a518ae --- /dev/null +++ b/cmd/upgrade/tools/zip.go @@ -0,0 +1,54 @@ +package tools + +import ( + "archive/zip" + "fmt" + "io" + "os" + "path/filepath" + "strings" +) + +func Unzip(src string, dest string) ([]string, error) { + var filenames []string + r, err := zip.OpenReader(src) + if err != nil { + return filenames, err + } + defer r.Close() + + for _, f := range r.File { + // Store filename/path for returning and using later on + fpath := filepath.Join(dest, f.Name) + // Check for ZipSlip. More Info: http://bit.ly/2MsjAWE + if !strings.HasPrefix(fpath, filepath.Clean(dest)+string(os.PathSeparator)) { + return filenames, fmt.Errorf("%s: illegal file path", fpath) + } + filenames = append(filenames, fpath) + if f.FileInfo().IsDir() { + // Make Folder + os.MkdirAll(fpath, os.ModePerm) + continue + } + // Make File + if err = os.MkdirAll(filepath.Dir(fpath), os.ModePerm); err != nil { + return filenames, err + } + outFile, err := os.OpenFile(fpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode()) + if err != nil { + return filenames, err + } + rc, err := f.Open() + if err != nil { + return filenames, err + } + _, err = io.Copy(outFile, rc) + // Close the file without defer to close before next iteration of loop + outFile.Close() + rc.Close() + if err != nil { + return filenames, err + } + } + return filenames, nil +} diff --git a/cmd/upgrade/views/index.html b/cmd/upgrade/views/index.html new file mode 100644 index 000000000..d8181e590 --- /dev/null +++ b/cmd/upgrade/views/index.html @@ -0,0 +1,69 @@ + + + + + + 上传 + + + + + + + + + + + +
+ +

上传原型，获取永久在线浏览地址

+ + +
+ +

点击或者拖拽上传

+
+ + + + 浏览地址：${ url } + + + + + 历史prd: +

+ + + ${ dir.name }    ${ dir.dateTime} + + +
+ + + + \ No newline at end of file