feat: add billing_information column to gcp_project and gcp_organizat…

…ion_project tables
turbot · Oct 7, 2024 · a44aa32 · a44aa32
1 parent 5c0cec5
commit a44aa32
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/gcp/table_gcp_organization_project.go b/gcp/table_gcp_organization_project.go
@@ -75,6 +75,13 @@ func tableGcpOrganizationProject(_ context.Context) *plugin.Table {
 				Hydrate:     getProjectAncestors,
 				Transform:   transform.FromValue(),
 			},
+			{
+				Name:        "billing_information",
+				Description: "The billing information of the project.",
+				Type:        proto.ColumnType_JSON,
+				Hydrate:     getProjectBillingInfo,
+				Transform:   transform.FromValue(),
+			},
 
 			// Steampipe standard columns
 			{

diff --git a/gcp/table_gcp_project.go b/gcp/table_gcp_project.go
@@ -75,6 +75,13 @@ func tableGcpProject(_ context.Context) *plugin.Table {
 				Hydrate:     getProjectAncestors,
 				Transform:   transform.FromValue(),
 			},
+			{
+				Name:        "billing_information",
+				Description: "The billing information of the project.",
+				Type:        proto.ColumnType_JSON,
+				Hydrate:     getProjectBillingInfo,
+				Transform:   transform.FromValue(),
+			},
 
 			// Steampipe standard columns
 			{
@@ -186,6 +193,28 @@ func getProjectAncestors(ctx context.Context, d *plugin.QueryData, h *plugin.Hyd
 	return resp.Ancestor, nil
 }
 
+func getProjectBillingInfo(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
+	// Create Service Connection
+	service, err := BillingService(ctx, d)
+	if err != nil {
+		plugin.Logger(ctx).Error("gcp_project.getProjectBillingInformation", "connection_error", err)
+		return nil, err
+	}
+
+	// Get project details
+	projectId := h.Item.(*cloudresourcemanager.Project).ProjectId
+
+	resp, err := service.Projects.GetBillingInfo("projects/" + projectId).Do()
+	if err != nil {
+		if strings.Contains(err.Error(), "404") {
+			return nil, nil
+		}
+		plugin.Logger(ctx).Error("gcp_project.getProjectBillingInformation", "api_err", err)
+		return nil, err
+	}
+	return resp, nil
+}
+
 func projectSelfLink(_ context.Context, d *transform.TransformData) (interface{}, error) {
 	data := d.HydrateItem.(*cloudresourcemanager.Project)
 	selfLink := "https://cloudresourcemanager.googleapis.com/v1/projects/" + data.Name