forked from urdt/ussd
81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"bufio"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"git.grassecon.net/urdt/ussd/initializers"
|
|
)
|
|
|
|
type AdminStore struct {
|
|
filePath string
|
|
}
|
|
|
|
// Creates a new Admin store
|
|
func NewAdminStore(filePath string) *AdminStore {
|
|
return &AdminStore{filePath: filePath}
|
|
}
|
|
|
|
// Seed initializes a list of phonenumbers with admin privileges
|
|
func (as *AdminStore) Seed() error {
|
|
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)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
writer := bufio.NewWriter(file)
|
|
for _, num := range adminNumbers {
|
|
_, err := writer.WriteString(strconv.FormatInt(num, 10) + "\n")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return writer.Flush()
|
|
}
|
|
|
|
func (as *AdminStore) load() ([]int64, error) {
|
|
file, err := os.Open(as.filePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer file.Close()
|
|
|
|
var numbers []int64
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
num, err := strconv.ParseInt(scanner.Text(), 10, 64)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
numbers = append(numbers, num)
|
|
}
|
|
return numbers, scanner.Err()
|
|
}
|
|
|
|
func (as *AdminStore) IsAdmin(phoneNumber int64) (bool, error) {
|
|
phoneNumbers, err := as.load()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
for _, phonenumber := range phoneNumbers {
|
|
if phonenumber == phoneNumber {
|
|
return true, nil
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|