Add config

Author: uetchy

config.go

@@ -0,0 +1,61 @@
+package main
+
+import (
+	"encoding/json"
+	"github.com/spf13/viper"
+	"os"
+	"os/user"
+	"path/filepath"
+)
+
+type config struct {
+	AccessToken string `json:"accessToken"`
+}
+
+var C config
+
+func getHomeDir() string {
+	currentUser, _ := user.Current()
+	return currentUser.HomeDir
+}
+
+func getDefaultConfigPath() string {
+	return filepath.Join(getHomeDir(), "Library/Application Support/Alfred 2/Workflow Data/", bundleId)
+}
+
+func loadConfig() error {
+	configPath := getDefaultConfigPath()
+	viper.SetConfigName("config")
+	viper.AddConfigPath(configPath)
+	viper.SetConfigType("json")
+	err := viper.ReadInConfig()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func saveConfig() error {
+	configPath := getDefaultConfigPath()
+	viper.Marshal(&C)
+
+	buf, err := json.MarshalIndent(C, "", "    ")
+	if err != nil {
+		return err
+	}
+
+	err = os.MkdirAll(configPath, 0755)
+	if err != nil {
+		return err
+	}
+
+	f, err := os.Create(configPath + "/config.json")
+	if err != nil {
+		return err
+	}
+
+	defer f.Close()
+
+	f.WriteString(string(buf))
+	return nil
+}

main.go

@@ -6,7 +6,7 @@ import (
 )
 
 const (
-	bundleId    = "co.randompaper.alfred-qiita-workflow"
+	bundleId = "co.randompaper.alfred-qiita-workflow"
 )
 
 func main() {
@@ -18,20 +18,10 @@ func main() {
 			Name:   "search",
 			Action: cmdSearch,
 		},
+		{
+			Name:   "setup",
+			Action: cmdSetup,
+		},
 	}
 	app.Run(os.Args)
 }
-
-func cmdSetToken(c *cli.Context) {
-	token := c.Args().First()
-
-	configPath := "~/Library/Application Support/Alfred 2/Workflow Data/" + bundleId + "/token"
-
-	fout, err := os.Create(configPath)
-	if err != nil {
-		return
-	}
-	defer fout.Close()
-
-	fout.WriteString(token)
-}

search.go

@@ -1,18 +1,20 @@
 package main
 
 import (
-	"github.com/pascalw/go-alfred"
 	"fmt"
 	"github.com/codegangsta/cli"
+	"github.com/pascalw/go-alfred"
+	"github.com/spf13/viper"
 	"github.com/uetchy/alfred-qiita-workflow/qiita"
 	"golang.org/x/oauth2"
 )
 
 func cmdSearch(c *cli.Context) {
 	query := c.Args().First()
 
+	loadConfig()
 	ts := oauth2.StaticTokenSource(
-		&oauth2.Token{AccessToken: accessToken},
+		&oauth2.Token{AccessToken: viper.GetString("accessToken")},
 	)
 	tc := oauth2.NewClient(oauth2.NoContext, ts)
 	client := qiita.NewClient(tc)

setup.go

@@ -0,0 +1,14 @@
+package main
+
+import (
+	"github.com/codegangsta/cli"
+	"github.com/spf13/viper"
+)
+
+func cmdSetup(c *cli.Context) {
+	token := c.Args().First()
+
+	loadConfig()
+	viper.Set("accessToken", token)
+	saveConfig()
+}