From bd2d399951b0ce41cd0feb8c18dc5e8b5515290b Mon Sep 17 00:00:00 2001 From: Chris Roberts Date: Thu, 2 Dec 2021 15:47:13 -0800 Subject: [PATCH] Use function to get component type --- internal/core/basis.go | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/internal/core/basis.go b/internal/core/basis.go index a009c1d03..74ee3a215 100644 --- a/internal/core/basis.go +++ b/internal/core/basis.go @@ -162,22 +162,15 @@ func (b *Basis) DataDir() (*datadir.Basis, error) { return b.dir, nil } -var StringComponentTypeMap = map[string]component.Type{ - "command": component.CommandType, - "communicator": component.CommunicatorType, - "guest": component.GuestType, - "host": component.HostType, - "provider": component.ProviderType, - "provisioner": component.ProvisionerType, - "synced_folder": component.SyncedFolderType, -} - // Implements core.Basis // Returns all the registered plugins of the types specified func (b *Basis) Plugins(types ...string) (plugins []*core.NamedPlugin, err error) { plugins = []*core.NamedPlugin{} for _, pluginType := range types { - typ := StringComponentTypeMap[pluginType] + typ, err := component.FindType(pluginType) + if err != nil { + return nil, err + } components, err := b.typeComponents(b.ctx, typ) if err != nil { return nil, err