steampipe/pkg/steampipeconfig/modconfig/runtime_dependency_provider_impl.go

package modconfig

import (
	"fmt"
	"github.com/hashicorp/hcl/v2"
	"golang.org/x/exp/maps"
)

type RuntimeDependencyProviderImpl struct {
	ModTreeItemImpl
	// required to allow partial decoding
	RuntimeDependencyProviderRemain hcl.Body `hcl:",remain" json:"-"`

	// map of withs keyed by unqualified name
	withs               map[string]*DashboardWith
	runtimeDependencies map[string]*RuntimeDependency
}

func (b *RuntimeDependencyProviderImpl) AddWith(with *DashboardWith) hcl.Diagnostics {
	if b.withs == nil {
		b.withs = make(map[string]*DashboardWith)
	}
	// if we already have this with, fail
	if _, ok := b.withs[with.UnqualifiedName]; ok {
		return hcl.Diagnostics{&hcl.Diagnostic{
			Severity: hcl.DiagError,
			Summary:  fmt.Sprintf("duplicate with block '%s'", with.ShortName),
			Subject:  with.GetDeclRange(),
		}}
	}
	b.withs[with.UnqualifiedName] = with
	return nil
}

func (b *RuntimeDependencyProviderImpl) GetWiths() []*DashboardWith {
	return maps.Values(b.withs)
}

func (b *RuntimeDependencyProviderImpl) AddRuntimeDependencies(dependencies []*RuntimeDependency) {
	if b.runtimeDependencies == nil {
		b.runtimeDependencies = make(map[string]*RuntimeDependency)
	}
	for _, dependency := range dependencies {
		// set the dependency provider (this is used if this resource is inherited via base)
		dependency.Provider = b
		b.runtimeDependencies[dependency.String()] = dependency
	}
}

func (b *RuntimeDependencyProviderImpl) GetRuntimeDependencies() map[string]*RuntimeDependency {
	return b.runtimeDependencies
}