diff options
| -rw-r--r-- | cmd/score.go | 40 | ||||
| -rw-r--r-- | cmd/version.go | 2 | ||||
| -rw-r--r-- | mal/episodes.go | 36 | ||||
| -rw-r--r-- | ui/score.go | 20 | ||||
| -rw-r--r-- | util/number_parser.go | 41 | 
5 files changed, 89 insertions, 50 deletions
diff --git a/cmd/score.go b/cmd/score.go index f97de87..27106e5 100644 --- a/cmd/score.go +++ b/cmd/score.go @@ -22,8 +22,11 @@ import (  	"fmt"  	"os"  	"strings" +	"strconv" +	"errors"  	"github.com/MikunoNaka/macli/ui"  	"github.com/MikunoNaka/macli/mal" +	"github.com/MikunoNaka/macli/util"  	"github.com/spf13/cobra"  ) @@ -41,7 +44,7 @@ var scoreCmd = &cobra.Command{  		mal.Init()  		searchInput := strings.Join(args, " ") -		scoreInput, err := cmd.Flags().GetInt("set-value") +		scoreInput, 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) @@ -62,7 +65,18 @@ var scoreCmd = &cobra.Command{  	},  } -func setAnimeScore(scoreInput int, searchInput string) { +func validateScore(input string, currentScore int) (int, error) { +	parsedScore := util.ParseNumeric(input, currentScore) +	if parsedScore > 10 { +        return currentScore, errors.New("\x1b[31mScore out of range (" + strconv.Itoa(parsedScore) + " > 10)\x1b[0m") +	} +	if parsedScore < 0 { +        return currentScore, errors.New("\x1b[31mScore out of range (" + strconv.Itoa(parsedScore) + " < 0)\x1b[0m") +	} +	return parsedScore, nil +} + +func setAnimeScore(scoreInput, searchInput string) {  	if searchInput == "" {  	    var promptText string  		if queryOnlyMode { @@ -82,15 +96,20 @@ func setAnimeScore(scoreInput int, searchInput string) {  		os.Exit(0)  	} -    if scoreInput < 0 { +    if scoreInput == "" {  		ui.ScoreInput(anime.Id, selectedAnime.MyListStatus.Score, anime.Title, false)      } else { -    	resp := mal.SetAnimeScore(anime.Id, scoreInput) +		score, err := validateScore(scoreInput, selectedAnime.MyListStatus.Score) +		if err != nil { +			fmt.Println(err) +			os.Exit(1) +		} +    	resp := mal.SetAnimeScore(anime.Id, score)      	fmt.Println(ui.CreateScoreUpdateConfirmationMessage(anime.Title, currentScore, resp.Score))      }  } -func setMangaScore(scoreInput int, searchInput string) { +func setMangaScore(scoreInput, searchInput string) {  	if searchInput == "" {  	    var promptText string  		if queryOnlyMode { @@ -110,15 +129,20 @@ func setMangaScore(scoreInput int, searchInput string) {  		os.Exit(0)  	} -    if scoreInput < 0 { +    if scoreInput == "" {  		ui.ScoreInput(manga.Id, selectedManga.MyListStatus.Score, manga.Title, true)      } else { -    	resp := mal.SetMangaScore(manga.Id, scoreInput) +		score, err := validateScore(scoreInput, selectedManga.MyListStatus.Score) +		if err != nil { +			fmt.Println(err) +			os.Exit(1) +		} +    	resp := mal.SetMangaScore(manga.Id, score)      	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") +    scoreCmd.Flags().StringP("set-value", "s", "", "Score to be set")  } diff --git a/cmd/version.go b/cmd/version.go index 0901980..174b7e3 100644 --- a/cmd/version.go +++ b/cmd/version.go @@ -24,7 +24,7 @@ import (  	"github.com/spf13/cobra"  ) -var version string = "v1.5.1" +var version string = "v1.5.3"  var versionCmd = &cobra.Command {  	Use:   "version", diff --git a/mal/episodes.go b/mal/episodes.go index 06bb7a0..22dbae4 100644 --- a/mal/episodes.go +++ b/mal/episodes.go @@ -21,28 +21,13 @@ package mal  import (    "fmt"    "os" -  "strconv"    a "github.com/MikunoNaka/MAL2Go/v2/user/anime"    m "github.com/MikunoNaka/MAL2Go/v2/user/manga" +  "github.com/MikunoNaka/macli/util"  )  func SetEpisodes(animeId, prevValue int, ep string) a.UpdateResponse { -  epInt, err := strconv.Atoi(ep) -  if err != nil { -    fmt.Println("Error while parsing episode input", err) -    os.Exit(1) -  } - -  var epValue int -  switch ep[0:1] { -  case "+", "-": -    // works both for increment and decrement -    epValue = prevValue + epInt -  default: -    epValue = epInt -  } - -  res, err := userAnimeClient.SetWatchedEpisodes(animeId, epValue) +  res, err := userAnimeClient.SetWatchedEpisodes(animeId, util.ParseNumeric(ep, prevValue))    if err != nil {      fmt.Println("MyAnimeList returned error while updating episodes:", err)      os.Exit(1) @@ -51,22 +36,7 @@ func SetEpisodes(animeId, prevValue int, ep string) a.UpdateResponse {  }  func SetChapters(mangaId, prevValue int, ch string) m.UpdateResponse { -  chInt, err := strconv.Atoi(ch) -  if err != nil { -    fmt.Println("Error while parsing chapter input", err) -    os.Exit(1) -  } - -  var chValue int -  switch ch[0:1] { -  case "+", "-": -    // works both for increment and decrement -    chValue = prevValue + chInt -  default: -    chValue = chInt -  } - -  res, err := userMangaClient.SetChaptersRead(mangaId, chValue) +  res, err := userMangaClient.SetChaptersRead(mangaId, util.ParseNumeric(ch, prevValue))    if err != nil {      fmt.Println("MyAnimeList returned error while updating chapters:", err)      os.Exit(1) diff --git a/ui/score.go b/ui/score.go index 4832f6a..fd83d0b 100644 --- a/ui/score.go +++ b/ui/score.go @@ -24,6 +24,7 @@ import (    "fmt"    "os"    "github.com/MikunoNaka/macli/mal" +  "github.com/MikunoNaka/macli/util"    // m "github.com/MikunoNaka/MAL2Go/v2/manga"    p "github.com/manifoldco/promptui"  ) @@ -70,9 +71,16 @@ func CreateScoreUpdateConfirmationMessage(title string, prevScore, score int) st  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 { +    if err != nil || i < -10 || i > 10 {        return errors.New("Input must be a number within 0-10.")      } +    newScore := util.ParseNumeric(input, currentScore) +    if newScore < 0 { +      return errors.New("Score out of range (" + strconv.Itoa(newScore) + " < 0)") +    } +    if newScore > 10 { +      return errors.New("Score out of range (" + strconv.Itoa(newScore) + " > 10)") +    }      return nil    } @@ -94,17 +102,13 @@ func ScoreInput(id, currentScore int, title string, isManga bool) {      os.Exit(1)    } -  score, err := strconv.Atoi(res) -  if err != nil { -    fmt.Println("Error while parsing score input:", err) -  } -    var newScore int +  parsedScore := util.ParseNumeric(res, currentScore)    if isManga { -    resp := mal.SetMangaScore(id, score) +    resp := mal.SetMangaScore(id, parsedScore)      newScore = resp.Score    } else { -    resp := mal.SetAnimeScore(id, score) +    resp := mal.SetAnimeScore(id, parsedScore)      newScore = resp.Score    } diff --git a/util/number_parser.go b/util/number_parser.go new file mode 100644 index 0000000..de46160 --- /dev/null +++ b/util/number_parser.go @@ -0,0 +1,41 @@ +/* +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 util + +import ( +  "fmt" +  "os" +  "strconv" +) + +func ParseNumeric(input string, prevValue int) int { +  inputInt, err := strconv.Atoi(input) +  if err != nil { +    fmt.Println("Error while parsing input", err) +    os.Exit(1) +  } + +  switch input[0:1] { +  case "+", "-": +    // works both for increment and decrement +    return prevValue + inputInt +  default: +    return inputInt +  } +}  |