add users command

Author: nasum

cmd/list.go

@@ -1,49 +1,24 @@
 package cmd
 
 import (
-	"fmt"
-
-	"github.com/dghubble/go-twitter/twitter"
-	"github.com/dghubble/oauth1"
+	"github.com/nasum/tt/cmd/list"
 	"github.com/nasum/tt/lib"
 	"github.com/spf13/cobra"
 )
 
 func listCmd(config lib.Config) *cobra.Command {
 
-	oauthConfig := oauth1.NewConfig(config.ConsumerKey, config.ConsumerSecret)
-	token := oauth1.NewToken(config.AccessToken, config.AccessSecret)
-
-	httpClient := oauthConfig.Client(oauth1.NoContext, token)
-	client := twitter.NewClient(httpClient)
-
-	displayConsole := &lib.DisplayConsole{}
-
 	cmd := &cobra.Command{
-		Use:   "list",
-		Short: "get your list",
-		RunE: func(cmd *cobra.Command, args []string) error {
-			user, res, err := client.Accounts.VerifyCredentials(&twitter.AccountVerifyParams{})
-
-			if err != nil {
-				return fmt.Errorf("cannot get VerifyCredentials: %v: %v", err, res.Status)
-			}
-
-			lists, res, err := client.Lists.List(&twitter.ListsListParams{
-				UserID: user.ID,
-			})
-
-			if err != nil {
-				return fmt.Errorf("cannot get list: %v: %v", err, res.Status)
-			}
-
-			for _, list := range lists {
-				displayConsole.ShowList(list.Name, list.URI)
-			}
-
-			return nil
-		},
+		Use:           "list",
+		Short:         "list command",
+		SilenceErrors: true,
+		SilenceUsage:  true,
 	}
 
+	cmd.AddCommand(
+		list.LSCmd(config),
+		list.UsersCmd(config),
+	)
+
 	return cmd
 }

cmd/list/ls.go

@@ -0,0 +1,48 @@
+package list
+
+import (
+	"fmt"
+
+	"github.com/dghubble/go-twitter/twitter"
+	"github.com/dghubble/oauth1"
+	"github.com/nasum/tt/lib"
+	"github.com/spf13/cobra"
+)
+
+func LSCmd(config lib.Config) *cobra.Command {
+
+	oauthConfig := oauth1.NewConfig(config.ConsumerKey, config.ConsumerSecret)
+	token := oauth1.NewToken(config.AccessToken, config.AccessSecret)
+
+	httpClient := oauthConfig.Client(oauth1.NoContext, token)
+	client := twitter.NewClient(httpClient)
+
+	displayConsole := &lib.DisplayConsole{}
+
+	cmd := &cobra.Command{
+		Use:   "ls",
+		Short: "get your list",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			user, res, err := client.Accounts.VerifyCredentials(&twitter.AccountVerifyParams{})
+
+			if err != nil {
+				return fmt.Errorf("cannot get VerifyCredentials: %v: %v", err, res.Status)
+			}
+
+			lists, res, err := client.Lists.List(&twitter.ListsListParams{
+				UserID: user.ID,
+			})
+
+			if err != nil {
+				return fmt.Errorf("cannot get list: %v: %v", err, res.Status)
+			}
+
+			for _, list := range lists {
+				displayConsole.ShowList(list.Name, list.URI, list.ID)
+			}
+
+			return nil
+		},
+	}
+	return cmd
+}

cmd/list/users.go

@@ -0,0 +1,55 @@
+package list
+
+import (
+	"fmt"
+
+	"github.com/dghubble/go-twitter/twitter"
+	"github.com/dghubble/oauth1"
+	"github.com/nasum/tt/lib"
+	"github.com/spf13/cobra"
+)
+
+// UsersParams is users command paramater
+type UsersParams struct {
+	ListID int64
+}
+
+// UsersCmd is users command
+func UsersCmd(config lib.Config) *cobra.Command {
+
+	oauthConfig := oauth1.NewConfig(config.ConsumerKey, config.ConsumerSecret)
+	token := oauth1.NewToken(config.AccessToken, config.AccessSecret)
+
+	httpClient := oauthConfig.Client(oauth1.NoContext, token)
+	client := twitter.NewClient(httpClient)
+
+	displayConsole := &lib.DisplayConsole{}
+
+	usersParams := &UsersParams{}
+
+	cmd := &cobra.Command{
+		Use:   "users",
+		Short: "get your list users",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var cursor int64 = -1
+			for cursor != 0 {
+
+				members, res, err := client.Lists.Members(&twitter.ListsMembersParams{ListID: usersParams.ListID, Cursor: cursor})
+
+				if err != nil {
+					return fmt.Errorf("cannot get users: %v: %v", err, res.Status)
+				}
+
+				for _, user := range members.Users {
+					displayConsole.ShowUser(user.Name, user.ScreenName, user.URL, user.FriendsCount, user.FollowersCount)
+				}
+				cursor = members.NextCursor
+			}
+
+			return nil
+		},
+	}
+	flags := cmd.Flags()
+	flags.Int64VarP(&usersParams.ListID, "list", "l", 0, "list id")
+	return cmd
+}

lib/display_console.go

@@ -24,6 +24,12 @@ func (d *DisplayConsole) TweetID(tweetID string) string {
 	return yellow(tweetID)
 }
 
+// ListID is output colored tweet id text
+func (d *DisplayConsole) ListID(listID string) string {
+	yellow := color.New(color.FgYellow).SprintFunc()
+	return yellow(listID)
+}
+
 // ReplyTo is output colored reply text
 func (d *DisplayConsole) ReplyTo(tweetID string) string {
 	cyan := color.New(color.FgCyan).SprintFunc()
@@ -36,6 +42,12 @@ func (d *DisplayConsole) URL(url string) string {
 	return green(url)
 }
 
+// Name is output colored name text
+func (d *DisplayConsole) Name(name string) string {
+	yellow := color.New(color.FgYellow).SprintFunc()
+	return yellow(name)
+}
+
 // ShowTweet is display tweet text
 func (d *DisplayConsole) ShowTweet(createdAt time.Time, tweetID int64, screenName string, text string) {
 	fmt.Fprintf(
@@ -49,11 +61,25 @@ func (d *DisplayConsole) ShowTweet(createdAt time.Time, tweetID int64, screenNam
 }
 
 // ShowList is display list text
-func (d *DisplayConsole) ShowList(title string, url string) {
+func (d *DisplayConsole) ShowList(title string, url string, id int64) {
 	fmt.Fprintf(
 		color.Output,
-		"%s\t%s\n",
+		"%s\t%s\t%s\n",
 		d.URL("https://twitter.com/"+url),
+		d.ListID(strconv.FormatInt(id, 10)),
 		title,
 	)
 }
+
+// ShowUser is display user
+func (d *DisplayConsole) ShowUser(name string, screenName string, url string, friendsCount int, followersCount int) {
+	fmt.Fprintf(
+		color.Output,
+		"%s\t%s\t%s\t%s\t%s\n",
+		d.URL("https://twitter.com/"+screenName),
+		d.Name(name),
+		d.Name("@"+screenName),
+		strconv.Itoa(friendsCount),
+		strconv.Itoa(followersCount),
+	)
+}