mirror of https://github.com/docker/docs.git
129 lines
3.0 KiB
Go
129 lines
3.0 KiB
Go
package provision
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/docker/machine/drivers"
|
|
"github.com/docker/machine/libmachine/auth"
|
|
"github.com/docker/machine/libmachine/engine"
|
|
"github.com/docker/machine/libmachine/provision/pkgaction"
|
|
"github.com/docker/machine/libmachine/swarm"
|
|
"github.com/docker/machine/log"
|
|
"github.com/docker/machine/utils"
|
|
)
|
|
|
|
func init() {
|
|
Register("Ubuntu", &RegisteredProvisioner{
|
|
New: NewUbuntuProvisioner,
|
|
})
|
|
}
|
|
|
|
func NewUbuntuProvisioner(d drivers.Driver) Provisioner {
|
|
return &UbuntuProvisioner{
|
|
GenericProvisioner{
|
|
DockerOptionsDir: "/etc/docker",
|
|
DaemonOptionsFile: "/etc/default/docker",
|
|
OsReleaseId: "ubuntu",
|
|
Packages: []string{
|
|
"curl",
|
|
},
|
|
Driver: d,
|
|
},
|
|
}
|
|
}
|
|
|
|
type UbuntuProvisioner struct {
|
|
GenericProvisioner
|
|
}
|
|
|
|
func (provisioner *UbuntuProvisioner) Service(name string, action pkgaction.ServiceAction) error {
|
|
command := fmt.Sprintf("sudo service %s %s", name, action.String())
|
|
|
|
if _, err := provisioner.SSHCommand(command); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (provisioner *UbuntuProvisioner) Package(name string, action pkgaction.PackageAction) error {
|
|
var packageAction string
|
|
|
|
switch action {
|
|
case pkgaction.Install:
|
|
packageAction = "install"
|
|
case pkgaction.Remove:
|
|
packageAction = "remove"
|
|
case pkgaction.Upgrade:
|
|
packageAction = "upgrade"
|
|
}
|
|
|
|
// TODO: This should probably have a const
|
|
switch name {
|
|
case "docker":
|
|
name = "lxc-docker"
|
|
}
|
|
|
|
command := fmt.Sprintf("DEBIAN_FRONTEND=noninteractive sudo -E apt-get %s -y %s", packageAction, name)
|
|
|
|
if _, err := provisioner.SSHCommand(command); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (provisioner *UbuntuProvisioner) dockerDaemonResponding() bool {
|
|
if _, err := provisioner.SSHCommand("sudo docker version"); err != nil {
|
|
log.Warn("Error getting SSH command to check if the daemon is up: %s", err)
|
|
return false
|
|
}
|
|
|
|
// The daemon is up if the command worked. Carry on.
|
|
return true
|
|
}
|
|
|
|
func (provisioner *UbuntuProvisioner) Provision(swarmOptions swarm.SwarmOptions, authOptions auth.AuthOptions, engineOptions engine.EngineOptions) error {
|
|
provisioner.SwarmOptions = swarmOptions
|
|
provisioner.AuthOptions = authOptions
|
|
provisioner.EngineOptions = engineOptions
|
|
|
|
if provisioner.EngineOptions.StorageDriver == "" {
|
|
provisioner.EngineOptions.StorageDriver = "aufs"
|
|
}
|
|
|
|
if err := provisioner.SetHostname(provisioner.Driver.GetMachineName()); err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, pkg := range provisioner.Packages {
|
|
if err := provisioner.Package(pkg, pkgaction.Install); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if err := installDockerGeneric(provisioner); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := utils.WaitFor(provisioner.dockerDaemonResponding); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := makeDockerOptionsDir(provisioner); err != nil {
|
|
return err
|
|
}
|
|
|
|
provisioner.AuthOptions = setRemoteAuthOptions(provisioner)
|
|
|
|
if err := ConfigureAuth(provisioner); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := configureSwarm(provisioner, swarmOptions); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|