Merge pull request #74 from Tonkpils/fetch-by-language

Fetch by language

Author: kytrinyx

fetch_endpoint.go

@@ -8,5 +8,11 @@ func FetchEndpoint(args []string) string {
 	if len(args) == 0 {
 		return FetchEndpoints["current"]
 	}
-	return fmt.Sprintf("%s/%s/%s", FetchEndpoints["exercise"], args[0], args[1])
+
+	endpoint := FetchEndpoints["exercise"]
+	for _, arg := range args {
+		endpoint = fmt.Sprintf("%s/%s", endpoint, arg)
+	}
+
+	return endpoint
 }

fetch_endpoint_test.go

@@ -16,3 +16,9 @@ func TestFetchExerciseEndpoint(t *testing.T) {
 	actual := FetchEndpoint([]string{"language", "slug"})
 	assert.Equal(t, expected, actual)
 }
+
+func TestFetchExerciseEndpointByLanguage(t *testing.T) {
+	expected := "/api/v1/assignments/language"
+	actual := FetchEndpoint([]string{"language"})
+	assert.Equal(t, expected, actual)
+}

main.go

@@ -92,14 +92,15 @@ func main() {
 			ShortName: "f",
 			Usage:     "Fetch assignments from exercism.io",
 			Action: func(c *cli.Context) {
-				if len(c.Args()) != 0 && len(c.Args()) != 2 {
-					fmt.Println("Usage: exercism fetch\n   or: exercism fetch LANGUAGE EXERCISE")
+				argCount := len(c.Args())
+				if argCount < 0 || argCount > 2 {
+					fmt.Println("Usage: exercism fetch\n   or: exercism fetch LANGUAGE\n   or: exercism fetch LANGUAGE EXERCISE")
 					return
 				}
 
 				config, err := configuration.FromFile(c.GlobalString("config"))
 				if err != nil {
-					if len(c.Args()) == 0 {
+					if argCount == 0 || argCount == 1 {
 						fmt.Println("Are you sure you are logged in? Please login again.")
 						return
 					} else {
@@ -117,6 +118,18 @@ func main() {
 					return
 				}
 
+				if len(assignments) == 0 {
+					noAssignmentMessage := "No assignments found"
+					if argCount == 2 {
+						fmt.Printf("%s for %s - %s\n", noAssignmentMessage, c.Args()[0], c.Args()[1])
+					} else if argCount == 1 {
+						fmt.Printf("%s for %s\n", noAssignmentMessage, c.Args()[0])
+					} else {
+						fmt.Printf("%s\n", noAssignmentMessage)
+					}
+					return
+				}
+
 				for _, a := range assignments {
 					err := SaveAssignment(config.ExercismDirectory, a)
 					if err != nil {