go-watchdog/cmd/watchdog/installer/install.go

//go:generate go run -mod=vendor github.com/UnnoTed/fileb0x b0x.toml

package installer

import (
	"os"
	"path/filepath"
	"strings"
)

// Config should describe the service well-enough for it to
// run on Mac, Linux, and Windows.
//
// 	&Config{
// 		// A human-friendy name
// 		Title: "Foobar App",
// 		// A computer-friendly name
// 		Name: "foobar-app",
// 		// A name for OS X plist
// 		ReverseDNS: "com.example.foobar-app",
// 		// A human-friendly description
// 		Desc: "Foobar App",
// 		// The app /service homepage
// 		URL: "https://example.com/foobar-app/",
// 		// The full path of the interpreter, if any (ruby, python, node, etc)
// 		Interpreter: "/opt/node/bin/node",
// 		// The name of the executable (or script)
// 		Exec: "foobar-app.js",
// 		// An array of arguments
// 		Argv: []string{"-c", "/path/to/config.json"},
// 		// A map of Environment variables that should be set
// 		Envs: map[string]string{
// 			PORT: "8080",
// 			ENV: "development",
// 		},
// 		// The user (Linux & Mac only).
// 		// This does not apply to userspace services.
// 		// There may be special considerations
// 		User: "www-data",
// 		// If different from User
// 		Group: "",
// 		// Whether to install as a system or user service
// 		System: false,
// 		// Whether or not the service may need privileged ports
// 		PrivilegedPorts: false,
// 	}
//
// Note that some fields are exported for templating,
// but not intended to be set by you.
// These are documented as omitted from JSON.
// Try to stick to what's outlined above.
type Config struct {
	Title               string            `json:"title"`
	Name                string            `json:"name"`
	Desc                string            `json:"desc"`
	URL                 string            `json:"url"`
	Interpreter         string            `json:"interpreter"` // i.e. node, python
	Exec                string            `json:"exec"`
	Argv                []string          `json:"argv"`
	Args                string            `json:"-"`
	Envs                map[string]string `json:"envs"`
	User                string            `json:"user"`
	Group               string            `json:"group"`
	home                string            `json:"-"`
	Local               string            `json:"-"`
	LogDir              string            `json:"-"`
	System              bool              `json:"system"`
	Restart             bool              `json:"restart"`
	Production          bool              `json:"production"`
	PrivilegedPorts     bool              `json:"privileged_ports"`
	MultiuserProtection bool              `json:"multiuser_protection"`
}

// Install will do a best-effort attempt to install a start-on-startup
// user or system service via systemd, launchd, or reg.exe
func Install(c *Config) error {
	if "" == c.Exec {
		c.Exec = c.Name
	}
	c.Args = strings.Join(c.Argv, " ")

	// TODO handle non-system installs
	// * ~/.local/opt/watchdog/watchdog
	// * ~/.local/share/watchdog/var/log/
	// * ~/.config/watchdog/watchdog.json
	if !c.System {
		home, err := os.UserHomeDir()
		if nil != err {
			return err
		}
		c.home = home
		c.Local = filepath.Join(c.home, ".local")
		c.LogDir = filepath.Join(c.home, ".local", "share", c.Name, "var", "log")
	} else {
		c.LogDir = "/var/log/" + c.Name
	}

	err := install(c)
	if nil != err {
		return err
	}

	err = os.MkdirAll(c.LogDir, 0750)
	if nil != err {
		return err
	}

	return nil
}