add init function for gcp compute image resources

Signed-off-by: Ivan Milchev <[email protected]>

motor/discovery/gcp/mql_assets.go

@@ -92,7 +92,7 @@ type gcpObjectPlatformInfo struct {
 	platform string
 }
 
-func MondooObjectID(o gcpObject) string {
+func GcpPlatformID(o gcpObject) string {
 	return "//platformid.api.mondoo.app/runtime/gcp/" + o.service + "/v1/projects/" + o.project + "/regions/" + o.region + "/" + o.objectType + "/" + o.name
 }
 
@@ -109,11 +109,11 @@ func MqlObjectToAsset(account string, mqlObject mqlObject, tc *providers.Config)
 		log.Error().Err(err).Msg("missing runtime info")
 		return nil
 	}
-	platformid := MondooObjectID(mqlObject.gcpObject)
+	platformid := GcpPlatformID(mqlObject.gcpObject)
 	t := tc.Clone()
 	t.PlatformId = platformid
 	return &asset.Asset{
-		PlatformIds: []string{platformid, mqlObject.gcpObject.id},
+		PlatformIds: []string{platformid},
 		Name:        mqlObject.name,
 		Platform: &platform.Platform{
 			Name:    info.platform,

resources/packs/gcp/common.go

@@ -4,6 +4,7 @@ import (
 	"strings"
 	"time"
 
+	"go.mondoo.com/cnquery/resources"
 	"google.golang.org/protobuf/types/known/timestamppb"
 )
 
@@ -20,3 +21,28 @@ func parseResourceName(fullPath string) string {
 	segments := strings.Split(fullPath, "/")
 	return segments[len(segments)-1]
 }
+
+type assetIdentifier struct {
+	name    string
+	region  string
+	project string
+}
+
+func getAssetIdentifier(runtime *resources.Runtime) *assetIdentifier {
+	a := runtime.Motor.GetAsset()
+	if a == nil {
+		return nil
+	}
+	var name, region, project string
+	for _, id := range a.PlatformIds {
+		if strings.HasPrefix(id, "//platformid.api.mondoo.app/runtime/gcp/") {
+			// "//platformid.api.mondoo.app/runtime/gcp/{o.service}/v1/projects/{project}/regions/{region}/{objectType}/{name}"
+			segments := strings.Split(id, "/")
+			name = segments[len(segments)-1]
+			region = segments[8]
+			project = segments[6]
+			break
+		}
+	}
+	return &assetIdentifier{name: name, region: region, project: project}
+}

resources/packs/gcp/compute.go

@@ -911,6 +911,44 @@ func (g *mqlGcpProjectComputeServiceImage) id() (string, error) {
 	return "gcloud.compute.image/" + id, nil
 }
 
+func (g *mqlGcpProjectComputeServiceImage) init(args *resources.Args) (*resources.Args, GcpProjectComputeServiceImage, error) {
+	if len(*args) > 2 {
+		return args, nil, nil
+	}
+
+	if ids := getAssetIdentifier(g.MotorRuntime); ids != nil {
+		(*args)["name"] = ids.name
+		(*args)["projectId"] = ids.project
+	}
+
+	obj, err := g.MotorRuntime.CreateResource("gcp.project.computeService", "projectId", (*args)["projectId"])
+	if err != nil {
+		return nil, nil, err
+	}
+	computeSvc := obj.(GcpProjectComputeService)
+	images, err := computeSvc.Images()
+	if err != nil {
+		return nil, nil, err
+	}
+
+	for _, i := range images {
+		image := i.(GcpProjectComputeServiceImage)
+		name, err := image.Name()
+		if err != nil {
+			return nil, nil, err
+		}
+		projectId, err := image.ProjectId()
+		if err != nil {
+			return nil, nil, err
+		}
+
+		if name == (*args)["name"] && projectId == (*args)["projectId"] {
+			return args, image, nil
+		}
+	}
+	return nil, nil, &resources.ResourceNotFound{}
+}
+
 func (g *mqlGcpProjectComputeServiceImage) GetSourceDisk() (interface{}, error) {
 	// TODO: implement
 	return nil, errors.New("not implemented")
@@ -945,6 +983,7 @@ func (g *mqlGcpProjectComputeService) GetImages() ([]interface{}, error) {
 		for _, image := range page.Items {
 			mqlImage, err := g.MotorRuntime.CreateResource("gcp.project.computeService.image",
 				"id", strconv.FormatUint(image.Id, 10),
+				"projectId", projectId,
 				"name", image.Name,
 				"description", image.Description,
 				"architecture", image.Architecture,

resources/packs/gcp/gcp.lr

@@ -434,6 +434,8 @@ private gcp.project.computeService.snapshot @defaults("name") {
 private gcp.project.computeService.image @defaults("id name") {
   // Unique identifier
   id string
+  // Project ID
+  projectId string
   // Name of the resource
   name string
   // Optional description