arimelody.space
Helper tool for stitching together livestream VOD segments and uploading them to YouTube!
fix oauth2 storage and reuse
pretty sure this isn't the *best* approach, but it's an approach that
works.
+24
-14
main.go
+24
-14
main.go
···
214
215
// youtube oauth flow
216
ctx := context.Background()
217
var token *oauth2.Token
218
if cfg.Token != nil {
219
token = cfg.Token
220
} else {
221
-
token, err = generateOAuthToken(&ctx, cfg)
222
if err != nil {
223
log.Fatalf("OAuth flow failed: %v", err)
224
os.Exit(1)
225
}
226
cfg.Token = token
227
-
err = config.WriteConfig(cfg, config.CONFIG_FILENAME)
228
-
if err != nil {
229
-
log.Fatalf("Failed to save OAuth token: %v", err)
230
-
}
231
}
232
233
// okay actually upload now!
234
-
ytVideo, err := yt.UploadVideo(ctx, token, video)
235
if err != nil {
236
log.Fatalf("Failed to upload video: %v", err)
237
os.Exit(1)
···
283
return err
284
}
285
286
-
func generateOAuthToken(ctx *context.Context, cfg *config.Config) (*oauth2.Token, error) {
287
-
oauth2Config := &oauth2.Config{
288
-
ClientID: cfg.Google.ClientID,
289
-
ClientSecret: cfg.Google.ClientSecret,
290
-
Endpoint: google.Endpoint,
291
-
Scopes: []string{ youtube.YoutubeScope },
292
-
RedirectURL: cfg.RedirectUri,
293
-
}
294
verifier := oauth2.GenerateVerifier()
295
296
var token *oauth2.Token
···
214
215
// youtube oauth flow
216
ctx := context.Background()
217
+
oauth2Config := &oauth2.Config{
218
+
ClientID: cfg.Google.ClientID,
219
+
ClientSecret: cfg.Google.ClientSecret,
220
+
Endpoint: google.Endpoint,
221
+
Scopes: []string{ youtube.YoutubeScope },
222
+
RedirectURL: cfg.RedirectUri,
223
+
}
224
var token *oauth2.Token
225
if cfg.Token != nil {
226
token = cfg.Token
227
} else {
228
+
token, err = generateOAuthToken(&ctx, oauth2Config, cfg)
229
if err != nil {
230
log.Fatalf("OAuth flow failed: %v", err)
231
os.Exit(1)
232
}
233
cfg.Token = token
234
+
}
235
+
tokenSource := oauth2Config.TokenSource(ctx, token)
236
+
if err != nil {
237
+
log.Fatalf("Failed to create OAuth2 token source: %v", err)
238
+
os.Exit(1)
239
+
}
240
+
241
+
err = config.WriteConfig(cfg, config.CONFIG_FILENAME)
242
+
if err != nil {
243
+
log.Fatalf("Failed to save OAuth token: %v", err)
244
}
245
246
// okay actually upload now!
247
+
ytVideo, err := yt.UploadVideo(ctx, tokenSource, video)
248
if err != nil {
249
log.Fatalf("Failed to upload video: %v", err)
250
os.Exit(1)
···
296
return err
297
}
298
299
+
func generateOAuthToken(
300
+
ctx *context.Context,
301
+
oauth2Config *oauth2.Config,
302
+
cfg *config.Config,
303
+
) (*oauth2.Token, error) {
304
verifier := oauth2.GenerateVerifier()
305
306
var token *oauth2.Token
+6
-2
youtube/youtube.go
+6
-2
youtube/youtube.go
···
168
return strings.TrimSpace(out.String()), err
169
}
170
171
-
func UploadVideo(ctx context.Context, token *oauth2.Token, video *Video) (*youtube.Video, error) {
172
title, err := BuildTitle(video)
173
if err != nil {
174
return nil, fmt.Errorf("failed to build title: %v", err)
···
181
service, err := youtube.NewService(
182
ctx,
183
option.WithScopes(youtube.YoutubeUploadScope),
184
-
option.WithTokenSource(oauth2.StaticTokenSource(token)),
185
)
186
if err != nil {
187
log.Fatalf("Failed to create youtube service: %v\n", err)
···
168
return strings.TrimSpace(out.String()), err
169
}
170
171
+
func UploadVideo(
172
+
ctx context.Context,
173
+
tokenSource oauth2.TokenSource,
174
+
video *Video,
175
+
) (*youtube.Video, error) {
176
title, err := BuildTitle(video)
177
if err != nil {
178
return nil, fmt.Errorf("failed to build title: %v", err)
···
185
service, err := youtube.NewService(
186
ctx,
187
option.WithScopes(youtube.YoutubeUploadScope),
188
+
option.WithTokenSource(tokenSource),
189
)
190
if err != nil {
191
log.Fatalf("Failed to create youtube service: %v\n", err)