From 92debdf072448c8d2d338d6090f43843644ac2af Mon Sep 17 00:00:00 2001 From: Maycon Santos Date: Sat, 30 Dec 2023 20:19:00 +0100 Subject: [PATCH] Add InitEventStore function extract the method from the management cmd package --- integrations/integrations.go | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/integrations/integrations.go b/integrations/integrations.go index def74e4..bc65eeb 100644 --- a/integrations/integrations.go +++ b/integrations/integrations.go @@ -3,7 +3,10 @@ package integrations import ( "github.com/gorilla/mux" "github.com/netbirdio/netbird/management/server" + "github.com/netbirdio/netbird/management/server/activity" + "github.com/netbirdio/netbird/management/server/activity/sqlite" "github.com/netbirdio/netbird/management/server/jwtclaims" + log "github.com/sirupsen/logrus" ) func RegisterHandlers( @@ -13,3 +16,16 @@ func RegisterHandlers( ) *mux.Router { return router } + +func InitEventStore(dataDir string, key string) (activity.Store, string, error) { + var err error + if key == "" { + log.Debugf("generate new activity store encryption key") + key, err = sqlite.GenerateKey() + if err != nil { + return nil, "", err + } + } + store, err := sqlite.NewSQLiteStore(dataDir, key) + return store, key, err +}