remove deprecated code

internal/handlers/ussd/menuhandler.go

@@ -68,7 +68,7 @@ type Handlers struct {
 
 func NewHandlers(parser *asm.FlagParser, pe *persist.Persister, userdataStore db.Db) (*Handlers, error) {
 	userDb := utils.UserDataStore{
-		Store: userdataStore,
+		Db: userdataStore,
 	}
 	if pe == nil {
 		return nil, fmt.Errorf("cannot create handler with nil persister")
@@ -78,7 +78,7 @@ func NewHandlers(parser *asm.FlagParser, pe *persist.Persister, userdataStore db
 	}
 	h := &Handlers{
 		pe:             pe,
-		userdataStore:  userDb,
+		userdataStore:  &userDb,
 		flagManager:    parser,
 		accountService: &server.AccountService{},
 	}
@@ -254,7 +254,7 @@ func (h *Handlers) VerifyPin(ctx context.Context, sym string, input []byte) (res
 	}
 
 	//AccountPin, _ := utils.ReadEntry(ctx, h.userdataStore, sessionId, utils.DATA_ACCOUNT_PIN)
-	store := h.userdataStore.(utils.UserDataStore)
+	store := h.userdataStore
 	AccountPin, err := store.ReadEntry(ctx, sessionId, utils.DATA_ACCOUNT_PIN)
 	if err != nil {
 		return res, err
@@ -523,7 +523,7 @@ func (h *Handlers) CheckAccountStatus(ctx context.Context, sym string, input []b
 	if !ok {
 		return res, fmt.Errorf("missing session")
 	}
-	store := h.userdataStore.(utils.UserDataStore)
+	store := h.userdataStore
 	trackingId, err := store.ReadEntry(ctx, sessionId, utils.DATA_TRACKING_ID)
 	if err != nil {
 		return res, err
@@ -610,7 +610,7 @@ func (h *Handlers) CheckBalance(ctx context.Context, sym string, input []byte) (
 		return res, fmt.Errorf("missing session")
 	}
 
-	store := h.userdataStore.(utils.UserDataStore)
+	store := h.userdataStore
 	publicKey, err := store.ReadEntry(ctx, sessionId, utils.DATA_PUBLIC_KEY)
 	if err != nil {
 		return res, err
@@ -862,7 +862,7 @@ func (h *Handlers) QuitWithBalance(ctx context.Context, sym string, input []byte
 	l := gotext.NewLocale(translationDir, code)
 	l.AddDomain("default")
 
-	store := h.userdataStore.(utils.UserDataStore)
+	store := h.userdataStore
 	publicKey, err := store.ReadEntry(ctx, sessionId, utils.DATA_PUBLIC_KEY)
 	if err != nil {
 		return res, err

internal/handlers/ussd/mocks/userdbmock.go

@@ -3,28 +3,16 @@ package mocks
 import (
 	"context"
 
-	"git.defalsify.org/vise.git/lang"
+	"git.defalsify.org/vise.git/db"
 	"git.grassecon.net/urdt/ussd/internal/utils"
 	"github.com/stretchr/testify/mock"
 )
 
 type MockUserDataStore struct {
+	db.Db
 	mock.Mock
 }
 
-func (m *MockUserDataStore) SetPrefix(prefix uint8) {
-	m.Called(prefix)
-}
-
-func (m *MockUserDataStore) SetSession(sessionId string) {
-	m.Called(sessionId)
-}
-
-func (m *MockUserDataStore) Get(ctx context.Context, key []byte) ([]byte, error) {
-	args := m.Called(ctx, key)
-	return args.Get(0).([]byte), args.Error(1)
-}
-
 func (m *MockUserDataStore) ReadEntry(ctx context.Context, sessionId string, typ utils.DataTyp) ([]byte, error) {
 	args := m.Called(ctx, sessionId, typ)
 	return args.Get(0).([]byte), args.Error(1)
@@ -34,36 +22,3 @@ func (m *MockUserDataStore) WriteEntry(ctx context.Context, sessionId string, ty
 	args := m.Called(ctx, sessionId, typ, value)
 	return args.Error(0)
 }
-
-func (m *MockUserDataStore) Prefix() uint8 {
-	args := m.Called()
-	return args.Get(0).(uint8)
-}
-
-func (m *MockUserDataStore) Safe() bool {
-	args := m.Called()
-	return args.Get(0).(bool)
-}
-
-func (m *MockUserDataStore) SetLanguage(language *lang.Language) {
-	m.Called(language)
-}
-
-func (m *MockUserDataStore) SetLock(uint8, bool) error {
-	args := m.Called()
-	return args.Error(0)
-}
-
-func (m *MockUserDataStore) Connect(ctx context.Context, connectionStr string) error {
-	args := m.Called(ctx, connectionStr)
-	return args.Error(0)
-}
-
-func (m *MockUserDataStore) Put(ctx context.Context, key, value []byte) error {
-	args := m.Called(ctx, key, value)
-	return args.Error(0)
-}
-func (m *MockUserDataStore) Close() error {
-	args := m.Called(nil)
-	return args.Error(0)
-}

internal/utils/account_utils.go

@@ -1,44 +0,0 @@
-package utils
-
-import (
-	"context"
-	"encoding/json"
-
-	"git.defalsify.org/vise.git/db"
-)
-
-type AccountFileHandler struct {
-	store db.Db
-}
-
-func NewAccountFileHandler(store db.Db) *AccountFileHandler {
-	return &AccountFileHandler{
-		store: store,
-	}
-}
-
-func (afh *AccountFileHandler) ReadAccountData(ctx context.Context, sessionId string) (map[string]string, error) {
-	var accountData map[string]string
-	jsonData, err := ReadEntry(ctx, afh.store, sessionId, DATA_ACCOUNT)
-	if err != nil {
-		return nil,err
-	}
-	err = json.Unmarshal(jsonData, &accountData)
-	if err != nil {
-		return nil, err
-	}
-	return accountData, nil
-}
-
-func (afh *AccountFileHandler) WriteAccountData(ctx context.Context, sessionId string, accountData map[string]string) error {
-	_, err := json.Marshal(accountData)
-	if err != nil {
-		return err
-	}
-
-	return nil
-}
-
-func (afh *AccountFileHandler) EnsureFileExists() error {
-	return nil
-}

internal/utils/userStore.go

@@ -4,78 +4,29 @@ import (
 	"context"
 
 	"git.defalsify.org/vise.git/db"
-	"git.defalsify.org/vise.git/lang"
 )
 
 type DataStore interface {
-	SetPrefix(prefix uint8)
-	SetSession(sessionId string)
-	Get(ctx context.Context, key []byte) ([]byte, error)
+	db.Db
 	ReadEntry(ctx context.Context, sessionId string, typ DataTyp) ([]byte, error)
 	WriteEntry(ctx context.Context, sessionId string, typ DataTyp, value []byte) error
-	Connect(ctx context.Context, connStr string) error
-	SetLanguage(*lang.Language)
-	Close() error
-	Prefix() uint8
-	Put(ctx context.Context, key []byte, val []byte) error
-	Safe() bool
-	SetLock(typ uint8, locked bool) error
 }
 
 type UserDataStore struct {
-	Store db.Db
-}
-
-func (store UserDataStore) SetPrefix(prefix uint8) {
-	store.Store.SetPrefix(prefix)
-}
-
-func (store UserDataStore) SetLanguage(lang *lang.Language) {
-	store.Store.SetLanguage(lang)
-}
-
-func (store UserDataStore) SetLock(typ uint8, locked bool) error {
-	return store.Store.SetLock(typ, locked)
-}
-func (store UserDataStore) Safe() bool {
-	return store.Store.Safe()
-}
-
-func (store UserDataStore) Put(ctx context.Context, key []byte, val []byte) error {
-	return store.Store.Put(ctx, key, val)
-}
-
-func (store UserDataStore) Connect(ctx context.Context, connectionStr string) error {
-	return store.Store.Connect(ctx, connectionStr)
-}
-
-func (store UserDataStore) Close() error {
-	return store.Store.Close()
-}
-
-func (store UserDataStore) Prefix() uint8 {
-	return store.Store.Prefix()
-}
-
-func (store UserDataStore) SetSession(sessionId string) {
-	store.Store.SetSession(sessionId)
-}
-
-func (store UserDataStore) Get(ctx context.Context, key []byte) ([]byte, error) {
-	return store.Store.Get(ctx, key)
+	db.Db
 }
 
 // ReadEntry retrieves an entry from the store based on the provided parameters.
-func (store UserDataStore) ReadEntry(ctx context.Context, sessionId string, typ DataTyp) ([]byte, error) {
-	store.Store.SetPrefix(db.DATATYPE_USERDATA)
-	store.Store.SetSession(sessionId)
+func (store *UserDataStore) ReadEntry(ctx context.Context, sessionId string, typ DataTyp) ([]byte, error) {
+	store.SetPrefix(db.DATATYPE_USERDATA)
+	store.SetSession(sessionId)
 	k := PackKey(typ, []byte(sessionId))
 	return store.Get(ctx, k)
 }
 
-func (store UserDataStore) WriteEntry(ctx context.Context, sessionId string, typ DataTyp, value []byte) error {
-	store.Store.SetPrefix(db.DATATYPE_USERDATA)
-	store.Store.SetSession(sessionId)
+func (store *UserDataStore) WriteEntry(ctx context.Context, sessionId string, typ DataTyp, value []byte) error {
+	store.SetPrefix(db.DATATYPE_USERDATA)
+	store.SetSession(sessionId)
 	k := PackKey(typ, []byte(sessionId))
-	return store.Store.Put(ctx, k, value)
+	return store.Put(ctx, k, value)
 }