2018-06-11 13:13:19 +00:00
|
|
|
package libcompose
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"path"
|
|
|
|
"path/filepath"
|
|
|
|
|
2018-07-12 07:17:07 +00:00
|
|
|
"github.com/portainer/libcompose/config"
|
2018-06-11 13:13:19 +00:00
|
|
|
"github.com/portainer/libcompose/docker"
|
|
|
|
"github.com/portainer/libcompose/docker/client"
|
|
|
|
"github.com/portainer/libcompose/docker/ctx"
|
|
|
|
"github.com/portainer/libcompose/lookup"
|
|
|
|
"github.com/portainer/libcompose/project"
|
|
|
|
"github.com/portainer/libcompose/project/options"
|
|
|
|
"github.com/portainer/portainer"
|
|
|
|
)
|
|
|
|
|
|
|
|
// ComposeStackManager represents a service for managing compose stacks.
|
|
|
|
type ComposeStackManager struct {
|
|
|
|
dataPath string
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewComposeStackManager initializes a new ComposeStackManager service.
|
|
|
|
func NewComposeStackManager(dataPath string) *ComposeStackManager {
|
|
|
|
return &ComposeStackManager{
|
|
|
|
dataPath: dataPath,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Up will deploy a compose stack (equivalent of docker-compose up)
|
|
|
|
func (manager *ComposeStackManager) Up(stack *portainer.Stack, endpoint *portainer.Endpoint) error {
|
|
|
|
clientFactory, err := client.NewDefaultFactory(client.Options{
|
|
|
|
TLS: endpoint.TLSConfig.TLS,
|
|
|
|
TLSVerify: endpoint.TLSConfig.TLSSkipVerify,
|
|
|
|
Host: endpoint.URL,
|
|
|
|
TLSCAFile: endpoint.TLSCACertPath,
|
|
|
|
TLSCertFile: endpoint.TLSCertPath,
|
|
|
|
TLSKeyFile: endpoint.TLSKeyPath,
|
2018-07-12 07:17:07 +00:00
|
|
|
APIVersion: portainer.SupportedDockerAPIVersion,
|
2018-06-11 13:13:19 +00:00
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2018-07-12 07:17:07 +00:00
|
|
|
env := make(map[string]string)
|
|
|
|
for _, envvar := range stack.Env {
|
|
|
|
env[envvar.Name] = envvar.Value
|
|
|
|
}
|
|
|
|
|
2018-06-11 13:13:19 +00:00
|
|
|
composeFilePath := path.Join(stack.ProjectPath, stack.EntryPoint)
|
|
|
|
proj, err := docker.NewProject(&ctx.Context{
|
|
|
|
ConfigDir: manager.dataPath,
|
|
|
|
Context: project.Context{
|
|
|
|
ComposeFiles: []string{composeFilePath},
|
2018-07-12 07:17:07 +00:00
|
|
|
EnvironmentLookup: &lookup.ComposableEnvLookup{
|
|
|
|
Lookups: []config.EnvironmentLookup{
|
|
|
|
&lookup.EnvfileLookup{
|
|
|
|
Path: filepath.Join(stack.ProjectPath, ".env"),
|
|
|
|
},
|
|
|
|
&lookup.MapLookup{
|
|
|
|
Vars: env,
|
|
|
|
},
|
|
|
|
},
|
2018-06-11 13:13:19 +00:00
|
|
|
},
|
|
|
|
ProjectName: stack.Name,
|
|
|
|
},
|
|
|
|
ClientFactory: clientFactory,
|
|
|
|
}, nil)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return proj.Up(context.Background(), options.Up{})
|
|
|
|
}
|
|
|
|
|
|
|
|
// Down will shutdown a compose stack (equivalent of docker-compose down)
|
|
|
|
func (manager *ComposeStackManager) Down(stack *portainer.Stack, endpoint *portainer.Endpoint) error {
|
|
|
|
clientFactory, err := client.NewDefaultFactory(client.Options{
|
|
|
|
TLS: endpoint.TLSConfig.TLS,
|
|
|
|
TLSVerify: endpoint.TLSConfig.TLSSkipVerify,
|
|
|
|
Host: endpoint.URL,
|
|
|
|
TLSCAFile: endpoint.TLSCACertPath,
|
|
|
|
TLSCertFile: endpoint.TLSCertPath,
|
|
|
|
TLSKeyFile: endpoint.TLSKeyPath,
|
2018-07-12 07:17:07 +00:00
|
|
|
APIVersion: portainer.SupportedDockerAPIVersion,
|
2018-06-11 13:13:19 +00:00
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
composeFilePath := path.Join(stack.ProjectPath, stack.EntryPoint)
|
|
|
|
proj, err := docker.NewProject(&ctx.Context{
|
|
|
|
Context: project.Context{
|
|
|
|
ComposeFiles: []string{composeFilePath},
|
|
|
|
ProjectName: stack.Name,
|
|
|
|
},
|
|
|
|
ClientFactory: clientFactory,
|
|
|
|
}, nil)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2018-06-20 15:20:16 +00:00
|
|
|
return proj.Down(context.Background(), options.Down{RemoveVolume: false, RemoveOrphans: true})
|
2018-06-11 13:13:19 +00:00
|
|
|
}
|