Support app telemetry toggle for docker deployer. (#168)

Author: Aaliya7516

cmd/apptelemetry-disable.go

@@ -0,0 +1,34 @@
+package cmd
+
+import (
+	"github.com/couchbaselabs/cbdinocluster/deployment/dockerdeploy"
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var appTelemetryDisableCmd = &cobra.Command{
+	Use:   "disable <cluster-id>",
+	Short: "Disables app telemetry",
+	Args:  cobra.MinimumNArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		helper := CmdHelper{}
+		logger := helper.GetLogger()
+		ctx := helper.GetContext()
+
+		_, deployer, cluster := helper.IdentifyCluster(ctx, args[0])
+
+		dockerDeployer, ok := deployer.(*dockerdeploy.Deployer)
+		if !ok {
+			logger.Fatal("Toggling app telemetry is only supported for docker deployer")
+		}
+
+		err := dockerDeployer.SetAppTelemetry(ctx, cluster.GetID(), false)
+		if err != nil {
+			logger.Fatal("failed to disable app telemetry", zap.Error(err))
+		}
+	},
+}
+
+func init() {
+	appTelemetryCmd.AddCommand(appTelemetryDisableCmd)
+}

cmd/apptelemetry-enable.go

@@ -0,0 +1,34 @@
+package cmd
+
+import (
+	"github.com/couchbaselabs/cbdinocluster/deployment/dockerdeploy"
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var appTelemetryEnableCmd = &cobra.Command{
+	Use:   "enable <cluster-id>",
+	Short: "Enables app telemetry",
+	Args:  cobra.MinimumNArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		helper := CmdHelper{}
+		logger := helper.GetLogger()
+		ctx := helper.GetContext()
+
+		_, deployer, cluster := helper.IdentifyCluster(ctx, args[0])
+
+		dockerDeployer, ok := deployer.(*dockerdeploy.Deployer)
+		if !ok {
+			logger.Fatal("Toggling app telemetry is only supported for docker deployer")
+		}
+
+		err := dockerDeployer.SetAppTelemetry(ctx, cluster.GetID(), true)
+		if err != nil {
+			logger.Fatal("failed to enable app telemetry", zap.Error(err))
+		}
+	},
+}
+
+func init() {
+	appTelemetryCmd.AddCommand(appTelemetryEnableCmd)
+}

cmd/apptelemetry.go

@@ -0,0 +1,15 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+)
+
+var appTelemetryCmd = &cobra.Command{
+	Use:   "app-telemetry",
+	Short: "Provides access to app telemetry settings",
+	Run:   nil,
+}
+
+func init() {
+	rootCmd.AddCommand(appTelemetryCmd)
+}

deployment/dockerdeploy/deployer.go

@@ -1367,6 +1367,17 @@ func (d *Deployer) KillCouchbase(ctx context.Context, clusterID string, nodeIDs
 	return nil
 }
 
+func (d *Deployer) SetAppTelemetry(ctx context.Context, clusterID string, enable bool) error {
+	controller, err := d.getController(ctx, clusterID)
+	if err != nil {
+		return errors.Wrap(err, "failed to get controller for cluster")
+	}
+
+	return controller.Controller().SetAppTelemetry(ctx, &clustercontrol.AppTelemetryOptions{
+		Enabled: enable,
+	})
+}
+
 func (d *Deployer) RedeployCluster(ctx context.Context, clusterID string) error {
 	return errors.New("docker deploy does not support redeploy cluster")
 }

utils/clustercontrol/controller.go

@@ -948,3 +948,17 @@ type SetupJwtOptions_Issuer struct {
 func (c *Controller) SetupJwtAuth(ctx context.Context, opts *SetupJwtAuthOptions) error {
 	return c.doJsonReq(ctx, "PUT", "/settings/jwt", opts, true, nil)
 }
+
+type AppTelemetryOptions struct {
+	Enabled bool
+}
+
+func (c *Controller) SetAppTelemetry(ctx context.Context, opts *AppTelemetryOptions) error {
+	form := make(url.Values)
+	enabled := false
+	if opts != nil {
+		enabled = opts.Enabled
+	}
+	form.Add("enabled", strconv.FormatBool(enabled))
+	return c.doFormPost(ctx, "/settings/appTelemetry", form, true, nil)
+}