added volumes command

3 files changed, 151 insertions, 1 deletions

diff --git a/cmd/volumes.go b/cmd/volumes.go
new file mode 100644
index 0000000..aa9ad7a
--- /dev/null
+++ b/cmd/volumes.go
@@ -0,0 +1,101 @@
+/*
+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 (
+	"os"
+	"fmt"
+	"strings"
+	"github.com/MikunoNaka/macli/ui"
+	"github.com/MikunoNaka/macli/mal"
+	m "github.com/MikunoNaka/MAL2Go/v4/manga"
+
+	"github.com/spf13/cobra"
+)
+
+// statusCmd represents the status command
+var volumesCmd = &cobra.Command{
+	Use:   "volumes",
+	Short: "Set the number of volumes read",
+	Long: "Set the number of volumes read" +
+	"\n" +
+    "Example Usage:\n" +
+	" - \x1b[33m`macli volumes <manga-name>`\x1b[0m For interactive prompt (manga-name can be omitted)\n" +
+	" - \x1b[33m`macli volumes -s 4 <manga-name>`\x1b[0m to set the volumes to 4\n" +
+	" - \x1b[33m`macli volumes -s +1 <manga-name>`\x1b[0m to increment the volumes by 1\n" +
+	" - \x1b[33m`macli volumes -s -2 <manga-name>`\x1b[0m to decrement the volumes by 2\n",
+	Run: func(cmd *cobra.Command, args []string) {
+    mal.Init()
+		var selectedManga m.Manga
+		if entryId > 0 {
+			selectedManga = mal.GetMangaData(entryId, []string{"my_list_status", "num_volumes"})
+		}
+
+		searchInput := strings.Join(args, " ")
+	    if searchInput == "" && entryId < 1 {
+		    var promptText string
+			if queryOnlyMode {
+				promptText = "Search Manga to Get Amount of Volumes Read For: "
+			} else {
+				promptText = "Search Manga to Set Volumes For: "
+			}
+	    	searchInput = ui.TextInput(promptText, "Search can't be blank.")
+	    }
+
+		var (
+			volInput string
+			err     error
+		)
+		if !queryOnlyMode {
+		  volInput, err = cmd.Flags().GetString("set-value")
+		  if err != nil {
+		  	fmt.Println("Error while reading \x1b[33m--set-value\x1b[0m flag.", err.Error())
+		  	os.Exit(1)
+		  }
+		}
+
+		if entryId < 1 {
+			manga := ui.MangaSearch("Select Manga:", searchInput)
+			selectedManga = mal.GetMangaData(manga.Id, []string{"my_list_status", "num_volumes"})
+		}
+
+		if queryOnlyMode {
+		    fmt.Printf("%s / \x1b[1;33m%d\n", ui.CreateVolumeUpdateConfirmationMessage(selectedManga.Title, selectedManga.MyListStatus.VolumesRead, -1), selectedManga.NumVolumes)
+			os.Exit(0)
+		}
+
+		if volInput == "" {
+			ui.VolumeInput(selectedManga)
+		} else {
+			resp := mal.SetVolumes(selectedManga.Id, selectedManga.MyListStatus.VolumesRead, volInput)
+		    fmt.Println(ui.CreateVolumeUpdateConfirmationMessage(selectedManga.Title, resp.VolumesRead, selectedManga.MyListStatus.VolumesRead))
+		}
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(volumesCmd)
+    volumesCmd.Flags().StringP("set-value", "s", "", "Number of voulmes")
+    volumesCmd.Flags().IntVarP(&ui.PromptLength, "prompt-length", "l", 5, "Length of select prompt")
+    volumesCmd.Flags().IntVarP(&mal.SearchLength, "search-length", "n", 10, "Amount of search results to load")
+    volumesCmd.Flags().IntVarP(&mal.SearchOffset, "search-offset", "o", 0, "Offset for the search results")
+    volumesCmd.Flags().BoolVarP(&mal.SearchNSFW, "search-nsfw", "", false, "Include NSFW-rated items in search results")
+    volumesCmd.Flags().BoolVarP(&queryOnlyMode, "query", "q", false, "Query only (don't update data)")
+    volumesCmd.Flags().IntVarP(&entryId, "id", "i", -1, "Manually specify the ID of manga (overrides search)")
+}
diff --git a/mal/episodes.go b/mal/episodes.go
index 43b69c0..0940af5 100644
--- a/mal/episodes.go
+++ b/mal/episodes.go
@@ -43,3 +43,12 @@ func SetChapters(mangaId, prevValue int, ch string) m.UpdateResponse {
   }
   return res
 }
+
+func SetVolumes(mangaId, prevValue int, vol string) m.UpdateResponse {
+  res, err := userMangaClient.SetVolumesRead(mangaId, util.ParseNumeric(vol, prevValue))
+  if err != nil {
+    fmt.Println("MyAnimeList returned error while updating volumes:", err)
+    os.Exit(1)
+  }
+  return res
+}
diff --git a/ui/episodes.go b/ui/episodes.go
index e6ac976..b4137ed 100644
--- a/ui/episodes.go
+++ b/ui/episodes.go
@@ -44,6 +44,13 @@ func CreateChapterUpdateConfirmationMessage(title string, chNum, prevChNum int)
   return fmt.Sprintf("\x1b[35m%s\x1b[0m Chapters Read :: \x1b[1;36m%d\x1b[0m", title, chNum)
 }
 
+func CreateVolumeUpdateConfirmationMessage(title string, volNum, prevVolNum int) string {
+  if prevVolNum >= 0 {
+    return fmt.Sprintf("\x1b[35m%s\x1b[0m Volumes Read :: \x1b[1;33m%d\x1b[0m -> \x1b[1;36m%d\x1b[0m", title, prevVolNum, volNum)
+  }
+  return fmt.Sprintf("\x1b[35m%s\x1b[0m Volumes Read :: \x1b[1;36m%d\x1b[0m", title, volNum)
+}
+
 func EpisodeInput(anime a.Anime) {
   epWatchedNum := anime.MyListStatus.EpWatched
   epTotalNum := anime.NumEpisodes
@@ -107,5 +114,38 @@ func ChapterInput(manga m.Manga) {
   }
 
   resp := mal.SetChapters(manga.Id, chReadNum, res)
-  fmt.Println(CreateEpisodeUpdateConfirmationMessage(manga.Title, resp.ChaptersRead, chReadNum))
+  fmt.Println(CreateChapterUpdateConfirmationMessage(manga.Title, resp.ChaptersRead, chReadNum))
+}
+
+func VolumeInput(manga m.Manga) {
+  volReadNum := manga.MyListStatus.VolumesRead
+  volTotalNum := manga.NumVolumes
+
+  validate := func(input string) error {
+    if _, err := strconv.ParseFloat(input, 64); err != nil {
+      return errors.New("Input must be a number.")
+    }
+    return nil
+  }
+
+  template := &p.PromptTemplates {
+    Valid: "\x1b[0m{{ . | magenta }}",
+    Invalid: "\x1b[0m{{ . | magenta }}\x1b[31m",
+    Success: "{{ . | cyan }}",
+  }
+
+  prompt := p.Prompt {
+    Label: fmt.Sprintf("Set Volume Number (%d/%d read): ", volReadNum, volTotalNum),
+    Templates: template,
+    Validate:  validate,
+  }
+
+  res, err := prompt.Run()
+  if err != nil {
+    fmt.Println("Error Running volume input Prompt.", err.Error())
+    os.Exit(1)
+  }
+
+  resp := mal.SetVolumes(manga.Id, volReadNum, res)
+  fmt.Println(CreateVolumeUpdateConfirmationMessage(manga.Title, resp.VolumesRead, volReadNum))
 }