add comand to create integrations

cmd/integration.go

@@ -0,0 +1,126 @@
+package main
+
+import (
+	"context"
+	"errors"
+
+	"github.com/ayinke-llc/hermes"
+	"github.com/ayinke-llc/malak"
+	"github.com/ayinke-llc/malak/config"
+	"github.com/ayinke-llc/malak/internal/datastore/postgres"
+	"github.com/charmbracelet/huh"
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+func addIntegrationCommand(c *cobra.Command, cfg *config.Config) {
+
+	cmd := &cobra.Command{
+		Use:   "integration",
+		Short: "Manage your system wide integrations",
+	}
+
+	cmd.AddCommand(createIntegration(c, cfg))
+
+	c.AddCommand(cmd)
+}
+
+func createIntegration(_ *cobra.Command, cfg *config.Config) *cobra.Command {
+
+	cmd := &cobra.Command{
+		Use:   "create",
+		Short: `Create a new integration`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+
+			var (
+				name                 string
+				description          string
+				isEnabledIntegration bool
+
+				integrationType malak.IntegrationType
+			)
+
+			logger, err := getLogger(hermes.DeRef(cfg))
+			if err != nil {
+				return err
+			}
+
+			db, err := postgres.New(cfg, logger)
+			if err != nil {
+				logger.Error("could not connect to postgres database",
+					zap.Error(err))
+				return err
+			}
+
+			form := huh.NewForm(
+				huh.NewGroup(
+					huh.NewInput().
+						Title("What’s the name of this integration?").
+						Value(&name).
+						Validate(func(str string) error {
+							if hermes.IsStringEmpty(str) {
+								return errors.New("please provide the name of your integration")
+							}
+
+							if len(str) > 30 {
+								return errors.New("integration name cannot be more than 30")
+							}
+
+							return nil
+						}),
+
+					huh.NewInput().
+						Title("Write a description for this integration?").
+						CharLimit(300).
+						Value(&description),
+
+					huh.NewSelect[malak.IntegrationType]().
+						Title("Pick an integration type").
+						Options(
+							huh.NewOption("oauth2", malak.IntegrationTypeOauth2),
+							huh.NewOption("api key", malak.IntegrationTypeApiKey),
+							huh.NewOption("malak ( internal )", malak.IntegrationTypeMalak),
+						).
+						Value(&integrationType),
+
+					huh.NewConfirm().
+						Title("Is enabled?").
+						Description("should this integration be enabled?").
+						Value(&isEnabledIntegration),
+				),
+			)
+
+			defer db.Close()
+
+			if err := form.Run(); err != nil {
+				logger.Error("could not take in data from user",
+					zap.Error(err))
+				return err
+			}
+
+			logger.Debug("creating a new integration")
+
+			integration := &malak.Integration{
+				Reference:       malak.NewReferenceGenerator().Generate(malak.EntityTypeIntegration),
+				Description:     description,
+				IsEnabled:       isEnabledIntegration,
+				Metadata:        malak.IntegrationMetadata{},
+				IntegrationType: integrationType,
+				IntegrationName: name,
+			}
+
+			integrationRepo := postgres.NewIntegrationRepo(db)
+
+			if err := integrationRepo.Create(context.Background(), integration); err != nil {
+				logger.Error("could not create integration", zap.Error(err))
+				return err
+			}
+
+			logger.Debug("created integration")
+
+			return nil
+		},
+	}
+
+	return cmd
+}

cmd/main.go

@@ -15,7 +15,6 @@ import (
 	"github.com/ayinke-llc/malak/internal/pkg/email/smtp"
 	"github.com/google/uuid"
 	"github.com/spf13/cobra"
-	"github.com/spf13/pflag"
 	"github.com/spf13/viper"
 	"go.uber.org/zap"
 )
@@ -75,14 +74,7 @@ func Execute() error {
 	addHTTPCommand(rootCmd, cfg)
 	addCronCommand(rootCmd, cfg)
 	addPlanCommand(rootCmd, cfg)
-
-	cmd, _, err := rootCmd.Find(os.Args[1:])
-	// default cmd if no cmd is given
-	// default to http
-	if err == nil && cmd.Use == rootCmd.Use && cmd.Flags().Parse(os.Args[1:]) != pflag.ErrHelp {
-		args := append([]string{"http"}, os.Args[1:]...)
-		rootCmd.SetArgs(args)
-	}
+	addIntegrationCommand(rootCmd, cfg)
 
 	return rootCmd.Execute()
 }

cmd/plan.go

@@ -95,7 +95,7 @@ func createPlan(_ *cobra.Command, cfg *config.Config) *cobra.Command {
 						Value(&defaultPlanID),
 
 					huh.NewConfirm().
-						Title("Default plan").
+						Title("Default plan?").
 						Description("should this plan be the default one for new users?").
 						Value(&isDefaultPlan),
 				),

integration.go

@@ -1,6 +1,7 @@
 package malak
 
 import (
+	"context"
 	"time"
 
 	"github.com/google/uuid"
@@ -54,4 +55,11 @@ type WorkspaceIntegration struct {
 }
 
 type IntegrationRepository interface {
+	Create(context.Context, *Integration) error
+
+	// ListSystem(context.Context) ([]Integration, error)
+	// DisableSystem(context.Context, *Integration) error
+	//
+	// List(context.Context, *Workspace) error
+	// Disable(context.Context, *Workspace, *Integration) error
 }

internal/datastore/postgres/integration.go

@@ -0,0 +1,34 @@
+package postgres
+
+import (
+	"context"
+	"database/sql"
+
+	"github.com/ayinke-llc/malak"
+	"github.com/uptrace/bun"
+)
+
+type integrationRepo struct {
+	inner *bun.DB
+}
+
+func NewIntegrationRepo(db *bun.DB) malak.IntegrationRepository {
+	return &integrationRepo{
+		inner: db,
+	}
+}
+
+func (i *integrationRepo) Create(ctx context.Context,
+	integration *malak.Integration) error {
+
+	ctx, cancelFn := withContext(ctx)
+	defer cancelFn()
+
+	return i.inner.RunInTx(ctx, &sql.TxOptions{},
+		func(ctx context.Context, tx bun.Tx) error {
+			_, err := tx.NewInsert().
+				Model(integration).
+				Exec(ctx)
+			return err
+		})
+}

internal/datastore/postgres/migrations/20250129121822_create_integrations_table.up.sql

@@ -15,7 +15,7 @@ CREATE TABLE integrations (
   deleted_at TIMESTAMP WITH TIME ZONE
 );
 
-ALTER TABLE integrations ADD CONSTRAINT integrations_reference_check_key CHECK (reference ~ 'integrations_[a-zA-Z0-9._]+');
+ALTER TABLE integrations ADD CONSTRAINT integration_reference_check_key CHECK (reference ~ 'integration_[a-zA-Z0-9._]+');
 
 CREATE TABLE workspace_integrations (
   id uuid PRIMARY KEY DEFAULT uuid_generate_v4(),
@@ -31,4 +31,4 @@ CREATE TABLE workspace_integrations (
   deleted_at TIMESTAMP WITH TIME ZONE
 );
 
-ALTER TABLE workspace_integrations ADD CONSTRAINT workspace_integrations_reference_check_key CHECK (reference ~ 'workspace_integrations_[a-zA-Z0-9._]+');
+ALTER TABLE workspace_integrations ADD CONSTRAINT workspace_integration_reference_check_key CHECK (reference ~ 'workspace_integration_[a-zA-Z0-9._]+');