serviceman/manager/install_windows.go

192 lines
4.5 KiB
Go
Raw Normal View History

2019-07-04 07:36:35 +00:00
package manager
2019-07-01 08:44:48 +00:00
import (
2019-07-03 05:51:30 +00:00
"encoding/json"
2019-07-01 08:44:48 +00:00
"fmt"
2019-07-03 05:51:30 +00:00
"io/ioutil"
2019-07-01 08:44:48 +00:00
"log"
2019-07-03 05:51:30 +00:00
"os"
2019-07-01 08:44:48 +00:00
"path/filepath"
"strings"
"git.rootprojects.org/root/go-serviceman/runner"
2019-07-03 05:51:30 +00:00
"git.rootprojects.org/root/go-serviceman/service"
2019-07-01 08:44:48 +00:00
"golang.org/x/sys/windows/registry"
)
2019-07-05 18:48:58 +00:00
var (
srvLen int
srvExt = ".json"
srvSysPath = "/opt/serviceman/etc"
srvUserPath = ".local/opt/serviceman/etc"
)
func init() {
srvLen = len(srvExt)
}
2019-07-03 05:51:30 +00:00
// TODO nab some goodness from https://github.com/takama/daemon
2019-07-01 08:44:48 +00:00
// TODO system service requires elevated privileges
// See https://coolaj86.com/articles/golang-and-windows-and-admins-oh-my/
2019-07-03 05:51:30 +00:00
func install(c *service.Service) error {
2019-07-01 08:44:48 +00:00
/*
// LEAVE THIS DOCUMENTATION HERE
reg.exe
/V <value name> - "Telebit"
/T <data type> - "REG_SZ" - String
/D <value data>
/C - case sensitive
/F <search data??> - not sure...
// Special Note:
"/c" is similar to -- (*nix), and required within the data string
So instead of setting "do.exe --do-arg1 --do-arg2"
you must set "do.exe /c --do-arg1 --do-arg2"
vars.telebitNode += '.exe';
var cmd = 'reg.exe add "HKCU\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run"'
+ ' /V "Telebit" /t REG_SZ /D '
+ '"' + things.argv[0] + ' /c ' // something like C:\Program Files (x64)\nodejs\node.exe
+ [ path.join(__dirname, 'bin/telebitd.js')
, 'daemon'
, '--config'
, path.join(os.homedir(), '.config/telebit/telebitd.yml')
].join(' ')
+ '" /F'
;
*/
autorunKey := `SOFTWARE\Microsoft\Windows\CurrentVersion\Run`
k, _, err := registry.CreateKey(
registry.CURRENT_USER,
autorunKey,
registry.SET_VALUE,
)
if err != nil {
log.Fatal(err)
}
defer k.Close()
// Try to stop before trying to copy the file
_ = runner.Stop(c)
2019-07-03 05:51:30 +00:00
args, err := installServiceman(c)
if nil != err {
return err
2019-07-01 08:44:48 +00:00
}
2019-07-03 05:51:30 +00:00
/*
setArgs := ""
args := c.Argv
exec := c.Exec
bin := c.Interpreter
if "" != bin {
// If this is something like node or python,
// the interpeter must be called as "the main thing"
// and "the app" must be an argument
args = append([]string{exec}, args...)
} else {
// Otherwise, if "the app" is a true binary,
// it can be "the main thing"
bin = exec
}
// The final string ends up looking something like one of these:
2019-07-03 08:11:50 +00:00
// `"C:\Users\aj\.local\opt\appname\appname.js" -p 8080`
// `"C:\Program Files (x64)\nodejs\node.exe" C:\Users\aj\.local\opt\appname\appname.js -p 8080`
2019-07-03 05:51:30 +00:00
regSZ := bin + setArgs + strings.Join(c.Argv, " ")
*/
2019-07-03 08:11:50 +00:00
regSZ := fmt.Sprintf(`"%s" %s`, args[0], strings.Join(args[1:], " "))
2019-07-01 08:44:48 +00:00
if len(regSZ) > 260 {
return fmt.Errorf("data value is too long for registry entry")
}
2019-07-03 08:11:50 +00:00
// In order for a windows gui program to not show a console,
// it has to not output any messages?
//fmt.Println("Set Registry Key:")
//fmt.Println(autorunKey, c.Title, regSZ)
2019-07-01 08:44:48 +00:00
k.SetStringValue(c.Title, regSZ)
// to return ErrDaemonize
return start(c)
2019-07-01 08:44:48 +00:00
}
func start(conf *service.Service) error {
args := getRunnerArgs(conf)
return &ErrDaemonize{
DaemonArgs: append(args, "--daemon"),
error: "Not as much an error as a bad value...",
}
//return runner.Start(conf)
}
func stop(conf *service.Service) error {
return runner.Stop(conf)
}
func getRunnerArgs(c *service.Service) []string {
2019-07-03 05:51:30 +00:00
self := os.Args[0]
2019-07-04 09:36:58 +00:00
debug := ""
if strings.Contains(self, "debug.exe") {
debug = "debug."
}
2019-07-03 05:51:30 +00:00
smdir := `\opt\serviceman`
// TODO support service level services (which probably wouldn't need serviceman)
smdir = filepath.Join(c.Home, ".local", smdir)
// for now we'll scope the runner to the name of the application
2019-07-04 09:36:58 +00:00
smbin := filepath.Join(smdir, `bin\serviceman.`+debug+c.Name+`.exe`)
2019-07-03 05:51:30 +00:00
confpath := filepath.Join(smdir, `etc`)
conffile := filepath.Join(confpath, c.Name+`.json`)
return []string{
smbin,
"run",
"--config",
conffile,
}
}
// copies self to install path and returns config path
func installServiceman(c *service.Service) ([]string, error) {
// TODO check version and upgrade or dismiss
self := os.Args[0]
args := getRunnerArgs(c)
smbin := args[0]
conffile := args[len(args)-1]
2019-07-03 05:51:30 +00:00
if smbin != self {
2019-07-03 06:05:12 +00:00
err := os.MkdirAll(filepath.Dir(smbin), 0755)
2019-07-03 05:51:30 +00:00
if nil != err {
2019-07-03 06:05:12 +00:00
return nil, err
2019-07-03 05:51:30 +00:00
}
bin, err := ioutil.ReadFile(self)
if nil != err {
2019-07-03 06:05:12 +00:00
return nil, err
2019-07-03 05:51:30 +00:00
}
2019-07-03 06:05:12 +00:00
err = ioutil.WriteFile(smbin, bin, 0755)
2019-07-03 05:51:30 +00:00
if nil != err {
2019-07-03 06:05:12 +00:00
return nil, err
2019-07-03 05:51:30 +00:00
}
}
b, err := json.Marshal(c)
if nil != err {
// this should be impossible, so we'll just panic
panic(err)
}
err = os.MkdirAll(filepath.Dir(conffile), 0755)
if nil != err {
return nil, err
}
err = ioutil.WriteFile(conffile, b, 0640)
2019-07-03 05:51:30 +00:00
if nil != err {
2019-07-03 06:05:12 +00:00
return nil, err
2019-07-03 05:51:30 +00:00
}
return args, nil
2019-07-03 05:51:30 +00:00
}