moby--moby/plugin/defs.go

package plugin // import "github.com/docker/docker/plugin"

import (
	"fmt"
	"strings"
	"sync"

	"github.com/docker/docker/pkg/plugins"
	v2 "github.com/docker/docker/plugin/v2"
	specs "github.com/opencontainers/runtime-spec/specs-go"
)

// Store manages the plugin inventory in memory and on-disk
type Store struct {
	sync.RWMutex
	plugins  map[string]*v2.Plugin
	specOpts map[string][]SpecOpt
	/* handlers are necessary for transition path of legacy plugins
	 * to the new model. Legacy plugins use Handle() for registering an
	 * activation callback.*/
	handlers map[string][]func(string, *plugins.Client)
}

// NewStore creates a Store.
func NewStore() *Store {
	return &Store{
		plugins:  make(map[string]*v2.Plugin),
		specOpts: make(map[string][]SpecOpt),
		handlers: make(map[string][]func(string, *plugins.Client)),
	}
}

// SpecOpt is used for subsystems that need to modify the runtime spec of a plugin
type SpecOpt func(*specs.Spec)

// CreateOpt is used to configure specific plugin details when created
type CreateOpt func(p *v2.Plugin)

// WithSwarmService is a CreateOpt that flags the passed in a plugin as a plugin
// managed by swarm
func WithSwarmService(id string) CreateOpt {
	return func(p *v2.Plugin) {
		p.SwarmServiceID = id
	}
}

// WithEnv is a CreateOpt that passes the user-provided environment variables
// to the plugin container, de-duplicating variables with the same names case
// sensitively and only appends valid key=value pairs
func WithEnv(env []string) CreateOpt {
	return func(p *v2.Plugin) {
		effectiveEnv := make(map[string]string)
		for _, penv := range p.PluginObj.Config.Env {
			if penv.Value != nil {
				effectiveEnv[penv.Name] = *penv.Value
			}
		}
		for _, line := range env {
			if pair := strings.SplitN(line, "=", 2); len(pair) > 1 {
				effectiveEnv[pair[0]] = pair[1]
			}
		}
		p.PluginObj.Settings.Env = make([]string, len(effectiveEnv))
		i := 0
		for key, value := range effectiveEnv {
			p.PluginObj.Settings.Env[i] = fmt.Sprintf("%s=%s", key, value)
			i++
		}
	}
}

// WithSpecMounts is a SpecOpt which appends the provided mounts to the runtime spec
func WithSpecMounts(mounts []specs.Mount) SpecOpt {
	return func(s *specs.Spec) {
		s.Mounts = append(s.Mounts, mounts...)
	}
}