Move FetchAll to method on *api.Client

Author: neslom

api/api.go

@@ -75,6 +75,24 @@ func (c *Client) Fetch(args []string) ([]*Problem, error) {
 	return payload.Problems, nil
 }
 
+// FetchAll retrieves all problems for a given language track from the API
+func (c *Client) FetchAll(trackID string) ([]*Problem, error) {
+	list, err := c.List(trackID)
+	if err != nil {
+		return nil, err
+	}
+
+	problems := make([]*Problem, len(list))
+	for i, prob := range list {
+		p, err := c.Fetch([]string{trackID, prob})
+		if err != nil {
+			return nil, err
+		}
+		problems[i] = p[0]
+	}
+	return problems, nil
+}
+
 // Restore fetches the latest revision of a solution and writes it to disk.
 func (c *Client) Restore() ([]*Problem, error) {
 	url := fmt.Sprintf("%s/v2/exercises/restore?key=%s", c.XAPIHost, c.APIKey)

cmd/fetch.go

@@ -19,19 +19,25 @@ func Fetch(ctx *cli.Context) error {
 	client := api.NewClient(c)
 
 	args := ctx.Args()
-	problems, err := client.Fetch(args)
+	var problems []*api.Problem
 
 	if ctx.Bool("all") {
 		if len(args) > 0 {
 			trackID := args[0]
-			problems = fetchAll(trackID, client)
+			p, err := client.FetchAll(trackID)
+			if err != nil {
+				log.Fatal(err)
+			}
+			problems = p
 		} else {
 			log.Fatalf("You must supply a track to fetch all exercises")
 		}
-	}
-
-	if err != nil {
-		log.Fatal(err)
+	} else {
+		p, err := client.Fetch(args)
+		if err != nil {
+			log.Fatal(err)
+		}
+		problems = p
 	}
 
 	submissionInfo, err := client.Submissions()