Delete overrides

Author: ajworkos

cmd/dev_server/dev_server.go

@@ -62,6 +62,7 @@ func NewDevServerCmd(client resources.Client, analyticsTrackerFn analytics.Track
 	cmd.AddGroup(&cobra.Group{ID: "overrides", Title: "Override commands:"})
 	cmd.AddCommand(NewAddOverrideCmd(client))
 	cmd.AddCommand(NewRemoveOverrideCmd(client))
+	cmd.AddCommand(NewDeleteOverridesCmd(client))
 	cmd.AddGroup(&cobra.Group{ID: "server", Title: "Server commands:"})
 
 	cmd.AddCommand(NewStartServerCmd(ldClient))

cmd/dev_server/overrides.go

@@ -73,6 +73,44 @@ func addOverride(client resources.Client) func(*cobra.Command, []string) error {
 	}
 }
 
+func NewDeleteOverridesCmd(client resources.Client) *cobra.Command {
+	cmd := &cobra.Command{
+		GroupID: "overrides",
+		Args:    validators.Validate(),
+		Long:    "remove all overrides for a project",
+		RunE:    deleteOverrides(client),
+		Short:   "remove all overrides for a project",
+		Use:     "remove-overrides",
+	}
+
+	cmd.SetUsageTemplate(resourcescmd.SubcommandUsageTemplate())
+
+	cmd.Flags().String(cliflags.ProjectFlag, "", "The project key")
+	_ = cmd.MarkFlagRequired(cliflags.ProjectFlag)
+	_ = cmd.Flags().SetAnnotation(cliflags.ProjectFlag, "required", []string{"true"})
+	_ = viper.BindPFlag(cliflags.ProjectFlag, cmd.Flags().Lookup(cliflags.ProjectFlag))
+
+	return cmd
+}
+
+func deleteOverrides(client resources.Client) func(*cobra.Command, []string) error {
+	return func(cmd *cobra.Command, args []string) error {
+		path := fmt.Sprintf("%s/dev/projects/%s/overrides", getDevServerUrl(), viper.GetString(cliflags.ProjectFlag))
+		res, err := client.MakeUnauthenticatedRequest(
+			"DELETE",
+			path,
+			nil,
+		)
+		if err != nil {
+			return output.NewCmdOutputError(err, viper.GetString(cliflags.OutputFlag))
+		}
+
+		fmt.Fprint(cmd.OutOrStdout(), string(res))
+
+		return nil
+	}
+}
+
 func NewRemoveOverrideCmd(client resources.Client) *cobra.Command {
 	cmd := &cobra.Command{
 		GroupID: "overrides",

internal/dev_server/api/api.yaml

@@ -113,6 +113,17 @@ paths:
           $ref: "#/components/responses/ErrorResponse"
         409:
           $ref: "#/components/responses/ErrorResponse"
+  /dev/projects/{projectKey}/overrides:
+    delete:
+      summary: remove all overrides for the given project
+      operationId: deleteOverrides
+      parameters:
+        - $ref: "#/components/parameters/projectKey"
+      responses:
+        204:
+          description: OK. All overrides were removed
+        404:
+          description: No project found          
   /dev/projects/{projectKey}/overrides/{flagKey}:
     put:
       summary: override flag value with value provided in the body

internal/dev_server/api/delete_overrides.go

@@ -0,0 +1,20 @@
+package api
+
+import (
+	"context"
+
+	"github.com/pkg/errors"
+
+	"github.com/launchdarkly/ldcli/internal/dev_server/model"
+)
+
+func (s server) DeleteOverrides(ctx context.Context, request DeleteOverridesRequestObject) (DeleteOverridesResponseObject, error) {
+	err := model.DeleteOverrides(ctx, request.ProjectKey)
+	if err != nil {
+		if errors.As(err, &model.ErrNotFound{}) {
+			return DeleteOverrides404Response{}, nil
+		}
+		return nil, err
+	}
+	return DeleteOverrides204Response{}, nil
+}

internal/dev_server/api/server.gen.go

@@ -301,6 +301,17 @@ func (s *Sqlite) GetOverridesForProject(ctx context.Context, projectKey string)
 	return overrides, nil
 }
 
+func (s *Sqlite) DeleteOverridesForProject(ctx context.Context, projectKey string) error {
+	_, err := s.database.ExecContext(ctx, `
+		DELETE FROM overrides
+		WHERE project_key = ?
+	`, projectKey)
+	if err != nil {
+		return errors.Wrap(err, "unable to delete overrides for project")
+	}
+	return nil
+}
+
 func (s *Sqlite) UpsertOverride(ctx context.Context, override model.Override) (model.Override, error) {
 	valueJson, err := override.Value.MarshalJSON()
 	if err != nil {

internal/dev_server/db/sqlite.go

@@ -328,4 +328,13 @@ func TestDBFunctions(t *testing.T) {
 
 		assert.True(t, found)
 	})
+
+	t.Run("DeleteOverridesForProject deletes overrides", func(t *testing.T) {
+		err := store.DeleteOverridesForProject(ctx, projects[0].Key)
+		assert.NoError(t, err)
+
+		overridesResult, err := store.GetOverridesForProject(ctx, projects[0].Key)
+		assert.NoError(t, err)
+		assert.Len(t, overridesResult, 0)
+	})
 }