Capstone/ui/controllers/actions.go

package controllers

import (
	"errors"
	"fmt"
	"strings"

	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/config"
	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/db/models"
	"git.preston-baxter.com/Preston_PLB/capstone/webhook-service/vendors/pco"
	"git.preston-baxter.com/Preston_PLB/capstone/webhook-service/vendors/pco/webhooks"
	"github.com/gin-gonic/gin"
	"golang.org/x/oauth2"
)

type actionFunc func(user *models.User) error

var (
	actionFuncs      map[string]actionFunc            = map[string]actionFunc{"pco.plan": setupPcoSubscriptions}
	webhooksTemplate map[string]webhooks.Subscription = map[string]webhooks.Subscription{
		"services.v2.events.plan.created": {
			Active: true,
			Name:   "services.v2.events.plan.created",
			Url:    "https://%s/pco/%s",
		},
		"services.v2.events.plan.updated": {
			Active: true,
			Name:   "services.v2.events.plan.updated",
			Url:    "https://%s/pco/%s",
		},
		"services.v2.events.plan.deleted": {
			Active: true,
			Name:   "services.v2.events.plan.destroyed",
			Url:    "https://%s/pco/%s",
		},
	}
)

func AddActionFromForm(c *gin.Context) {
	user := getUserFromContext(c)
	if user == nil {
		log.Warnf("Could not find user in context. Trying to redner Action form")
		badRequest(c, "No user available in context")
		return
	}
	//parse the form
	c.Request.ParseForm()
	var source []string
	var action []string

	//validate source
	if str := c.Request.FormValue("source"); str != "" {
		source = strings.Split(str, ".")
	} else {
		log.Warnf("Form request was partially or fully blank")
		badRequest(c, "Form request was partially or fully blank")
		return
	}

	//validate action
	if str := c.Request.FormValue("action"); str != "" {
		action = strings.Split(str, ".")
	} else {
		log.Warnf("Form request was partially or fully blank")
		badRequest(c, "Form request was partially or fully blank")
		return
	}

	//setup action listener
	if afunc, ok := actionFuncs[strings.Join(source, ".")]; ok {
		err := afunc(user)
		if err != nil {
			log.WithError(err).Error("Failed to setup actions")
			serverError(c, "Failed to setup actions")
			return
		}
	}

	//Build mappings
	am := &models.ActionMapping{
		UserId: user.Id,
		SourceEvent: &models.Event{
			VendorName: source[0],
			Key:        source[1],
			Fields:     map[string]string{},
		},
		Action: &models.Action{
			VendorName: action[0],
			Type:       action[1],
			Fields:     map[string]string{},
		},
	}

	err := mongo.SaveModel(am)
	if err != nil {
		log.WithError(err).Error("Failed to setup actions")
		serverError(c, "Failed to setup actions")
		return
	}

	c.Redirect(302, "/dashboard")
}

func setupPcoSubscriptions(user *models.User) error {
	// Get PCO vendor account
	conf := config.Config()
	pcoAccount, err := mongo.FindVendorAccountByUser(user.Id, models.PCO_VENDOR_NAME)
	if err != nil {
		return err
	}

	//build pco api
	tokenSource := oauth2.ReuseTokenSource(pcoAccount.Token(), mongo.NewVendorTokenSource(pcoAccount))
	pcoApi := pco.NewClientWithOauthConfig(conf.Vendors[models.PCO_VENDOR_NAME].OauthConfig(), tokenSource)

	//Check if subscriptions already exist
	webhookMap := make(map[string]webhooks.Subscription)
	subscriptions, err := pcoApi.GetSubscriptions()
	if err != nil {
		return errors.Join(fmt.Errorf("Failed to find subscriptions for user: %s", user.Id), err)
	}
	//Loop through found subscriptions
	for _, sub := range subscriptions {
		//if subsciption is in the templates look to add it to our map
		if templ, ok := webhooksTemplate[sub.Name]; ok {
			//if the subscription is for our url add it to our map
			url := fmt.Sprintf(templ.Url, conf.AppSettings.WebhookServiceUrl, user.Id.Hex())
			if url == sub.Url {
				webhookMap[sub.Name] = sub
			}
		}
	}

	builtHooks := make([]webhooks.Subscription, 0, len(webhooksTemplate))
	//Build subscriptions
	for _, templ := range webhooksTemplate {
		if _, ok := webhookMap[templ.Name]; !ok {
			builtHooks = append(builtHooks, webhooks.Subscription{
				Active: true,
				Name:   templ.Name,
				Url:    fmt.Sprintf(templ.Url, conf.AppSettings.WebhookServiceUrl, user.Id.Hex()),
			})
		}
	}

	//Todo: save subscriptions for succesfull hooksetups
	for index := range builtHooks {
		err = pcoApi.CreateSubscription(&builtHooks[index])
		if err != nil {
			return errors.Join(fmt.Errorf("Failed to create subscription: %s for user: %s", builtHooks[index].Name ,user.Id), err)
		}
	}

	//Save Subscriptions
	err = mongo.SaveSubscriptionsForUser(user.Id, builtHooks...)
	if err != nil {
		return errors.Join(fmt.Errorf("Failed to save subscriptions for user: %s", user.Id), err)
	}

	return nil
}