Refactor to use libcompose

compose/project.go

@@ -0,0 +1,162 @@
+package compose
+
+import (
+	log "github.com/Sirupsen/logrus"
+	"github.com/docker/libcompose/cli/logger"
+	"github.com/docker/libcompose/docker"
+	"github.com/docker/libcompose/project"
+	"github.com/rancherio/os/config"
+	rosDocker "github.com/rancherio/os/docker"
+	"github.com/rancherio/os/util"
+)
+
+func CreateService(cfg *config.CloudConfig, name string, serviceConfig *project.ServiceConfig) (project.Service, error) {
+	if cfg == nil {
+		var err error
+		cfg, err = config.LoadConfig()
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	p, err := RunServiceSet("once", cfg, map[string]*project.ServiceConfig{
+		name: serviceConfig,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return p.CreateService(name)
+}
+
+func RunServiceSet(name string, cfg *config.CloudConfig, configs map[string]*project.ServiceConfig) (*project.Project, error) {
+	p, err := newProject(name, cfg)
+	if err != nil {
+		return nil, err
+	}
+
+	addServices(p, cfg, map[string]string{}, configs)
+
+	return p, p.Up()
+}
+
+func RunServices(cfg *config.CloudConfig) error {
+	p, err := newCoreServiceProject(cfg)
+	if err != nil {
+		return err
+	}
+
+	return p.Up()
+}
+
+func newProject(name string, cfg *config.CloudConfig) (*project.Project, error) {
+	clientFactory, err := rosDocker.NewClientFactory(docker.ClientOpts{})
+	if err != nil {
+		return nil, err
+	}
+
+	serviceFactory := &rosDocker.ServiceFactory{
+		Deps: map[string][]string{},
+	}
+	context := &docker.Context{
+		ClientFactory: clientFactory,
+		Context: project.Context{
+			ProjectName:       name,
+			EnvironmentLookup: rosDocker.NewConfigEnvironment(cfg),
+			ServiceFactory:    serviceFactory,
+			Rebuild:           true,
+			Log:               cfg.Rancher.Log,
+			LoggerFactory:     logger.NewColorLoggerFactory(),
+		},
+	}
+	serviceFactory.Context = context
+
+	return docker.NewProject(context)
+}
+
+func addServices(p *project.Project, cfg *config.CloudConfig, enabled map[string]string, configs map[string]*project.ServiceConfig) {
+	// Note: we ignore errors while loading services
+	for name, serviceConfig := range cfg.Rancher.Services {
+		hash := project.GetServiceHash(name, *serviceConfig)
+
+		if enabled[name] == hash {
+			continue
+		}
+
+		if err := p.AddConfig(name, serviceConfig); err != nil {
+			log.Infof("Failed loading service %s", name)
+			continue
+		}
+
+		enabled[name] = hash
+	}
+}
+
+func newCoreServiceProject(cfg *config.CloudConfig) (*project.Project, error) {
+	network := false
+	projectEvents := make(chan project.ProjectEvent)
+	enabled := make(map[string]string)
+
+	p, err := newProject("os", cfg)
+	if err != nil {
+		return nil, err
+	}
+
+	p.AddListener(project.NewDefaultListener(p))
+	p.AddListener(projectEvents)
+
+	p.ReloadCallback = func() error {
+		err := cfg.Reload()
+		if err != nil {
+			return err
+		}
+
+		for service, serviceEnabled := range cfg.Rancher.ServicesInclude {
+			if enabled[service] != "" || !serviceEnabled {
+				continue
+			}
+
+			bytes, err := LoadServiceResource(service, network, cfg)
+			if err != nil {
+				if err == util.ErrNoNetwork {
+					log.Debugf("Can not load %s, networking not enabled", service)
+				} else {
+					log.Errorf("Failed to load %s : %v", service, err)
+				}
+				continue
+			}
+
+			err = p.Load(bytes)
+			if err != nil {
+				log.Errorf("Failed to load %s : %v", service, err)
+				continue
+			}
+
+			enabled[service] = service
+		}
+
+		addServices(p, cfg, enabled, cfg.Rancher.Services)
+
+		return nil
+	}
+
+	go func() {
+		for event := range projectEvents {
+			if event.Event == project.CONTAINER_STARTED && event.ServiceName == "network" {
+				network = true
+			}
+		}
+	}()
+
+	err = p.ReloadCallback()
+	if err != nil {
+		log.Errorf("Failed to reload os: %v", err)
+		return nil, err
+	}
+
+	return p, nil
+}
+
+func LoadServiceResource(name string, network bool, cfg *config.CloudConfig) ([]byte, error) {
+	return util.LoadResource(name, network, cfg.Rancher.Repositories.ToArray())
+}