APPS-8263: methods for managing App Platform dev DBs (#665)

apps.go

@@ -56,6 +56,19 @@ type AppsService interface {
 
 	ListBuildpacks(ctx context.Context) ([]*Buildpack, *Response, error)
 	UpgradeBuildpack(ctx context.Context, appID string, opts UpgradeBuildpackOptions) (*UpgradeBuildpackResponse, *Response, error)
+
+	GetAppDatabaseConnectionDetails(ctx context.Context, appID string) ([]*GetDatabaseConnectionDetailsResponse, *Response, error)
+	ResetDatabasePassword(ctx context.Context, appID string, component string) (*Deployment, *Response, error)
+	ToggleDatabaseTrustedSource(
+		ctx context.Context,
+		appID string,
+		component string,
+		opts ToggleDatabaseTrustedSourceOptions,
+	) (
+		*ToggleDatabaseTrustedSourceResponse,
+		*Response,
+		error,
+	)
 }
 
 // AppLogs represent app logs.
@@ -90,6 +103,12 @@ type UpgradeBuildpackOptions struct {
 	TriggerDeployment bool `json:"trigger_deployment,omitempty"`
 }
 
+// ToggleDatabaseTrustedSourceOptions provides optional parameters for ToggleDatabaseTrustedSource.
+type ToggleDatabaseTrustedSourceOptions struct {
+	// Enable, if true, indicates the database should enable the trusted sources firewall.
+	Enable bool
+}
+
 type appRoot struct {
 	App *App `json:"app"`
 }
@@ -498,6 +517,60 @@ func (s *AppsServiceOp) UpgradeBuildpack(ctx context.Context, appID string, opts
 	return root, resp, nil
 }
 
+// GetAppDatabaseConnectionDetails retrieves credentials for databases associated with the app.
+func (s *AppsServiceOp) GetAppDatabaseConnectionDetails(ctx context.Context, appID string) ([]*GetDatabaseConnectionDetailsResponse, *Response, error) {
+	path := fmt.Sprintf("%s/%s/database_connection_details", appsBasePath, appID)
+	req, err := s.client.NewRequest(ctx, http.MethodGet, path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+	root := new(GetAppDatabaseConnectionDetailsResponse)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+	return root.ConnectionDetails, resp, nil
+}
+
+// ResetDatabasePassword resets credentials for a database component associated with the app.
+func (s *AppsServiceOp) ResetDatabasePassword(ctx context.Context, appID string, component string) (*Deployment, *Response, error) {
+	path := fmt.Sprintf("%s/%s/components/%s/reset_password", appsBasePath, appID, component)
+	req, err := s.client.NewRequest(ctx, http.MethodPost, path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+	root := new(deploymentRoot)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+	return root.Deployment, resp, nil
+}
+
+// ToggleDatabaseTrustedSource enables/disables trusted sources on the specified dev database component.
+func (s *AppsServiceOp) ToggleDatabaseTrustedSource(
+	ctx context.Context,
+	appID string,
+	component string,
+	opts ToggleDatabaseTrustedSourceOptions,
+) (
+	*ToggleDatabaseTrustedSourceResponse,
+	*Response,
+	error,
+) {
+	path := fmt.Sprintf("%s/%s/components/%s/trusted_sources", appsBasePath, appID, component)
+	req, err := s.client.NewRequest(ctx, http.MethodPost, path, opts)
+	if err != nil {
+		return nil, nil, err
+	}
+	root := new(ToggleDatabaseTrustedSourceResponse)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+	return root, resp, nil
+}
+
 // AppComponentType is an app component type.
 type AppComponentType string
 

apps_test.go

@@ -225,6 +225,17 @@ var (
 			MajorVersion: 0,
 		},
 	}
+
+	testConnectionDetails = []*GetDatabaseConnectionDetailsResponse{
+		{
+			Host:          "db1.b.db.ondigitalocean.com",
+			Port:          3306,
+			Username:      "example",
+			Password:      "PASSWORD",
+			DatabaseName:  "db",
+			ComponentName: "example-service",
+		},
+	}
 )
 
 func TestApps_CreateApp(t *testing.T) {
@@ -741,6 +752,60 @@ func TestApps_UpgradeBuildpack(t *testing.T) {
 	assert.Equal(t, response, gotResponse)
 }
 
+func TestApps_GetAppDatabaseConnectionDetails(t *testing.T) {
+	setup()
+	defer teardown()
+
+	mux.HandleFunc(fmt.Sprintf("/v2/apps/%s/database_connection_details", testApp.ID), func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, http.MethodGet)
+
+		json.NewEncoder(w).Encode(&GetAppDatabaseConnectionDetailsResponse{ConnectionDetails: testConnectionDetails})
+	})
+
+	appConnectionDetails, _, err := client.Apps.GetAppDatabaseConnectionDetails(ctx, testApp.ID)
+	require.NoError(t, err)
+	assert.Equal(t, testConnectionDetails, appConnectionDetails)
+}
+
+func TestApps_ResetDatabasePassword(t *testing.T) {
+	setup()
+	defer teardown()
+
+	mux.HandleFunc(fmt.Sprintf("/v2/apps/%s/components/%s/reset_password",
+		testApp.ID, testApp.Spec.GetServices()[0].GetName(),
+	), func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, http.MethodPost)
+
+		json.NewEncoder(w).Encode(&ResetDatabasePasswordResponse{Deployment: &testDeployment})
+	})
+
+	deployment, _, err := client.Apps.ResetDatabasePassword(ctx, testApp.ID, testApp.Spec.GetServices()[0].GetName())
+	require.NoError(t, err)
+	assert.Equal(t, &testDeployment, deployment)
+}
+
+func TestApps_ToggleDatabaseTrustedSource(t *testing.T) {
+	setup()
+	defer teardown()
+
+	mux.HandleFunc(fmt.Sprintf("/v2/apps/%s/components/%s/trusted_sources",
+		testApp.ID, testApp.Spec.GetServices()[0].GetName(),
+	), func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, http.MethodPost)
+
+		json.NewEncoder(w).Encode(&ToggleDatabaseTrustedSourceResponse{IsEnabled: true})
+	})
+
+	resp, _, err := client.Apps.ToggleDatabaseTrustedSource(
+		ctx,
+		testApp.ID,
+		testApp.Spec.GetServices()[0].GetName(),
+		ToggleDatabaseTrustedSourceOptions{Enable: true},
+	)
+	require.NoError(t, err)
+	assert.Equal(t, &ToggleDatabaseTrustedSourceResponse{IsEnabled: true}, resp)
+}
+
 func TestApps_ToURN(t *testing.T) {
 	app := &App{
 		ID: "deadbeef-dead-4aa5-beef-deadbeef347d",