Add gcp compute node group table. Closes #58

gcp/plugin.go

@@ -33,6 +33,7 @@ func Plugin(ctx context.Context) *plugin.Plugin {
 			"gcp_compute_global_forwarding_rule":  tableGcpComputeGlobalForwardingRule(ctx),
 			"gcp_compute_image":                   tableGcpComputeImage(ctx),
 			"gcp_compute_instance":                tableGcpComputeInstance(ctx),
+			"gcp_compute_node_group":              tableGcpComputeNodeGroup(ctx),
 			"gcp_iam_policy":                      tableGcpIAMPolicy(ctx),
 			"gcp_iam_role":                        tableGcpIamRole(ctx),
 			"gcp_logging_exclusion":               tableGcpLoggingExclusion(ctx),

gcp/table_compute_node_group.go

@@ -0,0 +1,247 @@
+package gcp
+
+import (
+	"context"
+
+	"github.com/turbot/go-kit/types"
+	"github.com/turbot/steampipe-plugin-sdk/grpc/proto"
+	"github.com/turbot/steampipe-plugin-sdk/plugin"
+	"github.com/turbot/steampipe-plugin-sdk/plugin/transform"
+
+	"google.golang.org/api/compute/v1"
+)
+
+//// TABLE DEFINITION
+
+func tableGcpComputeNodeGroup(ctx context.Context) *plugin.Table {
+	return &plugin.Table{
+		Name:        "gcp_compute_node_group",
+		Description: "GCP Compute Node Group",
+		Get: &plugin.GetConfig{
+			KeyColumns: plugin.SingleColumn("name"),
+			Hydrate:    getComputeNodeGroup,
+		},
+		List: &plugin.ListConfig{
+			Hydrate: listComputeNodeGroups,
+		},
+		Columns: []*plugin.Column{
+			{
+				Name:        "name",
+				Description: "A friendly name that identifies the resource.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "id",
+				Description: "The unique identifier for the resource.",
+				Type:        proto.ColumnType_INT,
+			},
+			{
+				Name:        "status",
+				Description: "Specifies the current state of the node group.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "size",
+				Description: "The total number of nodes in the node group.",
+				Type:        proto.ColumnType_INT,
+			},
+			{
+				Name:        "description",
+				Description: "A user-specified, human-readable description of the node group.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "kind",
+				Description: "The type of the resource.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "autoscaling_policy_mode",
+				Description: "Specifies the autoscaling mode of the node group.",
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("AutoscalingPolicy.Mode"),
+			},
+			{
+				Name:        "autoscaling_policy_max_nodes",
+				Description: "The maximum number of nodes that the group should have. Must be set if autoscaling is enabled. Maximum value allowed is 100.",
+				Type:        proto.ColumnType_INT,
+				Transform:   transform.FromField("AutoscalingPolicy.MaxNodes"),
+			},
+			{
+				Name:        "autoscaling_policy_min_nodes",
+				Description: "The minimum number of nodes that the group should have.",
+				Type:        proto.ColumnType_INT,
+				Transform:   transform.FromField("AutoscalingPolicy.MinNodes"),
+			},
+			{
+				Name:        "creation_timestamp",
+				Description: "The creation timestamp of the resource.",
+				Type:        proto.ColumnType_TIMESTAMP,
+			},
+			{
+				Name:        "fingerprint",
+				Description: "An unique system generated string, to reduce conflicts when multiple users change any property of the resource.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "maintenance_policy",
+				Description: "Specifies how to handle instances when a node in the group undergoes maintenance.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "node_template",
+				Description: "The URL of the node template to create the node group from.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "self_link",
+				Description: "The server-defined URL for the resource.",
+				Type:        proto.ColumnType_STRING,
+			},
+			{
+				Name:        "zone",
+				Description: "The name of the zone where the node group resides.",
+				Type:        proto.ColumnType_STRING,
+			},
+			// zone_name is a simpler view of the zone, without the full path
+			{
+				Name:        "zone_name",
+				Description: "The zone name in which the node group resides.",
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("Zone").Transform(lastPathElement),
+			},
+
+			{
+				Name:        "iam_policy",
+				Description: "An Identity and Access Management (IAM) policy, which specifies access controls for Google Cloud resources. A `Policy` is a collection of `bindings`. A `binding` binds one or more `members` to a single `role`. Members can be user accounts, service accounts, Google groups, and domains (such as G Suite). A `role` is a named list of permissions; each `role` can be an IAM predefined role or a user-created custom role. For some types of Google Cloud resources, a `binding` can also specify a `condition`, which is a logical expression that allows access to a resource only if the expression evaluates to `true`.",
+				Type:        proto.ColumnType_JSON,
+				Hydrate:     getComputeNodeGroupIamPolicy,
+				Transform:   transform.FromValue(),
+			},
+
+			// standard steampipe columns
+			{
+				Name:        "title",
+				Description: ColumnDescriptionTitle,
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("Name"),
+			},
+			{
+				Name:        "akas",
+				Description: ColumnDescriptionAkas,
+				Type:        proto.ColumnType_JSON,
+				Transform:   transform.From(gcpComputeNodeGroupAka),
+			},
+
+			// standard gcp columns
+			{
+				Name:        "location",
+				Description: ColumnDescriptionLocation,
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromField("Zone").Transform(lastPathElement),
+			},
+			{
+				Name:        "project",
+				Description: ColumnDescriptionProject,
+				Type:        proto.ColumnType_STRING,
+				Transform:   transform.FromConstant(activeProject()),
+			},
+		},
+	}
+}
+
+//// LIST FUNCTION
+
+func listComputeNodeGroups(ctx context.Context, d *plugin.QueryData, _ *plugin.HydrateData) (interface{}, error) {
+	plugin.Logger(ctx).Trace("listComputeNodeGroups")
+	service, err := compute.NewService(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	project := activeProject()
+	resp := service.NodeGroups.AggregatedList(project)
+	if err := resp.Pages(ctx, func(page *compute.NodeGroupAggregatedList) error {
+		for _, item := range page.Items {
+			for _, nodeGroup := range item.NodeGroups {
+				d.StreamListItem(ctx, nodeGroup)
+			}
+		}
+		return nil
+	}); err != nil {
+		return nil, err
+	}
+
+	return nil, nil
+}
+
+//// HYDRATE FUNCTIONS
+
+func getComputeNodeGroup(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
+	service, err := compute.NewService(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var nodeGroup compute.NodeGroup
+	name := d.KeyColumnQuals["name"].GetStringValue()
+	project := activeProject()
+
+	resp := service.NodeGroups.AggregatedList(project).Filter("name=" + name)
+	if err := resp.Pages(
+		ctx,
+		func(page *compute.NodeGroupAggregatedList) error {
+			for _, item := range page.Items {
+				for _, i := range item.NodeGroups {
+					nodeGroup = *i
+				}
+			}
+			return nil
+		},
+	); err != nil {
+		return nil, err
+	}
+
+	// If the specified resource is not present, API does not return any not found errors
+	if len(nodeGroup.Name) < 1 {
+		return nil, nil
+	}
+
+	return &nodeGroup, nil
+}
+
+func getComputeNodeGroupIamPolicy(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
+	service, err := compute.NewService(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	nodeGroup := h.Item.(*compute.NodeGroup)
+	project := activeProject()
+	zoneName := getLastPathElement(types.SafeString(nodeGroup.Zone))
+
+	req, err := service.NodeGroups.GetIamPolicy(project, zoneName, nodeGroup.Name).Do()
+	if err != nil {
+		// Return nil, if the resource not present
+		result := isNotFoundError([]string{"404"})
+		if result != nil {
+			return nil, nil
+		}
+		return nil, err
+	}
+
+	return req, nil
+}
+
+//// TRANSFORM FUNCTIONS
+
+func gcpComputeNodeGroupAka(_ context.Context, d *transform.TransformData) (interface{}, error) {
+	nodeGroup := d.HydrateItem.(*compute.NodeGroup)
+
+	zoneName := getLastPathElement(types.SafeString(nodeGroup.Zone))
+	nodeGroupName := types.SafeString(nodeGroup.Name)
+
+	akas := []string{"gcp://compute.googleapis.com/projects/" + activeProject() + "/zones/" + zoneName + "/nodeGroups/" + nodeGroupName}
+
+	return akas, nil
+}