package main import ( "fmt" "io" "net/http" "os" "runtime" "strconv" "strings" "sync" "time" "github.com/fumiama/terasu/http2" "github.com/pkg/errors" ) func (c *config) download(path, prefix string, waits time.Duration, usecust, usetrs, force bool) error { for i, t := range c.Targets { if t.Refer != "" { refp := path[:strings.LastIndex(path, "/")+1] + t.Refer infof("#%s%d refer to target '%s'.", prefix, i+1, refp) refcfg, err := readconfig(refp, usecust) if err != nil { return err } err = refcfg.download(refp, prefix+strconv.Itoa(i+1)+".", waits, usecust, usetrs, force) if err != nil { return err } continue } if t.OS != "" && t.OS != runtime.GOOS { warnf("#%s%d target required OS: %s but you are %s, skip.", prefix, i+1, t.OS, runtime.GOOS) continue } if t.Arch != "" && t.Arch != runtime.GOARCH { warnf("#%s%d target required Arch: %s but you are %s, skip.", prefix, i+1, t.Arch, runtime.GOARCH) continue } err := os.MkdirAll(t.Folder, 0755) if err != nil { return errors.Wrap(err, fmt.Sprintf("#%s%d make target folder '%s'", prefix, i+1, t.Folder)) } infof("#%s%d open target folder '%s'.", prefix, i+1, t.Folder) if len(t.Copy) == 0 { warnf("#%s%d empty copy target.", prefix, i+1) continue } wg := sync.WaitGroup{} wg.Add(len(t.Copy)) infof("#%s%d download copy: '%v'.", prefix, i+1, t.Copy) for j, cp := range t.Copy { go func(i int, cp, prefix string) { defer wg.Done() sleep := waits * time.Duration(i) if sleep > time.Millisecond { time.Sleep(sleep) } fname := t.Folder + "/" + cp[strings.LastIndex(cp, "/")+1:] if !force { if _, err := os.Stat(fname); err == nil || os.IsExist(err) { warnf("#%s%d skip exist file %s", prefix, i+1, fname) return } } req, err := http.NewRequest("GET", c.BaseURL+"/"+cp, nil) if err != nil { errorf("#%s%d new request to %s err: %v", prefix, i+1, cp, err) return } infof("#%s%d get: %s", prefix, i+1, req.URL) req.Header.Add("user-agent", ua) var resp *http.Response if usetrs { resp, err = http2.DefaultClient.Do(req) } else { resp, err = http.DefaultClient.Do(req) } if err != nil { errorf("#%s%d get %s err: %v", prefix, i+1, req.URL, err) return } defer resp.Body.Close() if resp.StatusCode != http.StatusOK { err := errors.New(fmt.Sprintf("HTTP %d %s", resp.StatusCode, resp.Status)) errorf("#%s%d get %s err: %v", prefix, i+1, req.URL, err) return } f, err := os.Create(fname) if err != nil { errorf("#%s%d create file %s err: %v", prefix, i+1, fname, err) return } defer f.Close() infof("#%s%d writing file %s", prefix, i+1, fname) pm := newmeter(fmt.Sprintf("#%s%d", prefix, i+1), fname, int(resp.ContentLength)) defer pm.finish() _, err = io.Copy(io.MultiWriter(f, &pm), resp.Body) if err != nil { errorf("#%s%d download file %s err: %v", prefix, i+1, fname, err) return } infof("#%s%d finished download %s", prefix, i+1, fname) }(j, cp, fmt.Sprintf("%s%d.", prefix, i+1)) } wg.Wait() } return nil }