mirror of
https://github.com/safing/portmaster
synced 2025-04-16 17:09:09 +00:00
115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/safing/portmaster/base/api"
|
|
"github.com/safing/portmaster/base/info"
|
|
"github.com/safing/portmaster/base/metrics"
|
|
"github.com/safing/portmaster/cmds/cmdbase"
|
|
"github.com/safing/portmaster/service"
|
|
"github.com/safing/portmaster/service/configure"
|
|
"github.com/safing/portmaster/service/updates"
|
|
"github.com/safing/portmaster/spn/captain"
|
|
"github.com/safing/portmaster/spn/conf"
|
|
"github.com/safing/portmaster/spn/sluice"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
rootCmd = &cobra.Command{
|
|
Use: "observation-hub",
|
|
PersistentPreRun: initializeGlobals,
|
|
Run: cmdbase.RunService,
|
|
}
|
|
|
|
binDir string
|
|
dataDir string
|
|
|
|
logToStdout bool
|
|
logDir string
|
|
logLevel string
|
|
)
|
|
|
|
func init() {
|
|
// Add persistent flags for all commands.
|
|
rootCmd.PersistentFlags().StringVar(&binDir, "bin-dir", "", "set directory for executable binaries (rw/ro)")
|
|
rootCmd.PersistentFlags().StringVar(&dataDir, "data-dir", "", "set directory for variable data (rw)")
|
|
|
|
// Add flags for service only.
|
|
rootCmd.Flags().BoolVar(&logToStdout, "log-stdout", false, "log to stdout instead of file")
|
|
rootCmd.Flags().StringVar(&logDir, "log-dir", "", "set directory for logs")
|
|
rootCmd.Flags().StringVar(&logLevel, "log", "", "set log level to [trace|debug|info|warning|error|critical]")
|
|
rootCmd.Flags().BoolVar(&cmdbase.PrintStackOnExit, "print-stack-on-exit", false, "prints the stack before of shutting down")
|
|
rootCmd.Flags().BoolVar(&cmdbase.RebootOnRestart, "reboot-on-restart", false, "reboot server instead of service restart")
|
|
|
|
// Add other commands.
|
|
rootCmd.AddCommand(cmdbase.VersionCmd)
|
|
rootCmd.AddCommand(cmdbase.UpdateCmd)
|
|
}
|
|
|
|
func main() {
|
|
// Add Go's default flag set.
|
|
// TODO: Move flags throughout Portmaster to here and add their values to the service config.
|
|
rootCmd.Flags().AddGoFlagSet(flag.CommandLine)
|
|
|
|
if err := rootCmd.Execute(); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func initializeGlobals(cmd *cobra.Command, args []string) {
|
|
// Set version info.
|
|
info.Set("SPN Observation Hub", "", "GPLv3")
|
|
|
|
// Configure metrics.
|
|
_ = metrics.SetNamespace("observer")
|
|
|
|
// Configure user agent.
|
|
updates.UserAgent = fmt.Sprintf("SPN Observation Hub (%s %s)", runtime.GOOS, runtime.GOARCH)
|
|
|
|
// Configure SPN mode.
|
|
conf.EnableClient(true)
|
|
captain.DisableAccount = true
|
|
|
|
// Disable unneeded listeners.
|
|
sluice.EnableListener = false
|
|
api.EnableServer = false
|
|
|
|
// Configure service.
|
|
cmdbase.SvcFactory = func(svcCfg *service.ServiceConfig) (cmdbase.ServiceInstance, error) {
|
|
svc, err := service.New(svcCfg)
|
|
|
|
// Add additional modules.
|
|
observer, err := New(svc)
|
|
if err != nil {
|
|
fmt.Printf("error creating an instance: create observer module: %s\n", err)
|
|
os.Exit(2)
|
|
}
|
|
svc.AddModule(observer)
|
|
_, err = NewApprise(svc)
|
|
if err != nil {
|
|
fmt.Printf("error creating an instance: create apprise module: %s\n", err)
|
|
os.Exit(2)
|
|
}
|
|
svc.AddModule(observer)
|
|
|
|
return svc, err
|
|
}
|
|
cmdbase.SvcConfig = &service.ServiceConfig{
|
|
BinDir: binDir,
|
|
DataDir: dataDir,
|
|
|
|
LogToStdout: logToStdout,
|
|
LogDir: logDir,
|
|
LogLevel: logLevel,
|
|
|
|
BinariesIndexURLs: configure.DefaultStableBinaryIndexURLs,
|
|
IntelIndexURLs: configure.DefaultIntelIndexURLs,
|
|
VerifyBinaryUpdates: configure.BinarySigningTrustStore,
|
|
}
|
|
}
|