func updateDockerOpts(dd driver.DistroDriver, args string) error {
log.Printf("Updating daemon args to: %s", args)
if err := dd.ChangeOpts(args); err != nil {
return fmt.Errorf("error updating DOCKER_OPTS: %v", err)
}
return nil
}
func install(he vmextension.HandlerEnvironment, d driver.DistroDriver) error {
// Install docker daemon
log.Printf("++ install docker")
if _, err := exec.LookPath("docker"); err == nil {
log.Printf("docker already installed. not re-installing")
} else {
if err := d.InstallDocker(); err != nil {
return err
}
}
log.Printf("-- install docker")
// Install docker-compose
log.Printf("++ install docker-compose")
if err := installCompose(d.DockerComposeDir()); err != nil {
return fmt.Errorf("error installing docker-compose: %v", err)
}
log.Printf("-- install docker-compose")
// Add user to 'docker' group to user docker as non-root
u, err := util.GetAzureUser()
if err != nil {
return fmt.Errorf("failed to get provisioned user: %v", err)
}
log.Printf("++ add user to docker group")
if out, err := executil.Exec("usermod", "-aG", "docker", u); err != nil {
log.Printf("%s", string(out))
return err
}
log.Printf("-- add user to docker group")
return nil
}
func composeUp(d driver.DistroDriver, json map[string]interface{}) error {
if len(json) == 0 {
log.Println("docker-compose config not specified, noop")
return nil
}
// Convert json to yaml
yaml, err := yaml.Marshal(json)
if err != nil {
return fmt.Errorf("error converting to compose.yml: %v", err)
}
tmpDir, err := ioutil.TempDir(os.TempDir(), "compose")
if err != nil {
return fmt.Errorf("failed creating temp dir: %v", err)
}
log.Printf("Using compose yaml:---------\n%s\n----------", string(yaml))
ymlPath := filepath.Join(tmpDir, composeYml)
if err := ioutil.WriteFile(ymlPath, yaml, 0666); err != nil {
return fmt.Errorf("error writing yaml: %v", err)
}
compose := filepath.Join(d.DockerComposeDir(), composeBin)
return executil.ExecPipe(compose, "-f", ymlPath, "up", "-d")
}
func uninstall(he vmextension.HandlerEnvironment, d driver.DistroDriver) error {
log.Println("++ uninstall docker")
if err := d.UninstallDocker(); err != nil {
return err
}
log.Println("-- uninstall docker")
return nil
}
func disable(he vmextension.HandlerEnvironment, d driver.DistroDriver) error {
log.Printf("++ stop docker daemon")
if err := d.StopDocker(); err != nil {
return err
}
log.Printf("-- stop docker daemon")
return nil
}
func enable(he vmextension.HandlerEnvironment, d driver.DistroDriver) error {
settings, err := parseSettings(he.HandlerEnvironment.ConfigFolder)
if err != nil {
return err
}
// Install docker remote access certs
log.Printf("++ setup docker certs")
if err := installDockerCerts(*settings, dockerCfgDir); err != nil {
return fmt.Errorf("error installing docker certs: %v", err)
}
log.Printf("-- setup docker certs")
// Update dockeropts
log.Printf("++ update dockeropts")
if err := updateDockerOpts(d, getArgs(*settings, d)); err != nil {
return fmt.Errorf("failed to update dockeropts: %v", err)
}
log.Printf("-- update dockeropts")
// Restart docker
log.Printf("++ restart docker")
if err := d.RestartDocker(); err != nil {
return err
}
time.Sleep(3 * time.Second) // wait for instance to come up
log.Printf("-- restart docker")
// Login Docker registry server
log.Printf("++ login docker registry")
if err := loginRegistry(settings.Login); err != nil {
return err
}
log.Printf("-- login docker registry")
// Compose Up
log.Printf("++ compose up")
if err := composeUp(d, settings.ComposeJson); err != nil {
return fmt.Errorf("'compose up' failed: %v", err)
}
log.Printf("-- compose up")
return nil
}
func getArgs(s DockerHandlerSettings, dd driver.DistroDriver) string {
args := dd.BaseOpts()
if s.Certs.HasDockerCerts() {
tls := []string{"--tlsverify",
fmt.Sprintf("--tlscacert=%s", filepath.Join(dockerCfgDir, dockerCaCert)),
fmt.Sprintf("--tlscert=%s", filepath.Join(dockerCfgDir, dockerSrvCert)),
fmt.Sprintf("--tlskey=%s", filepath.Join(dockerCfgDir, dockerSrvKey)),
}
args = append(args, tls...)
}
if s.Docker.Port != "" {
args = append(args, fmt.Sprintf("-H=0.0.0.0:%s", s.Docker.Port))
}
if len(s.Docker.Options) > 0 {
args = append(args, s.Docker.Options...)
}
return strings.Join(args, " ")
}
func composeUp(d driver.DistroDriver, json map[string]interface{}) error {
if len(json) == 0 {
log.Println("docker-compose config not specified, noop")
return nil
}
// Convert json to yaml
yaml, err := yaml.Marshal(json)
if err != nil {
return fmt.Errorf("error converting to compose.yml: %v", err)
}
if err := os.MkdirAll(composeDir, 0777); err != nil {
return fmt.Errorf("failed creating %s: %v", composeDir, err)
}
log.Printf("Using compose yaml:>>>>>\n%s\n<<<<<", string(yaml))
ymlPath := filepath.Join(composeDir, composeYml)
if err := ioutil.WriteFile(ymlPath, yaml, 0666); err != nil {
return fmt.Errorf("error writing %s: %v", ymlPath, err)
}
compose := filepath.Join(d.DockerComposeDir(), composeBin)
return executil.ExecPipeToFds(executil.Fds{Out: ioutil.Discard}, compose, "-p", composeProject, "-f", ymlPath, "up", "-d")
}