urdt/ussd
5
0
Fork 0
Code Issues 51 Pull Requests 7 Actions Packages Projects Releases Wiki Activity

pin-reset #139

Merged
lash merged 61 commits from pin-reset into pre-mock-remove 2024-11-02 14:54:20 +01:00
Conversation 0 Commits 61 Files Changed 37 +3 -37
1 changed files with 3 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 7aab3cff8c - Show all commits

40
internal/utils/adminstore.go
View File

@ -2,8 +2,6 @@ package utils
import ( import (
"context" "context"
"encoding/json"
"os"
"git.defalsify.org/vise.git/db" "git.defalsify.org/vise.git/db"
fsdb "git.defalsify.org/vise.git/db/fs" fsdb "git.defalsify.org/vise.git/db/fs"
@ -14,17 +12,9 @@ var (
logg = logging.NewVanilla().WithDomain("adminstore") logg = logging.NewVanilla().WithDomain("adminstore")
) )
type Admin struct {
PhoneNumber string `json:"phonenumber"`
}
type Config struct {
Admins []Admin `json:"admins"`
}
type AdminStore struct { type AdminStore struct {
ctx context.Context ctx context.Context
fsStore db.Db FsStore db.Db
} }
func NewAdminStore(ctx context.Context, fileName string) (*AdminStore, error) { func NewAdminStore(ctx context.Context, fileName string) (*AdminStore, error) {
@ -32,7 +22,7 @@ func NewAdminStore(ctx context.Context, fileName string) (*AdminStore, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
return &AdminStore{ctx: ctx, fsStore: fsStore}, nil return &AdminStore{ctx: ctx, FsStore: fsStore}, nil
} }
func getFsStore(ctx context.Context, connectStr string) (db.Db, error) { func getFsStore(ctx context.Context, connectStr string) (db.Db, error) {
@ -45,33 +35,9 @@ func getFsStore(ctx context.Context, connectStr string) (db.Db, error) {
return fsStore, nil return fsStore, nil
} }
// Seed initializes a list of phonenumbers with admin privileges
func (as *AdminStore) Seed() error {
var config Config
store := as.fsStore
defer store.Close()
data, err := os.ReadFile("admin_numbers.json")
if err != nil {
return err
}
if err := json.Unmarshal(data, &config); err != nil {
return err
}
for _, admin := range config.Admins {
err := store.Put(as.ctx, []byte(admin.PhoneNumber), []byte("1"))
if err != nil {
logg.Printf(logging.LVL_DEBUG, "Failed to insert admin number", admin.PhoneNumber)
return err
}
}
return nil
}
// Checks if the given sessionId is listed as an admin. // Checks if the given sessionId is listed as an admin.
func (as *AdminStore) IsAdmin(sessionId string) (bool, error) { func (as *AdminStore) IsAdmin(sessionId string) (bool, error) {
_, err := as.fsStore.Get(as.ctx, []byte(sessionId)) _, err := as.FsStore.Get(as.ctx, []byte(sessionId))
if err != nil { if err != nil {
if db.IsNotFound(err) { if db.IsNotFound(err) {
logg.Printf(logging.LVL_INFO, "Returning false because session id was not found") logg.Printf(logging.LVL_INFO, "Returning false because session id was not found")