yutu/pkg/caption/caption.go

package caption

import (
	"errors"
	"fmt"
	"github.com/eat-pray-ai/yutu/pkg/auth"
	"github.com/eat-pray-ai/yutu/pkg/utils"
	"google.golang.org/api/youtube/v3"
	"io"
	"log"
	"os"
)

var (
	service            *youtube.Service
	errGetCaption      = errors.New("failed to get caption")
	errUpdateCaption   = errors.New("failed to update caption")
	errDeleteCaption   = errors.New("failed to delete caption")
	errInsertCaption   = errors.New("failed to insert caption")
	errDownloadCaption = errors.New("failed to download caption")
)

type caption struct {
	ID                     string `yaml:"id" json:"id"`
	File                   string `yaml:"file" json:"file"`
	AudioTrackType         string `yaml:"audio_track_type" json:"audio_track_type"`
	IsAutoSynced           *bool  `yaml:"is_auto_synced" json:"is_auto_synced"`
	IsCC                   *bool  `yaml:"is_cc" json:"is_cc"`
	IsDraft                *bool  `yaml:"is_draft" json:"is_draft"`
	IsEasyReader           *bool  `yaml:"is_easy_reader" json:"is_easy_reader"`
	IsLarge                *bool  `yaml:"is_large" json:"is_large"`
	Language               string `yaml:"language" json:"language"`
	Name                   string `yaml:"name" json:"name"`
	TrackKind              string `yaml:"track_kind" json:"track_kind"`
	OnBehalfOf             string `yaml:"on_behalf_of" json:"on_behalf_of"`
	OnBehalfOfContentOwner string `yaml:"on_behalf_of_content_owner" json:"on_behalf_of_content_owner"`
	VideoId                string `yaml:"video_id" json:"video_id"`
	Tfmt                   string `yaml:"tfmt" json:"tfmt"`
	Tlang                  string `yaml:"tlang" json:"tlang"`
}

type Caption interface {
	get(parts []string) []*youtube.Caption
	List(parts []string, output string)
	Insert(output string)
	Update(output string)
	Delete()
	Download()
}

type Option func(*caption)

func NewCation(opts ...Option) Caption {
	c := &caption{}
	for _, opt := range opts {
		opt(c)
	}
	return c
}

func (c *caption) get(parts []string) []*youtube.Caption {
	call := service.Captions.List(parts, c.VideoId)
	if c.ID != "" {
		call = call.Id(c.ID)
	}
	if c.OnBehalfOf != "" {
		call = call.OnBehalfOf(c.OnBehalfOf)
	}
	if c.OnBehalfOfContentOwner != "" {
		call = call.OnBehalfOfContentOwner(c.OnBehalfOfContentOwner)
	}

	res, err := call.Do()
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errGetCaption, err))
	}

	return res.Items
}

func (c *caption) List(parts []string, output string) {
	captions := c.get(parts)
	switch output {
	case "json":
		utils.PrintJSON(captions)
	case "yaml":
		utils.PrintYAML(captions)
	default:
		fmt.Println("ID\tName")
		for _, caption := range captions {
			fmt.Printf("%s\t%s\n", caption.Id, caption.Snippet.Name)
		}
	}
}

func (c *caption) Insert(output string) {
	file, err := os.Open(c.File)
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errInsertCaption, err))
	}
	defer file.Close()

	caption := &youtube.Caption{
		Snippet: &youtube.CaptionSnippet{
			AudioTrackType: c.AudioTrackType,
			IsAutoSynced:   *c.IsAutoSynced,
			IsCC:           *c.IsCC,
			IsDraft:        *c.IsDraft,
			IsEasyReader:   *c.IsEasyReader,
			IsLarge:        *c.IsLarge,
			Language:       c.Language,
			Name:           c.Name,
			TrackKind:      c.TrackKind,
			VideoId:        c.VideoId,
		},
	}

	call := service.Captions.Insert([]string{"snippet"}, caption).Media(file)
	if c.OnBehalfOf != "" {
		call = call.OnBehalfOf(c.OnBehalfOf)
	}
	if c.OnBehalfOfContentOwner != "" {
		call = call.OnBehalfOfContentOwner(c.OnBehalfOfContentOwner)
	}

	res, err := call.Do()
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errInsertCaption, err))
	}

	switch output {
	case "json":
		utils.PrintJSON(res)
	case "yaml":
		utils.PrintYAML(res)
	case "silent":
	default:
		fmt.Printf("Caption inserted: %s\n", res.Id)
	}
}

