2021-11-11

Author: netscane

main.go

@@ -14,13 +14,20 @@ import (
 )
 
 var (
-	trackerUrl = "https://cdn.jsdelivr.net/gh/ngosang/trackerslist/trackers_best_ip.txt"
+	tracker_url         string
+	default_tracker_url = "https://cdn.jsdelivr.net/gh/ngosang/trackerslist/trackers_best_ip.txt"
+	conf_path           string
 	//trackerUrl = "https://raw.githubusercontent.com/ngosang/trackerslist/master/trackers_best.txt" //need http proxy
 )
 
+func init() {
+	flag.StringVar(&tracker_url, "tracker-url", default_tracker_url, "tracker url")
+	flag.StringVar(&conf_path, "conf-path", "", "aria2 conf path")
+}
+
 func getTrackers() ([]string, error) {
 	client := &http.Client{}
-	request, err := http.NewRequest("GET", trackerUrl, nil)
+	request, err := http.NewRequest("GET", tracker_url, nil)
 	if err != nil {
 		return nil, err
 	}
@@ -46,18 +53,18 @@ func getTrackers() ([]string, error) {
 	return trackers, nil
 }
 
-func getConfFilePath() (string, error) {
+func getDefaultConfPath() (string, error) {
 	var (
-		home     string
-		confPath string
-		err      error
+		home string
+		path string
+		err  error
 	)
 	home, err = os.UserHomeDir()
 	if err != nil {
 		return "", err
 	}
-	confPath = filepath.Join(home, "aria2", "aria2.conf")
-	return confPath, nil
+	path = filepath.Join(home, ".aria2c", "aria2.conf")
+	return path, nil
 }
 
 func readConfLines(confPath string) ([]string, error) {
@@ -117,8 +124,14 @@ func updateConfFile(confPath string, lines []string, trackers []string) error {
 		bufw.WriteString(line)
 		if "windows" == runtime.GOOS {
 			bufw.WriteString("\r\n")
+			if !strings.HasPrefix(line, "#") {
+				bufw.WriteString("\r\n")
+			}
 		} else {
 			bufw.WriteString("\n")
+			if !strings.HasPrefix(line, "#") {
+				bufw.WriteString("\n")
+			}
 		}
 	}
 	_, err = file.WriteString(bufw.String())
@@ -131,28 +144,26 @@ func updateConfFile(confPath string, lines []string, trackers []string) error {
 
 func main() {
 	flag.Parse()
-	if len(flag.Arg(0)) > 0 {
-		trackerUrl = flag.Arg(0)
-	}
 	var (
 		trackers  []string
 		confLines []string
 		err       error
-		confPath  string
 	)
-	trackers, err = getTrackers()
-	if err != nil {
-		panic(err.Error())
+	if len(conf_path) == 0 {
+		conf_path, err = getDefaultConfPath()
+		if err != nil {
+			panic(err.Error())
+		}
 	}
-	confPath, err = getConfFilePath()
+	trackers, err = getTrackers()
 	if err != nil {
 		panic(err.Error())
 	}
-	confLines, err = readConfLines(confPath)
+	confLines, err = readConfLines(conf_path)
 	if err != nil {
 		panic(err.Error())
 	}
-	err = updateConfFile(confPath, confLines, trackers)
+	err = updateConfFile(conf_path, confLines, trackers)
 	if err != nil {
 		panic(err.Error())
 	}