112 lines
2.4 KiB
Go
112 lines
2.4 KiB
Go
package schema
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/blang/semver"
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/pkg/errors"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/resource/plugin"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/tokens"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/workspace"
|
|
)
|
|
|
|
type Loader interface {
|
|
LoadPackage(pkg string, version *semver.Version) (*Package, error)
|
|
}
|
|
|
|
type pluginLoader struct {
|
|
m sync.RWMutex
|
|
|
|
host plugin.Host
|
|
entries map[string]*Package
|
|
}
|
|
|
|
func NewPluginLoader(host plugin.Host) Loader {
|
|
return &pluginLoader{
|
|
host: host,
|
|
entries: map[string]*Package{},
|
|
}
|
|
}
|
|
|
|
func (l *pluginLoader) getPackage(key string) (*Package, bool) {
|
|
l.m.RLock()
|
|
defer l.m.RUnlock()
|
|
|
|
p, ok := l.entries[key]
|
|
return p, ok
|
|
}
|
|
|
|
// ensurePlugin downloads and installs the specified plugin if it does not already exist.
|
|
func (l *pluginLoader) ensurePlugin(pkg string, version *semver.Version) error {
|
|
// TODO: schema and provider versions
|
|
// hack: Some of the hcl2 code isn't yet handling versions, so bail out if the version is nil to avoid failing
|
|
// the download. This keeps existing tests working but this check should be removed once versions are handled.
|
|
if version == nil {
|
|
return nil
|
|
}
|
|
|
|
pkgPlugin := workspace.PluginInfo{
|
|
Kind: workspace.ResourcePlugin,
|
|
Name: pkg,
|
|
Version: version,
|
|
}
|
|
if !workspace.HasPlugin(pkgPlugin) {
|
|
tarball, _, err := pkgPlugin.Download()
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to download plugin: %s", pkgPlugin)
|
|
}
|
|
if err := pkgPlugin.Install(tarball); err != nil {
|
|
return errors.Wrapf(err, "failed to install plugin %s", pkgPlugin)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (l *pluginLoader) LoadPackage(pkg string, version *semver.Version) (*Package, error) {
|
|
key := pkg + "@"
|
|
if version != nil {
|
|
key += version.String()
|
|
}
|
|
|
|
if p, ok := l.getPackage(key); ok {
|
|
return p, nil
|
|
}
|
|
|
|
if err := l.ensurePlugin(pkg, version); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
provider, err := l.host.Provider(tokens.Package(pkg), version)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
schemaFormatVersion := 0
|
|
schemaBytes, err := provider.GetSchema(schemaFormatVersion)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var spec PackageSpec
|
|
if err := jsoniter.Unmarshal(schemaBytes, &spec); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
p, err := importSpec(spec, nil, l)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
l.m.Lock()
|
|
defer l.m.Unlock()
|
|
|
|
if p, ok := l.entries[pkg]; ok {
|
|
return p, nil
|
|
}
|
|
l.entries[key] = p
|
|
|
|
return p, nil
|
|
}
|