Can now refresh a single anime in refresh-episodes

jobs/refresh-episodes/refresh-episodes.go

@@ -12,6 +12,10 @@ import (
 func main() {
 	color.Yellow("Refreshing episode information for each anime.")
 
+	if InvokeShellArgs() {
+		return
+	}
+
 	highPriority := []*arn.Anime{}
 	mediumPriority := []*arn.Anime{}
 	lowPriority := []*arn.Anime{}
@@ -32,19 +36,24 @@ func main() {
 	}
 
 	color.Cyan("High priority queue (%d):", len(highPriority))
-	refresh(highPriority)
+	refreshQueue(highPriority)
 
 	color.Cyan("Medium priority queue (%d):", len(mediumPriority))
-	refresh(mediumPriority)
+	refreshQueue(mediumPriority)
 
 	color.Cyan("Low priority queue (%d):", len(lowPriority))
-	refresh(lowPriority)
+	refreshQueue(lowPriority)
 
 	color.Green("Finished.")
 }
 
-func refresh(queue []*arn.Anime) {
+func refreshQueue(queue []*arn.Anime) {
 	for _, anime := range queue {
+		refresh(anime)
+	}
+}
+
+func refresh(anime *arn.Anime) {
 	fmt.Println(anime.ID, "|", anime.Title.Canonical, "|", anime.GetMapping("shoboi/anime"))
 
 	episodeCount := len(anime.Episodes().Items)
@@ -54,7 +63,7 @@ func refresh(queue []*arn.Anime) {
 
 	if err != nil {
 		if strings.Contains(err.Error(), "missing a Shoboi ID") {
-				continue
+			return
 		}
 
 		color.Red(err.Error())
@@ -62,4 +71,3 @@ func refresh(queue []*arn.Anime) {
 		fmt.Println("+"+strconv.Itoa(len(anime.Episodes().Items)-episodeCount)+" airing", "|", "+"+strconv.Itoa(anime.Episodes().AvailableCount()-availableEpisodeCount)+" available")
 	}
 }
-}

jobs/refresh-episodes/shell.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"flag"
+
+	"github.com/animenotifier/arn"
+)
+
+// Shell parameters
+var animeID string
+
+// Shell flags
+func init() {
+	flag.StringVar(&animeID, "id", "", "ID of the anime you want to refresh")
+	flag.Parse()
+}
+
+// InvokeShellArgs ...
+func InvokeShellArgs() bool {
+	if animeID != "" {
+		anime, err := arn.GetAnime(animeID)
+
+		if err != nil {
+			panic(err)
+		}
+
+		refresh(anime)
+		return true
+	}
+
+	return false
+}