[CLOUDGA-24196] Put DB audit log feature under FF (#121)

Put DB audit log feature under FF

Author: Siddarth-Baldwa

docs/data-sources/associate_db_audit_export_config_cluster.md

@@ -8,7 +8,7 @@ require (
 	github.com/hashicorp/terraform-plugin-framework-validators v0.4.0
 	github.com/hashicorp/terraform-plugin-log v0.9.0
 	github.com/sethvargo/go-retry v0.2.3
-	github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241005174727-b357e4b80b13
+	github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241018110715-ae57a1f1fb73
 )
 
 require (

docs/resources/associate_db_audit_export_config_cluster.md

@@ -159,8 +159,8 @@ github.com/vmihailenco/msgpack/v5 v5.3.5 h1:5gO0H1iULLWGhs2H5tbAHIZTV8/cYafcFOr9
 github.com/vmihailenco/msgpack/v5 v5.3.5/go.mod h1:7xyJ9e+0+9SaZT0Wt1RGleJXzli6Q/V5KbhBonMG9jc=
 github.com/vmihailenco/tagparser/v2 v2.0.0 h1:y09buUbR+b5aycVFQs/g70pqKVZNBmxwAhO7/IwNM9g=
 github.com/vmihailenco/tagparser/v2 v2.0.0/go.mod h1:Wri+At7QHww0WTrCBeu4J6bNtoV6mEfg5OIWRZA9qds=
-github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241005174727-b357e4b80b13 h1:X5ElhDvBDzXcg14eObrBoaD6Di13n2137DgUHzc0FcQ=
-github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241005174727-b357e4b80b13/go.mod h1:5vW0xIzIZw+1djkiWKx0qqNmqbRBSf4mjc4qw8lIMik=
+github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241018110715-ae57a1f1fb73 h1:ZKr+fauJieZ4tYqoGJqSFhgNxFNYUgjnVED18iqQEKc=
+github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20241018110715-ae57a1f1fb73/go.mod h1:5vW0xIzIZw+1djkiWKx0qqNmqbRBSf4mjc4qw8lIMik=
 github.com/yuin/goldmark v1.1.25/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.32/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=

go.mod

@@ -14,10 +14,12 @@ type FeatureFlag string
 
 const (
 	GOOGLECLOUD_INTEGRATION_ENABLED FeatureFlag = "GOOGLECLOUD_INTEGRATION_ENABLED"
+	DB_AUDIT_LOGGING                FeatureFlag = "DB_AUDIT_LOGGING"
 )
 
 var flagEnabled = map[FeatureFlag]bool{
 	GOOGLECLOUD_INTEGRATION_ENABLED: false,
+	DB_AUDIT_LOGGING:                false,
 }
 
 func (f FeatureFlag) String() string {

go.sum

@@ -11,6 +11,7 @@ import (
 	"github.com/hashicorp/terraform-plugin-framework/diag"
 	"github.com/hashicorp/terraform-plugin-framework/tfsdk"
 	"github.com/hashicorp/terraform-plugin-framework/types"
+	"github.com/yugabyte/terraform-provider-ybm/managed/fflags"
 	openapiclient "github.com/yugabyte/yugabytedb-managed-go-client-internal"
 )
 
@@ -142,7 +143,7 @@ func (p *provider) Configure(ctx context.Context, req tfsdk.ConfigureProviderReq
 }
 
 func (p *provider) GetResources(_ context.Context) (map[string]tfsdk.ResourceType, diag.Diagnostics) {
-	return map[string]tfsdk.ResourceType{
+	resources := map[string]tfsdk.ResourceType{
 		"ybm_cluster":                            resourceClusterType{},
 		"ybm_allow_list":                         resourceAllowListType{},
 		"ybm_backup":                             resourceBackupType{},
@@ -155,18 +156,30 @@ func (p *provider) GetResources(_ context.Context) (map[string]tfsdk.ResourceTyp
 		"ybm_api_key":                            resourceApiKeyType{},
 		"ybm_metrics_exporter":                   resourceMetricsExporterType{},
 		"ybm_associate_metrics_exporter_cluster": resourceAssociateMetricsExporterClusterType{},
-		"ybm_associate_db_audit_export_config_cluster": resourceAssociateDbAuditExportConfigClusterType{},
-		"ybm_integration": resourceIntegrationType{},
-	}, nil
+		"ybm_integration":                        resourceIntegrationType{},
+	}
+
+	// Add DB Audit logging resource only if the feature flag is enabled
+	if fflags.IsFeatureFlagEnabled(fflags.DB_AUDIT_LOGGING) {
+		resources["ybm_associate_db_audit_export_config_cluster"] = resourceAssociateDbAuditExportConfigClusterType{}
+	}
+
+	return resources, nil
 }
 
 func (p *provider) GetDataSources(_ context.Context) (map[string]tfsdk.DataSourceType, diag.Diagnostics) {
-	return map[string]tfsdk.DataSourceType{
+	dataSources := map[string]tfsdk.DataSourceType{
 		"ybm_backup":      dataSourceBackupType{},
 		"ybm_cluster":     dataClusterNameType{},
 		"ybm_vpc":         dataSourceVPCType{},
 		"ybm_allow_list":  dataSourceAllowListType{},
 		"ybm_integration": dataSourceIntegrationType{},
-		"ybm_associate_db_audit_export_config_cluster": dataSourceAssociateDbAuditExportConfigClusterType{},
-	}, nil
+	}
+
+	// Add DB Audit logging data source only if the feature flag is enabled
+	if fflags.IsFeatureFlagEnabled(fflags.DB_AUDIT_LOGGING) {
+		dataSources["ybm_associate_db_audit_export_config_cluster"] = dataSourceAssociateDbAuditExportConfigClusterType{}
+	}
+
+	return dataSources, nil
 }