🧹 Add function to load the asset url schema for all known providers (#…

…3633)

* 🧹 Add function to load the asset url schema for all known providers

* make it safe to create schema multiple times

providers-sdk/v1/inventory/asset_url.go

@@ -149,10 +149,34 @@ func (a *AssetUrlSchema) Add(branch *AssetUrlBranch) error {
 	}
 
 	branch.setDepth(found.Depth + 1)
-	found.Values[lastKey] = branch
+	found.Values[lastKey] = copyBranch(branch)
 	return nil
 }
 
+func copyBranch(b *AssetUrlBranch) *AssetUrlBranch {
+	if b == nil {
+		return nil
+	}
+
+	res := &AssetUrlBranch{
+		PathSegments: make([]string, len(b.PathSegments)),
+		Key:          b.Key,
+		Values:       make(map[string]*AssetUrlBranch, len(b.Values)),
+		Title:        b.Title,
+		References:   make([]string, len(b.References)),
+		Depth:        b.Depth,
+	}
+
+	copy(res.PathSegments, b.PathSegments)
+	copy(res.References, b.References)
+
+	for k, v := range b.Values {
+		res.Values[k] = copyBranch(v)
+	}
+
+	return res
+}
+
 func (a *AssetUrlBranch) setDepth(i uint32) {
 	a.Depth = i
 	next := i + 1

providers/providers.go

@@ -24,6 +24,7 @@ import (
 	"github.com/spf13/afero"
 	"github.com/ulikunitz/xz"
 	"go.mondoo.com/cnquery/v10/cli/config"
+	"go.mondoo.com/cnquery/v10/providers-sdk/v1/inventory"
 	"go.mondoo.com/cnquery/v10/providers-sdk/v1/plugin"
 	"go.mondoo.com/cnquery/v10/providers-sdk/v1/resources"
 	"go.mondoo.com/cnquery/v10/providers/core/resources/versions/semver"
@@ -942,3 +943,29 @@ func convertToFields(keysAndValues ...interface{}) map[string]interface{} {
 	}
 	return fields
 }
+
+func LoadAssetUrlSchema() (*inventory.AssetUrlSchema, error) {
+	providers, err := ListAll()
+	if err != nil {
+		return nil, err
+	}
+
+	s, err := inventory.NewAssetUrlSchema("technology")
+	if err != nil {
+		return nil, err
+	}
+
+	for _, provider := range providers {
+		for _, b := range provider.AssetUrlTrees {
+			if err := s.Add(b); err != nil {
+				return nil, err
+			}
+		}
+	}
+
+	if err := s.RefreshCache(); err != nil {
+		return nil, err
+	}
+
+	return s, nil
+}