302 lines
6.6 KiB
Go
302 lines
6.6 KiB
Go
// Copyright 2015 Daniel Theophanes.
|
|
// Use of this source code is governed by a zlib-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package service
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"syscall"
|
|
"text/template"
|
|
)
|
|
|
|
func isSystemd() bool {
|
|
if _, err := os.Stat("/run/systemd/system"); err == nil {
|
|
return true
|
|
}
|
|
if _, err := os.Stat("/proc/1/comm"); err == nil {
|
|
filerc, err := os.Open("/proc/1/comm")
|
|
if err != nil {
|
|
return false
|
|
}
|
|
defer filerc.Close()
|
|
|
|
buf := new(bytes.Buffer)
|
|
buf.ReadFrom(filerc)
|
|
contents := buf.String()
|
|
|
|
if strings.Trim(contents, " \r\n") == "systemd" {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
type systemd struct {
|
|
i Interface
|
|
platform string
|
|
*Config
|
|
}
|
|
|
|
func newSystemdService(i Interface, platform string, c *Config) (Service, error) {
|
|
s := &systemd{
|
|
i: i,
|
|
platform: platform,
|
|
Config: c,
|
|
}
|
|
|
|
return s, nil
|
|
}
|
|
|
|
func (s *systemd) String() string {
|
|
if len(s.DisplayName) > 0 {
|
|
return s.DisplayName
|
|
}
|
|
return s.Name
|
|
}
|
|
|
|
func (s *systemd) Platform() string {
|
|
return s.platform
|
|
}
|
|
|
|
func (s *systemd) configPath() (cp string, err error) {
|
|
if !s.Option.bool(optionUserService, optionUserServiceDefault) {
|
|
cp = "/etc/systemd/system/" + s.Config.Name + ".service"
|
|
return
|
|
}
|
|
homeDir, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return
|
|
}
|
|
systemdUserDir := filepath.Join(homeDir, ".config/systemd/user")
|
|
err = os.MkdirAll(systemdUserDir, os.ModePerm)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cp = filepath.Join(systemdUserDir, s.Config.Name + ".service")
|
|
return
|
|
}
|
|
|
|
func (s *systemd) getSystemdVersion() int64 {
|
|
_, out, err := runWithOutput("systemctl", "--version")
|
|
if err != nil {
|
|
return -1
|
|
}
|
|
|
|
re := regexp.MustCompile(`systemd ([0-9]+)`)
|
|
matches := re.FindStringSubmatch(out)
|
|
if len(matches) != 2 {
|
|
return -1
|
|
}
|
|
|
|
v, err := strconv.ParseInt(matches[1], 10, 64)
|
|
if err != nil {
|
|
return -1
|
|
}
|
|
|
|
return v
|
|
}
|
|
|
|
func (s *systemd) hasOutputFileSupport() bool {
|
|
defaultValue := true
|
|
version := s.getSystemdVersion()
|
|
if version == -1 {
|
|
return defaultValue
|
|
}
|
|
|
|
if version < 236 {
|
|
return false
|
|
}
|
|
|
|
return defaultValue
|
|
}
|
|
|
|
func (s *systemd) template() *template.Template {
|
|
customScript := s.Option.string(optionSystemdScript, "")
|
|
|
|
if customScript != "" {
|
|
return template.Must(template.New("").Funcs(tf).Parse(customScript))
|
|
} else {
|
|
return template.Must(template.New("").Funcs(tf).Parse(systemdScript))
|
|
}
|
|
}
|
|
|
|
func (s *systemd) Install() error {
|
|
confPath, err := s.configPath()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = os.Stat(confPath)
|
|
if err == nil {
|
|
return fmt.Errorf("Init already exists: %s", confPath)
|
|
}
|
|
|
|
f, err := os.OpenFile(confPath, os.O_WRONLY|os.O_CREATE, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
path, err := s.execPath()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var to = &struct {
|
|
*Config
|
|
Path string
|
|
HasOutputFileSupport bool
|
|
ReloadSignal string
|
|
PIDFile string
|
|
LimitNOFILE int
|
|
Restart string
|
|
SuccessExitStatus string
|
|
LogOutput bool
|
|
}{
|
|
s.Config,
|
|
path,
|
|
s.hasOutputFileSupport(),
|
|
s.Option.string(optionReloadSignal, ""),
|
|
s.Option.string(optionPIDFile, ""),
|
|
s.Option.int(optionLimitNOFILE, optionLimitNOFILEDefault),
|
|
s.Option.string(optionRestart, "always"),
|
|
s.Option.string(optionSuccessExitStatus, ""),
|
|
s.Option.bool(optionLogOutput, optionLogOutputDefault),
|
|
}
|
|
|
|
err = s.template().Execute(f, to)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if s.Option.bool(optionUserService, optionUserServiceDefault) {
|
|
err = run("systemctl", "enable", "--user", s.Name+".service")
|
|
} else {
|
|
err = run("systemctl", "enable", s.Name+".service")
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if s.Option.bool(optionUserService, optionUserServiceDefault) {
|
|
err = run("systemctl", "daemon-reload", "--user")
|
|
} else {
|
|
err = run("systemctl", "daemon-reload")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (s *systemd) Uninstall() error {
|
|
var err error
|
|
if s.Option.bool(optionUserService, optionUserServiceDefault) {
|
|
err = run("systemctl", "disable", "--user", s.Name+".service")
|
|
} else {
|
|
err = run("systemctl", "disable", s.Name+".service")
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cp, err := s.configPath()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := os.Remove(cp); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *systemd) Logger(errs chan<- error) (Logger, error) {
|
|
if system.Interactive() {
|
|
return ConsoleLogger, nil
|
|
}
|
|
return s.SystemLogger(errs)
|
|
}
|
|
func (s *systemd) SystemLogger(errs chan<- error) (Logger, error) {
|
|
return newSysLogger(s.Name, errs)
|
|
}
|
|
|
|
func (s *systemd) Run() (err error) {
|
|
err = s.i.Start(s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
s.Option.funcSingle(optionRunWait, func() {
|
|
var sigChan = make(chan os.Signal, 3)
|
|
signal.Notify(sigChan, syscall.SIGTERM, os.Interrupt)
|
|
<-sigChan
|
|
})()
|
|
|
|
return s.i.Stop(s)
|
|
}
|
|
|
|
func (s *systemd) Status() (Status, error) {
|
|
exitCode, out, err := runWithOutput("systemctl", "is-active", s.Name)
|
|
if exitCode == 0 && err != nil {
|
|
return StatusUnknown, err
|
|
}
|
|
|
|
switch {
|
|
case strings.HasPrefix(out, "active"):
|
|
return StatusRunning, nil
|
|
case strings.HasPrefix(out, "inactive"):
|
|
return StatusStopped, nil
|
|
case strings.HasPrefix(out, "activating"):
|
|
return StatusRunning, nil
|
|
case strings.HasPrefix(out, "failed"):
|
|
return StatusUnknown, errors.New("service in failed state")
|
|
default:
|
|
return StatusUnknown, ErrNotInstalled
|
|
}
|
|
}
|
|
|
|
func (s *systemd) Start() error {
|
|
return run("systemctl", "start", s.Name+".service")
|
|
}
|
|
|
|
func (s *systemd) Stop() error {
|
|
return run("systemctl", "stop", s.Name+".service")
|
|
}
|
|
|
|
func (s *systemd) Restart() error {
|
|
return run("systemctl", "restart", s.Name+".service")
|
|
}
|
|
|
|
const systemdScript = `[Unit]
|
|
Description={{.Description}}
|
|
ConditionFileIsExecutable={{.Path|cmdEscape}}
|
|
{{range $i, $dep := .Dependencies}}
|
|
{{$dep}} {{end}}
|
|
|
|
[Service]
|
|
StartLimitInterval=5
|
|
StartLimitBurst=10
|
|
ExecStart={{.Path|cmdEscape}}{{range .Arguments}} {{.|cmd}}{{end}}
|
|
{{if .ChRoot}}RootDirectory={{.ChRoot|cmd}}{{end}}
|
|
{{if .WorkingDirectory}}WorkingDirectory={{.WorkingDirectory|cmdEscape}}{{end}}
|
|
{{if .UserName}}User={{.UserName}}{{end}}
|
|
{{if .ReloadSignal}}ExecReload=/bin/kill -{{.ReloadSignal}} "$MAINPID"{{end}}
|
|
{{if .PIDFile}}PIDFile={{.PIDFile|cmd}}{{end}}
|
|
{{if and .LogOutput .HasOutputFileSupport -}}
|
|
StandardOutput=file:/var/log/{{.Name}}.out
|
|
StandardError=file:/var/log/{{.Name}}.err
|
|
{{- end}}
|
|
{{if gt .LimitNOFILE -1 }}LimitNOFILE={{.LimitNOFILE}}{{end}}
|
|
{{if .Restart}}Restart={{.Restart}}{{end}}
|
|
{{if .SuccessExitStatus}}SuccessExitStatus={{.SuccessExitStatus}}{{end}}
|
|
RestartSec=120
|
|
EnvironmentFile=-/etc/sysconfig/{{.Name}}
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
`
|