package manager import ( "encoding/json" "fmt" "io/ioutil" "log" "os" "path/filepath" "strings" "git.rootprojects.org/root/go-serviceman/runner" "git.rootprojects.org/root/go-serviceman/service" "golang.org/x/sys/windows/registry" ) var ( srvLen int srvExt = ".json" srvSysPath = "/opt/serviceman/etc" srvUserPath = ".local/opt/serviceman/etc" ) func init() { srvLen = len(srvExt) } // TODO nab some goodness from https://github.com/takama/daemon // TODO system service requires elevated privileges // See https://coolaj86.com/articles/golang-and-windows-and-admins-oh-my/ func install(c *service.Service) error { /* // LEAVE THIS DOCUMENTATION HERE reg.exe /V - "Telebit" /T - "REG_SZ" - String /D /C - case sensitive /F - 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) args, err := installServiceman(c) if nil != err { return err } /* 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: // `"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` regSZ := bin + setArgs + strings.Join(c.Argv, " ") */ regSZ := fmt.Sprintf(`"%s" %s`, args[0], strings.Join(args[1:], " ")) if len(regSZ) > 260 { return fmt.Errorf("data value is too long for registry entry") } // 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) k.SetStringValue(c.Title, regSZ) // to return ErrDaemonize return start(c) } 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 { self := os.Args[0] debug := "" if strings.Contains(self, "debug.exe") { debug = "debug." } 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 smbin := filepath.Join(smdir, `bin\serviceman.`+debug+c.Name+`.exe`) 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] if smbin != self { err := os.MkdirAll(filepath.Dir(smbin), 0755) if nil != err { return nil, err } bin, err := ioutil.ReadFile(self) if nil != err { return nil, err } err = ioutil.WriteFile(smbin, bin, 0755) if nil != err { return nil, err } } 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) if nil != err { return nil, err } return args, nil }