mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 18:00:49 +01:00
120 lines
2.2 KiB
Go
120 lines
2.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var installCmd = &cobra.Command{
|
|
Use: "install",
|
|
Short: "installs Netbird service",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
SetFlagsFromEnvVars(rootCmd)
|
|
|
|
cmd.SetOut(cmd.OutOrStdout())
|
|
|
|
err := handleRebrand(cmd)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
svcConfig := newSVCConfig()
|
|
|
|
svcConfig.Arguments = []string{
|
|
"service",
|
|
"run",
|
|
"--config",
|
|
configPath,
|
|
"--log-level",
|
|
logLevel,
|
|
"--daemon-addr",
|
|
daemonAddr,
|
|
}
|
|
|
|
if managementURL != "" {
|
|
svcConfig.Arguments = append(svcConfig.Arguments, "--management-url", managementURL)
|
|
}
|
|
|
|
if logFile != "console" {
|
|
svcConfig.Arguments = append(svcConfig.Arguments, "--log-file", logFile)
|
|
}
|
|
|
|
if runtime.GOOS == "linux" {
|
|
// Respected only by systemd systems
|
|
svcConfig.Dependencies = []string{"After=network.target syslog.target"}
|
|
|
|
if logFile != "console" {
|
|
setStdLogPath := true
|
|
dir := filepath.Dir(logFile)
|
|
|
|
_, err := os.Stat(dir)
|
|
if err != nil {
|
|
err = os.MkdirAll(dir, 0750)
|
|
if err != nil {
|
|
setStdLogPath = false
|
|
}
|
|
}
|
|
|
|
if setStdLogPath {
|
|
svcConfig.Option["LogOutput"] = true
|
|
svcConfig.Option["LogDirectory"] = dir
|
|
}
|
|
}
|
|
}
|
|
|
|
if runtime.GOOS == "windows" {
|
|
svcConfig.Option["OnFailure"] = "restart"
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(cmd.Context())
|
|
|
|
s, err := newSVC(newProgram(ctx, cancel), svcConfig)
|
|
if err != nil {
|
|
cmd.PrintErrln(err)
|
|
return err
|
|
}
|
|
|
|
err = s.Install()
|
|
if err != nil {
|
|
cmd.PrintErrln(err)
|
|
return err
|
|
}
|
|
|
|
cmd.Println("Netbird service has been installed")
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var uninstallCmd = &cobra.Command{
|
|
Use: "uninstall",
|
|
Short: "uninstalls Netbird service from system",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
SetFlagsFromEnvVars(rootCmd)
|
|
|
|
cmd.SetOut(cmd.OutOrStdout())
|
|
|
|
err := handleRebrand(cmd)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(cmd.Context())
|
|
|
|
s, err := newSVC(newProgram(ctx, cancel), newSVCConfig())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = s.Uninstall()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cmd.Println("Netbird service has been uninstalled")
|
|
return nil
|
|
},
|
|
}
|