feat: colored output is adult

lpsm-dev · Jan 3, 2022 · fd53a20 · fd53a20
1 parent 4ab38d9
commit fd53a20
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/internal/helpers/helpers.go b/internal/helpers/helpers.go
@@ -82,10 +82,10 @@ func AnimeSimilarity(similarity string) string {
 }
 
 // AnimeIsAdult is for colorful output
-func AnimeIsAdult(isAdult bool) {
+func AnimeIsAdult(isAdult bool) string {
 	if isAdult {
-		fmt.Println(color.GreenString("true"))
+		return fmt.Sprint(color.GreenString("true"))
 	} else {
-		fmt.Println(color.RedString("false"))
+		return fmt.Sprint(color.RedString("false"))
 	}
 }
diff --git a/internal/trace/search_file.go b/internal/trace/search_file.go
@@ -92,7 +92,7 @@ func SearchAnimeByFile(animeFile string, pretty bool) {
 			{"🗽 Title English", animeResp.Result[0].Anilist.Title.English},
 			{"🗻 Title Romaji", animeResp.Result[0].Anilist.Title.Romaji},
 			{"📺 Episode Number", color.MagentaString(strconv.Itoa(animeResp.Result[0].Episode))},
-			{"😈 Is Adult", animeResp.Result[0].Anilist.IsAdult},
+			{"😈 Is Adult", helpers.AnimeIsAdult(animeResp.Result[0].Anilist.IsAdult)},
 		})
 		versionTable.SetStyle(table.StyleColoredBlueWhiteOnBlack)
 		versionTable.Render()
@@ -102,6 +102,6 @@ func SearchAnimeByFile(animeFile string, pretty bool) {
 		fmt.Println("🗽 Title English: " + animeResp.Result[0].Anilist.Title.English)
 		fmt.Println("🗻 Title Romaji: " + animeResp.Result[0].Anilist.Title.Romaji)
 		fmt.Println("📺 Episode Number: " + color.MagentaString(strconv.Itoa(animeResp.Result[0].Episode)))
-		fmt.Println("😈 Is Adult: " + fmt.Sprintf("%v", animeResp.Result[0].Anilist.IsAdult))
+		fmt.Println("😈 Is Adult: " + helpers.AnimeIsAdult(animeResp.Result[0].Anilist.IsAdult))
 	}
 }
diff --git a/internal/trace/search_link.go b/internal/trace/search_link.go
@@ -80,7 +80,7 @@ func SearchAnimeByLink(animeLink string, pretty bool) {
 			{"🗽 Title English", animeResp.Result[0].Anilist.Title.English},
 			{"🗻 Title Romaji", animeResp.Result[0].Anilist.Title.Romaji},
 			{"📺 Episode Number", color.MagentaString(strconv.Itoa(animeResp.Result[0].Episode))},
-			{"😈 Is Adult", animeResp.Result[0].Anilist.IsAdult},
+			{"😈 Is Adult", helpers.AnimeIsAdult(animeResp.Result[0].Anilist.IsAdult)},
 		})
 		versionTable.SetStyle(table.StyleColoredBlueWhiteOnBlack)
 		versionTable.Render()
@@ -90,6 +90,6 @@ func SearchAnimeByLink(animeLink string, pretty bool) {
 		fmt.Println("🗽 Title English: " + animeResp.Result[0].Anilist.Title.English)
 		fmt.Println("🗻 Title Romaji: " + animeResp.Result[0].Anilist.Title.Romaji)
 		fmt.Println("📺 Episode Number: " + color.MagentaString(strconv.Itoa(animeResp.Result[0].Episode)))
-		fmt.Println("😈 Is Adult: " + fmt.Sprintf("%v", animeResp.Result[0].Anilist.IsAdult))
+		fmt.Println("😈 Is Adult: " + helpers.AnimeIsAdult(animeResp.Result[0].Anilist.IsAdult))
 	}
 }