Merge pull request #12 from mist8kengas/develop

[✨] Add command flags to update command

Author: CosmicPredator

cmd/cmd_add.go

@@ -24,8 +24,8 @@ func handleMediaAdd(cmd *cobra.Command, args []string) {
 	err = viewmodel.HandleMediaUpdate(
 		viewmodel.MediaUpdateParams{
 			IsNewAddition: true,
-			MediaId: mediaId,
-			Status: mediaAddStatus,
+			MediaId:       mediaId,
+			Status:        mediaAddStatus,
 		},
 	)
 	if err != nil {

cmd/cmd_list.go

@@ -22,7 +22,7 @@ var mediaListCmd = &cobra.Command{
 	Use:     "list",
 	Short:   "List your current anime/manga list",
 	Aliases: []string{"ls"},
-	Run: handleLs,
+	Run:     handleLs,
 }
 
 func init() {

cmd/cmd_media_search.go

@@ -86,7 +86,7 @@ var mediaSearchCmd = &cobra.Command{
 	Use:   "search [query...]",
 	Short: "Search for anime and manga",
 	Args:  cobra.MinimumNArgs(1),
-	Run: handleMediaSearch,
+	Run:   handleMediaSearch,
 }
 
 func init() {

cmd/cmd_update.go

@@ -11,6 +11,9 @@ import (
 
 // // TODO: Update progress relatively. For example "+2", "-10" etc.,
 var progress string
+var updateStatus string
+var notes string
+var scoreString string
 
 // func handleUpdate(mediaId int) {
 // 	CheckIfTokenExists()
@@ -48,15 +51,32 @@ func handleUpdate(cmd *cobra.Command, args []string) {
 		)
 	}
 
-	err = viewmodel.HandleMediaUpdate(
-		viewmodel.MediaUpdateParams{
-			IsNewAddition: false,
-			MediaId: id,
-			Progress: progress,
-			Status: "none",
-			StartDate: "none",
-		},
-	)
+	var scoreFloat *float64
+	if scoreString != "" {
+		rawScoreFloat, err := strconv.ParseFloat(scoreString, 32)
+		if err != nil {
+			fmt.Println(ui.ErrorText(err))
+			return
+		}
+		scoreFloat = &rawScoreFloat
+	}
+
+	params := viewmodel.MediaUpdateParams{
+		IsNewAddition: false,
+		MediaId:       id,
+		Progress:      progress,
+		Status:        updateStatus,
+		StartDate:     "none",
+	}
+	// TODO: add a way to differentiate between an
+	// empty notes value vs. an unset notes value
+	if notes != "\n" {
+		params.Notes = notes
+	}
+	if scoreFloat != nil {
+		params.Score = float32(*scoreFloat)
+	}
+	err = viewmodel.HandleMediaUpdate(params)
 
 	if err != nil {
 		fmt.Println(ui.ErrorText(err))
@@ -67,15 +87,28 @@ var mediaUpdateCmd = &cobra.Command{
 	Use:   "update [id]",
 	Short: "Update a list entry",
 	Args:  cobra.MinimumNArgs(1),
-	Run: handleUpdate,
+	Run:   handleUpdate,
 }
 
 func init() {
 	mediaUpdateCmd.Flags().StringVarP(
 		&progress,
 		"progress",
 		"p",
-		"0",
+		"",
 		"The number of episodes/chapter to update",
 	)
+	mediaUpdateCmd.Flags().StringVarP(
+		&updateStatus, "status", "s", "none", "Status of the media. Can be 'watching/w or reading/r', 'planning/p', 'completed/c', 'dropped/d', 'paused/ps'",
+	)
+	mediaUpdateCmd.Flags().StringVarP(
+		&notes,
+		"notes",
+		"n",
+		"\n",
+		"Text notes. Note: you can add multiple lines by typing \"\\n\" and wrapping the note in double quotes",
+	)
+	mediaUpdateCmd.Flags().StringVarP(
+		&scoreString, "score", "r", "", "The score of the entry. If your score is in emoji, type 1 for 😞, 2 for 😐 and 3 for 😊",
+	)
 }

internal/api/anilist.go

@@ -145,4 +145,4 @@ func UpdateMediaEntry(params map[string]any) (*responses.MediaUpdateResponse, er
 	}
 
 	return &responseStruct, nil
-}
+}

internal/api/mutations.go

@@ -37,4 +37,4 @@ const mediaEntryUpdateMutation = `mutation(
 			}
 		}
     }
-}`
+}`

internal/api/responses/media_list.go

@@ -11,9 +11,9 @@ type ListCollection struct {
 				Title struct {
 					UserPreferred string `json:"userPreferred"`
 				} `json:"title"`
-				Chapters *int `json:"chapters"`
-				Volumes  *int `json:"volumes"`
-				Episodes *int `json:"episodes"`
+				Chapters    *int   `json:"chapters"`
+				Volumes     *int   `json:"volumes"`
+				Episodes    *int   `json:"episodes"`
 				MediaFormat string `json:"format"`
 			} `json:"media"`
 		} `json:"entries"`

internal/api/responses/media_search.go

@@ -6,8 +6,8 @@ type MediaSearchList struct {
 		UserPreferred string `json:"userPreferred"`
 	} `json:"title"`
 	AverageScore *float64 `json:"averageScore"`
-	MediaType    string  `json:"type"`
-	MediaFormat  string  `json:"format"`
+	MediaType    string   `json:"type"`
+	MediaFormat  string   `json:"format"`
 }
 
 type MediaSearch struct {

internal/api/responses/media_update_response.go

@@ -4,11 +4,11 @@ type MediaUpdateResponse struct {
 	Data struct {
 		SaveMediaListEntry struct {
 			Media struct {
-				Id int `json:"id"`
+				Id    int `json:"id"`
 				Title struct {
 					UserPreferred string `json:"userPreferred"`
 				} `json:"title"`
 			} `json:"media"`
 		} `json:"SaveMediaListEntry"`
 	} `json:"data"`
-}
+}

internal/constants.go

@@ -4,16 +4,16 @@ package internal
 
 // Base URL is not gonna get changed for a while.
 // So, keeping it as constant is not gonna hurt anyone.
-const(
-	API_ENDPOINT = "https://graphql.anilist.co"
-	AUTH_URL = "https://anilist.co/api/v2/oauth/authorize?client_id=4593&response_type=token"
+const (
+	API_ENDPOINT     = "https://graphql.anilist.co"
+	AUTH_URL         = "https://anilist.co/api/v2/oauth/authorize?client_id=4593&response_type=token"
 	BOLT_BUCKET_NAME = "ChibiConfig"
-	BOLT_DB_NAME = "chibi_config.db"
+	BOLT_DB_NAME     = "chibi_config.db"
 )
 
-type MediaType string 
+type MediaType string
 
 const (
 	ANIME MediaType = "ANIME"
 	MANGA MediaType = "MANGA"
-)
+)