node-api/docker/types.go

package docker

import (
	"log"
	"path"

	"github.com/rosti-cz/node-api/apps"
)

// username in the containers under which all containers run
const appUsername = "app"
const passwordFile = "/srv/.rosti"

// Container extends App struct from App
type Container struct {
	App *apps.App `json:"app"`
}

func (c *Container) getDriver() *Driver {
	driver := &Driver{}
	return driver
}

// volumeHostPath each container has one volume mounted into it,
func (c *Container) volumeHostPath() string {
	return path.Join("/srv", c.App.Name)
}

// GetRawResourceStats returns RAW CPU and memory usage directly from Docker API
func (c *Container) GetRawResourceStats() (int64, int, error) {
	driver := c.getDriver()
	cpu, memory, err := driver.RawStats(c.App.Name)
	return cpu, memory, err
}

// GetState returns app state object with populated state fields
func (c *Container) GetState() (*apps.AppState, error) {
	status, err := c.Status()
	if err != nil {
		return nil, err
	}

	cpu, memory, err := c.ResourceUsage()
	if err != nil {
		return nil, err
	}

	bytes, inodes, err := c.DiskUsage()
	if err != nil {
		return nil, err
	}

	state := apps.AppState{
		State:           status,
		CPUUsage:        cpu,
		MemoryUsage:     memory,
		DiskUsageBytes:  bytes,
		DiskUsageInodes: inodes,
	}

	return &state, nil
}

// Status returns state of the container
// Possible values: running, stopped, no-container, unknown
func (c *Container) Status() (string, error) {
	status := "unknown"

	// if _, err := os.Stat(path.Join("/srv", c.App.Name)); !os.IsNotExist(err) {
	// 	status = "data-only"
	// }

	driver := c.getDriver()
	containerStatus, err := driver.Status(c.App.Name)
	if err != nil && err.Error() == "no container found" {
		return "no-container", nil
	}
	if err != nil {
		return status, err
	}

	status = containerStatus

	return status, nil
}

// DiskUsage returns number of bytes and inodes used by the container in it's mounted volume
func (c *Container) DiskUsage() (int, int, error) {
	return du(c.volumeHostPath())
}

// ResourceUsage returns amount of memory in B and CPU in % that the app occupies
func (c *Container) ResourceUsage() (float64, int, error) {
	driver := c.getDriver()

	cpu, memory, err := driver.Stats(c.App.Name)
	if err != nil {
		return 0.0, 0, err
	}

	return cpu, memory, nil
}

// Create creates the container
func (c *Container) Create() error {
	driver := c.getDriver()
	_, err := driver.Create(
		c.App.Name,
		c.App.Image,
		c.volumeHostPath(),
		c.App.HTTPPort,
		c.App.SSHPort,
		c.App.CPU,
		c.App.Memory,
		[]string{},
	)

	return err
}

// Start starts the container
func (c *Container) Start() error {
	driver := c.getDriver()

	return driver.Start(c.App.Name)
}

// Stop stops the container
func (c *Container) Stop() error {
	driver := c.getDriver()

	return driver.Stop(c.App.Name)
}

// Restart restarts the container
func (c *Container) Restart() error {
	driver := c.getDriver()

	err := driver.Stop(c.App.Name)
	if err != nil {
		return err
	}

	return driver.Start(c.App.Name)
}

// Destroy removes the container but keeps the data so it can be created again
func (c *Container) Destroy() error {
	driver := c.getDriver()

	return driver.Remove(c.App.Name)
}

// Delete removes both data and the container
func (c *Container) Delete() error {
	err := c.Destroy()
	if err != nil {
		return err
	}

	volumePath := path.Join("/srv", c.App.Name)
	err = removeDirectory(volumePath)
	if err != nil {
		log.Println(err)
	}

	return nil
}

// SetPassword configures password for system user app in the container
func (c *Container) SetPassword(password string) error {
	driver := c.getDriver()

	err := driver.Exec(c.App.Name, []string{"chpasswd"}, appUsername+":"+password)
	if err != nil {
		return err
	}

	err = driver.Exec(c.App.Name, []string{"tee", passwordFile}, password)
	if err != nil {
		return err
	}

	return err
}

// SetFileContent uploads text into a file inside the container. It's greate for uploading SSH keys.
// The method creates the diretory where the file is located and sets mode of the final file
func (c *Container) SetFileContent(filename string, text string, mode string) error {
	driver := c.getDriver()

	directory := path.Dir(filename)

	err := driver.Exec(c.App.Name, []string{"mkdir", "-p", directory}, "")
	if err != nil {
		return err
	}

	err = driver.Exec(c.App.Name, []string{"tee", filename}, text)
	if err != nil {
		return err
	}

	err = driver.Exec(c.App.Name, []string{"chown", directory, "app:app"}, "")
	if err != nil {
		return err
	}

	err = driver.Exec(c.App.Name, []string{"chown", filename, "app:app"}, "")
	if err != nil {
		return err
	}

	err = driver.Exec(c.App.Name, []string{"chmod", mode, filename}, "")
	return err
}