|
|
@@ -0,0 +1,190 @@
|
|
|
+package main
|
|
|
+
|
|
|
+import (
|
|
|
+ "encoding/json"
|
|
|
+ "fmt"
|
|
|
+ "io"
|
|
|
+ "io/ioutil"
|
|
|
+ "log"
|
|
|
+ "net/http"
|
|
|
+ "os"
|
|
|
+ "strings"
|
|
|
+ "sync"
|
|
|
+)
|
|
|
+
|
|
|
+// 配置文件
|
|
|
+type configuration struct {
|
|
|
+ Proxy string
|
|
|
+}
|
|
|
+
|
|
|
+var G_config configuration
|
|
|
+
|
|
|
+func readConfig() {
|
|
|
+ // 打开文件
|
|
|
+ file, err := os.Open("config.json")
|
|
|
+ if err != nil {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ // 关闭文件
|
|
|
+ defer file.Close()
|
|
|
+ //NewDecoder创建一个从file读取并解码json对象的*Decoder，解码器有自己的缓冲，并可能超前读取部分json数据。
|
|
|
+ decoder := json.NewDecoder(file)
|
|
|
+ G_config = configuration{}
|
|
|
+ //Decode从输入流读取下一个json编码值并保存在v指向的值里
|
|
|
+ err = decoder.Decode(&G_config)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("Error:", err)
|
|
|
+ }
|
|
|
+ fmt.Println("proxy:" + G_config.Proxy)
|
|
|
+}
|
|
|
+
|
|
|
+// 下载
|
|
|
+type people struct {
|
|
|
+ Tag_name string `json:"tag_name"` //Tag_name 小写变成私有
|
|
|
+}
|
|
|
+
|
|
|
+func api_get_version(project string) string {
|
|
|
+ git_api := "https://api.github.com/repos/" + project + "/releases/latest"
|
|
|
+
|
|
|
+ // 目标URL
|
|
|
+ url := git_api
|
|
|
+
|
|
|
+ // 发送HTTP GET请求
|
|
|
+ resp, err := http.Get(url)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+ defer resp.Body.Close()
|
|
|
+
|
|
|
+ // 读取响应体
|
|
|
+ body, err := ioutil.ReadAll(resp.Body)
|
|
|
+ if err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ // 解析JSON数据
|
|
|
+ responseData := people{}
|
|
|
+ if err := json.Unmarshal(body, &responseData); err != nil {
|
|
|
+ panic(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ // 打印解析后的数据
|
|
|
+ // fmt.Println("tag_name:" + responseData.Tag_name)
|
|
|
+ // latest_version = responseData.tag_name
|
|
|
+ return responseData.Tag_name
|
|
|
+}
|
|
|
+
|
|
|
+type Downloader struct {
|
|
|
+ io.Reader
|
|
|
+ Total int64
|
|
|
+ Current int64
|
|
|
+}
|
|
|
+
|
|
|
+func printProgressBar(progress int, total int) {
|
|
|
+ bar := ""
|
|
|
+ numSlashes := progress * 100 / total / 5
|
|
|
+ for i := 0; i < numSlashes; i++ {
|
|
|
+ bar += "\u2588" // 这是一个完整的块字符
|
|
|
+ }
|
|
|
+ fmt.Printf("下载进度: [%-21s] %.2f%%\r", bar, float64(progress*100.0/total))
|
|
|
+}
|
|
|
+
|
|
|
+func (d *Downloader) Read(p []byte) (n int, err error) {
|
|
|
+ n, err = d.Reader.Read(p)
|
|
|
+ d.Current += int64(n)
|
|
|
+
|
|
|
+ printProgressBar(int(d.Current*10000/d.Total), 10000)
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func downloadFile(url, filePath string) {
|
|
|
+ defer wg.Done()
|
|
|
+ fmt.Println("下载", filePath)
|
|
|
+ resp, err := http.Get(url)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatalln(err)
|
|
|
+ }
|
|
|
+ defer func() {
|
|
|
+ _ = resp.Body.Close()
|
|
|
+ }()
|
|
|
+ file, err := os.Create(filePath)
|
|
|
+ defer func() {
|
|
|
+ _ = file.Close()
|
|
|
+ }()
|
|
|
+ downloader := &Downloader{
|
|
|
+ Reader: resp.Body,
|
|
|
+ Total: resp.ContentLength,
|
|
|
+ }
|
|
|
+ if _, err := io.Copy(file, downloader); err != nil {
|
|
|
+ log.Fatalln(err)
|
|
|
+ }
|
|
|
+ fmt.Println() // 新的一行以清除进度条打印
|
|
|
+}
|
|
|
+
|
|
|
+var wg sync.WaitGroup
|
|
|
+
|
|
|
+func http_down(project string, down_list string, proxy string) {
|
|
|
+ // fmt.Println(project, down_list, proxy)
|
|
|
+
|
|
|
+ // 获取最新版本
|
|
|
+ latest_version := api_get_version(project)
|
|
|
+
|
|
|
+ // 创建文件夹
|
|
|
+ folderPath := latest_version
|
|
|
+ err := os.MkdirAll(folderPath, os.ModePerm)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("创建文件夹错误: %v", err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // 分割下载列表
|
|
|
+ git_down_url := proxy + project + "/releases/download/" + latest_version + "/"
|
|
|
+ parts := strings.Split(down_list, ",")
|
|
|
+ task := make(map[string]string)
|
|
|
+ for i := 0; i < len(parts); i++ {
|
|
|
+ // 替换
|
|
|
+ version_str := strings.ReplaceAll(latest_version, "v", "")
|
|
|
+ version_str = strings.ReplaceAll(version_str, "V", "")
|
|
|
+ filename := strings.ReplaceAll(parts[i], "[version]", version_str)
|
|
|
+ // 存入
|
|
|
+ task[git_down_url+filename] = folderPath + "/" + filename
|
|
|
+ }
|
|
|
+
|
|
|
+ for k, v := range task {
|
|
|
+ wg.Add(1)
|
|
|
+ downloadFile(k, v)
|
|
|
+ }
|
|
|
+ wg.Wait()
|
|
|
+}
|
|
|
+
|
|
|
+func main() {
|
|
|
+ // // 参数解析
|
|
|
+ // var name = flag.String("n", "yisier/nps", "项目，譬如 yisier/nps")
|
|
|
+ // var list = flag.String("l", "windows_amd64_client.tar.gz,windows_amd64_server.tar.gz", "下载列表，譬如 windows_amd64_client.tar.gz,windows_amd64_server.tar.gz, [version]可代替版本")
|
|
|
+ // var proxy = flag.String("p", "https://github.com/", "代理，譬如 https://ghproxy.com/https://github.com/")
|
|
|
+ // // 解析命令行参数
|
|
|
+ // flag.Parse()
|
|
|
+
|
|
|
+ name := ""
|
|
|
+ list := ""
|
|
|
+ proxy := "https://github.com/"
|
|
|
+ for i, num := range os.Args[1:] {
|
|
|
+ switch i {
|
|
|
+ case 0:
|
|
|
+ name = num
|
|
|
+ case 1:
|
|
|
+ list = num
|
|
|
+ case 2:
|
|
|
+ proxy = num
|
|
|
+ default:
|
|
|
+ fmt.Println("未知参数")
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ readConfig()
|
|
|
+ if proxy == "https://github.com/" && G_config.Proxy != "" {
|
|
|
+ proxy = G_config.Proxy
|
|
|
+ }
|
|
|
+
|
|
|
+ http_down(name, list, proxy)
|
|
|
+}
|
