From 61aacac37c52833cbb90ea029f7e2cd2bf93bb96 Mon Sep 17 00:00:00 2001
From: Preston Baxter <preston@preston-baxter.com>
Date: Tue, 14 Nov 2023 13:08:53 -0600
Subject: [PATCH] B: Start youtube scheduling work

---
 service/controllers/pco_webhook.go | 41 +++++++++++++++++++++++++++---
 ui/db/vendors.go                   |  2 +-
 2 files changed, 38 insertions(+), 5 deletions(-)

diff --git a/service/controllers/pco_webhook.go b/service/controllers/pco_webhook.go
index 3a12af6..ea89369 100644
--- a/service/controllers/pco_webhook.go
+++ b/service/controllers/pco_webhook.go
@@ -12,6 +12,7 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/google/jsonapi"
 	"go.mongodb.org/mongo-driver/bson/primitive"
+	"golang.org/x/oauth2"
 	"google.golang.org/api/option"
 	"google.golang.org/api/youtube/v3"
 )
@@ -98,15 +99,47 @@ func eventMatch(event string) bool {
 }
 
 func youtubeServiceForUser(userId primitive.ObjectID) (*youtube.Service, error) {
+	if client, ok := ytClientMap[userId]; !ok {
+		ytAccount, err := mongo.FindVendorAccountByUser(userId, models.YOUTUBE_VENDOR_NAME)
+		if err != nil {
+			return nil, err
+		}
 
-	client, err := youtube.NewService(context.Background(), option.WithTokenSource(mongo.NewVendorTokenSource()))
-	if err != nil {
-		log.WithError(err).Error("Failed to init youtube service")
-		panic(err)
+		//Build our fancy token source
+		tokenSource := oauth2.ReuseTokenSource(ytAccount.Token(), mongo.NewVendorTokenSource(ytAccount))
+
+		//init service
+		client, err := youtube.NewService(context.Background(), option.WithTokenSource(tokenSource))
+		if err != nil {
+			log.WithError(err).Error("Failed to init youtube service")
+			return nil, err
+		}
+
+		//add user to map
+		ytClientMap[userId] = client
+
+		return client, nil
+	} else {
+		return client, nil
 	}
 }
 
 func ScheduleLiveStreamFromWebhook(c *gin.Context, body *webhooks.EventDelivery) error {
+	//get uid from context. Lots of sanitizing just incase
+	var uid primitive.ObjectID
+	if raw, ok := c.Get("user_bson_id"); ok {
+		uid, ok = raw.(primitive.ObjectID)
+		if !ok {
+			log.Errorf("failed to parse user id to bson object id: %v", raw)
+			return errors.New("Failed to case user id as bson object id")
+		}
+	}
+
+	client, err := youtubeServiceForUser(uid)
+	if err != nil {
+		log.WithError(err).Error("Failed to initialize youtube client")
+		return err
+	}
 
 	return nil
 }
diff --git a/ui/db/vendors.go b/ui/db/vendors.go
index c97b8d5..f63f605 100644
--- a/ui/db/vendors.go
+++ b/ui/db/vendors.go
@@ -34,7 +34,7 @@ func (db *DB) FindAllVendorAccountsByUser(userId primitive.ObjectID) ([]models.V
 }
 
 //find vendor for user by name
-func (db *DB) FindVendorAccountsByUser(userId primitive.ObjectID, name string) (*models.VendorAccount, error) {
+func (db *DB) FindVendorAccountByUser(userId primitive.ObjectID, name string) (*models.VendorAccount, error) {
 	conf := config.Config()
 
 	opts := options.FindOne()