Add webhook create and update commands

Author: joshmgross

internal/cmd/webhook/create.go

@@ -0,0 +1,70 @@
+package webhook
+
+import (
+	"fmt"
+
+	"github.com/planetscale/cli/internal/cmdutil"
+	"github.com/planetscale/cli/internal/printer"
+	"github.com/planetscale/planetscale-go/planetscale"
+	"github.com/spf13/cobra"
+)
+
+func CreateCmd(ch *cmdutil.Helper) *cobra.Command {
+	var flags struct {
+		url     string
+		events  []string
+		enabled bool
+	}
+
+	cmd := &cobra.Command{
+		Use:   "create <database>",
+		Short: "Create a webhook for a database",
+		Args:  cmdutil.RequiredArgs("database"),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			database := args[0]
+
+			client, err := ch.Client()
+			if err != nil {
+				return err
+			}
+
+			req := &planetscale.CreateWebhookRequest{
+				Organization: ch.Config.Organization,
+				Database:     database,
+				URL:          flags.url,
+				Events:       flags.events,
+			}
+
+			if cmd.Flags().Changed("enabled") {
+				req.Enabled = &flags.enabled
+			}
+
+			end := ch.Printer.PrintProgress(fmt.Sprintf("Creating webhook for %s", printer.BoldBlue(database)))
+			defer end()
+
+			webhook, err := client.Webhooks.Create(ctx, req)
+			if err != nil {
+				switch cmdutil.ErrCode(err) {
+				case planetscale.ErrNotFound:
+					return fmt.Errorf("database %s does not exist in organization %s",
+						printer.BoldBlue(database), printer.BoldBlue(ch.Config.Organization))
+				default:
+					return cmdutil.HandleError(err)
+				}
+			}
+
+			end()
+
+			return ch.Printer.PrintResource(toWebhook(webhook))
+		},
+	}
+
+	cmd.Flags().StringVar(&flags.url, "url", "", "The URL to send webhook events to (required)")
+	cmd.Flags().StringSliceVar(&flags.events, "events", nil, "Comma-separated list of events to subscribe to")
+	cmd.Flags().BoolVar(&flags.enabled, "enabled", true, "Whether the webhook is enabled")
+
+	cmd.MarkFlagRequired("url") // nolint:errcheck
+
+	return cmd
+}

internal/cmd/webhook/create_test.go

@@ -0,0 +1,99 @@
+package webhook
+
+import (
+	"bytes"
+	"context"
+	"testing"
+	"time"
+
+	qt "github.com/frankban/quicktest"
+	"github.com/planetscale/cli/internal/cmdutil"
+	"github.com/planetscale/cli/internal/config"
+	"github.com/planetscale/cli/internal/mock"
+	"github.com/planetscale/cli/internal/printer"
+	ps "github.com/planetscale/planetscale-go/planetscale"
+)
+
+func TestWebhook_CreateCmd(t *testing.T) {
+	c := qt.New(t)
+
+	var buf bytes.Buffer
+	format := printer.JSON
+	p := printer.NewPrinter(&format)
+	p.SetResourceOutput(&buf)
+
+	org := "planetscale"
+	db := "mydb"
+	url := "https://example.com/webhook"
+	events := []string{"branch.created", "branch.deleted"}
+	createdAt := time.Date(2025, 1, 15, 10, 30, 0, 0, time.UTC)
+
+	webhook := &ps.Webhook{
+		ID:        "webhook-123",
+		URL:       url,
+		Enabled:   true,
+		Events:    events,
+		CreatedAt: createdAt,
+	}
+
+	svc := &mock.WebhooksService{
+		CreateFn: func(ctx context.Context, req *ps.CreateWebhookRequest) (*ps.Webhook, error) {
+			c.Assert(req.Organization, qt.Equals, org)
+			c.Assert(req.Database, qt.Equals, db)
+			c.Assert(req.URL, qt.Equals, url)
+			c.Assert(req.Events, qt.DeepEquals, events)
+			return webhook, nil
+		},
+	}
+
+	ch := &cmdutil.Helper{
+		Printer: p,
+		Config: &config.Config{
+			Organization: org,
+		},
+		Client: func() (*ps.Client, error) {
+			return &ps.Client{
+				Webhooks: svc,
+			}, nil
+		},
+	}
+
+	cmd := CreateCmd(ch)
+	cmd.SetArgs([]string{db, "--url", url, "--events", "branch.created,branch.deleted"})
+	err := cmd.Execute()
+
+	c.Assert(err, qt.IsNil)
+	c.Assert(svc.CreateFnInvoked, qt.IsTrue)
+
+	res := &Webhook{orig: webhook}
+	c.Assert(buf.String(), qt.JSONEquals, res)
+}
+
+func TestWebhook_CreateCmd_RequiresURL(t *testing.T) {
+	c := qt.New(t)
+
+	var buf bytes.Buffer
+	format := printer.Human
+	p := printer.NewPrinter(&format)
+	p.SetHumanOutput(&buf)
+
+	org := "planetscale"
+	db := "mydb"
+
+	ch := &cmdutil.Helper{
+		Printer: p,
+		Config: &config.Config{
+			Organization: org,
+		},
+		Client: func() (*ps.Client, error) {
+			return &ps.Client{}, nil
+		},
+	}
+
+	cmd := CreateCmd(ch)
+	cmd.SetArgs([]string{db})
+	err := cmd.Execute()
+
+	c.Assert(err, qt.IsNotNil)
+	c.Assert(err.Error(), qt.Contains, "required flag")
+}

