Add server keys details in table azure_postgresql_server closes #287

turbot · Sep 7, 2021 · 03fbd56 · 03fbd56
1 parent 34f9d17
commit 03fbd56
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/azure/table_azure_postgresql_server.go b/azure/table_azure_postgresql_server.go
@@ -206,6 +206,13 @@ func tableAzurePostgreSqlServer(_ context.Context) *plugin.Table {
 				Hydrate:     getPostgreSQLServerConfigurations,
 				Transform:   transform.FromValue(),
 			},
+			{
+				Name:        "server_keys",
+				Description: "A list of server keys for a server.",
+				Type:        proto.ColumnType_JSON,
+				Hydrate:     getPostgreSQLServerKeys,
+				Transform:   transform.FromValue(),
+			},
 
 			// Steampipe standard columns
 			{
@@ -350,6 +357,32 @@ func getPostgreSQLServerFirewallRules(ctx context.Context, d *plugin.QueryData,
 	return firewallRules, nil
 }
 
+func getPostgreSQLServerKeys(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
+	plugin.Logger(ctx).Trace("getPostgreSQLServerKeys")
+	server := h.Item.(postgresql.Server)
+
+	session, err := GetNewSession(ctx, d, "MANAGEMENT")
+	if err != nil {
+		return nil, err
+	}
+	subscriptionID := session.SubscriptionID
+	resourceGroupName := strings.Split(string(*server.ID), "/")[4]
+
+	client := postgresql.NewServerKeysClient(subscriptionID)
+	client.Authorizer = session.Authorizer
+
+	op, err := client.List(ctx, resourceGroupName, *server.Name)
+	if err != nil {
+		return nil, err
+	}
+
+	var serverKeys []postgresql.ServerKey
+	for _, key := range op.Values() {
+		serverKeys = append(serverKeys, key)
+	}
+	return serverKeys, nil
+}
+
 func getPostgreSQLServerAdministrator(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
 	plugin.Logger(ctx).Trace("getPostgreSQLServerAdministrator")
 	server := h.Item.(postgresql.Server)