mirror of
https://github.com/gravitl/netmaker.git
synced 2024-11-14 05:40:50 +08:00
142 lines
3.4 KiB
Go
142 lines
3.4 KiB
Go
package daemon
|
|
|
|
import (
|
|
//"github.com/davecgh/go-spew/spew"
|
|
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/gravitl/netmaker/netclient/ncutils"
|
|
)
|
|
|
|
// SetupSystemDDaemon - sets system daemon for supported machines
|
|
func SetupSystemDDaemon(interval string) error {
|
|
|
|
if ncutils.IsWindows() {
|
|
return nil
|
|
}
|
|
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
binarypath := dir + "/netclient"
|
|
|
|
_, err = os.Stat("/etc/netclient/config")
|
|
if os.IsNotExist(err) {
|
|
os.MkdirAll("/etc/netclient/config", 0744)
|
|
} else if err != nil {
|
|
log.Println("couldnt find or create /etc/netclient")
|
|
return err
|
|
}
|
|
|
|
if !ncutils.FileExists("/usr/local/bin/netclient") {
|
|
os.Symlink("/etc/netclient/netclient", "/usr/local/bin/netclient")
|
|
}
|
|
if !ncutils.FileExists("/etc/netclient/netclient") {
|
|
_, err = ncutils.Copy(binarypath, "/etc/netclient/netclient")
|
|
if err != nil {
|
|
log.Println(err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
systemservice := `[Unit]
|
|
Description=Network Check
|
|
Wants=netclient.timer
|
|
|
|
[Service]
|
|
Type=simple
|
|
ExecStart=/etc/netclient/netclient checkin -n all
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
`
|
|
|
|
systemtimer := `[Unit]
|
|
Description=Calls the Netmaker Mesh Client Service
|
|
Requires=netclient.service
|
|
|
|
[Timer]
|
|
Unit=netclient.service
|
|
|
|
`
|
|
systemtimer = systemtimer + "OnCalendar=*:*:0/" + interval
|
|
|
|
systemtimer = systemtimer +
|
|
`
|
|
|
|
[Install]
|
|
WantedBy=timers.target
|
|
`
|
|
|
|
servicebytes := []byte(systemservice)
|
|
timerbytes := []byte(systemtimer)
|
|
|
|
if !ncutils.FileExists("/etc/systemd/system/netclient.service") {
|
|
err = ioutil.WriteFile("/etc/systemd/system/netclient.service", servicebytes, 0644)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
if !ncutils.FileExists("/etc/systemd/system/netclient.timer") {
|
|
err = ioutil.WriteFile("/etc/systemd/system/netclient.timer", timerbytes, 0644)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
_, _ = ncutils.RunCmd("systemctl enable netclient.service", true)
|
|
_, _ = ncutils.RunCmd("systemctl daemon-reload", true)
|
|
_, _ = ncutils.RunCmd("systemctl enable netclient.timer", true)
|
|
_, _ = ncutils.RunCmd("systemctl start netclient.timer", true)
|
|
return nil
|
|
}
|
|
|
|
func CleanupLinux() {
|
|
err := os.RemoveAll(ncutils.GetNetclientPath())
|
|
if err != nil {
|
|
ncutils.PrintLog("Removing netclient binary: "+err.Error(), 1)
|
|
}
|
|
}
|
|
|
|
// RemoveSystemDServices - removes the systemd services on a machine
|
|
func RemoveSystemDServices() error {
|
|
//sysExec, err := exec.LookPath("systemctl")
|
|
var err error
|
|
if !ncutils.IsWindows() && isOnlyService() {
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
ncutils.RunCmd("systemctl disable netclient.service", false)
|
|
ncutils.RunCmd("systemctl disable netclient.timer", false)
|
|
if ncutils.FileExists("/etc/systemd/system/netclient.service") {
|
|
err = os.Remove("/etc/systemd/system/netclient.service")
|
|
if err != nil {
|
|
ncutils.Log("Error removing /etc/systemd/system/netclient.service. Please investigate.")
|
|
}
|
|
}
|
|
if ncutils.FileExists("/etc/systemd/system/netclient.timer") {
|
|
err = os.Remove("/etc/systemd/system/netclient.timer")
|
|
if err != nil {
|
|
ncutils.Log("Error removing /etc/systemd/system/netclient.timer. Please investigate.")
|
|
}
|
|
}
|
|
ncutils.RunCmd("systemctl daemon-reload", false)
|
|
ncutils.RunCmd("systemctl reset-failed", false)
|
|
ncutils.Log("removed systemd remnants if any existed")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func isOnlyService() bool {
|
|
files, err := filepath.Glob("/etc/netclient/config/netconfig-*")
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return len(files) == 0
|
|
}
|