internal/cmd/webhook/update.go

@@ -0,0 +1,85 @@
+package webhook
+
+import (
+	"fmt"
+
+	"github.com/planetscale/cli/internal/cmdutil"
+	"github.com/planetscale/cli/internal/printer"
+	"github.com/planetscale/planetscale-go/planetscale"
+	"github.com/spf13/cobra"
+)
+
+func UpdateCmd(ch *cmdutil.Helper) *cobra.Command {
+	var flags struct {
+		url     string
+		events  []string
+		enabled bool
+	}
+
+	cmd := &cobra.Command{
+		Use:   "update <database> <webhook-id>",
+		Short: "Update a webhook for a database",
+		Args:  cmdutil.RequiredArgs("database", "webhook-id"),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			database := args[0]
+			webhookID := args[1]
+
+			client, err := ch.Client()
+			if err != nil {
+				return err
+			}
+
+			req := &planetscale.UpdateWebhookRequest{
+				Organization: ch.Config.Organization,
+				Database:     database,
+				ID:           webhookID,
+			}
+
+			changed := false
+
+			if cmd.Flags().Changed("url") {
+				req.URL = &flags.url
+				changed = true
+			}
+
+			if cmd.Flags().Changed("events") {
+				req.Events = flags.events
+				changed = true
+			}
+
+			if cmd.Flags().Changed("enabled") {
+				req.Enabled = &flags.enabled
+				changed = true
+			}
+
+			if !changed {
+				return fmt.Errorf("at least one of --url, --events, or --enabled must be provided")
+			}
+
+			end := ch.Printer.PrintProgress(fmt.Sprintf("Updating webhook %s for %s", printer.BoldBlue(webhookID), printer.BoldBlue(database)))
+			defer end()
+
+			webhook, err := client.Webhooks.Update(ctx, req)
+			if err != nil {
+				switch cmdutil.ErrCode(err) {
+				case planetscale.ErrNotFound:
+					return fmt.Errorf("webhook %s does not exist in database %s (organization: %s)",
+						printer.BoldBlue(webhookID), printer.BoldBlue(database), printer.BoldBlue(ch.Config.Organization))
+				default:
+					return cmdutil.HandleError(err)
+				}
+			}
+
+			end()
+
+			return ch.Printer.PrintResource(toWebhook(webhook))
+		},
+	}
+
+	cmd.Flags().StringVar(&flags.url, "url", "", "The URL to send webhook events to")
+	cmd.Flags().StringSliceVar(&flags.events, "events", nil, "Comma-separated list of events to subscribe to")
+	cmd.Flags().BoolVar(&flags.enabled, "enabled", true, "Whether the webhook is enabled")
+
+	return cmd
+}