github.com/Redstoneguy129/cli@v0.0.0-20230211220159-15dca4e91917/internal/secrets/set/set.go

package set

import (
	"context"
	"errors"
	"fmt"
	"net/http"
	"strings"

	"github.com/Redstoneguy129/cli/internal/utils"
	"github.com/Redstoneguy129/cli/pkg/api"
	"github.com/joho/godotenv"
	"github.com/spf13/afero"
)

func Run(ctx context.Context, envFilePath string, args []string, fsys afero.Fs) error {
	// 1. Sanity checks.
	{
		if err := utils.AssertSupabaseCliIsSetUpFS(fsys); err != nil {
			return err
		}
	}

	// 2. Set secret(s).
	{
		projectRef, err := utils.LoadProjectRef(fsys)
		if err != nil {
			return err
		}

		var secrets api.CreateSecretsJSONBody
		if envFilePath != "" {
			envMap, err := godotenv.Read(envFilePath)
			if err != nil {
				return err
			}
			for name, value := range envMap {
				secret := api.CreateSecretBody{
					Name:  name,
					Value: value,
				}
				secrets = append(secrets, secret)
			}
		} else if len(args) == 0 {
			return errors.New("No arguments found. Use --env-file to read from a .env file.")
		} else {
			for _, pair := range args {
				name, value, found := strings.Cut(pair, "=")
				if !found {
					return errors.New("Invalid secret pair: " + utils.Aqua(pair) + ". Must be NAME=VALUE.")
				}

				secret := api.CreateSecretBody{
					Name:  name,
					Value: value,
				}
				secrets = append(secrets, secret)
			}
		}

		resp, err := utils.GetSupabase().CreateSecretsWithResponse(ctx, projectRef, secrets)
		if err != nil {
			return err
		}

		// TODO: remove the StatusOK case after 2022-08-20
		if resp.StatusCode() != http.StatusCreated && resp.StatusCode() != http.StatusOK {
			return errors.New("Unexpected error setting project secrets: " + string(resp.Body))
		}
	}

	fmt.Println("Finished " + utils.Aqua("supabase secrets set") + ".")
	return nil
}