mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-01-23 22:42:15 +08:00
430 lines
11 KiB
Go
430 lines
11 KiB
Go
// Copyright (c) 2020 tickstep.
|
||
//
|
||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
// Unless required by applicable law or agreed to in writing, software
|
||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
// See the License for the specific language governing permissions and
|
||
// limitations under the License.
|
||
package panupdate
|
||
|
||
import (
|
||
"archive/zip"
|
||
"bytes"
|
||
"fmt"
|
||
jsoniter "github.com/json-iterator/go"
|
||
"github.com/tickstep/aliyunpan/cmder/cmdliner"
|
||
"github.com/tickstep/aliyunpan/cmder/cmdutil"
|
||
"github.com/tickstep/aliyunpan/internal/config"
|
||
"github.com/tickstep/aliyunpan/internal/utils"
|
||
"github.com/tickstep/aliyunpan/library/requester/transfer"
|
||
"github.com/tickstep/library-go/cachepool"
|
||
"github.com/tickstep/library-go/checkaccess"
|
||
"github.com/tickstep/library-go/converter"
|
||
"github.com/tickstep/library-go/getip"
|
||
"github.com/tickstep/library-go/jsonhelper"
|
||
"github.com/tickstep/library-go/logger"
|
||
"github.com/tickstep/library-go/requester"
|
||
"io/ioutil"
|
||
"net/http"
|
||
"path/filepath"
|
||
"regexp"
|
||
"runtime"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
const (
|
||
ReleaseName = "aliyunpan"
|
||
)
|
||
|
||
type info struct {
|
||
filename string
|
||
size int64
|
||
downloadURL string
|
||
}
|
||
|
||
type tsResp struct {
|
||
Code int `json:"code"`
|
||
Data interface{} `json:"data"`
|
||
Msg string `json:"msg"`
|
||
}
|
||
|
||
func getReleaseFromTicstep(client *requester.HTTPClient, showPrompt bool) *ReleaseInfo {
|
||
tsReleaseInfo := &ReleaseInfo{}
|
||
tsRespObj := &tsResp{Data: tsReleaseInfo}
|
||
fullUrl := strings.Builder{}
|
||
ipAddr, err := getip.IPInfoFromTechainBaidu()
|
||
if err != nil {
|
||
ipAddr = "127.0.0.1"
|
||
}
|
||
fmt.Fprintf(&fullUrl, "http://api.tickstep.com/update/tickstep/aliyunpan/releases/latest?ip=%s&os=%s&arch=%s&version=%s",
|
||
ipAddr, runtime.GOOS, runtime.GOARCH, config.AppVersion)
|
||
resp, err := client.Req(http.MethodGet, fullUrl.String(), nil, nil)
|
||
if resp != nil {
|
||
defer resp.Body.Close()
|
||
}
|
||
if err != nil {
|
||
if showPrompt {
|
||
logger.Verbosef("获取数据错误: %s\n", err)
|
||
}
|
||
return nil
|
||
}
|
||
respBytes, _ := ioutil.ReadAll(resp.Body)
|
||
|
||
// 解析响应返回
|
||
tsRespEntity := &tsResp{}
|
||
if err = jsoniter.Unmarshal(respBytes, tsRespEntity); err == nil {
|
||
if tsRespEntity.Code != 0 {
|
||
if showPrompt {
|
||
fmt.Printf("错误: %s\n", tsRespEntity.Msg)
|
||
}
|
||
logger.Verboseln(string(respBytes))
|
||
return nil
|
||
}
|
||
}
|
||
|
||
// 解析可用版本返回
|
||
err = jsoniter.Unmarshal(respBytes, tsRespObj)
|
||
if err != nil {
|
||
if showPrompt {
|
||
fmt.Printf("json数据解析失败: %s\n", err)
|
||
}
|
||
return nil
|
||
}
|
||
if tsRespObj.Code == 0 {
|
||
if tsReleaseInfo != nil && len(tsReleaseInfo.Assets) > 0 {
|
||
return tsReleaseInfo
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func getReleaseFromGithub(client *requester.HTTPClient, showPrompt bool) *ReleaseInfo {
|
||
resp, err := client.Req(http.MethodGet, "https://api.github.com/repos/tickstep/aliyunpan/releases/latest", nil, nil)
|
||
if resp != nil {
|
||
defer resp.Body.Close()
|
||
}
|
||
if err != nil {
|
||
if showPrompt {
|
||
logger.Verbosef("获取数据错误: %s\n", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
releaseInfo := ReleaseInfo{}
|
||
err = jsonhelper.UnmarshalData(resp.Body, &releaseInfo)
|
||
if err != nil {
|
||
if showPrompt {
|
||
fmt.Printf("json数据解析失败: %s\n", err)
|
||
}
|
||
return nil
|
||
}
|
||
return &releaseInfo
|
||
}
|
||
|
||
func GetLatestReleaseInfo(showPrompt bool) *ReleaseInfo {
|
||
client := config.Config.HTTPClient("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.116 Safari/537.36")
|
||
client.SetTimeout(time.Duration(0) * time.Second)
|
||
client.SetKeepAlive(true)
|
||
|
||
// check tickstep srv
|
||
var tsReleaseInfo *ReleaseInfo = nil
|
||
for idx := 0; idx < 3; idx++ {
|
||
tsReleaseInfo = getReleaseFromTicstep(client, showPrompt)
|
||
if tsReleaseInfo != nil {
|
||
break
|
||
}
|
||
time.Sleep(time.Duration(5) * time.Second)
|
||
}
|
||
|
||
// github
|
||
var ghReleaseInfo *ReleaseInfo = nil
|
||
for idx := 0; idx < 3; idx++ {
|
||
ghReleaseInfo = getReleaseFromGithub(client, showPrompt)
|
||
if ghReleaseInfo != nil {
|
||
break
|
||
}
|
||
time.Sleep(time.Duration(5) * time.Second)
|
||
}
|
||
|
||
var releaseInfo *ReleaseInfo = nil
|
||
if config.Config.UpdateCheckInfo.PreferUpdateSrv == "tickstep" {
|
||
// theoretically, tickstep server will be more faster at mainland
|
||
releaseInfo = tsReleaseInfo
|
||
} else {
|
||
releaseInfo = ghReleaseInfo
|
||
if ghReleaseInfo == nil || ghReleaseInfo.TagName == "" {
|
||
releaseInfo = tsReleaseInfo
|
||
}
|
||
}
|
||
return releaseInfo
|
||
}
|
||
|
||
// CheckUpdate 检测更新
|
||
func CheckUpdate(version string, yes bool) {
|
||
if !checkaccess.AccessRDWR(cmdutil.ExecutablePath()) {
|
||
fmt.Printf("程序目录不可写, 无法更新.\n")
|
||
return
|
||
}
|
||
fmt.Println("检测更新中, 稍候...")
|
||
client := config.Config.HTTPClient("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.116 Safari/537.36")
|
||
client.SetTimeout(time.Duration(0) * time.Second)
|
||
client.SetKeepAlive(true)
|
||
|
||
releaseInfo := GetLatestReleaseInfo(true)
|
||
if releaseInfo == nil {
|
||
fmt.Printf("获取版本信息失败!\n")
|
||
return
|
||
}
|
||
|
||
// 没有更新, 或忽略 Beta 版本, 和版本前缀不符的
|
||
if strings.Contains(releaseInfo.TagName, "Beta") || !strings.HasPrefix(releaseInfo.TagName, "v") || utils.ParseVersionNum(version) >= utils.ParseVersionNum(releaseInfo.TagName) {
|
||
fmt.Printf("未检测到更新!\n")
|
||
return
|
||
}
|
||
|
||
fmt.Printf("检测到新版本: %s\n", releaseInfo.TagName)
|
||
|
||
line := cmdliner.NewLiner()
|
||
defer line.Close()
|
||
|
||
if !yes {
|
||
y, err := line.State.Prompt("是否进行更新 (y/n): ")
|
||
if err != nil {
|
||
fmt.Printf("输入错误: %s\n", err)
|
||
return
|
||
}
|
||
|
||
if y != "y" && y != "Y" {
|
||
fmt.Printf("更新取消.\n")
|
||
return
|
||
}
|
||
}
|
||
|
||
builder := &strings.Builder{}
|
||
builder.WriteString(ReleaseName + "-" + releaseInfo.TagName + "-" + runtime.GOOS + "-.*?")
|
||
if runtime.GOOS == "darwin" && (runtime.GOARCH == "arm" || runtime.GOARCH == "arm64") {
|
||
builder.WriteString("arm")
|
||
} else {
|
||
switch runtime.GOARCH {
|
||
case "amd64":
|
||
builder.WriteString("(amd64|x86_64|x64)")
|
||
case "386":
|
||
builder.WriteString("(386|x86)")
|
||
case "arm":
|
||
builder.WriteString("(armv5|armv7|arm)")
|
||
case "arm64":
|
||
builder.WriteString("arm64")
|
||
case "mips":
|
||
builder.WriteString("mips")
|
||
case "mips64":
|
||
builder.WriteString("mips64")
|
||
case "mipsle":
|
||
builder.WriteString("(mipsle|mipsel)")
|
||
case "mips64le":
|
||
builder.WriteString("(mips64le|mips64el)")
|
||
default:
|
||
builder.WriteString(runtime.GOARCH)
|
||
}
|
||
}
|
||
builder.WriteString("\\.zip")
|
||
|
||
exp := regexp.MustCompile(builder.String())
|
||
|
||
var targetList []*info
|
||
for _, asset := range releaseInfo.Assets {
|
||
if asset == nil || asset.State != "uploaded" {
|
||
continue
|
||
}
|
||
|
||
if exp.MatchString(asset.Name) {
|
||
targetList = append(targetList, &info{
|
||
filename: asset.Name,
|
||
size: asset.Size,
|
||
downloadURL: asset.BrowserDownloadURL,
|
||
})
|
||
}
|
||
}
|
||
|
||
var target info
|
||
switch len(targetList) {
|
||
case 0:
|
||
fmt.Printf("未匹配到当前系统的程序更新文件, GOOS: %s, GOARCH: %s\n", runtime.GOOS, runtime.GOARCH)
|
||
return
|
||
case 1:
|
||
target = *targetList[0]
|
||
default:
|
||
fmt.Println()
|
||
for k := range targetList {
|
||
fmt.Printf("%d: %s\n", k, targetList[k].filename)
|
||
}
|
||
|
||
fmt.Println()
|
||
t, err := line.State.Prompt("输入序号以下载更新: ")
|
||
if err != nil {
|
||
fmt.Printf("%s\n", err)
|
||
return
|
||
}
|
||
|
||
i, err := strconv.Atoi(t)
|
||
if err != nil {
|
||
fmt.Printf("输入错误: %s\n", err)
|
||
return
|
||
}
|
||
|
||
if i < 0 || i >= len(targetList) {
|
||
fmt.Printf("输入错误: 序号不在范围内\n")
|
||
return
|
||
}
|
||
|
||
target = *targetList[i]
|
||
}
|
||
|
||
if target.size > 0x7fffffff {
|
||
fmt.Printf("file size too large: %d\n", target.size)
|
||
return
|
||
}
|
||
|
||
fmt.Printf("准备下载更新: %s\n", target.filename)
|
||
|
||
// 开始下载
|
||
buf := cachepool.RawMallocByteSlice(int(target.size))
|
||
client.CheckRedirect = func(req *http.Request, via []*http.Request) error {
|
||
logger.Verboseln("下载文件：" + req.URL.String())
|
||
req.Header.Del("Referer") // aliyundrive会检测盗链行为，所以删除Referer
|
||
return nil
|
||
}
|
||
header := map[string]string{}
|
||
header["Referer"] = ""
|
||
header["User-Agent"] = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/106.0.0.0 Safari/537.36 Edg/106.0.1370.42"
|
||
resp, err := client.Req("GET", target.downloadURL, nil, header)
|
||
if err != nil {
|
||
fmt.Printf("下载更新文件发生错误: %s\n", err)
|
||
return
|
||
}
|
||
total, _ := strconv.Atoi(resp.Header.Get("Content-Length"))
|
||
if total > 0 {
|
||
if int64(total) != target.size {
|
||
if es, e := ioutil.ReadAll(resp.Body); e == nil {
|
||
// 发生错误
|
||
logger.Verboseln(string(es))
|
||
}
|
||
fmt.Printf("下载更新文件发生错误: %s\n", err)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 初始化数据
|
||
var readErr error
|
||
downloadSize := 0
|
||
nn := 0
|
||
nn64 := int64(0)
|
||
downloadStatus := transfer.NewDownloadStatus()
|
||
downloadStatus.AddTotalSize(target.size)
|
||
|
||
statusIndicator := func(status *transfer.DownloadStatus) {
|
||
status.UpdateSpeeds() // 更新速度
|
||
var leftStr string
|
||
left := status.TimeLeft()
|
||
if left < 0 {
|
||
leftStr = "-"
|
||
} else {
|
||
leftStr = left.String()
|
||
}
|
||
|
||
fmt.Printf("\r ↓ %s/%s %s/s in %s, left %s ............",
|
||
converter.ConvertFileSize(status.Downloaded(), 2),
|
||
converter.ConvertFileSize(status.TotalSize(), 2),
|
||
converter.ConvertFileSize(status.SpeedsPerSecond(), 2),
|
||
status.TimeElapsed()/1e7*1e7, leftStr,
|
||
)
|
||
}
|
||
|
||
// 读取数据
|
||
wg := sync.WaitGroup{}
|
||
wg.Add(1)
|
||
go func() {
|
||
defer wg.Done()
|
||
for downloadSize < len(buf) && readErr == nil {
|
||
nn, readErr = resp.Body.Read(buf[downloadSize:])
|
||
nn64 = int64(nn)
|
||
|
||
// 更新速度统计
|
||
downloadStatus.AddSpeedsDownloaded(nn64)
|
||
downloadStatus.AddDownloaded(nn64)
|
||
downloadSize += nn
|
||
|
||
if statusIndicator != nil {
|
||
statusIndicator(downloadStatus)
|
||
}
|
||
}
|
||
}()
|
||
wg.Wait()
|
||
|
||
if int64(downloadSize) == target.size {
|
||
// 下载完成
|
||
fmt.Printf("\n下载完毕\n")
|
||
} else {
|
||
fmt.Printf("\n下载更新文件失败\n")
|
||
return
|
||
}
|
||
|
||
// 读取文件
|
||
reader, err := zip.NewReader(bytes.NewReader(buf), target.size)
|
||
if err != nil {
|
||
fmt.Printf("读取更新文件发生错误: %s\n", err)
|
||
return
|
||
}
|
||
|
||
execPath := cmdutil.ExecutablePath()
|
||
|
||
var fileNum, errTimes int
|
||
for _, zipFile := range reader.File {
|
||
if zipFile == nil {
|
||
continue
|
||
}
|
||
|
||
info := zipFile.FileInfo()
|
||
|
||
if info.IsDir() {
|
||
continue
|
||
}
|
||
|
||
rc, err := zipFile.Open()
|
||
if err != nil {
|
||
fmt.Printf("解析 zip 文件错误: %s\n", err)
|
||
continue
|
||
}
|
||
|
||
fileNum++
|
||
|
||
name := zipFile.Name[strings.Index(zipFile.Name, "/")+1:]
|
||
if name == ReleaseName {
|
||
err = update(cmdutil.Executable(), rc)
|
||
} else {
|
||
err = update(filepath.Join(execPath, name), rc)
|
||
}
|
||
|
||
if err != nil {
|
||
errTimes++
|
||
fmt.Printf("发生错误, zip 路径: %s, 错误: %s\n", zipFile.Name, err)
|
||
continue
|
||
}
|
||
}
|
||
|
||
if errTimes == fileNum {
|
||
fmt.Printf("更新失败\n")
|
||
return
|
||
}
|
||
|
||
fmt.Printf("更新完毕, 请重启程序\n")
|
||
}
|