package youtube import ( "context" "fmt" "path" "strings" "time" "github.com/sirupsen/logrus" "google.golang.org/api/option" "google.golang.org/api/youtube/v3" "git.luzifer.io/luzifer/publish-vod/pkg/uploader" vaultoauth2 "git.luzifer.io/luzifer/publish-vod/pkg/vault-oauth2" oauth4youtube "git.luzifer.io/luzifer/publish-vod/pkg/vault-oauth2/youtube" "github.com/Luzifer/go_helpers/v2/fieldcollection" ) type ( Uploader struct{} ) var ( ptrStringEmpty = func(v string) *string { return &v }("") _ uploader.Uploader = Uploader{} ) func (Uploader) Prepare(ctx context.Context, opts uploader.UploaderOpts) error { _, ts, err := oauth4youtube.GetOAuth2Client(ctx, opts.Config.MustString("vaultKey", nil)) if err != nil { return fmt.Errorf("getting oauth credentials: %w", err) } t, err := ts.Token() if err != nil { return fmt.Errorf("getting token to store: %w", err) } if err := vaultoauth2.SaveTokenToVault(opts.Config.MustString("vaultKey", nil), t); err != nil { return fmt.Errorf("storing token to vault: %w", err) } return nil } func (Uploader) UploadFile(ctx context.Context, opts uploader.UploaderOpts) error { client, ts, err := oauth4youtube.GetOAuth2Client(ctx, opts.Config.MustString("vaultKey", nil)) if err != nil { return fmt.Errorf("getting oauth credentials: %w", err) } defer func() { t, err := ts.Token() if err != nil { logrus.WithError(err).Error("getting token to store back") } if err := vaultoauth2.SaveTokenToVault(opts.Config.MustString("vaultKey", nil), t); err != nil { logrus.WithError(err).Error("storing token back to vault") } }() service, err := youtube.NewService(ctx, option.WithHTTPClient(client)) if err != nil { return fmt.Errorf("creating Youtube client: %w", err) } upload := &youtube.Video{ Snippet: &youtube.VideoSnippet{ CategoryId: "20", // Gaming ChannelId: opts.Config.MustString("channel", nil), DefaultAudioLanguage: "de", DefaultLanguage: "de", Description: opts.Config.MustString("description", ptrStringEmpty), LiveBroadcastContent: "none", Title: strings.ReplaceAll(path.Base(opts.Filename), path.Ext(opts.Filename), ""), }, Status: &youtube.VideoStatus{ Embeddable: true, License: "youtube", MadeForKids: false, PrivacyStatus: "private", PublicStatsViewable: true, }, RecordingDetails: &youtube.VideoRecordingDetails{ RecordingDate: time.Now().Format(time.RFC3339), }, } progressReader := opts.ProgressBar.NewProxyReader(opts.Content) resp, err := service.Videos. Insert([]string{"snippet", "status", "recordingDetails"}, upload). Media(progressReader). Do() if err != nil { return fmt.Errorf("inserting video: %w", err) } opts.FinalMessage("Video uploaded: https://youtu.be/%s", resp.Id) return nil } func (Uploader) ValidateConfig(config *fieldcollection.FieldCollection) error { for _, key := range []string{"channel", "vaultKey"} { if v, err := config.String(key); err != nil || v == "" { return fmt.Errorf("key %q must be non-empty string", key) } } return nil }