Improve status module

status/database.go

@@ -0,0 +1,47 @@
+package status
+
+import (
+	"github.com/Safing/portbase/database"
+	"github.com/Safing/portbase/database/query"
+	"github.com/Safing/portbase/database/record"
+)
+
+const (
+	statusDBKey = "core:status/status"
+)
+
+var (
+	statusDB = database.NewInterface(nil)
+	hook     *database.RegisteredHook
+)
+
+type statusHook struct {
+	database.HookBase
+}
+
+// UsesPrePut implements the Hook interface.
+func (sh *statusHook) UsesPrePut() bool {
+	return true
+}
+
+// PrePut implements the Hook interface.
+func (sh *statusHook) PrePut(r record.Record) (record.Record, error) {
+	newStatus, err := EnsureSystemStatus(r)
+	if err != nil {
+		return nil, err
+	}
+	newStatus.Lock()
+	defer newStatus.Unlock()
+
+	// apply applicable settings
+	setSelectedSecurityLevel(newStatus.SelectedSecurityLevel)
+	// TODO: allow setting of Gate17 status (on/off)
+
+	// return original status
+	return status, nil
+}
+
+func initStatusHook() (err error) {
+	hook, err = database.RegisterHook(query.New(statusDBKey), &statusHook{})
+	return err
+}