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

load admin numbers defined in the .env

Browse Source
This commit is contained in:
Carlosokumu 2024-10-29 14:26:24 +03:00
parent 41da099933
commit d83962c0ba
Signed by: carlos
GPG Key ID: 7BD6BC8160A5C953
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
internal/utils/adminstore.go
View File

@ -2,8 +2,12 @@ package utils
import ( import (
"bufio" "bufio"
"log"
"os" "os"
"strconv" "strconv"
"strings"
"git.grassecon.net/urdt/ussd/initializers"
) )
type AdminStore struct { type AdminStore struct {
@ -15,9 +19,18 @@ func NewAdminStore(filePath string) *AdminStore {
return &AdminStore{filePath: filePath} return &AdminStore{filePath: filePath}
} }
// Seed saves a list of phonumbers with admin privileges // Seed initializes a list of phonenumbers with admin privileges
func (as *AdminStore) Seed() error { func (as *AdminStore) Seed() error {
adminNumbers := []int64{254705136690, 123456789012, 987654321098} var adminNumbers []int64
numbersEnv := initializers.GetEnv("ADMIN_NUMBERS", "")
for _, numStr := range strings.Split(numbersEnv, ",") {
if num, err := strconv.ParseInt(strings.TrimSpace(numStr), 10, 64); err == nil {
adminNumbers = append(adminNumbers, num)
} else {
log.Printf("Skipping invalid number: %s", numStr)
}
}
file, err := os.Create(as.filePath) file, err := os.Create(as.filePath)
if err != nil { if err != nil {
return err return err
@ -34,7 +47,6 @@ func (as *AdminStore) Seed() error {
return writer.Flush() return writer.Flush()
} }
func (as *AdminStore) load() ([]int64, error) { func (as *AdminStore) load() ([]int64, error) {
file, err := os.Open(as.filePath) file, err := os.Open(as.filePath)
if err != nil { if err != nil {