1 files changed, 123 insertions, 0 deletions
diff --git a/cmd/score.go b/cmd/score.go
new file mode 100644
index 0000000..15a1e03
--- /dev/null
+++ b/cmd/score.go
@@ -0,0 +1,123 @@
+/*
+macli - Unofficial CLI-Based MyAnimeList Client
+Copyright © 2022 Vidhu Kant Sharma <vidhukant@vidhukant.xyz>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"strings"
+	"github.com/MikunoNaka/macli/ui"
+	"github.com/MikunoNaka/macli/mal"
+
+	"github.com/spf13/cobra"
+)
+
+// statusCmd represents the status command
+var scoreCmd = &cobra.Command{
+	Use:   "score",
+	Short: "Set an anime/manga's status",
+	Long: "Set an anime's status\n" +
+	"\n" +
+    "Example Usage:\n" +
+	" - \x1b[33m`macli status <anime-name>`\x1b[0m For interactive prompt (anime-name can be omitted)\n" +
+	" - \x1b[33m`macli status -s \x1b[34mwatching|plan_to_watch|dropped|on_hold|completed\x1b[33m <anime-name>`\x1b[0m to specify status from command\n",
+	Run: func(cmd *cobra.Command, args []string) {
+		searchInput := strings.Join(args, " ")
+
+		scoreInput, err := cmd.Flags().GetInt("set-value")
+		if err != nil {
+			fmt.Println("Error while reading \x1b[33m--set-value\x1b[0m flag.", err.Error())
+			os.Exit(1)
+		}
+
+		mangaMode, err := cmd.Flags().GetBool("manga")
+		if err != nil {
+			fmt.Println("Error while reading \x1b[33m--manga\x1b[0m flag.", err.Error())
+			os.Exit(1)
+		}
+
+		if mangaMode {
+			setMangaScore(scoreInput, searchInput)
+		} else {
+			setAnimeScore(scoreInput, searchInput)
+		}
+
+	},
+}
+
+func setAnimeScore(scoreInput int, searchInput string) {
+	if searchInput == "" {
+	    var promptText string
+		if queryOnlyMode {
+			promptText = "Search Anime to Get Score of: "
+		} else {
+			promptText = "Search Anime to Set Score of: "
+		}
+	   	searchInput = ui.TextInput(promptText, "Search can't be blank.")
+	}
+
+	anime := ui.AnimeSearch("Select Anime: ", searchInput)
+	selectedAnime := mal.GetAnimeData(anime.Id, []string{"my_list_status"})
+	currentScore := selectedAnime.MyListStatus.Score
+
+	if queryOnlyMode {
+		fmt.Printf("\x1b[35m%s\x1b[0m Score :: %s\n", anime.Title, ui.FormatScore(currentScore))
+		os.Exit(0)
+	}
+
+    if scoreInput < 0 {
+		ui.ScoreInput(anime.Id, selectedAnime.MyListStatus.Score, anime.Title, false)
+    } else {
+    	resp := mal.SetAnimeScore(anime.Id, scoreInput)
+    	fmt.Println(ui.CreateScoreUpdateConfirmationMessage(anime.Title, currentScore, resp.Score))
+    }
+}
+
+func setMangaScore(scoreInput int, searchInput string) {
+	if searchInput == "" {
+	    var promptText string
+		if queryOnlyMode {
+			promptText = "Search Manga to Get Score of: "
+		} else {
+			promptText = "Search Manga to Set Score of: "
+		}
+	   	searchInput = ui.TextInput(promptText, "Search can't be blank.")
+	}
+
+	manga := ui.MangaSearch("Select Manga: ", searchInput)
+	selectedManga := mal.GetMangaData(manga.Id, []string{"my_list_status"})
+	currentScore := selectedManga.MyListStatus.Score
+
+	if queryOnlyMode {
+		fmt.Printf("\x1b[35m%s\x1b[0m :: Score -> %s\n", manga.Title, ui.FormatScore(currentScore))
+		os.Exit(0)
+	}
+
+    if scoreInput < 0 {
+		ui.ScoreInput(manga.Id, selectedManga.MyListStatus.Score, manga.Title, false)
+    } else {
+    	resp := mal.SetAnimeScore(manga.Id, scoreInput)
+    	fmt.Println(ui.CreateScoreUpdateConfirmationMessage(manga.Title, currentScore, resp.Score))
+    }
+}
+
+func init() {
+	rootCmd.AddCommand(scoreCmd)
+    scoreCmd.Flags().IntP("set-value", "s", -1, "Score to be set")
+}