package docker

import (
	"os"

	plugin_exec "github.com/thegeeklab/wp-plugin-go/v3/exec"
	"github.com/urfave/cli/v2"
)

const dockerdBin = "/usr/local/bin/dockerd"

// Daemon defines Docker daemon parameters.
type Daemon struct {
	Registry             string          // Docker registry
	Mirror               string          // Docker registry mirror
	Insecure             bool            // Docker daemon enable insecure registries
	StorageDriver        string          // Docker daemon storage driver
	StoragePath          string          // Docker daemon storage path
	Disabled             bool            // DOcker daemon is disabled (already running)
	Debug                bool            // Docker daemon started in debug mode
	Bip                  string          // Docker daemon network bridge IP address
	DNS                  cli.StringSlice // Docker daemon dns server
	DNSSearch            cli.StringSlice // Docker daemon dns search domain
	MTU                  string          // Docker daemon mtu setting
	IPv6                 bool            // Docker daemon IPv6 networking
	Experimental         bool            // Docker daemon enable experimental mode
	BuildkitConfigFile   string          // Docker buildkit config file
	MaxConcurrentUploads string          // Docker daemon max concurrent uploads
}

// helper function to create the docker daemon command.
func (d *Daemon) Start() *plugin_exec.Cmd {
	args := []string{
		"--data-root", d.StoragePath,
		"--host=unix:///var/run/docker.sock",
	}

	if d.StorageDriver != "" {
		args = append(args, "-s", d.StorageDriver)
	}

	if d.Insecure && d.Registry != "" {
		args = append(args, "--insecure-registry", d.Registry)
	}

	if d.IPv6 {
		args = append(args, "--ipv6")
	}

	if d.Mirror != "" {
		args = append(args, "--registry-mirror", d.Mirror)
	}

	if d.Bip != "" {
		args = append(args, "--bip", d.Bip)
	}

	for _, dns := range d.DNS.Value() {
		args = append(args, "--dns", dns)
	}

	for _, dnsSearch := range d.DNSSearch.Value() {
		args = append(args, "--dns-search", dnsSearch)
	}

	if d.MTU != "" {
		args = append(args, "--mtu", d.MTU)
	}

	if d.Experimental {
		args = append(args, "--experimental")
	}

	if d.MaxConcurrentUploads != "" {
		args = append(args, "--max-concurrent-uploads", d.MaxConcurrentUploads)
	}

	cmd := plugin_exec.Command(dockerdBin, args...)

	if d.Debug {
		cmd.Stdout = os.Stdout
		cmd.Stderr = os.Stderr
	}

	return cmd
}

func (d *Daemon) CreateBuilder() *plugin_exec.Cmd {
	args := []string{
		"buildx",
		"create",
		"--use",
	}

	if d.BuildkitConfigFile != "" {
		args = append(args, "--config", d.BuildkitConfigFile)
	}

	cmd := plugin_exec.Command(dockerBin, args...)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	return cmd
}

func (d *Daemon) ListBuilder() *plugin_exec.Cmd {
	cmd := plugin_exec.Command(dockerBin, "buildx", "ls")
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	return cmd
}

func (d *Daemon) StartCoreDNS() *plugin_exec.Cmd {
	cmd := plugin_exec.Command("coredns", "-conf", "/etc/coredns/Corefile")

	if d.Debug {
		cmd.Stdout = os.Stdout
		cmd.Stderr = os.Stderr
	}

	return cmd
}