mirror of
https://github.com/gravitl/netmaker.git
synced 2024-09-21 15:56:07 +08:00
147 lines
3.3 KiB
Go
147 lines
3.3 KiB
Go
package daemon
|
|
|
|
import (
|
|
//"github.com/davecgh/go-spew/spew"
|
|
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/gravitl/netmaker/netclient/ncutils"
|
|
)
|
|
|
|
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 RemoveSystemDServices(network string) error {
|
|
//sysExec, err := exec.LookPath("systemctl")
|
|
if !ncutils.IsWindows() {
|
|
fullremove, err := isOnlyService(network)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
if fullremove {
|
|
_, err = ncutils.RunCmd("systemctl disable netclient.service", true)
|
|
}
|
|
_, _ = ncutils.RunCmd("systemctl daemon-reload", true)
|
|
|
|
if ncutils.FileExists("/etc/systemd/system/netclient.timer") {
|
|
_, _ = ncutils.RunCmd("systemctl disable netclient.timer", true)
|
|
}
|
|
if fullremove {
|
|
if ncutils.FileExists("/etc/systemd/system/netclient.service") {
|
|
err = os.Remove("/etc/systemd/system/netclient.service")
|
|
}
|
|
}
|
|
if ncutils.FileExists("/etc/systemd/system/netclient.timer") {
|
|
err = os.Remove("/etc/systemd/system/netclient.timer")
|
|
}
|
|
if err != nil {
|
|
log.Println("Error removing file. Please investigate.")
|
|
log.Println(err)
|
|
}
|
|
_, _ = ncutils.RunCmd("systemctl daemon-reload", true)
|
|
_, _ = ncutils.RunCmd("systemctl reset-failed", true)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
|
|
func isOnlyService(network string) (bool, error) {
|
|
isonly := false
|
|
files, err := filepath.Glob("/etc/netclient/config/netconfig-*")
|
|
if err != nil {
|
|
return isonly, err
|
|
}
|
|
count := len(files)
|
|
if count == 0 {
|
|
isonly = true
|
|
}
|
|
return isonly, err
|
|
|
|
}
|