feat: add validations

Caixetadev · Jan 6, 2024 · 3f73c61 · 3f73c61
1 parent 34d0a44
commit 3f73c61
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 10 deletions.
diff --git a/cmd/gophimation/main.go b/cmd/gophimation/main.go
@@ -3,7 +3,6 @@ package main
 // Fazer opção para voltar para a lista de animes.
 
 import (
-	"fmt"
 	"os"
 
 	"github.com/Caixetadev/gophimation/pkg/constants"
@@ -37,15 +36,19 @@ func main() {
 
 	case len(os.Args) > 1:
 		animeSearch := search.Search()
-		episodeSelected, _ := episode.SelectEpisode(animeSearch)
-		selectVideo.SelectVideo(episodeSelected)
-
+		episodeSelected, nextEpisode := episode.SelectEpisode(animeSearch)
+		videoSelected := selectVideo.SelectVideo(episodeSelected)
+		if nextEpisode != nil {
+			go selectVideo.SelectVideo(*nextEpisode)
+		}
+		util.PlayVideo(videoSelected.Url, videoSelected.Name)
 	default:
 		animeMostWatched := mostWatched.MostWatched()
 		episodeSelected, nextEpisode := episode.SelectEpisode(animeMostWatched)
-		fmt.Printf("EP SELECIONADO: %s. PROXIMO EP: %s\n", episodeSelected, nextEpisode)
 		videoSelected := selectVideo.SelectVideo(episodeSelected)
-		go selectVideo.SelectVideo(nextEpisode)
+		if nextEpisode != nil {
+			go selectVideo.SelectVideo(*nextEpisode)
+		}
 		util.PlayVideo(videoSelected.Url, videoSelected.Name)
 	}
 

diff --git a/pkg/episode/episode.go b/pkg/episode/episode.go
@@ -13,7 +13,7 @@ import (
 	"github.com/gocolly/colly"
 )
 
-func SelectEpisode(URL string) (string, string) {
+func SelectEpisode(URL string) (string, *string) {
 	var selectedOption int
 	var episodes []models.Anime
 
@@ -56,5 +56,9 @@ func SelectEpisode(URL string) (string, string) {
 
 	go presence.Presence("https:"+image, nameAnime, fmt.Sprintf("Episódio %02d", selectedOption), "https://www.stickersdevs.com.br/wp-content/uploads/2022/01/gopher-adesivo-sticker.png")
 
-	return episodes[selectedOption-1].URL, episodes[(selectedOption+1)-1].URL
+	if selectedOption == len(episodes) {
+		return episodes[selectedOption-1].URL, nil
+	}
+
+	return episodes[selectedOption-1].URL, &episodes[(selectedOption+1)-1].URL
 }
diff --git a/pkg/random/random.go b/pkg/random/random.go
@@ -3,9 +3,16 @@ package random
 import (
 	"github.com/Caixetadev/gophimation/pkg/episode"
 	"github.com/Caixetadev/gophimation/pkg/selectVideo"
+	"github.com/Caixetadev/gophimation/pkg/util"
 )
 
 func Random() {
-	s, _ := episode.SelectEpisode("random")
-	selectVideo.SelectVideo(s)
+	episodeSelected, nextEpisode := episode.SelectEpisode("random")
+
+	videoSelected := selectVideo.SelectVideo(episodeSelected)
+	if nextEpisode != nil {
+		go selectVideo.SelectVideo(*nextEpisode)
+	}
+
+	util.PlayVideo(videoSelected.Url, videoSelected.Name)
 }