added score command

3 files changed, 115 insertions, 3 deletions

diff --git a/ui/episodes.go b/ui/episodes.go
index 8b1bcff..69e2463 100644
--- a/ui/episodes.go
+++ b/ui/episodes.go
@@ -31,11 +31,11 @@ import (
 
 // very short name I know
 func CreateEpisodeUpdateConfirmationMessage(title string, prevEpNum, epNum int) string {
-  return fmt.Sprintf("Set Episodes Watched for \x1b[35m%s\x1b[0m from \x1b[1;33m%d\x1b[0m to \x1b[1;36m%d\x1b[0m.", title, prevEpNum, epNum)
+  return fmt.Sprintf("\x1b[35m%s\x1b[0m Episodes :: \x1b[1;33m%d\x1b[0m -> \x1b[1;36m%d\x1b[0m", title, prevEpNum, epNum)
 }
 
 func CreateChapterUpdateConfirmationMessage(title string, prevChNum, chNum int) string {
-  return fmt.Sprintf("Set Chapters Read for \x1b[35m%s\x1b[0m from \x1b[1;33m%d\x1b[0m to \x1b[1;36m%d\x1b[0m.", title, prevChNum, chNum)
+  return fmt.Sprintf("\x1b[35m%s\x1b[0m Chapters :: \x1b[1;33m%d\x1b[0m -> \x1b[1;36m%d\x1b[0m", title, prevChNum, chNum)
 }
 
 func EpisodeInput(anime a.Anime) {
diff --git a/ui/score.go b/ui/score.go
new file mode 100644
index 0000000..6f0d92a
--- /dev/null
+++ b/ui/score.go
@@ -0,0 +1,112 @@
+/*
+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 ui
+
+import (
+  "strconv"
+  "errors"
+  "fmt"
+  "os"
+  "github.com/MikunoNaka/macli/mal"
+  // m "github.com/MikunoNaka/MAL2Go/v2/manga"
+  p "github.com/manifoldco/promptui"
+)
+
+// to print dropped in red color, etc
+func GetColorCodeByScore(score int) string {
+  switch score {
+    case 0:
+      return "\x1b[37m"
+    case 1:
+      return "\x1b[31m"
+    case 2:
+      return "\x1b[1;31m"
+    case 3:
+      return "\x1b[35m"
+    case 4:
+      return "\x1b[1;35m"
+    case 5:
+      return "\x1b[33m"
+    case 6:
+      return "\x1b[1;33m"
+    case 7:
+      return "\x1b[36m"
+    case 8:
+      return "\x1b[1;36m"
+    case 9:
+      return "\x1b[32m"
+    case 10:
+      return "\x1b[1;32m"
+    default:
+      return ""
+  }
+}
+
+func FormatScore(score int) string {
+  return fmt.Sprintf("%s%d\x1b[0m", GetColorCodeByScore(score), score)
+}
+
+// very short name I know
+func CreateScoreUpdateConfirmationMessage(title string, prevScore, score int) string {
+  return fmt.Sprintf("\x1b[35m%s\x1b[0m Score :: %s -> %s", title, FormatScore(prevScore), FormatScore(score))
+}
+
+func ScoreInput(id, currentScore int, title string, isManga bool) {
+  validate := func(input string) error {
+    i, err := strconv.ParseFloat(input, 64)
+    if err != nil || i < 0 || i > 10 {
+      return errors.New("Input must be a number within 0-10.")
+    }
+    return nil
+  }
+
+  template := &p.PromptTemplates {
+    Valid: "\x1b[0m{{ . | magenta }}",
+    Invalid: "\x1b[0m{{ . | magenta }}\x1b[31m ",
+    Success: "{{ . | cyan }}",
+  }
+
+  prompt := p.Prompt {
+    Label: fmt.Sprintf("Set Score (Current: %d): ", currentScore),
+    Templates: template,
+    Validate:  validate,
+  }
+
+  res, err := prompt.Run()
+  if err != nil {
+    fmt.Println("Error Running score input Prompt.", err.Error())
+    os.Exit(1)
+  }
+
+  score, err := strconv.Atoi(res)
+  if err != nil {
+    fmt.Println("Error while parsing score input:", err)
+  }
+
+  var newScore int
+  if isManga {
+    resp := mal.SetMangaScore(id, score)
+    newScore = resp.Score
+  } else {
+    resp := mal.SetAnimeScore(id, score)
+    newScore = resp.Score
+  }
+
+  fmt.Println(CreateScoreUpdateConfirmationMessage(title, currentScore, newScore))
+}
diff --git a/ui/status.go b/ui/status.go
index 7e8687e..7a0e538 100644
--- a/ui/status.go
+++ b/ui/status.go
@@ -47,7 +47,7 @@ func GetColorCodeByStatus(status string) string {
     case "plan_to_watch", "plan_to_read":
       return "\x1b[36m"
     case "":
-      return "\x1b[38;5;7m"
+      return "\x1b[37m"
     default:
       return ""
   }