wip: migrate to mono-repo. SPN has already been moved to spn/

service/profile/database.go

@@ -0,0 +1,144 @@
+package profile
+
+import (
+	"context"
+	"errors"
+	"strings"
+
+	"github.com/safing/portbase/config"
+	"github.com/safing/portbase/database"
+	"github.com/safing/portbase/database/query"
+	"github.com/safing/portbase/database/record"
+	"github.com/safing/portbase/log"
+)
+
+// Database paths:
+// core:profiles/<scope>/<id>
+// cache:profiles/index/<identifier>/<value>
+
+// ProfilesDBPath is the base database path for profiles.
+const ProfilesDBPath = "core:profiles/"
+
+var profileDB = database.NewInterface(&database.Options{
+	Local:    true,
+	Internal: true,
+})
+
+// MakeScopedID returns a scoped profile ID.
+func MakeScopedID(source ProfileSource, id string) string {
+	return string(source) + "/" + id
+}
+
+// MakeProfileKey returns a profile key.
+func MakeProfileKey(source ProfileSource, id string) string {
+	return ProfilesDBPath + string(source) + "/" + id
+}
+
+func registerValidationDBHook() (err error) {
+	_, err = database.RegisterHook(query.New(ProfilesDBPath), &databaseHook{})
+	return
+}
+
+func startProfileUpdateChecker() error {
+	module.StartServiceWorker("update active profiles", 0, func(ctx context.Context) (err error) {
+		profilesSub, err := profileDB.Subscribe(query.New(ProfilesDBPath))
+		if err != nil {
+			return err
+		}
+		defer func() {
+			err := profilesSub.Cancel()
+			if err != nil {
+				log.Warningf("profile: failed to cancel subscription for updating active profiles: %s", err)
+			}
+		}()
+
+	profileFeed:
+		for {
+			select {
+			case r := <-profilesSub.Feed:
+				// Check if subscription was canceled.
+				if r == nil {
+					return errors.New("subscription canceled")
+				}
+
+				// Get active profile.
+				scopedID := strings.TrimPrefix(r.Key(), ProfilesDBPath)
+				activeProfile := getActiveProfile(scopedID)
+				if activeProfile == nil {
+					// Check if profile is being deleted.
+					if r.Meta().IsDeleted() {
+						meta.MarkDeleted(scopedID)
+					}
+
+					// Don't do any additional actions if the profile is not active.
+					continue profileFeed
+				}
+
+				// Always increase the revision counter of the layer profile.
+				// This marks previous connections in the UI as decided with outdated settings.
+				if activeProfile.layeredProfile != nil {
+					activeProfile.layeredProfile.increaseRevisionCounter(true)
+				}
+
+				// Always mark as outdated if the record is being deleted.
+				if r.Meta().IsDeleted() {
+					activeProfile.outdated.Set()
+
+					meta.MarkDeleted(scopedID)
+					module.TriggerEvent(DeletedEvent, scopedID)
+					continue
+				}
+
+				// If the profile is saved externally (eg. via the API), have the
+				// next one to use it reload the profile from the database.
+				receivedProfile, err := EnsureProfile(r)
+				if err != nil || !receivedProfile.savedInternally {
+					activeProfile.outdated.Set()
+					module.TriggerEvent(ConfigChangeEvent, scopedID)
+				}
+			case <-ctx.Done():
+				return nil
+			}
+		}
+	})
+
+	return nil
+}
+
+type databaseHook struct {
+	database.HookBase
+}
+
+// UsesPrePut implements the Hook interface and returns false.
+func (h *databaseHook) UsesPrePut() bool {
+	return true
+}
+
+// PrePut implements the Hook interface.
+func (h *databaseHook) PrePut(r record.Record) (record.Record, error) {
+	// Do not intervene with metadata key.
+	if r.Key() == profilesMetadataKey {
+		return r, nil
+	}
+
+	// convert
+	profile, err := EnsureProfile(r)
+	if err != nil {
+		return nil, err
+	}
+
+	// clean config
+	config.CleanHierarchicalConfig(profile.Config)
+
+	// prepare profile
+	profile.prepProfile()
+
+	// parse config
+	err = profile.parseConfig()
+	if err != nil {
+		// error here, warning when loading
+		return nil, err
+	}
+
+	return profile, nil
+}