safing-portmaster/status/module.go

package status

import (
	"context"
	"fmt"

	"github.com/safing/portbase/modules"
	"github.com/safing/portbase/utils/debug"
	"github.com/safing/portmaster/netenv"
)

var (
	module *modules.Module
)

func init() {
	module = modules.Register("status", prepare, start, nil, "base", "config")
}

func start() error {
	if err := setupRuntimeProvider(); err != nil {
		return err
	}

	module.StartWorker("auto-pilot", autoPilot)

	triggerAutopilot()

	if err := module.RegisterEventHook(
		netenv.ModuleName,
		netenv.OnlineStatusChangedEvent,
		"update online status in system status",
		func(_ context.Context, _ interface{}) error {
			triggerAutopilot()
			return nil
		},
	); err != nil {
		return err
	}

	if err := module.RegisterEventHook(
		"config",
		"config change",
		"Update network rating system",
		func(_ context.Context, _ interface{}) error {
			if !NetworkRatingEnabled() && ActiveSecurityLevel() != SecurityLevelNormal {
				setSelectedLevel(SecurityLevelNormal)
				triggerAutopilot()
			}
			return nil
		},
	); err != nil {
		return err
	}

	return nil
}

func prepare() error {
	if err := registerConfig(); err != nil {
		return err
	}

	return nil
}

// AddToDebugInfo adds the system status to the given debug.Info.
func AddToDebugInfo(di *debug.Info) {
	di.AddSection(
		fmt.Sprintf("Status: %s", SecurityLevelString(ActiveSecurityLevel())),
		debug.UseCodeSection|debug.AddContentLineBreaks,
		fmt.Sprintf("ActiveSecurityLevel:   %s", SecurityLevelString(ActiveSecurityLevel())),
		fmt.Sprintf("SelectedSecurityLevel: %s", SecurityLevelString(SelectedSecurityLevel())),
		fmt.Sprintf("ThreatMitigationLevel: %s", SecurityLevelString(getHighestMitigationLevel())),
		fmt.Sprintf("CaptivePortal:         %s", netenv.GetCaptivePortal().URL),
		fmt.Sprintf("OnlineStatus:          %s", netenv.GetOnlineStatus()),
	)
}