mirror of
https://github.com/ayflying/p2p.git
synced 2026-03-04 17:29:22 +00:00
110 lines
3.4 KiB
Go
110 lines
3.4 KiB
Go
package cmd
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"fmt"
|
||
"path"
|
||
"runtime"
|
||
|
||
systemV1 "github.com/ayflying/p2p/api/system/v1"
|
||
"github.com/ayflying/p2p/internal/service"
|
||
"github.com/gogf/gf/v2/crypto/gsha1"
|
||
"github.com/gogf/gf/v2/encoding/gcompress"
|
||
"github.com/gogf/gf/v2/encoding/gjson"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/os/gcfg"
|
||
"github.com/gogf/gf/v2/os/gcmd"
|
||
"github.com/gogf/gf/v2/os/gfile"
|
||
)
|
||
|
||
var (
|
||
Update = gcmd.Command{
|
||
Name: "update",
|
||
Usage: "update",
|
||
Brief: "更新版本",
|
||
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
|
||
g.Log().Info(ctx, "准备上传更新文件")
|
||
//加载编辑配置文件
|
||
g.Cfg("hack").GetAdapter().(*gcfg.AdapterFile).SetFileName("hack/config.yaml")
|
||
//获取文件名
|
||
getName, err := g.Cfg("hack").Get(ctx, "gfcli.build.name")
|
||
name := getName.String()
|
||
|
||
getPath, err := g.Cfg("hack").Get(ctx, "gfcli.build.path")
|
||
pathMain := getPath.String()
|
||
|
||
//获取版本号
|
||
getVersion, err := g.Cfg("hack").Get(ctx, "gfcli.build.version")
|
||
version := getVersion.String()
|
||
|
||
// 拼接操作系统和架构（格式：OS_ARCH）
|
||
platform := fmt.Sprintf("%s_%s", runtime.GOOS, runtime.GOARCH)
|
||
|
||
rootDir := "server_update"
|
||
|
||
var versionFile = make(map[string]string)
|
||
var filePath = path.Join(pathMain, version, platform, name)
|
||
dirList, _ := gfile.ScanDir(path.Join(pathMain, version), "*", false)
|
||
for _, v := range dirList {
|
||
updatePlatform := gfile.Name(v)
|
||
updateFilePath := path.Join(rootDir, name, version, updatePlatform)
|
||
|
||
var obj bytes.Buffer
|
||
g.Log().Debugf(ctx, "读取目录成功:%v", v)
|
||
fileMian := path.Join(v, name)
|
||
g.Log().Debugf(ctx, "判断当前文件是否存在：%v", fileMian)
|
||
if gfile.IsFile(fileMian) {
|
||
// 写入文件哈希
|
||
versionFile[updatePlatform] = gsha1.MustEncryptFile(fileMian)
|
||
err = gcompress.GzipPathWriter(fileMian, &obj)
|
||
service.S3().PutObject(ctx, &obj, updateFilePath+".gz")
|
||
g.Log().Debugf(ctx, "成功上传文件到：%v", updateFilePath+".gz")
|
||
}
|
||
if gfile.IsFile(fileMian + ".exe") {
|
||
// 写入文件哈希
|
||
versionFile[updatePlatform] = gsha1.MustEncryptFile(fileMian + ".exe")
|
||
err = gcompress.GzipPathWriter(fileMian+".exe", &obj)
|
||
service.S3().PutObject(ctx, &obj, updateFilePath+".gz")
|
||
g.Log().Debugf(ctx, "成功上传文件到：%v", updateFilePath+".gz")
|
||
}
|
||
|
||
// 写入文件版本文件
|
||
fileByte := gjson.MustEncode(versionFile)
|
||
service.S3().PutObject(ctx, bytes.NewReader(fileByte), path.Join(rootDir, name, "version.json"))
|
||
if err != nil {
|
||
g.Log().Error(ctx, err)
|
||
}
|
||
}
|
||
g.Log().Debugf(ctx, "当前获取到的地址为：%v", filePath)
|
||
|
||
versionUrl := service.S3().GetCdnUrl(path.Join(rootDir, name))
|
||
listVar := g.Cfg().MustGet(ctx, "p2p.list")
|
||
var p2pItem []struct {
|
||
Host string `json:"host"`
|
||
Port int `json:"port"`
|
||
SSL bool `json:"ssl"`
|
||
Ws string `json:"ws"`
|
||
}
|
||
listVar.Scan(&p2pItem)
|
||
for _, v := range p2pItem {
|
||
|
||
url := "http"
|
||
if v.SSL == true {
|
||
url = "https"
|
||
}
|
||
url = fmt.Sprintf("%s://%s:%d/system/update", url, v.Host, v.Port)
|
||
|
||
g.Log().Debugf(ctx, "开始上传到服务器：%v,file=%v", url, versionUrl)
|
||
_, err := g.Client().Get(ctx, url, systemV1.UpdateReq{
|
||
Url: versionUrl,
|
||
Version: version,
|
||
})
|
||
if err != nil {
|
||
g.Log().Error(ctx, err)
|
||
}
|
||
}
|
||
return
|
||
}}
|
||
)
|