func (c *caption) Update(output string) {
	caption := c.get([]string{"snippet"})[0]
	if c.AudioTrackType != "" {
		caption.Snippet.AudioTrackType = c.AudioTrackType
	}
	if c.IsAutoSynced != nil {
		caption.Snippet.IsAutoSynced = *c.IsAutoSynced
	}
	if c.IsCC != nil {
		caption.Snippet.IsCC = *c.IsCC
	}
	if c.IsDraft != nil {
		caption.Snippet.IsDraft = *c.IsDraft
	}
	if c.IsEasyReader != nil {
		caption.Snippet.IsEasyReader = *c.IsEasyReader
	}
	if c.IsLarge != nil {
		caption.Snippet.IsLarge = *c.IsLarge
	}
	if c.Language != "" {
		caption.Snippet.Language = c.Language
	}
	if c.Name != "" {
		caption.Snippet.Name = c.Name
	}
	if c.TrackKind != "" {
		caption.Snippet.TrackKind = c.TrackKind
	}
	if c.VideoId != "" {
		caption.Snippet.VideoId = c.VideoId
	}

	call := service.Captions.Update([]string{"snippet"}, caption)
	if c.File != "" {
		file, err := os.Open(c.File)
		if err != nil {
			utils.PrintJSON(c)
			log.Fatalln(errors.Join(errUpdateCaption, err))
		}
		defer file.Close()
		call = call.Media(file)
	}
	if c.OnBehalfOf != "" {
		call = call.OnBehalfOf(c.OnBehalfOf)
	}
	if c.OnBehalfOfContentOwner != "" {
		call = call.OnBehalfOfContentOwner(c.OnBehalfOfContentOwner)
	}

	res, err := call.Do()
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errUpdateCaption, err))
	}

	switch output {
	case "json":
		utils.PrintJSON(res)
	case "yaml":
		utils.PrintYAML(res)
	case "silent":
	default:
		fmt.Printf("Caption updated: %s\n", res.Id)
	}
}

func (c *caption) Delete() {
	call := service.Captions.Delete(c.ID)
	if c.OnBehalfOf != "" {
		call = call.OnBehalfOf(c.OnBehalfOf)
	}
	if c.OnBehalfOfContentOwner != "" {
		call = call.OnBehalfOfContentOwner(c.OnBehalfOfContentOwner)
	}

	err := call.Do()
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errDeleteCaption, err))
	}

	fmt.Printf("Caption %s deleted\n", c.ID)
}

func (c *caption) Download() {
	call := service.Captions.Download(c.ID)
	if c.Tfmt != "" {
		call = call.Tfmt(c.Tfmt)
	}
	if c.Tlang != "" {
		call = call.Tlang(c.Tlang)
	}
	if c.OnBehalfOf != "" {
		call = call.OnBehalfOf(c.OnBehalfOf)
	}
	if c.OnBehalfOfContentOwner != "" {
		call = call.OnBehalfOfContentOwner(c.OnBehalfOfContentOwner)
	}

	res, err := call.Download()
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errDownloadCaption, err))
	}
	defer res.Body.Close()
	body, err := io.ReadAll(res.Body)
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errDownloadCaption, err))
	}

	file, err := os.Create(c.File)
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errDownloadCaption, err))
	}
	defer file.Close()
	_, err = file.Write(body)
	if err != nil {
		utils.PrintJSON(c)
		log.Fatalln(errors.Join(errDownloadCaption, err))
	}

	fmt.Printf("Caption %s downloaded to %s\n", c.ID, c.File)
}

func WithID(id string) Option {
	return func(c *caption) {
		c.ID = id
	}
}

func WithFile(file string) Option {
	return func(c *caption) {
		c.File = file
	}
}

func WithAudioTrackType(audioTrackType string) Option {
	return func(c *caption) {
		c.AudioTrackType = audioTrackType
	}
}

func WithIsAutoSynced(isAutoSynced bool, changed bool) Option {
	return func(c *caption) {
		if changed {
			c.IsAutoSynced = &isAutoSynced
		}
	}
}

func WithIsCC(isCC bool, changed bool) Option {
	return func(c *caption) {
		if changed {
			c.IsCC = &isCC
		}
	}
}

func WithIsDraft(isDraft bool, changed bool) Option {
	return func(c *caption) {
		if changed {
			c.IsDraft = &isDraft
		}
	}
}

func WithIsEasyReader(isEasyReader bool, changed bool) Option {
	return func(c *caption) {
		if changed {
			c.IsEasyReader = &isEasyReader
		}
	}
}

func WithIsLarge(isLarge bool, changed bool) Option {
	return func(c *caption) {
		if changed {
			c.IsLarge = &isLarge
		}
	}
}

func WithLanguage(language string) Option {
	return func(c *caption) {
		c.Language = language
	}
}

func WithName(name string) Option {
	return func(c *caption) {
		c.Name = name
	}
}

func WithTrackKind(trackKind string) Option {
	return func(c *caption) {
		c.TrackKind = trackKind
	}
}

func WithOnBehalfOf(onBehalfOf string) Option {
	return func(c *caption) {
		c.OnBehalfOf = onBehalfOf
	}
}

func WithOnBehalfOfContentOwner(onBehalfOfContentOwner string) Option {
	return func(c *caption) {
		c.OnBehalfOfContentOwner = onBehalfOfContentOwner
	}
}

func WithVideoId(videoId string) Option {
	return func(c *caption) {
		c.VideoId = videoId
	}
}

func WithTfmt(tfmt string) Option {
	return func(c *caption) {
		c.Tfmt = tfmt
	}
}

func WithTlang(tlang string) Option {
	return func(c *caption) {
		c.Tlang = tlang
	}
}

func WithService(svc *youtube.Service) Option {
	return func(_ *caption) {
		if svc == nil {
			svc = auth.NewY2BService(
				auth.WithCredential(""),
				auth.WithCacheToken(""),
			)
		}
		service = svc
	}
}