Add tables aws_quicksight_user, aws_quicksight_namespace, aws_quicksight_group, aws_quicksight_data_source, aws_quicksight_data_set, aws_quicksight_vpc_connection and aws_quicksight_account_settings Closes #2457

aws/plugin.go

@@ -405,6 +405,13 @@ func Plugin(ctx context.Context) *plugin.Plugin {
 			"aws_pipes_pipe":                                               tableAwsPipes(ctx),
 			"aws_pricing_product":                                          tableAwsPricingProduct(ctx),
 			"aws_pricing_service_attribute":                                tableAwsPricingServiceAttribute(ctx),
+			"aws_quicksight_account_setting":                               tableAwsQuickSightAccountSetting(ctx),
+			"aws_quicksight_data_set":                                      tableAwsQuickSightDataset(ctx),
+			"aws_quicksight_data_source":                                   tableAwsQuickSightDatasource(ctx),
+			"aws_quicksight_group":                                         tableAwsQuickSightGroup(ctx),
+			"aws_quicksight_namespace":                                     tableAwsQuickSightNamespace(ctx),
+			"aws_quicksight_user":                                          tableAwsQuickSightUser(ctx),
+			"aws_quicksight_vpc_connection":                                tableAwsQuickSightVpcConnection(ctx),
 			"aws_ram_principal_association":                                tableAwsRAMPrincipalAssociation(ctx),
 			"aws_ram_resource_association":                                 tableAwsRAMResourceAssociation(ctx),
 			"aws_rds_db_cluster":                                           tableAwsRDSDBCluster(ctx),

aws/service.go

@@ -107,6 +107,7 @@ import (
 	"github.com/aws/aws-sdk-go-v2/service/pinpoint"
 	"github.com/aws/aws-sdk-go-v2/service/pipes"
 	"github.com/aws/aws-sdk-go-v2/service/pricing"
+	"github.com/aws/aws-sdk-go-v2/service/quicksight"
 	"github.com/aws/aws-sdk-go-v2/service/ram"
 	"github.com/aws/aws-sdk-go-v2/service/rds"
 	"github.com/aws/aws-sdk-go-v2/service/redshift"
@@ -1187,6 +1188,15 @@ func PricingClient(ctx context.Context, d *plugin.QueryData) (*pricing.Client, e
 	return pricing.NewFromConfig(*cfg), nil
 }
 
+// QuickSightClient returns the client for the Amazon QuickSight service
+func QuickSightClient(ctx context.Context, d *plugin.QueryData) (*quicksight.Client, error) {
+	cfg, err := getClientForQueryRegion(ctx, d)
+	if err != nil {
+		return nil, err
+	}
+	return quicksight.NewFromConfig(*cfg), nil
+}
+
 func RAMClient(ctx context.Context, d *plugin.QueryData) (*ram.Client, error) {
 	cfg, err := getClientForQueryRegion(ctx, d)
 	if err != nil {

aws/table_aws_quicksight_account_setting.go

@@ -0,0 +1,125 @@
+package aws
+
+import (
+	"context"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/quicksight"
+
+	"github.com/turbot/steampipe-plugin-sdk/v5/grpc/proto"
+	"github.com/turbot/steampipe-plugin-sdk/v5/plugin"
+	"github.com/turbot/steampipe-plugin-sdk/v5/plugin/transform"
+)
+
+//// TABLE DEFINITION
+
+func tableAwsQuickSightAccountSetting(_ context.Context) *plugin.Table {
+	return &plugin.Table{
+		Name:        "aws_quicksight_account_setting",
+		Description: "AWS QuickSight Account Setting",
+		List: &plugin.ListConfig{
+			Hydrate: listAwsQuickSightAccountSettings,
+			IgnoreConfig: &plugin.IgnoreConfig{
+				ShouldIgnoreErrorFunc: shouldIgnoreErrors([]string{"ResourceNotFoundException"}),
+			},
+			KeyColumns: []*plugin.KeyColumn{
+				{Name: "region", Require: plugin.Required},
+				{Name: "quicksight_account_id", Require: plugin.Optional},
+			},
+			Tags: map[string]string{"service": "quicksight", "action": "DescribeAccountSettings"},
+		},
+		GetMatrixItemFunc: SupportedRegionMatrix(AWS_QUICKSIGHT_SERVICE_ID),
+		Columns: awsRegionalColumns([]*plugin.Column{
+			{
+				Name:        "account_name",
+				Description: "The account name displayed for the account.",
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("AccountSettings.AccountName"),
+			},
+			// As we have already a column "account_id" as a common column for all the tables, we have renamed the column to "quicksight_account_id"
+			{
+				Name:        "quicksight_account_id",
+				Description: "The ID for the Amazon Web Services account that contains the settings that you want to list.",
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromQual("quicksight_account_id"),
+			},
+			{
+				Name:        "edition",
+				Description: "The edition of Amazon QuickSight that you're currently subscribed to: Enterprise edition or Standard edition.",
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("AccountSettings.Edition"),
+			},
+			{
+				Name:        "default_namespace",
+				Description: "The default namespace for this AWS account.",
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("AccountSettings.DefaultNamespace"),
+			},
+			{
+				Name:        "notification_email",
+				Description: "The email address that QuickSight uses to send notifications.",
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("AccountSettings.NotificationEmail"),
+			},
+			{
+				Name:        "termination_protection_enabled",
+				Description: "A boolean value that indicates whether termination protection is enabled.",
+				Type:        proto.ColumnType_BOOL,
+				Transform:   transform.FromField("AccountSettings.TerminationProtectionEnabled"),
+			},
+			{
+				Name:        "public_sharing_enabled",
+				Description: "A boolean value that indicates whether public sharing is enabled.",
+				Type:        proto.ColumnType_BOOL,
+				Transform:   transform.FromField("AccountSettings.PublicSharingEnabled"),
+			},
+
+			// Standard columns
+			{
+				Name:        "title",
+				Description: resourceInterfaceDescription("title"),
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("AccountSettings.AccountName"),
+			},
+		}),
+	}
+}
+
+//// LIST FUNCTION
+
+func listAwsQuickSightAccountSettings(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
+	// Create client
+	svc, err := QuickSightClient(ctx, d)
+	if err != nil {
+		plugin.Logger(ctx).Error("aws_quicksight_account_setting.listAwsQuickSightAccountSettings", "connection_error", err)
+		return nil, err
+	}
+
+	accountId := d.EqualsQuals["quicksight_account_id"].GetStringValue()
+	// Get AWS Account ID
+	commonData, err := getCommonColumns(ctx, d, h)
+	if err != nil {
+		return nil, err
+	}
+	commonColumnData := commonData.(*awsCommonColumnData)
+
+	if accountId == "" {
+		accountId = commonColumnData.AccountId
+	}
+
+	// Build the params
+	params := &quicksight.DescribeAccountSettingsInput{
+		AwsAccountId: aws.String(accountId),
+	}
+
+	// Get call
+	data, err := svc.DescribeAccountSettings(ctx, params)
+	if err != nil {
+		plugin.Logger(ctx).Error("aws_quicksight_account_setting.listAwsQuickSightAccountSettings", "api_error", err)
+		return nil, err
+	}
+
+	d.StreamListItem(ctx, data)
+
+	return nil, nil
+}