✨ Support more flags of video category and ARR

eat-pray-ai · May 17, 2024 · 87435ac · 87435ac
1 parent 1a8b1f8
commit 87435ac
Show file tree

Hide file tree

Showing 7 changed files with 66 additions and 10 deletions.
diff --git a/cmd/i18nLanguage/list.go b/cmd/i18nLanguage/list.go
@@ -17,7 +17,7 @@ var listCmd = &cobra.Command{
 
 func init() {
 	i18nLanguageCmd.AddCommand(listCmd)
-	listCmd.Flags().StringVarP(&hl, "hl", "l", "", "host language")
+	listCmd.Flags().StringVarP(&hl, "hl", "l", "", "Host language")
 	listCmd.Flags().StringSliceVarP(
 		&parts, "parts", "p", []string{"id", "snippet"}, "Comma separated parts",
 	)

diff --git a/cmd/videoAbuseReportReason/list.go b/cmd/videoAbuseReportReason/list.go
@@ -10,14 +10,23 @@ var listCmd = &cobra.Command{
 	Short: "list YouTube video abuse report reasons",
 	Long:  "list YouTube video abuse report reasons",
 	Run: func(cmd *cobra.Command, args []string) {
-		va := yutuber.NewVideoAbuseReportReason()
+		va := yutuber.NewVideoAbuseReportReason(
+			yutuber.WithVideoAbuseReportReasonHL(hl),
+		)
 		va.List(parts, output)
 	},
 }
 
 func init() {
 	videoAbuseReportReasonCmd.AddCommand(listCmd)
 
-	listCmd.Flags().StringSliceVarP(&parts, "parts", "p", []string{"id", "snippet"}, "Comma separated parts")
-	listCmd.Flags().StringVarP(&output, "output", "o", "", "Output format: json or yaml")
+	listCmd.Flags().StringVarP(
+		&hl, "hl", "l", "", "Host language",
+	)
+	listCmd.Flags().StringSliceVarP(
+		&parts, "parts", "p", []string{"id", "snippet"}, "Comma separated parts",
+	)
+	listCmd.Flags().StringVarP(
+		&output, "output", "o", "", "Output format: json or yaml",
+	)
 }
diff --git a/cmd/videoAbuseReportReason/videoAbuseReportReason.go b/cmd/videoAbuseReportReason/videoAbuseReportReason.go
@@ -6,6 +6,7 @@ import (
 )
 
 var (
+	hl     string
 	parts  []string
 	output string
 )

diff --git a/cmd/videoCategory/list.go b/cmd/videoCategory/list.go
@@ -11,7 +11,9 @@ var listCmd = &cobra.Command{
 	Long:  "list video categories' info, such as ID, title, assignable, etc.",
 	Run: func(cmd *cobra.Command, args []string) {
 		vc := yutuber.NewVideoCategory(
-			yutuber.WithRegionCode(regionCode),
+			yutuber.WithVideoCategoryId(id),
+			yutuber.WithVideoCategoryHl(hl),
+			yutuber.WithVideoCategoryRegionCode(regionCode),
 		)
 		vc.List(parts, output)
 	},
@@ -20,7 +22,13 @@ var listCmd = &cobra.Command{
 func init() {
 	videoCategoryCmd.AddCommand(listCmd)
 
+	listCmd.Flags().StringVarP(&id, "id", "i", "", "ID of the video category")
+	listCmd.Flags().StringVarP(&hl, "hl", "l", "", "Host language")
 	listCmd.Flags().StringVarP(&regionCode, "regionCode", "r", "US", "Region code")
-	listCmd.Flags().StringSliceVarP(&parts, "parts", "p", []string{"id", "snippet"}, "Comma separated parts")
-	listCmd.Flags().StringVarP(&output, "output", "o", "", "Output format: json or yaml")
+	listCmd.Flags().StringSliceVarP(
+		&parts, "parts", "p", []string{"id", "snippet"}, "Comma separated parts",
+	)
+	listCmd.Flags().StringVarP(
+		&output, "output", "o", "", "Output format: json or yaml",
+	)
 }
diff --git a/cmd/videoCategory/videoCategory.go b/cmd/videoCategory/videoCategory.go
@@ -6,6 +6,8 @@ import (
 )
 
 var (
+	id         string
+	hl         string
 	regionCode string
 	parts      []string
 	output     string

diff --git a/pkg/yutuber/videoAbuseReportReason.go b/pkg/yutuber/videoAbuseReportReason.go
@@ -13,7 +13,9 @@ var (
 	errGetVideoAbuseReportReason = errors.New("failed to get video abuse report reason")
 )
 
-type videoAbuseReportReason struct{}
+type videoAbuseReportReason struct {
+	hl string
+}
 
 type VideoAbuseReportReason interface {
 	get([]string) []*youtube.VideoAbuseReportReason
@@ -33,6 +35,10 @@ func NewVideoAbuseReportReason(opt ...VideoAbuseReportReasonOption) VideoAbuseRe
 
 func (vc *videoAbuseReportReason) get(parts []string) []*youtube.VideoAbuseReportReason {
 	call := service.VideoAbuseReportReasons.List(parts)
+	if vc.hl != "" {
+		call = call.Hl(vc.hl)
+	}
+
 	response, err := call.Do()
 	if err != nil {
 		log.Fatalln(errors.Join(errGetVideoAbuseReportReason, err))
@@ -58,3 +64,9 @@ func (vc *videoAbuseReportReason) List(parts []string, output string) {
 		}
 	}
 }
+
+func WithVideoAbuseReportReasonHL(hl string) VideoAbuseReportReasonOption {
+	return func(vc *videoAbuseReportReason) {
+		vc.hl = hl
+	}
+}
diff --git a/pkg/yutuber/videoCategory.go b/pkg/yutuber/videoCategory.go
@@ -15,6 +15,8 @@ var (
 )
 
 type videoCategory struct {
+	id         string
+	hl         string
 	regionCode string
 }
 
@@ -35,7 +37,17 @@ func NewVideoCategory(opt ...VideoCategoryOption) VideoCategory {
 }
 
 func (vc *videoCategory) get(parts []string) []*youtube.VideoCategory {
-	call := service.VideoCategories.List(parts).RegionCode(vc.regionCode)
+	call := service.VideoCategories.List(parts)
+	if vc.id != "" {
+		call = call.Id(vc.id)
+	}
+	if vc.hl != "" {
+		call = call.Hl(vc.hl)
+	}
+	if vc.regionCode != "" {
+		call = call.RegionCode(vc.regionCode)
+	}
+
 	response, err := call.Do()
 	if err != nil {
 		log.Fatalln(errors.Join(errGetVideoCategory, err), vc.regionCode)
@@ -62,7 +74,19 @@ func (vc *videoCategory) List(parts []string, output string) {
 	}
 }
 
-func WithRegionCode(regionCode string) VideoCategoryOption {
+func WithVideoCategoryId(id string) VideoCategoryOption {
+	return func(vc *videoCategory) {
+		vc.id = id
+	}
+}
+
+func WithVideoCategoryHl(hl string) VideoCategoryOption {
+	return func(vc *videoCategory) {
+		vc.hl = hl
+	}
+}
+
+func WithVideoCategoryRegionCode(regionCode string) VideoCategoryOption {
 	return func(vc *videoCategory) {
 		vc.regionCode = regionCode
 	}
-Original file line number
+Diff line change
@@ Expand Up / @@ -6,6 +6,7 @@ import ( @@
     )
     var (
+    	hl     string
     	parts  []string
     	output string
     )
@@ Expand Down @@