Introduce config package

Author: citizen428

cmd/unsavory/main.go

@@ -1,44 +1,11 @@
 package main
 
 import (
-	"flag"
-	"log"
-	"os"
-	"runtime"
-
+	"github.com/citizen428/unsavory/internal/config"
 	un "github.com/citizen428/unsavory/internal/unsavory"
 )
 
-var (
-	dryRun   bool
-	proxyURL string
-	token    string
-)
-
-func init() {
-	if cpu := runtime.NumCPU(); cpu == 1 {
-		runtime.GOMAXPROCS(2)
-	} else {
-		runtime.GOMAXPROCS(cpu)
-	}
-
-	log.SetFlags(0)
-
-	flag.BoolVar(&dryRun, "dry-run", false, "Enables dry run mode")
-	flag.StringVar(&proxyURL, "proxy-url", "", "HTTP proxy URL")
-	flag.StringVar(&token, "token", "", "Pinboard API token")
-}
-
 func main() {
-	flag.Parse()
-	if token == "" {
-		log.Fatalln("Missing required API token")
-	}
-
-	if proxyURL != "" {
-		os.Setenv("HTTP_PROXY", proxyURL)
-	}
-
-	un := un.NewClient(token, dryRun)
+	un := un.NewClient(config.Token, config.DryRun)
 	un.Run()
 }

internal/config/config.go

@@ -0,0 +1,40 @@
+package config
+
+import (
+	"flag"
+	"log"
+	"os"
+	"runtime"
+)
+
+var (
+	DryRun bool
+	Token  string
+
+	proxyURL string
+)
+
+func init() {
+	runtime.GOMAXPROCS(runtime.NumCPU())
+
+	log.SetFlags(0)
+
+	flag.BoolVar(&DryRun, "dry-run", false, "Enables dry run mode")
+	flag.StringVar(&Token, "token", "", "Pinboard API token")
+	flag.StringVar(&proxyURL, "proxy-url", "", "HTTP proxy URL")
+
+	flag.Parse()
+
+	if len(os.Args) < 2 {
+		flag.Usage()
+		os.Exit(0)
+	}
+
+	if Token == "" {
+		log.Fatalln("Missing required API token")
+	}
+
+	if proxyURL != "" {
+		os.Setenv("HTTP_PROXY", proxyURL)
+	}
+}