Fix #4

Author: uetchy

setup.go

@@ -4,16 +4,33 @@ import (
 	"fmt"
 	"github.com/codegangsta/cli"
 	"github.com/spf13/viper"
+	"github.com/uetchy/go-qiita/qiita"
+	"golang.org/x/oauth2"
+	"os"
 )
 
 func cmdSetup(c *cli.Context) {
 	token := c.Args().First()
 
+	ts := oauth2.StaticTokenSource(
+		&oauth2.Token{AccessToken: token},
+	)
+	tc := oauth2.NewClient(oauth2.NoContext, ts)
+	client := qiita.NewClient(tc)
+	user, err := client.AuthenticatedUser.Show()
+	if err != nil {
+		fmt.Println("Auth failed")
+		os.Exit(1)
+	}
+
 	loadConfig()
 	viper.Set("accessToken", token)
-	err := saveConfig()
+	viper.Set("id", user.Id)
+
+	err = saveConfig()
 	if err != nil {
 		fmt.Println(err)
+		os.Exit(1)
 	}
 
 	fmt.Println("Token saved")

stocks.go

@@ -3,6 +3,7 @@ package main
 import (
 	"github.com/codegangsta/cli"
 	"github.com/pascalw/go-alfred"
+	"github.com/spf13/viper"
 )
 
 func cmdStocks(c *cli.Context) {
@@ -11,7 +12,8 @@ func cmdStocks(c *cli.Context) {
 	if err != nil {
 		return
 	}
-	items, _, _ := client.Users.Stocks("uetchy", nil)
+
+	items, _, _ := client.Users.Stocks(viper.GetString("id"), nil)
 
 	alfred.InitTerms(query)
 	response := alfred.NewResponse()