Compare commits
	
		
			No commits in common. "master" and "lash/reverse-session" have entirely different histories.
		
	
	
		
			master
			...
			lash/rever
		
	
		
							
								
								
									
										24
									
								
								.env.example
									
									
									
									
									
								
							
							
						
						
									
										24
									
								
								.env.example
									
									
									
									
									
								
							@ -2,19 +2,17 @@
 | 
			
		||||
PORT=7123
 | 
			
		||||
HOST=127.0.0.1
 | 
			
		||||
 | 
			
		||||
#AfricasTalking USSD POST endpoint
 | 
			
		||||
AT_ENDPOINT=/ussd/africastalking
 | 
			
		||||
 | 
			
		||||
#PostgreSQL
 | 
			
		||||
DB_CONN=postgres://postgres:strongpass@localhost:5432/urdt_ussd
 | 
			
		||||
#DB_TIMEZONE=Africa/Nairobi
 | 
			
		||||
#DB_SCHEMA=vise
 | 
			
		||||
DB_HOST=localhost
 | 
			
		||||
DB_USER=postgres
 | 
			
		||||
DB_PASSWORD=strongpass
 | 
			
		||||
DB_NAME=urdt_ussd
 | 
			
		||||
DB_PORT=5432
 | 
			
		||||
DB_SSLMODE=disable
 | 
			
		||||
DB_TIMEZONE=Africa/Nairobi
 | 
			
		||||
 | 
			
		||||
#External API Calls
 | 
			
		||||
CUSTODIAL_URL_BASE=http://localhost:5003
 | 
			
		||||
BEARER_TOKEN=eyJeSIsInRcCI6IkpXVCJ.yJwdWJsaWNLZXkiOiIwrrrrrr
 | 
			
		||||
DATA_URL_BASE=http://localhost:5006
 | 
			
		||||
 | 
			
		||||
#Language
 | 
			
		||||
DEFAULT_LANGUAGE=eng
 | 
			
		||||
LANGUAGES=eng, swa
 | 
			
		||||
CREATE_ACCOUNT_URL=http://localhost:5003/api/v2/account/create
 | 
			
		||||
TRACK_STATUS_URL=https://custodial.sarafu.africa/api/track/
 | 
			
		||||
BALANCE_URL=https://custodial.sarafu.africa/api/account/status/
 | 
			
		||||
TRACK_URL=http://localhost:5003/api/v2/account/status
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							@ -6,4 +6,3 @@ go.work*
 | 
			
		||||
cmd/.state/
 | 
			
		||||
id_*
 | 
			
		||||
*.gdbm
 | 
			
		||||
*.log
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										9
									
								
								README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								README.md
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,9 @@
 | 
			
		||||
# ussd
 | 
			
		||||
 | 
			
		||||
> USSD
 | 
			
		||||
 | 
			
		||||
USSD service.
 | 
			
		||||
 | 
			
		||||
## License
 | 
			
		||||
 | 
			
		||||
[AGPL-3.0](LICENSE).
 | 
			
		||||
							
								
								
									
										180
									
								
								cmd/africastalking/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										180
									
								
								cmd/africastalking/main.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,180 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"os"
 | 
			
		||||
	"os/signal"
 | 
			
		||||
	"path"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"syscall"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/config"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/initializers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
	httpserver "git.grassecon.net/urdt/ussd/internal/http"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/storage"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg      = logging.NewVanilla()
 | 
			
		||||
	scriptDir = path.Join("services", "registration")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	initializers.LoadEnvVariables()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type atRequestParser struct{}
 | 
			
		||||
 | 
			
		||||
func (arp *atRequestParser) GetSessionId(rq any) (string, error) {
 | 
			
		||||
	rqv, ok := rq.(*http.Request)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return "", handlers.ErrInvalidRequest
 | 
			
		||||
	}
 | 
			
		||||
	if err := rqv.ParseForm(); err != nil {
 | 
			
		||||
		return "", fmt.Errorf("failed to parse form data: %v", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	phoneNumber := rqv.FormValue("phoneNumber")
 | 
			
		||||
	if phoneNumber == "" {
 | 
			
		||||
		return "", fmt.Errorf("no phone number found")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return phoneNumber, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (arp *atRequestParser) GetInput(rq any) ([]byte, error) {
 | 
			
		||||
	rqv, ok := rq.(*http.Request)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, handlers.ErrInvalidRequest
 | 
			
		||||
	}
 | 
			
		||||
	if err := rqv.ParseForm(); err != nil {
 | 
			
		||||
		return nil, fmt.Errorf("failed to parse form data: %v", err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	text := rqv.FormValue("text")
 | 
			
		||||
 | 
			
		||||
	parts := strings.Split(text, "*")
 | 
			
		||||
	if len(parts) == 0 {
 | 
			
		||||
		return nil, fmt.Errorf("no input found")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return []byte(parts[len(parts)-1]), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	config.LoadConfig()
 | 
			
		||||
 | 
			
		||||
	var dbDir string
 | 
			
		||||
	var resourceDir string
 | 
			
		||||
	var size uint
 | 
			
		||||
	var database string
 | 
			
		||||
	var engineDebug bool
 | 
			
		||||
	var host string
 | 
			
		||||
	var port uint
 | 
			
		||||
	flag.StringVar(&dbDir, "dbdir", ".state", "database dir to read from")
 | 
			
		||||
	flag.StringVar(&resourceDir, "resourcedir", path.Join("services", "registration"), "resource dir")
 | 
			
		||||
	flag.StringVar(&database, "db", "gdbm", "database to be used")
 | 
			
		||||
	flag.BoolVar(&engineDebug, "d", false, "use engine debug output")
 | 
			
		||||
	flag.UintVar(&size, "s", 160, "max size of output")
 | 
			
		||||
	flag.StringVar(&host, "h", initializers.GetEnv("HOST", "127.0.0.1"), "http host")
 | 
			
		||||
	flag.UintVar(&port, "p", initializers.GetEnvUint("PORT", 7123), "http port")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
 | 
			
		||||
	logg.Infof("start command", "dbdir", dbDir, "resourcedir", resourceDir, "outputsize", size)
 | 
			
		||||
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	ctx = context.WithValue(ctx, "Database", database)
 | 
			
		||||
	pfp := path.Join(scriptDir, "pp.csv")
 | 
			
		||||
 | 
			
		||||
	cfg := engine.Config{
 | 
			
		||||
		Root:       "root",
 | 
			
		||||
		OutputSize: uint32(size),
 | 
			
		||||
		FlagCount:  uint32(128),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if engineDebug {
 | 
			
		||||
		cfg.EngineDebug = true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	menuStorageService := storage.NewMenuStorageService(dbDir, resourceDir)
 | 
			
		||||
	rs, err := menuStorageService.GetResource(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = menuStorageService.EnsureDbDir()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	userdataStore, err := menuStorageService.GetUserdataDb(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
	defer userdataStore.Close()
 | 
			
		||||
 | 
			
		||||
	dbResource, ok := rs.(*resource.DbResource)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lhs, err := handlers.NewLocalHandlerService(pfp, true, dbResource, cfg, rs)
 | 
			
		||||
	lhs.SetDataStore(&userdataStore)
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	accountService := server.AccountService{}
 | 
			
		||||
	hl, err := lhs.GetHandler(&accountService)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	stateStore, err := menuStorageService.GetStateStore(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
	defer stateStore.Close()
 | 
			
		||||
 | 
			
		||||
	rp := &atRequestParser{}
 | 
			
		||||
	bsh := handlers.NewBaseSessionHandler(cfg, rs, stateStore, userdataStore, rp, hl)
 | 
			
		||||
	sh := httpserver.NewATSessionHandler(bsh)
 | 
			
		||||
	s := &http.Server{
 | 
			
		||||
		Addr:    fmt.Sprintf("%s:%s", host, strconv.Itoa(int(port))),
 | 
			
		||||
		Handler: sh,
 | 
			
		||||
	}
 | 
			
		||||
	s.RegisterOnShutdown(sh.Shutdown)
 | 
			
		||||
 | 
			
		||||
	cint := make(chan os.Signal)
 | 
			
		||||
	cterm := make(chan os.Signal)
 | 
			
		||||
	signal.Notify(cint, os.Interrupt, syscall.SIGINT)
 | 
			
		||||
	signal.Notify(cterm, os.Interrupt, syscall.SIGTERM)
 | 
			
		||||
	go func() {
 | 
			
		||||
		select {
 | 
			
		||||
		case _ = <-cint:
 | 
			
		||||
		case _ = <-cterm:
 | 
			
		||||
		}
 | 
			
		||||
		s.Shutdown(ctx)
 | 
			
		||||
	}()
 | 
			
		||||
	err = s.ListenAndServe()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.Infof("Server closed with error", "err", err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										174
									
								
								cmd/async/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										174
									
								
								cmd/async/main.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,174 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"os/signal"
 | 
			
		||||
	"path"
 | 
			
		||||
	"syscall"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/config"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/initializers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/storage"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg      = logging.NewVanilla()
 | 
			
		||||
	scriptDir = path.Join("services", "registration")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	initializers.LoadEnvVariables()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type asyncRequestParser struct {
 | 
			
		||||
	sessionId string
 | 
			
		||||
	input     []byte
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *asyncRequestParser) GetSessionId(r any) (string, error) {
 | 
			
		||||
	return p.sessionId, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *asyncRequestParser) GetInput(r any) ([]byte, error) {
 | 
			
		||||
	return p.input, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	config.LoadConfig()
 | 
			
		||||
 | 
			
		||||
	var sessionId string
 | 
			
		||||
	var dbDir string
 | 
			
		||||
	var resourceDir string
 | 
			
		||||
	var size uint
 | 
			
		||||
	var database string
 | 
			
		||||
	var engineDebug bool
 | 
			
		||||
	var host string
 | 
			
		||||
	var port uint
 | 
			
		||||
	flag.StringVar(&sessionId, "session-id", "075xx2123", "session id")
 | 
			
		||||
	flag.StringVar(&dbDir, "dbdir", ".state", "database dir to read from")
 | 
			
		||||
	flag.StringVar(&resourceDir, "resourcedir", path.Join("services", "registration"), "resource dir")
 | 
			
		||||
	flag.StringVar(&database, "db", "gdbm", "database to be used")
 | 
			
		||||
	flag.BoolVar(&engineDebug, "d", false, "use engine debug output")
 | 
			
		||||
	flag.UintVar(&size, "s", 160, "max size of output")
 | 
			
		||||
	flag.StringVar(&host, "h", initializers.GetEnv("HOST", "127.0.0.1"), "http host")
 | 
			
		||||
	flag.UintVar(&port, "p", initializers.GetEnvUint("PORT", 7123), "http port")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
 | 
			
		||||
	logg.Infof("start command", "dbdir", dbDir, "resourcedir", resourceDir, "outputsize", size, "sessionId", sessionId)
 | 
			
		||||
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	ctx = context.WithValue(ctx, "Database", database)
 | 
			
		||||
	pfp := path.Join(scriptDir, "pp.csv")
 | 
			
		||||
 | 
			
		||||
	cfg := engine.Config{
 | 
			
		||||
		Root:       "root",
 | 
			
		||||
		OutputSize: uint32(size),
 | 
			
		||||
		FlagCount:  uint32(128),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if engineDebug {
 | 
			
		||||
		cfg.EngineDebug = true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	menuStorageService := storage.NewMenuStorageService(dbDir, resourceDir)
 | 
			
		||||
	rs, err := menuStorageService.GetResource(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = menuStorageService.EnsureDbDir()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	userdataStore, err := menuStorageService.GetUserdataDb(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
	defer userdataStore.Close()
 | 
			
		||||
 | 
			
		||||
	dbResource, ok := rs.(*resource.DbResource)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lhs, err := handlers.NewLocalHandlerService(pfp, true, dbResource, cfg, rs)
 | 
			
		||||
	lhs.SetDataStore(&userdataStore)
 | 
			
		||||
	accountService := server.AccountService{}
 | 
			
		||||
 | 
			
		||||
	hl, err := lhs.GetHandler(&accountService)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	stateStore, err := menuStorageService.GetStateStore(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
	defer stateStore.Close()
 | 
			
		||||
 | 
			
		||||
	rp := &asyncRequestParser{
 | 
			
		||||
		sessionId: sessionId,
 | 
			
		||||
	}
 | 
			
		||||
	sh := handlers.NewBaseSessionHandler(cfg, rs, stateStore, userdataStore, rp, hl)
 | 
			
		||||
	cfg.SessionId = sessionId
 | 
			
		||||
	rqs := handlers.RequestSession{
 | 
			
		||||
		Ctx:    ctx,
 | 
			
		||||
		Writer: os.Stdout,
 | 
			
		||||
		Config: cfg,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	cint := make(chan os.Signal)
 | 
			
		||||
	cterm := make(chan os.Signal)
 | 
			
		||||
	signal.Notify(cint, os.Interrupt, syscall.SIGINT)
 | 
			
		||||
	signal.Notify(cterm, os.Interrupt, syscall.SIGTERM)
 | 
			
		||||
	go func() {
 | 
			
		||||
		select {
 | 
			
		||||
		case _ = <-cint:
 | 
			
		||||
		case _ = <-cterm:
 | 
			
		||||
		}
 | 
			
		||||
		sh.Shutdown()
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	for true {
 | 
			
		||||
		rqs, err = sh.Process(rqs)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logg.ErrorCtxf(ctx, "error in process: %v", "err", err)
 | 
			
		||||
			fmt.Errorf("error in process: %v", err)
 | 
			
		||||
			os.Exit(1)
 | 
			
		||||
		}
 | 
			
		||||
		rqs, err = sh.Output(rqs)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logg.ErrorCtxf(ctx, "error in output: %v", "err", err)
 | 
			
		||||
			fmt.Errorf("error in output: %v", err)
 | 
			
		||||
			os.Exit(1)
 | 
			
		||||
		}
 | 
			
		||||
		rqs, err = sh.Reset(rqs)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logg.ErrorCtxf(ctx, "error in reset: %v", "err", err)
 | 
			
		||||
			fmt.Errorf("error in reset: %v", err)
 | 
			
		||||
			os.Exit(1)
 | 
			
		||||
		}
 | 
			
		||||
		fmt.Println("")
 | 
			
		||||
		_, err = fmt.Scanln(&rqs.Input)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logg.ErrorCtxf(ctx, "error in input", "err", err)
 | 
			
		||||
			fmt.Errorf("error in input: %v", err)
 | 
			
		||||
			os.Exit(1)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										140
									
								
								cmd/http/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										140
									
								
								cmd/http/main.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,140 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"os"
 | 
			
		||||
	"os/signal"
 | 
			
		||||
	"path"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"syscall"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/config"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/initializers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
	httpserver "git.grassecon.net/urdt/ussd/internal/http"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/storage"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg      = logging.NewVanilla()
 | 
			
		||||
	scriptDir = path.Join("services", "registration")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	initializers.LoadEnvVariables()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	config.LoadConfig()
 | 
			
		||||
 | 
			
		||||
	var dbDir string
 | 
			
		||||
	var resourceDir string
 | 
			
		||||
	var size uint
 | 
			
		||||
	var database string
 | 
			
		||||
	var engineDebug bool
 | 
			
		||||
	var host string
 | 
			
		||||
	var port uint
 | 
			
		||||
	flag.StringVar(&dbDir, "dbdir", ".state", "database dir to read from")
 | 
			
		||||
	flag.StringVar(&resourceDir, "resourcedir", path.Join("services", "registration"), "resource dir")
 | 
			
		||||
	flag.StringVar(&database, "db", "gdbm", "database to be used")
 | 
			
		||||
	flag.BoolVar(&engineDebug, "d", false, "use engine debug output")
 | 
			
		||||
	flag.UintVar(&size, "s", 160, "max size of output")
 | 
			
		||||
	flag.StringVar(&host, "h", initializers.GetEnv("HOST", "127.0.0.1"), "http host")
 | 
			
		||||
	flag.UintVar(&port, "p", initializers.GetEnvUint("PORT", 7123), "http port")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
 | 
			
		||||
	logg.Infof("start command", "dbdir", dbDir, "resourcedir", resourceDir, "outputsize", size)
 | 
			
		||||
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	ctx = context.WithValue(ctx, "Database", database)
 | 
			
		||||
	pfp := path.Join(scriptDir, "pp.csv")
 | 
			
		||||
 | 
			
		||||
	cfg := engine.Config{
 | 
			
		||||
		Root:       "root",
 | 
			
		||||
		OutputSize: uint32(size),
 | 
			
		||||
		FlagCount:  uint32(128),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if engineDebug {
 | 
			
		||||
		cfg.EngineDebug = true
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	menuStorageService := storage.NewMenuStorageService(dbDir, resourceDir)
 | 
			
		||||
	rs, err := menuStorageService.GetResource(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = menuStorageService.EnsureDbDir()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	userdataStore, err := menuStorageService.GetUserdataDb(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
	defer userdataStore.Close()
 | 
			
		||||
 | 
			
		||||
	dbResource, ok := rs.(*resource.DbResource)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lhs, err := handlers.NewLocalHandlerService(pfp, true, dbResource, cfg, rs)
 | 
			
		||||
	lhs.SetDataStore(&userdataStore)
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
	accountService := server.AccountService{}
 | 
			
		||||
	hl, err := lhs.GetHandler(&accountService)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	stateStore, err := menuStorageService.GetStateStore(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
	defer stateStore.Close()
 | 
			
		||||
 | 
			
		||||
	rp := &httpserver.DefaultRequestParser{}
 | 
			
		||||
	bsh := handlers.NewBaseSessionHandler(cfg, rs, stateStore, userdataStore, rp, hl)
 | 
			
		||||
	sh := httpserver.ToSessionHandler(bsh)
 | 
			
		||||
	s := &http.Server{
 | 
			
		||||
		Addr:    fmt.Sprintf("%s:%s", host, strconv.Itoa(int(port))),
 | 
			
		||||
		Handler: sh,
 | 
			
		||||
	}
 | 
			
		||||
	s.RegisterOnShutdown(sh.Shutdown)
 | 
			
		||||
 | 
			
		||||
	cint := make(chan os.Signal)
 | 
			
		||||
	cterm := make(chan os.Signal)
 | 
			
		||||
	signal.Notify(cint, os.Interrupt, syscall.SIGINT)
 | 
			
		||||
	signal.Notify(cterm, os.Interrupt, syscall.SIGTERM)
 | 
			
		||||
	go func() {
 | 
			
		||||
		select {
 | 
			
		||||
		case _ = <-cint:
 | 
			
		||||
		case _ = <-cterm:
 | 
			
		||||
		}
 | 
			
		||||
		s.Shutdown(ctx)
 | 
			
		||||
	}()
 | 
			
		||||
	err = s.ListenAndServe()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.Infof("Server closed with error", "err", err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										118
									
								
								cmd/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										118
									
								
								cmd/main.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,118 @@
 | 
			
		||||
package main
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"flag"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/config"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/initializers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/storage"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg      = logging.NewVanilla()
 | 
			
		||||
	scriptDir = path.Join("services", "registration")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	initializers.LoadEnvVariables()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	config.LoadConfig()
 | 
			
		||||
 | 
			
		||||
	var dbDir string
 | 
			
		||||
	var size uint
 | 
			
		||||
	var sessionId string
 | 
			
		||||
	var database string
 | 
			
		||||
	var engineDebug bool
 | 
			
		||||
	flag.StringVar(&sessionId, "session-id", "075xx2123", "session id")
 | 
			
		||||
	flag.StringVar(&database, "db", "gdbm", "database to be used")
 | 
			
		||||
	flag.StringVar(&dbDir, "dbdir", ".state", "database dir to read from")
 | 
			
		||||
	flag.BoolVar(&engineDebug, "d", false, "use engine debug output")
 | 
			
		||||
	flag.UintVar(&size, "s", 160, "max size of output")
 | 
			
		||||
	flag.Parse()
 | 
			
		||||
 | 
			
		||||
	logg.Infof("start command", "dbdir", dbDir, "outputsize", size)
 | 
			
		||||
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	ctx = context.WithValue(ctx, "SessionId", sessionId)
 | 
			
		||||
	ctx = context.WithValue(ctx, "Database", database)
 | 
			
		||||
	pfp := path.Join(scriptDir, "pp.csv")
 | 
			
		||||
 | 
			
		||||
	cfg := engine.Config{
 | 
			
		||||
		Root:       "root",
 | 
			
		||||
		SessionId:  sessionId,
 | 
			
		||||
		OutputSize: uint32(size),
 | 
			
		||||
		FlagCount:  uint32(128),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	resourceDir := scriptDir
 | 
			
		||||
	menuStorageService := storage.NewMenuStorageService(dbDir, resourceDir)
 | 
			
		||||
 | 
			
		||||
	err := menuStorageService.EnsureDbDir()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rs, err := menuStorageService.GetResource(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	pe, err := menuStorageService.GetPersister(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	userdatastore, err := menuStorageService.GetUserdataDb(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	dbResource, ok := rs.(*resource.DbResource)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lhs, err := handlers.NewLocalHandlerService(pfp, true, dbResource, cfg, rs)
 | 
			
		||||
	lhs.SetDataStore(&userdatastore)
 | 
			
		||||
	lhs.SetPersister(pe)
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	accountService := server.AccountService{}
 | 
			
		||||
	hl, err := lhs.GetHandler(&accountService)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	en := lhs.GetEngine()
 | 
			
		||||
	en = en.WithFirst(hl.Init)
 | 
			
		||||
	if engineDebug {
 | 
			
		||||
		en = en.WithDebug(nil)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = engine.Loop(ctx, en, os.Stdin, os.Stdout, nil)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, "loop exited with error: %v\n", err)
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										18
									
								
								common/hex.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								common/hex.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,18 @@
 | 
			
		||||
package common
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/hex"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func NormalizeHex(s string) (string, error) {
 | 
			
		||||
	if len(s) >= 2 {
 | 
			
		||||
		if s[:2] == "0x" {
 | 
			
		||||
			s = s[2:]
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	r, err := hex.DecodeString(s)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return hex.EncodeToString(r), nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										166
									
								
								config/config.go
									
									
									
									
									
								
							
							
						
						
									
										166
									
								
								config/config.go
									
									
									
									
									
								
							@ -1,164 +1,18 @@
 | 
			
		||||
package config
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/env"
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/storage"
 | 
			
		||||
	slogging "github.com/grassrootseconomics/go-vise/slog"
 | 
			
		||||
)
 | 
			
		||||
import "git.grassecon.net/urdt/ussd/initializers"
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg               = slogging.Get().With("component", "visedriver-config")
 | 
			
		||||
	defaultLanguage    = "eng"
 | 
			
		||||
	languages          []string
 | 
			
		||||
	DefaultLanguage    string
 | 
			
		||||
	dbConn             string
 | 
			
		||||
	dbConnMissing      bool
 | 
			
		||||
	dbConnMode         storage.DbMode
 | 
			
		||||
	stateDbConn        string
 | 
			
		||||
	stateDbConnMode    storage.DbMode
 | 
			
		||||
	resourceDbConn     string
 | 
			
		||||
	resourceDbConnMode storage.DbMode
 | 
			
		||||
	userDbConn         string
 | 
			
		||||
	userDbConnMode     storage.DbMode
 | 
			
		||||
	Languages          []string
 | 
			
		||||
	configManager      *Config
 | 
			
		||||
	CreateAccountURL string
 | 
			
		||||
	TrackStatusURL   string
 | 
			
		||||
	BalanceURL       string
 | 
			
		||||
    TrackURL         string
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Override struct {
 | 
			
		||||
	DbConn           string
 | 
			
		||||
	DbConnMode       storage.DbMode
 | 
			
		||||
	StateConn        string
 | 
			
		||||
	StateConnMode    storage.DbMode
 | 
			
		||||
	ResourceConn     string
 | 
			
		||||
	ResourceConnMode storage.DbMode
 | 
			
		||||
	UserConn         string
 | 
			
		||||
	UserConnMode     storage.DbMode
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func setLanguage() error {
 | 
			
		||||
	defaultLanguage = env.GetEnv("DEFAULT_LANGUAGE", defaultLanguage)
 | 
			
		||||
	languages = strings.Split(env.GetEnv("LANGUAGES", defaultLanguage), ",")
 | 
			
		||||
	haveDefaultLanguage := false
 | 
			
		||||
	for i, v := range languages {
 | 
			
		||||
		languages[i] = strings.ReplaceAll(v, " ", "")
 | 
			
		||||
		if languages[i] == defaultLanguage {
 | 
			
		||||
			haveDefaultLanguage = true
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if !haveDefaultLanguage {
 | 
			
		||||
		languages = append([]string{defaultLanguage}, languages...)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func setConn() error {
 | 
			
		||||
	dbConn = env.GetEnv("DB_CONN", "?")
 | 
			
		||||
	stateDbConn = env.GetEnv("DB_CONN_STATE", dbConn)
 | 
			
		||||
	resourceDbConn = env.GetEnv("DB_CONN_RESOURCE", dbConn)
 | 
			
		||||
	userDbConn = env.GetEnv("DB_CONN_USER", dbConn)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func ApplyConn(override *Override) {
 | 
			
		||||
	if override.DbConn != "?" {
 | 
			
		||||
		dbConn = override.DbConn
 | 
			
		||||
		stateDbConn = override.StateConn
 | 
			
		||||
		resourceDbConn = override.ResourceConn
 | 
			
		||||
		userDbConn = override.UserConn
 | 
			
		||||
	}
 | 
			
		||||
	dbConnMode = override.DbConnMode
 | 
			
		||||
	if override.StateConn != "?" {
 | 
			
		||||
		stateDbConn = override.StateConn
 | 
			
		||||
	}
 | 
			
		||||
	if override.ResourceConn != "?" {
 | 
			
		||||
		resourceDbConn = override.ResourceConn
 | 
			
		||||
	}
 | 
			
		||||
	if override.UserConn != "?" {
 | 
			
		||||
		userDbConn = override.UserConn
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if dbConn == "?" {
 | 
			
		||||
		dbConn = ""
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if stateDbConn == "?" {
 | 
			
		||||
		stateDbConn = dbConn
 | 
			
		||||
		stateDbConnMode = dbConnMode
 | 
			
		||||
	}
 | 
			
		||||
	if resourceDbConn == "?" {
 | 
			
		||||
		resourceDbConn = dbConn
 | 
			
		||||
		resourceDbConnMode = dbConnMode
 | 
			
		||||
	}
 | 
			
		||||
	if userDbConn == "?" {
 | 
			
		||||
		userDbConn = dbConn
 | 
			
		||||
		userDbConnMode = dbConnMode
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	logg.Debugf("conns", "conn", dbConn, "user", userDbConn)
 | 
			
		||||
	if override.DbConnMode != storage.DBMODE_ANY {
 | 
			
		||||
		dbConnMode = override.DbConnMode
 | 
			
		||||
	}
 | 
			
		||||
	if override.StateConnMode != storage.DBMODE_ANY {
 | 
			
		||||
		stateDbConnMode = override.StateConnMode
 | 
			
		||||
	}
 | 
			
		||||
	if override.ResourceConnMode != storage.DBMODE_ANY {
 | 
			
		||||
		resourceDbConnMode = override.ResourceConnMode
 | 
			
		||||
	}
 | 
			
		||||
	if override.UserConnMode != storage.DBMODE_ANY {
 | 
			
		||||
		userDbConnMode = override.UserConnMode
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetConns() (storage.Conns, error) {
 | 
			
		||||
	o := storage.NewConns()
 | 
			
		||||
	c, err := storage.ToConnDataMode(stateDbConn, stateDbConnMode)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return o, err
 | 
			
		||||
	}
 | 
			
		||||
	o.Set(c, storage.STORETYPE_STATE)
 | 
			
		||||
	c, err = storage.ToConnDataMode(resourceDbConn, resourceDbConnMode)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return o, err
 | 
			
		||||
	}
 | 
			
		||||
	o.Set(c, storage.STORETYPE_RESOURCE)
 | 
			
		||||
	c, err = storage.ToConnDataMode(userDbConn, userDbConnMode)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return o, err
 | 
			
		||||
	}
 | 
			
		||||
	o.Set(c, storage.STORETYPE_USER)
 | 
			
		||||
	return o, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// LoadConfig initializes the configuration values after environment variables are loaded.
 | 
			
		||||
func LoadConfig() error {
 | 
			
		||||
	configManager = NewConfig()
 | 
			
		||||
 | 
			
		||||
	// Add configuration keys with validation
 | 
			
		||||
	configManager.AddKey("HOST", "127.0.0.1", false, nil)
 | 
			
		||||
	configManager.AddKey("PORT", "7123", false, func(v string) error {
 | 
			
		||||
		_, err := strconv.Atoi(v)
 | 
			
		||||
		return err
 | 
			
		||||
	})
 | 
			
		||||
	configManager.AddKey("DB_CONN", "", true, nil)
 | 
			
		||||
	// ... add other keys ?  or is enough :/ ...
 | 
			
		||||
 | 
			
		||||
	err := setConn()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	err = setLanguage()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	DefaultLanguage = defaultLanguage
 | 
			
		||||
	Languages = languages
 | 
			
		||||
 | 
			
		||||
	// Report configuration
 | 
			
		||||
	configManager.Report("INFO")
 | 
			
		||||
	return nil
 | 
			
		||||
func LoadConfig() {
 | 
			
		||||
	CreateAccountURL = initializers.GetEnv("CREATE_ACCOUNT_URL", "http://localhost:5003/api/v2/account/create")
 | 
			
		||||
	TrackStatusURL = initializers.GetEnv("TRACK_STATUS_URL", "https://custodial.sarafu.africa/api/track/")
 | 
			
		||||
	BalanceURL = initializers.GetEnv("BALANCE_URL", "https://custodial.sarafu.africa/api/account/status/")
 | 
			
		||||
    TrackURL = initializers.GetEnv("TRACK_URL", "http://localhost:5003/api/v2/account/status")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@ -1,64 +0,0 @@
 | 
			
		||||
//go:build configreport
 | 
			
		||||
// +build configreport
 | 
			
		||||
 | 
			
		||||
package config
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"testing"
 | 
			
		||||
 | 
			
		||||
	slogging "github.com/grassrootseconomics/go-vise/slog"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// go test -tags configreport ./config/...   ---> run with tag
 | 
			
		||||
func TestConfig(t *testing.T) {
 | 
			
		||||
	logger := slogging.Get().With("component", "test")
 | 
			
		||||
	cfg := NewConfig(logger)
 | 
			
		||||
 | 
			
		||||
	t.Run("Default Values", func(t *testing.T) {
 | 
			
		||||
		cfg.AddKey("TEST_KEY", "default", false, nil)
 | 
			
		||||
		value, err := cfg.GetValue("TEST_KEY")
 | 
			
		||||
		t.Logf("Got value: %q, error: %v", value, err)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			t.Errorf("unexpected error: %v", err)
 | 
			
		||||
		}
 | 
			
		||||
		if value != "default" {
 | 
			
		||||
			t.Errorf("expected 'default', got '%s'", value)
 | 
			
		||||
		}
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	t.Run("Environment Override", func(t *testing.T) {
 | 
			
		||||
		os.Setenv("TEST_ENV_KEY", "override")
 | 
			
		||||
		defer os.Unsetenv("TEST_ENV_KEY")
 | 
			
		||||
 | 
			
		||||
		cfg.AddKey("TEST_ENV_KEY", "default", false, nil)
 | 
			
		||||
		value, err := cfg.GetValue("TEST_ENV_KEY")
 | 
			
		||||
		t.Logf("Got value: %q, error: %v", value, err)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			t.Errorf("unexpected error: %v", err)
 | 
			
		||||
		}
 | 
			
		||||
		if value != "override" {
 | 
			
		||||
			t.Errorf("expected 'override', got '%s'", value)
 | 
			
		||||
		}
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	t.Run("Validation", func(t *testing.T) {
 | 
			
		||||
		validator := func(v string) error {
 | 
			
		||||
			if v != "valid" {
 | 
			
		||||
				return fmt.Errorf("invalid value")
 | 
			
		||||
			}
 | 
			
		||||
			return nil
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		cfg.AddKey("VALIDATED_KEY", "valid", false, validator)
 | 
			
		||||
		os.Setenv("VALIDATED_KEY", "invalid")
 | 
			
		||||
		defer os.Unsetenv("VALIDATED_KEY")
 | 
			
		||||
 | 
			
		||||
		value, err := cfg.GetValue("VALIDATED_KEY")
 | 
			
		||||
		t.Logf("Got value: %q, error: %v", value, err)
 | 
			
		||||
		if err == nil {
 | 
			
		||||
			t.Error("expected validation error, got nil")
 | 
			
		||||
		}
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
@ -1,84 +0,0 @@
 | 
			
		||||
//go:build configreport
 | 
			
		||||
 | 
			
		||||
package config
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/env"
 | 
			
		||||
	slogging "github.com/grassrootseconomics/go-vise/slog"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// ConfigValue represents a configuration key-value pair
 | 
			
		||||
type ConfigValue struct {
 | 
			
		||||
	Key       string
 | 
			
		||||
	Default   string
 | 
			
		||||
	Validator func(string) error
 | 
			
		||||
	Sensitive bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Config handles configuration management and reporting
 | 
			
		||||
type Config struct {
 | 
			
		||||
	values map[string]ConfigValue
 | 
			
		||||
	logger *slogging.Slog
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewConfig(logging interface{}) *Config {
 | 
			
		||||
	return &Config{
 | 
			
		||||
		values: make(map[string]ConfigValue),
 | 
			
		||||
		logger: slogging.Get().With("component", "visedriver-config-reporter"),
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// AddKey registers a new configuration key with optional validation
 | 
			
		||||
func (c *Config) AddKey(key string, defaultValue string, sensitive bool, validator func(string) error) {
 | 
			
		||||
	c.values[key] = ConfigValue{
 | 
			
		||||
		Key:       key,
 | 
			
		||||
		Default:   defaultValue,
 | 
			
		||||
		Validator: validator,
 | 
			
		||||
		Sensitive: sensitive,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetValue returns the value for a given key, applying environment override if present
 | 
			
		||||
func (c *Config) GetValue(key string) (string, error) {
 | 
			
		||||
	// Find config value by key
 | 
			
		||||
	var cv ConfigValue
 | 
			
		||||
	for _, v := range c.values {
 | 
			
		||||
		if v.Key == key {
 | 
			
		||||
			cv = v
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if cv.Key == "" {
 | 
			
		||||
		return "", fmt.Errorf("configuration key not found: %s", key)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Get value from environment or default
 | 
			
		||||
	value := env.GetEnv(key, cv.Default)
 | 
			
		||||
 | 
			
		||||
	// Validate if validator exists
 | 
			
		||||
	if cv.Validator != nil && cv.Validator(value) != nil {
 | 
			
		||||
		return "", fmt.Errorf("invalid value for key %s", key)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return value, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Report outputs all configuration values at the specified log level
 | 
			
		||||
func (c *Config) Report(level string) {
 | 
			
		||||
	for _, cv := range c.values {
 | 
			
		||||
		value, err := c.GetValue(cv.Key)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			c.logger.Errorf("Error getting value for %s: %v", cv.Key, err)
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if cv.Sensitive {
 | 
			
		||||
			value = "****"
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		c.logger.Debugf("config set", cv.Key, value)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@ -1,18 +0,0 @@
 | 
			
		||||
//go:build !configreport
 | 
			
		||||
 | 
			
		||||
package config
 | 
			
		||||
 | 
			
		||||
type Config struct{}
 | 
			
		||||
 | 
			
		||||
func NewConfig() *Config {
 | 
			
		||||
	return &Config{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) AddKey(key string, defaultValue string, sensitive bool, validator func(string) error) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) GetValue(key string) (string, error) {
 | 
			
		||||
	return "", nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *Config) Report(level string) {}
 | 
			
		||||
							
								
								
									
										28
									
								
								doc/data.md
									
									
									
									
									
								
							
							
						
						
									
										28
									
								
								doc/data.md
									
									
									
									
									
								
							@ -1,28 +0,0 @@
 | 
			
		||||
# Internals
 | 
			
		||||
 | 
			
		||||
## Version
 | 
			
		||||
 | 
			
		||||
This document describes component versions:
 | 
			
		||||
 | 
			
		||||
* `urdt-ussd` `v0.5.0-beta`
 | 
			
		||||
* `go-vise` `v0.2.2`
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
## User profile data
 | 
			
		||||
 | 
			
		||||
All user profile items are stored under keys matching the user's session id, prefixed with the 8-bit value `github.com/grassrootseconomics/go-vise/db.DATATYPE_USERDATA` (32), and followed with a 16-big big-endian value subprefix.
 | 
			
		||||
 | 
			
		||||
For example, given the sessionId `+254123` and the key `git.grassecon.net/urdt-ussd/common.DATA_PUBLIC_KEY` (2) will be stored under the key:
 | 
			
		||||
 | 
			
		||||
```
 | 
			
		||||
0x322b3235343132330002
 | 
			
		||||
 | 
			
		||||
prefix   sessionid       subprefix
 | 
			
		||||
32       2b323534313233  0002
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Sub-prefixes
 | 
			
		||||
 | 
			
		||||
All sub-prefixes are defined as constants in the `git.grassecon.net/urdt-ussd/common` package. The constant names have the prefix `DATA_`
 | 
			
		||||
 | 
			
		||||
Please refer to inline godoc documentation for the `git.grassecon.net/urdt-ussd/common` package for details on each data item.
 | 
			
		||||
@ -1,14 +0,0 @@
 | 
			
		||||
package entry
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/persist"
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/resource"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type EntryHandler interface {
 | 
			
		||||
	Init(context.Context, string, []byte) (resource.Result, error) // HandlerFunc
 | 
			
		||||
	Exit()
 | 
			
		||||
	SetPersister(*persist.Persister)
 | 
			
		||||
}
 | 
			
		||||
@ -1,15 +0,0 @@
 | 
			
		||||
package errors
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	ErrInvalidRequest = errors.New("invalid request for context")
 | 
			
		||||
	ErrSessionMissing = errors.New("missing session")
 | 
			
		||||
	ErrInvalidInput   = errors.New("invalid input")
 | 
			
		||||
	ErrStorage        = errors.New("storage retrieval fail")
 | 
			
		||||
	ErrEngineType     = errors.New("incompatible engine")
 | 
			
		||||
	ErrEngineInit     = errors.New("engine init fail")
 | 
			
		||||
	ErrEngineExec     = errors.New("engine exec fail")
 | 
			
		||||
)
 | 
			
		||||
							
								
								
									
										48
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										48
									
								
								go.mod
									
									
									
									
									
								
							@ -1,27 +1,43 @@
 | 
			
		||||
module git.grassecon.net/grassrootseconomics/visedriver
 | 
			
		||||
module git.grassecon.net/urdt/ussd
 | 
			
		||||
 | 
			
		||||
go 1.24
 | 
			
		||||
go 1.23.0
 | 
			
		||||
 | 
			
		||||
toolchain go1.24.6
 | 
			
		||||
toolchain go1.23.2
 | 
			
		||||
 | 
			
		||||
require (
 | 
			
		||||
	github.com/grassrootseconomics/go-vise v0.5.0
 | 
			
		||||
	github.com/jackc/pgx/v5 v5.7.5
 | 
			
		||||
	github.com/joho/godotenv v1.5.1
 | 
			
		||||
	git.defalsify.org/vise.git v0.2.1-0.20241017112704-307fa6fcdc6b
 | 
			
		||||
	github.com/alecthomas/assert/v2 v2.2.2
 | 
			
		||||
	github.com/grassrootseconomics/eth-custodial v1.3.0-beta
 | 
			
		||||
	github.com/peteole/testdata-loader v0.3.0
 | 
			
		||||
	gopkg.in/leonelquinteros/gotext.v1 v1.3.1
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
require (
 | 
			
		||||
	github.com/barbashov/iso639-3 v0.0.0-20211020172741-1f4ffb2d8d1c // indirect
 | 
			
		||||
	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 | 
			
		||||
	github.com/fxamacker/cbor/v2 v2.9.0 // indirect
 | 
			
		||||
	github.com/jackc/pgpassfile v1.0.0 // indirect
 | 
			
		||||
	github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
 | 
			
		||||
	github.com/jackc/pgx/v5 v5.7.1 // indirect
 | 
			
		||||
	github.com/jackc/puddle/v2 v2.2.2 // indirect
 | 
			
		||||
	github.com/leonelquinteros/gotext v1.7.2 // indirect
 | 
			
		||||
	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 | 
			
		||||
	github.com/stretchr/testify v1.9.0 // indirect
 | 
			
		||||
	github.com/x448/float16 v0.8.4 // indirect
 | 
			
		||||
	golang.org/x/crypto v0.40.0 // indirect
 | 
			
		||||
	golang.org/x/sync v0.16.0 // indirect
 | 
			
		||||
	golang.org/x/text v0.27.0 // indirect
 | 
			
		||||
	github.com/joho/godotenv v1.5.1
 | 
			
		||||
	github.com/kr/text v0.2.0 // indirect
 | 
			
		||||
	github.com/rogpeppe/go-internal v1.13.1 // indirect
 | 
			
		||||
	golang.org/x/crypto v0.27.0 // indirect
 | 
			
		||||
	golang.org/x/sync v0.8.0 // indirect
 | 
			
		||||
	golang.org/x/text v0.18.0 // indirect
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
require (
 | 
			
		||||
	github.com/alecthomas/participle/v2 v2.0.0 // indirect
 | 
			
		||||
	github.com/alecthomas/repr v0.2.0 // indirect
 | 
			
		||||
	github.com/barbashov/iso639-3 v0.0.0-20211020172741-1f4ffb2d8d1c // indirect
 | 
			
		||||
	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 | 
			
		||||
	github.com/fxamacker/cbor/v2 v2.4.0 // indirect
 | 
			
		||||
	github.com/gofrs/uuid v4.4.0+incompatible
 | 
			
		||||
	github.com/graygnuorg/go-gdbm v0.0.0-20220711140707-71387d66dce4 // indirect
 | 
			
		||||
	github.com/hexops/gotextdiff v1.0.3 // indirect
 | 
			
		||||
	github.com/mattn/kinako v0.0.0-20170717041458-332c0a7e205a // indirect
 | 
			
		||||
	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 | 
			
		||||
	github.com/stretchr/objx v0.5.2 // indirect
 | 
			
		||||
	github.com/stretchr/testify v1.9.0
 | 
			
		||||
	github.com/x448/float16 v0.8.4 // indirect
 | 
			
		||||
	gopkg.in/yaml.v3 v3.0.1 // indirect
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										61
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										61
									
								
								go.sum
									
									
									
									
									
								
							@ -1,47 +1,72 @@
 | 
			
		||||
git.defalsify.org/vise.git v0.2.1-0.20241017112704-307fa6fcdc6b h1:dxBplsIlzJHV+5EH+gzB+w08Blt7IJbb2jeRe1OEjLU=
 | 
			
		||||
git.defalsify.org/vise.git v0.2.1-0.20241017112704-307fa6fcdc6b/go.mod h1:jyBMe1qTYUz3mmuoC9JQ/TvFeW0vTanCUcPu3H8p4Ck=
 | 
			
		||||
github.com/alecthomas/assert/v2 v2.2.2 h1:Z/iVC0xZfWTaFNE6bA3z07T86hd45Xe2eLt6WVy2bbk=
 | 
			
		||||
github.com/alecthomas/assert/v2 v2.2.2/go.mod h1:pXcQ2Asjp247dahGEmsZ6ru0UVwnkhktn7S0bBDLxvQ=
 | 
			
		||||
github.com/alecthomas/participle/v2 v2.0.0 h1:Fgrq+MbuSsJwIkw3fEj9h75vDP0Er5JzepJ0/HNHv0g=
 | 
			
		||||
github.com/alecthomas/participle/v2 v2.0.0/go.mod h1:rAKZdJldHu8084ojcWevWAL8KmEU+AT+Olodb+WoN2Y=
 | 
			
		||||
github.com/alecthomas/repr v0.2.0 h1:HAzS41CIzNW5syS8Mf9UwXhNH1J9aix/BvDRf1Ml2Yk=
 | 
			
		||||
github.com/alecthomas/repr v0.2.0/go.mod h1:Fr0507jx4eOXV7AlPV6AVZLYrLIuIeSOWtW57eE/O/4=
 | 
			
		||||
github.com/barbashov/iso639-3 v0.0.0-20211020172741-1f4ffb2d8d1c h1:H9Nm+I7Cg/YVPpEV1RzU3Wq2pjamPc/UtHDgItcb7lE=
 | 
			
		||||
github.com/barbashov/iso639-3 v0.0.0-20211020172741-1f4ffb2d8d1c/go.mod h1:rGod7o6KPeJ+hyBpHfhi4v7blx9sf+QsHsA7KAsdN6U=
 | 
			
		||||
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
 | 
			
		||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 | 
			
		||||
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 | 
			
		||||
github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 | 
			
		||||
github.com/fxamacker/cbor/v2 v2.9.0 h1:NpKPmjDBgUfBms6tr6JZkTHtfFGcMKsw3eGcmD/sapM=
 | 
			
		||||
github.com/fxamacker/cbor/v2 v2.9.0/go.mod h1:vM4b+DJCtHn+zz7h3FFp/hDAI9WNWCsZj23V5ytsSxQ=
 | 
			
		||||
github.com/grassrootseconomics/go-vise v0.5.0 h1:FRg2de55Eb5SisrgTBeFWfWX+sXwp5q9r7YWtKWDwsk=
 | 
			
		||||
github.com/grassrootseconomics/go-vise v0.5.0/go.mod h1:b2/q4jfTu2i1wyUwYUu7FYq4m2f1AZv8MpiDM3ZcoGo=
 | 
			
		||||
github.com/fxamacker/cbor/v2 v2.4.0 h1:ri0ArlOR+5XunOP8CRUowT0pSJOwhW098ZCUyskZD88=
 | 
			
		||||
github.com/fxamacker/cbor/v2 v2.4.0/go.mod h1:TA1xS00nchWmaBnEIxPSE5oHLuJBAVvqrtAnWBwBCVo=
 | 
			
		||||
github.com/gofrs/uuid v4.4.0+incompatible h1:3qXRTX8/NbyulANqlc0lchS1gqAVxRgsuW1YrTJupqA=
 | 
			
		||||
github.com/gofrs/uuid v4.4.0+incompatible/go.mod h1:b2aQJv3Z4Fp6yNu3cdSllBxTCLRxnplIgP/c0N/04lM=
 | 
			
		||||
github.com/grassrootseconomics/eth-custodial v1.3.0-beta h1:twrMBhl89GqDUL9PlkzQxMP/6OST1BByrNDj+rqXDmU=
 | 
			
		||||
github.com/grassrootseconomics/eth-custodial v1.3.0-beta/go.mod h1:7uhRcdnJplX4t6GKCEFkbeDhhjlcaGJeJqevbcvGLZo=
 | 
			
		||||
github.com/graygnuorg/go-gdbm v0.0.0-20220711140707-71387d66dce4 h1:U4kkNYryi/qfbBF8gh7Vsbuz+cVmhf5kt6pE9bYYyLo=
 | 
			
		||||
github.com/graygnuorg/go-gdbm v0.0.0-20220711140707-71387d66dce4/go.mod h1:zpZDgZFzeq9s0MIeB1P50NIEWDFFHSFBohI/NbaTD/Y=
 | 
			
		||||
github.com/hexops/gotextdiff v1.0.3 h1:gitA9+qJrrTCsiCl7+kh75nPqQt1cx4ZkudSTLoUqJM=
 | 
			
		||||
github.com/hexops/gotextdiff v1.0.3/go.mod h1:pSWU5MAI3yDq+fZBTazCSJysOMbxWL1BSow5/V2vxeg=
 | 
			
		||||
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
 | 
			
		||||
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
 | 
			
		||||
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
 | 
			
		||||
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=
 | 
			
		||||
github.com/jackc/pgx/v5 v5.7.5 h1:JHGfMnQY+IEtGM63d+NGMjoRpysB2JBwDr5fsngwmJs=
 | 
			
		||||
github.com/jackc/pgx/v5 v5.7.5/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M=
 | 
			
		||||
github.com/jackc/pgx/v5 v5.7.1 h1:x7SYsPBYDkHDksogeSmZZ5xzThcTgRz++I5E+ePFUcs=
 | 
			
		||||
github.com/jackc/pgx/v5 v5.7.1/go.mod h1:e7O26IywZZ+naJtWWos6i6fvWK+29etgITqrqHLfoZA=
 | 
			
		||||
github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo=
 | 
			
		||||
github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
 | 
			
		||||
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
 | 
			
		||||
github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 | 
			
		||||
github.com/leonelquinteros/gotext v1.7.2 h1:bDPndU8nt+/kRo1m4l/1OXiiy2v7Z7dfPQ9+YP7G1Mc=
 | 
			
		||||
github.com/leonelquinteros/gotext v1.7.2/go.mod h1:9/haCkm5P7Jay1sxKDGJ5WIg4zkz8oZKw4ekNpALob8=
 | 
			
		||||
github.com/lmittmann/tint v1.1.2 h1:2CQzrL6rslrsyjqLDwD11bZ5OpLBPU+g3G/r5LSfS8w=
 | 
			
		||||
github.com/lmittmann/tint v1.1.2/go.mod h1:HIS3gSy7qNwGCj+5oRjAutErFBl4BzdQP6cJZ0NfMwE=
 | 
			
		||||
github.com/pashagolub/pgxmock/v4 v4.7.0 h1:de2ORuFYyjwOQR7NBm57+321RnZxpYiuUjsmqRiqgh8=
 | 
			
		||||
github.com/pashagolub/pgxmock/v4 v4.7.0/go.mod h1:9L57pC193h2aKRHVyiiE817avasIPZnPwPlw3JczWvM=
 | 
			
		||||
github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
 | 
			
		||||
github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
 | 
			
		||||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
 | 
			
		||||
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
 | 
			
		||||
github.com/mattn/kinako v0.0.0-20170717041458-332c0a7e205a h1:0Q3H0YXzMHiciXtRcM+j0jiCe8WKPQHoRgQiRTnfcLY=
 | 
			
		||||
github.com/mattn/kinako v0.0.0-20170717041458-332c0a7e205a/go.mod h1:CdTTBOYzS5E4mWS1N8NWP6AHI19MP0A2B18n3hLzRMk=
 | 
			
		||||
github.com/pashagolub/pgxmock/v4 v4.3.0 h1:DqT7fk0OCK6H0GvqtcMsLpv8cIwWqdxWgfZNLeHCb/s=
 | 
			
		||||
github.com/pashagolub/pgxmock/v4 v4.3.0/go.mod h1:9VoVHXwS3XR/yPtKGzwQvwZX1kzGB9sM8SviDcHDa3A=
 | 
			
		||||
github.com/peteole/testdata-loader v0.3.0 h1:8jckE9KcyNHgyv/VPoaljvKZE0Rqr8+dPVYH6rfNr9I=
 | 
			
		||||
github.com/peteole/testdata-loader v0.3.0/go.mod h1:Mt0ZbRtb56u8SLJpNP+BnQbENljMorYBpqlvt3cS83U=
 | 
			
		||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 | 
			
		||||
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
 | 
			
		||||
github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 | 
			
		||||
github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII=
 | 
			
		||||
github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
 | 
			
		||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 | 
			
		||||
github.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=
 | 
			
		||||
github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
 | 
			
		||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
 | 
			
		||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 | 
			
		||||
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
 | 
			
		||||
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 | 
			
		||||
github.com/x448/float16 v0.8.4 h1:qLwI1I70+NjRFUR3zs1JPUCgaCXSh3SW62uAKT1mSBM=
 | 
			
		||||
github.com/x448/float16 v0.8.4/go.mod h1:14CWIYCyZA/cWjXOioeEpHeN/83MdbZDRQHoFcYsOfg=
 | 
			
		||||
golang.org/x/crypto v0.40.0 h1:r4x+VvoG5Fm+eJcxMaY8CQM7Lb0l1lsmjGBQ6s8BfKM=
 | 
			
		||||
golang.org/x/crypto v0.40.0/go.mod h1:Qr1vMER5WyS2dfPHAlsOj01wgLbsyWtFn/aY+5+ZdxY=
 | 
			
		||||
golang.org/x/sync v0.16.0 h1:ycBJEhp9p4vXvUZNszeOq0kGTPghopOL8q0fq3vstxw=
 | 
			
		||||
golang.org/x/sync v0.16.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
 | 
			
		||||
golang.org/x/text v0.27.0 h1:4fGWRpyh641NLlecmyl4LOe6yDdfaYNrGb2zdfo4JV4=
 | 
			
		||||
golang.org/x/text v0.27.0/go.mod h1:1D28KMCvyooCX9hBiosv5Tz/+YLxj0j7XhWjpSUF7CU=
 | 
			
		||||
golang.org/x/crypto v0.27.0 h1:GXm2NjJrPaiv/h1tb2UH8QfgC/hOf/+z0p6PT8o1w7A=
 | 
			
		||||
golang.org/x/crypto v0.27.0/go.mod h1:1Xngt8kV6Dvbssa53Ziq6Eqn0HqbZi5Z6R0ZpwQzt70=
 | 
			
		||||
golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ=
 | 
			
		||||
golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 | 
			
		||||
golang.org/x/text v0.18.0 h1:XvMDiNzPAl0jr17s6W9lcaIhGUfUORdGCNsuLmPG224=
 | 
			
		||||
golang.org/x/text v0.18.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
 | 
			
		||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 | 
			
		||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 | 
			
		||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
 | 
			
		||||
gopkg.in/leonelquinteros/gotext.v1 v1.3.1 h1:8d9/fdTG0kn/B7NNGV1BsEyvektXFAbkMsTZS2sFSCc=
 | 
			
		||||
gopkg.in/leonelquinteros/gotext.v1 v1.3.1/go.mod h1:X1WlGDeAFIYsW6GjgMm4VwUwZ2XjI7Zan2InxSUQWrU=
 | 
			
		||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 | 
			
		||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 | 
			
		||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 | 
			
		||||
 | 
			
		||||
@ -1,32 +1,26 @@
 | 
			
		||||
package env
 | 
			
		||||
package initializers
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"log"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
	"strconv"
 | 
			
		||||
 | 
			
		||||
	"github.com/joho/godotenv"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func LoadEnvVariables() {
 | 
			
		||||
	LoadEnvVariablesPath(".")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func LoadEnvVariablesPath(dir string) {
 | 
			
		||||
	fp := path.Join(dir, ".env")
 | 
			
		||||
	err := godotenv.Load(fp)
 | 
			
		||||
	err := godotenv.Load()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatal("Error loading .env file", err)
 | 
			
		||||
		log.Fatal("Error loading .env file")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Helper to get environment variables with a default fallback
 | 
			
		||||
func GetEnv(key, defaultVal string) string {
 | 
			
		||||
	if value, exists := os.LookupEnv(key); exists {
 | 
			
		||||
		return value
 | 
			
		||||
  if value, exists := os.LookupEnv(key); exists {
 | 
			
		||||
   	return value
 | 
			
		||||
	}
 | 
			
		||||
	return defaultVal
 | 
			
		||||
  return defaultVal
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Helper to safely convert environment variables to uint
 | 
			
		||||
							
								
								
									
										105
									
								
								internal/handlers/base.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										105
									
								
								internal/handlers/base.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,105 @@
 | 
			
		||||
package handlers
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"git.defalsify.org/vise.git/db"
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/persist"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/ussd"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/storage"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type BaseSessionHandler struct {
 | 
			
		||||
	cfgTemplate engine.Config
 | 
			
		||||
	rp RequestParser
 | 
			
		||||
	rs resource.Resource
 | 
			
		||||
	hn *ussd.Handlers
 | 
			
		||||
	provider storage.StorageProvider
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewBaseSessionHandler(cfg engine.Config, rs resource.Resource, stateDb db.Db, userdataDb db.Db, rp RequestParser, hn *ussd.Handlers) *BaseSessionHandler {
 | 
			
		||||
	return &BaseSessionHandler{
 | 
			
		||||
		cfgTemplate: cfg,
 | 
			
		||||
		rs: rs,
 | 
			
		||||
		hn: hn,
 | 
			
		||||
		rp: rp,
 | 
			
		||||
		provider: storage.NewSimpleStorageProvider(stateDb, userdataDb),
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f* BaseSessionHandler) Shutdown() {
 | 
			
		||||
	err := f.provider.Close()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.Errorf("handler shutdown error", "err", err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *BaseSessionHandler) GetEngine(cfg engine.Config, rs resource.Resource, pr *persist.Persister) engine.Engine {
 | 
			
		||||
	en := engine.NewEngine(cfg, rs)
 | 
			
		||||
	en = en.WithPersister(pr)
 | 
			
		||||
	return en
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *BaseSessionHandler) Process(rqs RequestSession) (RequestSession, error) {
 | 
			
		||||
	var r bool
 | 
			
		||||
	var err error
 | 
			
		||||
	var ok bool
 | 
			
		||||
	
 | 
			
		||||
	logg.InfoCtxf(rqs.Ctx, "new request", "data", rqs)
 | 
			
		||||
 | 
			
		||||
	rqs.Storage, err = f.provider.Get(rqs.Config.SessionId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "storage get error", err)
 | 
			
		||||
		return rqs, ErrStorage
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	f.hn = f.hn.WithPersister(rqs.Storage.Persister)
 | 
			
		||||
	eni := f.GetEngine(rqs.Config, f.rs, rqs.Storage.Persister)
 | 
			
		||||
	en, ok := eni.(*engine.DefaultEngine)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		perr := f.provider.Put(rqs.Config.SessionId, rqs.Storage)
 | 
			
		||||
		rqs.Storage = nil
 | 
			
		||||
		if perr != nil {
 | 
			
		||||
			logg.ErrorCtxf(rqs.Ctx, "", "storage put error", perr)
 | 
			
		||||
		}
 | 
			
		||||
		return rqs, ErrEngineType
 | 
			
		||||
	}
 | 
			
		||||
	en = en.WithFirst(f.hn.Init)
 | 
			
		||||
	if rqs.Config.EngineDebug {
 | 
			
		||||
		en = en.WithDebug(nil)
 | 
			
		||||
	}
 | 
			
		||||
	rqs.Engine = en
 | 
			
		||||
 | 
			
		||||
	r, err = rqs.Engine.Exec(rqs.Ctx, rqs.Input)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		perr := f.provider.Put(rqs.Config.SessionId, rqs.Storage)
 | 
			
		||||
		rqs.Storage = nil
 | 
			
		||||
		if perr != nil {
 | 
			
		||||
			logg.ErrorCtxf(rqs.Ctx, "", "storage put error", perr)
 | 
			
		||||
		}
 | 
			
		||||
		return rqs, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rqs.Continue = r 
 | 
			
		||||
	return rqs, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *BaseSessionHandler) Output(rqs RequestSession) (RequestSession,  error) {
 | 
			
		||||
	var err error
 | 
			
		||||
	_, err = rqs.Engine.Flush(rqs.Ctx, rqs.Writer)
 | 
			
		||||
	return rqs, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *BaseSessionHandler) Reset(rqs RequestSession) (RequestSession, error) {
 | 
			
		||||
	defer f.provider.Put(rqs.Config.SessionId, rqs.Storage)
 | 
			
		||||
	return rqs, rqs.Engine.Finish()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *BaseSessionHandler) GetConfig() engine.Config {
 | 
			
		||||
	return f.cfgTemplate
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *BaseSessionHandler) GetRequestParser() RequestParser {
 | 
			
		||||
	return f.rp
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										110
									
								
								internal/handlers/handlerservice.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										110
									
								
								internal/handlers/handlerservice.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,110 @@
 | 
			
		||||
package handlers
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"git.defalsify.org/vise.git/asm"
 | 
			
		||||
	"git.defalsify.org/vise.git/db"
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/persist"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/ussd"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type HandlerService interface {
 | 
			
		||||
	GetHandler() (*ussd.Handlers, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getParser(fp string, debug bool) (*asm.FlagParser, error) {
 | 
			
		||||
	flagParser := asm.NewFlagParser().WithDebug()
 | 
			
		||||
	_, err := flagParser.Load(fp)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return flagParser, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type LocalHandlerService struct {
 | 
			
		||||
	Parser        *asm.FlagParser
 | 
			
		||||
	DbRs          *resource.DbResource
 | 
			
		||||
	Pe            *persist.Persister
 | 
			
		||||
	UserdataStore *db.Db
 | 
			
		||||
	Cfg           engine.Config
 | 
			
		||||
	Rs            resource.Resource
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewLocalHandlerService(fp string, debug bool, dbResource *resource.DbResource, cfg engine.Config, rs resource.Resource) (*LocalHandlerService, error) {
 | 
			
		||||
	parser, err := getParser(fp, debug)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return &LocalHandlerService{
 | 
			
		||||
		Parser: parser,
 | 
			
		||||
		DbRs:   dbResource,
 | 
			
		||||
		Cfg:    cfg,
 | 
			
		||||
		Rs:     rs,
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ls *LocalHandlerService) SetPersister(Pe *persist.Persister) {
 | 
			
		||||
	ls.Pe = Pe
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ls *LocalHandlerService) SetDataStore(db *db.Db) {
 | 
			
		||||
	ls.UserdataStore = db
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ls *LocalHandlerService) GetHandler(accountService server.AccountServiceInterface) (*ussd.Handlers, error) {
 | 
			
		||||
	ussdHandlers, err := ussd.NewHandlers(ls.Parser, *ls.UserdataStore,accountService)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	ussdHandlers = ussdHandlers.WithPersister(ls.Pe)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("set_language", ussdHandlers.SetLanguage)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("create_account", ussdHandlers.CreateAccount)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("save_temporary_pin", ussdHandlers.SaveTemporaryPin)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("verify_create_pin", ussdHandlers.VerifyCreatePin)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("check_identifier", ussdHandlers.CheckIdentifier)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("check_account_status", ussdHandlers.CheckAccountStatus)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("authorize_account", ussdHandlers.Authorize)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("quit", ussdHandlers.Quit)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("check_balance", ussdHandlers.CheckBalance)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("validate_recipient", ussdHandlers.ValidateRecipient)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("transaction_reset", ussdHandlers.TransactionReset)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("max_amount", ussdHandlers.MaxAmount)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("validate_amount", ussdHandlers.ValidateAmount)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("reset_transaction_amount", ussdHandlers.ResetTransactionAmount)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("get_recipient", ussdHandlers.GetRecipient)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("get_sender", ussdHandlers.GetSender)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("get_amount", ussdHandlers.GetAmount)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("reset_incorrect", ussdHandlers.ResetIncorrectPin)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("save_firstname", ussdHandlers.SaveFirstname)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("save_familyname", ussdHandlers.SaveFamilyname)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("save_gender", ussdHandlers.SaveGender)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("save_location", ussdHandlers.SaveLocation)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("save_yob", ussdHandlers.SaveYob)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("save_offerings", ussdHandlers.SaveOfferings)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("reset_account_authorized", ussdHandlers.ResetAccountAuthorized)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("reset_allow_update", ussdHandlers.ResetAllowUpdate)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("get_profile_info", ussdHandlers.GetProfileInfo)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("verify_yob", ussdHandlers.VerifyYob)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("reset_incorrect_date_format", ussdHandlers.ResetIncorrectYob)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("initiate_transaction", ussdHandlers.InitiateTransaction)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("verify_new_pin", ussdHandlers.VerifyNewPin)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("confirm_pin_change", ussdHandlers.ConfirmPinChange)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("quit_with_help", ussdHandlers.QuitWithHelp)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("fetch_custodial_balances", ussdHandlers.FetchCustodialBalances)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("set_default_voucher", ussdHandlers.SetDefaultVoucher)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("check_vouchers", ussdHandlers.CheckVouchers)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("get_vouchers", ussdHandlers.GetVoucherList)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("view_voucher", ussdHandlers.ViewVoucher)
 | 
			
		||||
	ls.DbRs.AddLocalFunc("set_voucher", ussdHandlers.SetVoucher)
 | 
			
		||||
 | 
			
		||||
	return ussdHandlers, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// TODO: enable setting of sessionId on engine init time
 | 
			
		||||
func (ls *LocalHandlerService) GetEngine() *engine.DefaultEngine {
 | 
			
		||||
	en := engine.NewEngine(ls.Cfg, ls.Rs)
 | 
			
		||||
	en = en.WithPersister(ls.Pe)
 | 
			
		||||
	return en
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										185
									
								
								internal/handlers/server/accountservice.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										185
									
								
								internal/handlers/server/accountservice.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,185 @@
 | 
			
		||||
package server
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"os"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/config"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/models"
 | 
			
		||||
	"github.com/grassrootseconomics/eth-custodial/pkg/api"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	okResponse  api.OKResponse
 | 
			
		||||
	errResponse api.ErrResponse
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type AccountServiceInterface interface {
 | 
			
		||||
	CheckBalance(ctx context.Context, publicKey string) (*models.BalanceResponse, error)
 | 
			
		||||
	CreateAccount(ctx context.Context) (*api.OKResponse, error)
 | 
			
		||||
	CheckAccountStatus(ctx context.Context, trackingId string) (*models.TrackStatusResponse, error)
 | 
			
		||||
	TrackAccountStatus(ctx context.Context, publicKey string) (*api.OKResponse, error)
 | 
			
		||||
	FetchVouchers(ctx context.Context, publicKey string) (*models.VoucherHoldingResponse, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type AccountService struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Parameters:
 | 
			
		||||
//   - trackingId: A unique identifier for the account.This should be obtained from a previous call to
 | 
			
		||||
//     CreateAccount or a similar function that returns an AccountResponse. The `trackingId` field in the
 | 
			
		||||
//     AccountResponse struct can be used here to check the account status during a transaction.
 | 
			
		||||
//
 | 
			
		||||
// Returns:
 | 
			
		||||
//   - string: The status of the transaction as a string. If there is an error during the request or processing, this will be an empty string.
 | 
			
		||||
//   - error: An error if any occurred during the HTTP request, reading the response, or unmarshalling the JSON data.
 | 
			
		||||
//     If no error occurs, this will be nil
 | 
			
		||||
func (as *AccountService) CheckAccountStatus(ctx context.Context, trackingId string) (*models.TrackStatusResponse, error) {
 | 
			
		||||
	resp, err := http.Get(config.BalanceURL + trackingId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	body, err := io.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var trackResp models.TrackStatusResponse
 | 
			
		||||
	err = json.Unmarshal(body, &trackResp)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return &trackResp, nil
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (as *AccountService) TrackAccountStatus(ctx context.Context, publicKey string) (*api.OKResponse, error) {
 | 
			
		||||
	var err error
 | 
			
		||||
	// Construct the URL with the path parameter
 | 
			
		||||
	url := fmt.Sprintf("%s/%s", config.TrackURL, publicKey)
 | 
			
		||||
	req, err := http.NewRequest("GET", url, nil)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	req.Header.Set("Content-Type", "application/json")
 | 
			
		||||
	req.Header.Set("X-GE-KEY", "xd")
 | 
			
		||||
 | 
			
		||||
	resp, err := http.DefaultClient.Do(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	body, err := io.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		errResponse.Description = err.Error()
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if resp.StatusCode >= http.StatusBadRequest {
 | 
			
		||||
		err := json.Unmarshal([]byte(body), &errResponse)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		return nil, errors.New(errResponse.Description)
 | 
			
		||||
	}
 | 
			
		||||
	err = json.Unmarshal([]byte(body), &okResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if len(okResponse.Result) == 0 {
 | 
			
		||||
		return nil, errors.New("Empty api result")
 | 
			
		||||
	}
 | 
			
		||||
	return &okResponse, nil
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CheckBalance retrieves the balance for a given public key from the custodial balance API endpoint.
 | 
			
		||||
// Parameters:
 | 
			
		||||
//   - publicKey: The public key associated with the account whose balance needs to be checked.
 | 
			
		||||
func (as *AccountService) CheckBalance(ctx context.Context, publicKey string) (*models.BalanceResponse, error) {
 | 
			
		||||
	resp, err := http.Get(config.BalanceURL + publicKey)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
	body, err := io.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	var balanceResp models.BalanceResponse
 | 
			
		||||
	err = json.Unmarshal(body, &balanceResp)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return &balanceResp, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CreateAccount creates a new account in the custodial system.
 | 
			
		||||
// Returns:
 | 
			
		||||
//   - *models.AccountResponse: A pointer to an AccountResponse struct containing the details of the created account.
 | 
			
		||||
//     If there is an error during the request or processing, this will be nil.
 | 
			
		||||
//   - error: An error if any occurred during the HTTP request, reading the response, or unmarshalling the JSON data.
 | 
			
		||||
//     If no error occurs, this will be nil.
 | 
			
		||||
func (as *AccountService) CreateAccount(ctx context.Context) (*api.OKResponse, error) {
 | 
			
		||||
	var err error
 | 
			
		||||
 | 
			
		||||
	// Create a new request
 | 
			
		||||
	req, err := http.NewRequest("POST", config.CreateAccountURL, nil)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	req.Header.Set("Content-Type", "application/json")
 | 
			
		||||
	req.Header.Set("X-GE-KEY", "xd")
 | 
			
		||||
 | 
			
		||||
	resp, err := http.DefaultClient.Do(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		errResponse.Description = err.Error()
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer resp.Body.Close()
 | 
			
		||||
 | 
			
		||||
	body, err := io.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if resp.StatusCode >= http.StatusBadRequest {
 | 
			
		||||
		err := json.Unmarshal([]byte(body), &errResponse)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		return nil, errors.New(errResponse.Description)
 | 
			
		||||
	}
 | 
			
		||||
	err = json.Unmarshal([]byte(body), &okResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if len(okResponse.Result) == 0 {
 | 
			
		||||
		return nil, errors.New("Empty api result")
 | 
			
		||||
	}
 | 
			
		||||
	return &okResponse, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// FetchVouchers retrieves the token holdings for a given public key from the custodial holdings API endpoint
 | 
			
		||||
// Parameters:
 | 
			
		||||
//   - publicKey: The public key associated with the account.
 | 
			
		||||
func (as *AccountService) FetchVouchers(ctx context.Context, publicKey string) (*models.VoucherHoldingResponse, error) {
 | 
			
		||||
	file, err := os.Open("sample_tokens.json")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	defer file.Close()
 | 
			
		||||
	var holdings models.VoucherHoldingResponse
 | 
			
		||||
 | 
			
		||||
	if err := json.NewDecoder(file).Decode(&holdings); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return &holdings, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										54
									
								
								internal/handlers/single.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								internal/handlers/single.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,54 @@
 | 
			
		||||
package handlers
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"io"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
	"git.defalsify.org/vise.git/persist"
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/storage"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg = logging.NewVanilla().WithDomain("handlers")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	ErrInvalidRequest = errors.New("invalid request for context")
 | 
			
		||||
	ErrSessionMissing = errors.New("missing session")
 | 
			
		||||
	ErrInvalidInput = errors.New("invalid input")
 | 
			
		||||
	ErrStorage = errors.New("storage retrieval fail")
 | 
			
		||||
	ErrEngineType = errors.New("incompatible engine")
 | 
			
		||||
	ErrEngineInit = errors.New("engine init fail")
 | 
			
		||||
	ErrEngineExec = errors.New("engine exec fail")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type RequestSession struct {
 | 
			
		||||
	Ctx context.Context
 | 
			
		||||
	Config engine.Config
 | 
			
		||||
	Engine engine.Engine
 | 
			
		||||
	Input []byte
 | 
			
		||||
	Storage *storage.Storage
 | 
			
		||||
	Writer io.Writer
 | 
			
		||||
	Continue bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// TODO: seems like can remove this.
 | 
			
		||||
type RequestParser interface {
 | 
			
		||||
	GetSessionId(rq any) (string, error)
 | 
			
		||||
	GetInput(rq any) ([]byte, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type RequestHandler interface {
 | 
			
		||||
	GetConfig() engine.Config
 | 
			
		||||
	GetRequestParser() RequestParser
 | 
			
		||||
	GetEngine(cfg engine.Config, rs resource.Resource, pe *persist.Persister) engine.Engine 
 | 
			
		||||
	Process(rs RequestSession) (RequestSession, error)
 | 
			
		||||
	Output(rs RequestSession) (RequestSession, error)
 | 
			
		||||
	Reset(rs RequestSession) (RequestSession, error)
 | 
			
		||||
	Shutdown()
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										1307
									
								
								internal/handlers/ussd/menuhandler.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1307
									
								
								internal/handlers/ussd/menuhandler.go
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										1936
									
								
								internal/handlers/ussd/menuhandler_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1936
									
								
								internal/handlers/ussd/menuhandler_test.go
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										92
									
								
								internal/http/at_session_handler.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										92
									
								
								internal/http/at_session_handler.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,92 @@
 | 
			
		||||
package http
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type ATSessionHandler struct {
 | 
			
		||||
	*SessionHandler
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewATSessionHandler(h handlers.RequestHandler) *ATSessionHandler {
 | 
			
		||||
	return &ATSessionHandler{
 | 
			
		||||
		SessionHandler: ToSessionHandler(h),
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ash *ATSessionHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 | 
			
		||||
	var code int
 | 
			
		||||
	var err error
 | 
			
		||||
 | 
			
		||||
	rqs := handlers.RequestSession{
 | 
			
		||||
		Ctx:    req.Context(),
 | 
			
		||||
		Writer: w,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rp := ash.GetRequestParser()
 | 
			
		||||
	cfg := ash.GetConfig()
 | 
			
		||||
	cfg.SessionId, err = rp.GetSessionId(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "header processing error", err)
 | 
			
		||||
		ash.writeError(w, 400, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	rqs.Config = cfg
 | 
			
		||||
	rqs.Input, err = rp.GetInput(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "header processing error", err)
 | 
			
		||||
		ash.writeError(w, 400, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rqs, err = ash.Process(rqs) 
 | 
			
		||||
	switch err {
 | 
			
		||||
	case nil: // set code to 200 if no err
 | 
			
		||||
		code = 200
 | 
			
		||||
	case handlers.ErrStorage, handlers.ErrEngineInit, handlers.ErrEngineExec, handlers.ErrEngineType:
 | 
			
		||||
		code = 500
 | 
			
		||||
	default:
 | 
			
		||||
		code = 500
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if code != 200 {
 | 
			
		||||
		ash.writeError(w, 500, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	w.WriteHeader(200)
 | 
			
		||||
	w.Header().Set("Content-Type", "text/plain")
 | 
			
		||||
	rqs, err = ash.Output(rqs)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		ash.writeError(w, 500, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rqs, err = ash.Reset(rqs)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		ash.writeError(w, 500, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ash *ATSessionHandler) Output(rqs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
	var err error
 | 
			
		||||
	var prefix string
 | 
			
		||||
 | 
			
		||||
	if rqs.Continue {
 | 
			
		||||
		prefix = "CON "
 | 
			
		||||
	} else {
 | 
			
		||||
		prefix = "END "
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	_, err = io.WriteString(rqs.Writer, prefix)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return rqs, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	_, err = rqs.Engine.Flush(rqs.Ctx, rqs.Writer)
 | 
			
		||||
	return rqs, err
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										449
									
								
								internal/http/http_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										449
									
								
								internal/http/http_test.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,449 @@
 | 
			
		||||
package http
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/httptest"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"testing"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/testutil/mocks/httpmocks"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// invalidRequestType is a custom type to test invalid request scenarios
 | 
			
		||||
type invalidRequestType struct{}
 | 
			
		||||
 | 
			
		||||
// errorReader is a helper type that always returns an error when Read is called
 | 
			
		||||
type errorReader struct{}
 | 
			
		||||
 | 
			
		||||
func (e *errorReader) Read(p []byte) (n int, err error) {
 | 
			
		||||
	return 0, errors.New("read error")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestNewATSessionHandler(t *testing.T) {
 | 
			
		||||
	mockHandler := &httpmocks.MockRequestHandler{}
 | 
			
		||||
	ash := NewATSessionHandler(mockHandler)
 | 
			
		||||
 | 
			
		||||
	if ash == nil {
 | 
			
		||||
		t.Fatal("NewATSessionHandler returned nil")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if ash.SessionHandler == nil {
 | 
			
		||||
		t.Fatal("SessionHandler is nil")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestATSessionHandler_ServeHTTP(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name           string
 | 
			
		||||
		setupMocks     func(*httpmocks.MockRequestHandler, *httpmocks.MockRequestParser, *httpmocks.MockEngine)
 | 
			
		||||
		formData       url.Values
 | 
			
		||||
		expectedStatus int
 | 
			
		||||
		expectedBody   string
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name: "Successful request",
 | 
			
		||||
			setupMocks: func(mh *httpmocks.MockRequestHandler, mrp *httpmocks.MockRequestParser, me *httpmocks.MockEngine) {
 | 
			
		||||
				mrp.GetSessionIdFunc = func(rq any) (string, error) {
 | 
			
		||||
					req := rq.(*http.Request)
 | 
			
		||||
					return req.FormValue("phoneNumber"), nil
 | 
			
		||||
				}
 | 
			
		||||
				mrp.GetInputFunc = func(rq any) ([]byte, error) {
 | 
			
		||||
					req := rq.(*http.Request)
 | 
			
		||||
					text := req.FormValue("text")
 | 
			
		||||
					parts := strings.Split(text, "*")
 | 
			
		||||
					return []byte(parts[len(parts)-1]), nil
 | 
			
		||||
				}
 | 
			
		||||
				mh.ProcessFunc = func(rqs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
					rqs.Continue = true
 | 
			
		||||
					rqs.Engine = me
 | 
			
		||||
					return rqs, nil
 | 
			
		||||
				}
 | 
			
		||||
				mh.GetConfigFunc = func() engine.Config { return engine.Config{} }
 | 
			
		||||
				mh.GetRequestParserFunc = func() handlers.RequestParser { return mrp }
 | 
			
		||||
				mh.OutputFunc = func(rs handlers.RequestSession) (handlers.RequestSession, error) { return rs, nil }
 | 
			
		||||
				mh.ResetFunc = func(rs handlers.RequestSession) (handlers.RequestSession, error) { return rs, nil }
 | 
			
		||||
				me.FlushFunc = func(context.Context, io.Writer) (int, error) { return 0, nil }
 | 
			
		||||
			},
 | 
			
		||||
			formData: url.Values{
 | 
			
		||||
				"phoneNumber": []string{"+1234567890"},
 | 
			
		||||
				"text":        []string{"1*2*3"},
 | 
			
		||||
			},
 | 
			
		||||
			expectedStatus: http.StatusOK,
 | 
			
		||||
			expectedBody:   "CON ",
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name: "GetSessionId error",
 | 
			
		||||
			setupMocks: func(mh *httpmocks.MockRequestHandler, mrp *httpmocks.MockRequestParser, me *httpmocks.MockEngine) {
 | 
			
		||||
				mrp.GetSessionIdFunc = func(rq any) (string, error) {
 | 
			
		||||
					return "", errors.New("no phone number found")
 | 
			
		||||
				}
 | 
			
		||||
				mh.GetConfigFunc = func() engine.Config { return engine.Config{} }
 | 
			
		||||
				mh.GetRequestParserFunc = func() handlers.RequestParser { return mrp }
 | 
			
		||||
			},
 | 
			
		||||
			formData: url.Values{
 | 
			
		||||
				"text": []string{"1*2*3"},
 | 
			
		||||
			},
 | 
			
		||||
			expectedStatus: http.StatusBadRequest,
 | 
			
		||||
			expectedBody:   "",
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name: "GetInput error",
 | 
			
		||||
			setupMocks: func(mh *httpmocks.MockRequestHandler, mrp *httpmocks.MockRequestParser, me *httpmocks.MockEngine) {
 | 
			
		||||
				mrp.GetSessionIdFunc = func(rq any) (string, error) {
 | 
			
		||||
					req := rq.(*http.Request)
 | 
			
		||||
					return req.FormValue("phoneNumber"), nil
 | 
			
		||||
				}
 | 
			
		||||
				mrp.GetInputFunc = func(rq any) ([]byte, error) {
 | 
			
		||||
					return nil, errors.New("no input found")
 | 
			
		||||
				}
 | 
			
		||||
				mh.GetConfigFunc = func() engine.Config { return engine.Config{} }
 | 
			
		||||
				mh.GetRequestParserFunc = func() handlers.RequestParser { return mrp }
 | 
			
		||||
			},
 | 
			
		||||
			formData: url.Values{
 | 
			
		||||
				"phoneNumber": []string{"+1234567890"},
 | 
			
		||||
			},
 | 
			
		||||
			expectedStatus: http.StatusBadRequest,
 | 
			
		||||
			expectedBody:   "",
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name: "Process error",
 | 
			
		||||
			setupMocks: func(mh *httpmocks.MockRequestHandler, mrp *httpmocks.MockRequestParser, me *httpmocks.MockEngine) {
 | 
			
		||||
				mrp.GetSessionIdFunc = func(rq any) (string, error) {
 | 
			
		||||
					req := rq.(*http.Request)
 | 
			
		||||
					return req.FormValue("phoneNumber"), nil
 | 
			
		||||
				}
 | 
			
		||||
				mrp.GetInputFunc = func(rq any) ([]byte, error) {
 | 
			
		||||
					req := rq.(*http.Request)
 | 
			
		||||
					text := req.FormValue("text")
 | 
			
		||||
					parts := strings.Split(text, "*")
 | 
			
		||||
					return []byte(parts[len(parts)-1]), nil
 | 
			
		||||
				}
 | 
			
		||||
				mh.ProcessFunc = func(rqs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
					return rqs, handlers.ErrStorage
 | 
			
		||||
				}
 | 
			
		||||
				mh.GetConfigFunc = func() engine.Config { return engine.Config{} }
 | 
			
		||||
				mh.GetRequestParserFunc = func() handlers.RequestParser { return mrp }
 | 
			
		||||
			},
 | 
			
		||||
			formData: url.Values{
 | 
			
		||||
				"phoneNumber": []string{"+1234567890"},
 | 
			
		||||
				"text":        []string{"1*2*3"},
 | 
			
		||||
			},
 | 
			
		||||
			expectedStatus: http.StatusInternalServerError,
 | 
			
		||||
			expectedBody:   "",
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			mockHandler := &httpmocks.MockRequestHandler{}
 | 
			
		||||
			mockRequestParser := &httpmocks.MockRequestParser{}
 | 
			
		||||
			mockEngine := &httpmocks.MockEngine{}
 | 
			
		||||
			tt.setupMocks(mockHandler, mockRequestParser, mockEngine)
 | 
			
		||||
 | 
			
		||||
			ash := NewATSessionHandler(mockHandler)
 | 
			
		||||
 | 
			
		||||
			req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(tt.formData.Encode()))
 | 
			
		||||
			req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
 | 
			
		||||
			w := httptest.NewRecorder()
 | 
			
		||||
 | 
			
		||||
			ash.ServeHTTP(w, req)
 | 
			
		||||
 | 
			
		||||
			if w.Code != tt.expectedStatus {
 | 
			
		||||
				t.Errorf("Expected status %d, got %d", tt.expectedStatus, w.Code)
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if tt.expectedBody != "" && w.Body.String() != tt.expectedBody {
 | 
			
		||||
				t.Errorf("Expected body %q, got %q", tt.expectedBody, w.Body.String())
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestATSessionHandler_Output(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name           string
 | 
			
		||||
		input          handlers.RequestSession
 | 
			
		||||
		expectedPrefix string
 | 
			
		||||
		expectedError  bool
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name: "Continue true",
 | 
			
		||||
			input: handlers.RequestSession{
 | 
			
		||||
				Continue: true,
 | 
			
		||||
				Engine: &httpmocks.MockEngine{
 | 
			
		||||
					FlushFunc: func(context.Context, io.Writer) (int, error) {
 | 
			
		||||
						return 0, nil
 | 
			
		||||
					},
 | 
			
		||||
				},
 | 
			
		||||
				Writer: &httpmocks.MockWriter{},
 | 
			
		||||
			},
 | 
			
		||||
			expectedPrefix: "CON ",
 | 
			
		||||
			expectedError:  false,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name: "Continue false",
 | 
			
		||||
			input: handlers.RequestSession{
 | 
			
		||||
				Continue: false,
 | 
			
		||||
				Engine: &httpmocks.MockEngine{
 | 
			
		||||
					FlushFunc: func(context.Context, io.Writer) (int, error) {
 | 
			
		||||
						return 0, nil
 | 
			
		||||
					},
 | 
			
		||||
				},
 | 
			
		||||
				Writer: &httpmocks.MockWriter{},
 | 
			
		||||
			},
 | 
			
		||||
			expectedPrefix: "END ",
 | 
			
		||||
			expectedError:  false,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name: "Flush error",
 | 
			
		||||
			input: handlers.RequestSession{
 | 
			
		||||
				Continue: true,
 | 
			
		||||
				Engine: &httpmocks.MockEngine{
 | 
			
		||||
					FlushFunc: func(context.Context, io.Writer) (int, error) {
 | 
			
		||||
						return 0, errors.New("write error")
 | 
			
		||||
					},
 | 
			
		||||
				},
 | 
			
		||||
				Writer: &httpmocks.MockWriter{},
 | 
			
		||||
			},
 | 
			
		||||
			expectedPrefix: "CON ",
 | 
			
		||||
			expectedError:  true,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			ash := &ATSessionHandler{}
 | 
			
		||||
			_, err := ash.Output(tt.input)
 | 
			
		||||
 | 
			
		||||
			if tt.expectedError && err == nil {
 | 
			
		||||
				t.Error("Expected an error, but got nil")
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if !tt.expectedError && err != nil {
 | 
			
		||||
				t.Errorf("Unexpected error: %v", err)
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			mw := tt.input.Writer.(*httpmocks.MockWriter)
 | 
			
		||||
			if !mw.WriteStringCalled {
 | 
			
		||||
				t.Error("WriteString was not called")
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if mw.WrittenString != tt.expectedPrefix {
 | 
			
		||||
				t.Errorf("Expected prefix %q, got %q", tt.expectedPrefix, mw.WrittenString)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestSessionHandler_ServeHTTP(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name           string
 | 
			
		||||
		sessionID      string
 | 
			
		||||
		input          []byte
 | 
			
		||||
		parserErr      error
 | 
			
		||||
		processErr     error
 | 
			
		||||
		outputErr      error
 | 
			
		||||
		resetErr       error
 | 
			
		||||
		expectedStatus int
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Success",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			expectedStatus: http.StatusOK,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Missing Session ID",
 | 
			
		||||
			sessionID:      "",
 | 
			
		||||
			parserErr:      handlers.ErrSessionMissing,
 | 
			
		||||
			expectedStatus: http.StatusBadRequest,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Process Error",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			processErr:     handlers.ErrStorage,
 | 
			
		||||
			expectedStatus: http.StatusInternalServerError,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Output Error",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			outputErr:      errors.New("output error"),
 | 
			
		||||
			expectedStatus: http.StatusOK,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Reset Error",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			resetErr:       errors.New("reset error"),
 | 
			
		||||
			expectedStatus: http.StatusOK,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			mockRequestParser := &httpmocks.MockRequestParser{
 | 
			
		||||
				GetSessionIdFunc: func(any) (string, error) {
 | 
			
		||||
					return tt.sessionID, tt.parserErr
 | 
			
		||||
				},
 | 
			
		||||
				GetInputFunc: func(any) ([]byte, error) {
 | 
			
		||||
					return tt.input, nil
 | 
			
		||||
				},
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			mockRequestHandler := &httpmocks.MockRequestHandler{
 | 
			
		||||
				ProcessFunc: func(rs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
					return rs, tt.processErr
 | 
			
		||||
				},
 | 
			
		||||
				OutputFunc: func(rs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
					return rs, tt.outputErr
 | 
			
		||||
				},
 | 
			
		||||
				ResetFunc: func(rs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
					return rs, tt.resetErr
 | 
			
		||||
				},
 | 
			
		||||
				GetRequestParserFunc: func() handlers.RequestParser {
 | 
			
		||||
					return mockRequestParser
 | 
			
		||||
				},
 | 
			
		||||
				GetConfigFunc: func() engine.Config {
 | 
			
		||||
					return engine.Config{}
 | 
			
		||||
				},
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			sessionHandler := ToSessionHandler(mockRequestHandler)
 | 
			
		||||
 | 
			
		||||
			req := httptest.NewRequest(http.MethodPost, "/", bytes.NewBuffer(tt.input))
 | 
			
		||||
			req.Header.Set("X-Vise-Session", tt.sessionID)
 | 
			
		||||
 | 
			
		||||
			rr := httptest.NewRecorder()
 | 
			
		||||
 | 
			
		||||
			sessionHandler.ServeHTTP(rr, req)
 | 
			
		||||
 | 
			
		||||
			if status := rr.Code; status != tt.expectedStatus {
 | 
			
		||||
				t.Errorf("handler returned wrong status code: got %v want %v",
 | 
			
		||||
					status, tt.expectedStatus)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestSessionHandler_writeError(t *testing.T) {
 | 
			
		||||
	handler := &SessionHandler{}
 | 
			
		||||
	mockWriter := &httpmocks.MockWriter{}
 | 
			
		||||
	err := errors.New("test error")
 | 
			
		||||
 | 
			
		||||
	handler.writeError(mockWriter, http.StatusBadRequest, err)
 | 
			
		||||
 | 
			
		||||
	if mockWriter.WrittenString != "" {
 | 
			
		||||
		t.Errorf("Expected empty body, got %s", mockWriter.WrittenString)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestDefaultRequestParser_GetSessionId(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name          string
 | 
			
		||||
		request       any
 | 
			
		||||
		expectedID    string
 | 
			
		||||
		expectedError error
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name: "Valid Session ID",
 | 
			
		||||
			request: func() *http.Request {
 | 
			
		||||
				req := httptest.NewRequest(http.MethodPost, "/", nil)
 | 
			
		||||
				req.Header.Set("X-Vise-Session", "123456")
 | 
			
		||||
				return req
 | 
			
		||||
			}(),
 | 
			
		||||
			expectedID:    "123456",
 | 
			
		||||
			expectedError: nil,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Missing Session ID",
 | 
			
		||||
			request:       httptest.NewRequest(http.MethodPost, "/", nil),
 | 
			
		||||
			expectedID:    "",
 | 
			
		||||
			expectedError: handlers.ErrSessionMissing,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Invalid Request Type",
 | 
			
		||||
			request:       invalidRequestType{},
 | 
			
		||||
			expectedID:    "",
 | 
			
		||||
			expectedError: handlers.ErrInvalidRequest,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	parser := &DefaultRequestParser{}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			id, err := parser.GetSessionId(tt.request)
 | 
			
		||||
 | 
			
		||||
			if id != tt.expectedID {
 | 
			
		||||
				t.Errorf("Expected session ID %s, got %s", tt.expectedID, id)
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if err != tt.expectedError {
 | 
			
		||||
				t.Errorf("Expected error %v, got %v", tt.expectedError, err)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestDefaultRequestParser_GetInput(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name          string
 | 
			
		||||
		request       any
 | 
			
		||||
		expectedInput []byte
 | 
			
		||||
		expectedError error
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name: "Valid Input",
 | 
			
		||||
			request: func() *http.Request {
 | 
			
		||||
				return httptest.NewRequest(http.MethodPost, "/", bytes.NewBufferString("test input"))
 | 
			
		||||
			}(),
 | 
			
		||||
			expectedInput: []byte("test input"),
 | 
			
		||||
			expectedError: nil,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Empty Input",
 | 
			
		||||
			request:       httptest.NewRequest(http.MethodPost, "/", nil),
 | 
			
		||||
			expectedInput: []byte{},
 | 
			
		||||
			expectedError: nil,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Invalid Request Type",
 | 
			
		||||
			request:       invalidRequestType{},
 | 
			
		||||
			expectedInput: nil,
 | 
			
		||||
			expectedError: handlers.ErrInvalidRequest,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name: "Read Error",
 | 
			
		||||
			request: func() *http.Request {
 | 
			
		||||
				return httptest.NewRequest(http.MethodPost, "/", &errorReader{})
 | 
			
		||||
			}(),
 | 
			
		||||
			expectedInput: nil,
 | 
			
		||||
			expectedError: errors.New("read error"),
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	parser := &DefaultRequestParser{}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			input, err := parser.GetInput(tt.request)
 | 
			
		||||
 | 
			
		||||
			if !bytes.Equal(input, tt.expectedInput) {
 | 
			
		||||
				t.Errorf("Expected input %s, got %s", tt.expectedInput, input)
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if err != tt.expectedError && (err == nil || err.Error() != tt.expectedError.Error()) {
 | 
			
		||||
				t.Errorf("Expected error %v, got %v", tt.expectedError, err)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										122
									
								
								internal/http/server.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										122
									
								
								internal/http/server.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,122 @@
 | 
			
		||||
package http
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strconv"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg = logging.NewVanilla().WithDomain("httpserver")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type DefaultRequestParser struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
func(rp *DefaultRequestParser) GetSessionId(rq any) (string, error) {
 | 
			
		||||
	rqv, ok := rq.(*http.Request)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return "", handlers.ErrInvalidRequest
 | 
			
		||||
	}
 | 
			
		||||
	v := rqv.Header.Get("X-Vise-Session")
 | 
			
		||||
	if v == "" {
 | 
			
		||||
		return "", handlers.ErrSessionMissing
 | 
			
		||||
	}
 | 
			
		||||
	return v, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(rp *DefaultRequestParser) GetInput(rq any) ([]byte, error) {
 | 
			
		||||
	rqv, ok := rq.(*http.Request)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, handlers.ErrInvalidRequest
 | 
			
		||||
	}
 | 
			
		||||
	defer rqv.Body.Close()
 | 
			
		||||
	v, err := ioutil.ReadAll(rqv.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return v, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type SessionHandler struct {
 | 
			
		||||
	handlers.RequestHandler
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func ToSessionHandler(h handlers.RequestHandler) *SessionHandler {
 | 
			
		||||
	return &SessionHandler{
 | 
			
		||||
		RequestHandler: h,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *SessionHandler) writeError(w http.ResponseWriter, code int, err error) {
 | 
			
		||||
	s := err.Error()
 | 
			
		||||
	w.Header().Set("Content-Length", strconv.Itoa(len(s)))
 | 
			
		||||
	w.WriteHeader(code)
 | 
			
		||||
	_, err = w.Write([]byte{})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.Errorf("error writing error!!", "err", err, "olderr", s)
 | 
			
		||||
		w.WriteHeader(500)
 | 
			
		||||
	}
 | 
			
		||||
	return 
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(f *SessionHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 | 
			
		||||
	var code int
 | 
			
		||||
	var err error
 | 
			
		||||
	var perr error
 | 
			
		||||
 | 
			
		||||
	rqs := handlers.RequestSession{
 | 
			
		||||
		Ctx: req.Context(),
 | 
			
		||||
		Writer: w,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rp := f.GetRequestParser()
 | 
			
		||||
	cfg := f.GetConfig()
 | 
			
		||||
	cfg.SessionId, err = rp.GetSessionId(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "header processing error", err)
 | 
			
		||||
		f.writeError(w, 400, err)
 | 
			
		||||
	}
 | 
			
		||||
	rqs.Config = cfg
 | 
			
		||||
	rqs.Input, err = rp.GetInput(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "header processing error", err)
 | 
			
		||||
		f.writeError(w, 400, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rqs, err = f.Process(rqs)
 | 
			
		||||
	switch err {
 | 
			
		||||
	case handlers.ErrStorage:
 | 
			
		||||
		code = 500
 | 
			
		||||
	case handlers.ErrEngineInit:
 | 
			
		||||
		code = 500
 | 
			
		||||
	case handlers.ErrEngineExec:
 | 
			
		||||
		code = 500
 | 
			
		||||
	default:
 | 
			
		||||
		code = 200
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if code != 200 {
 | 
			
		||||
		f.writeError(w, 500, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	w.WriteHeader(200)
 | 
			
		||||
	w.Header().Set("Content-Type", "text/plain")
 | 
			
		||||
	rqs, err = f.Output(rqs)
 | 
			
		||||
	rqs, perr = f.Reset(rqs)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		f.writeError(w, 500, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if perr != nil {
 | 
			
		||||
		f.writeError(w, 500, perr)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										10
									
								
								internal/models/accountresponse.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								internal/models/accountresponse.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,10 @@
 | 
			
		||||
package models
 | 
			
		||||
 | 
			
		||||
type AccountResponse struct {
 | 
			
		||||
	Ok          bool   `json:"ok"`
 | 
			
		||||
	Description string `json:"description"` // Include the description field
 | 
			
		||||
	Result      struct {
 | 
			
		||||
		PublicKey  string `json:"publicKey"`
 | 
			
		||||
		TrackingId string `json:"trackingId"`
 | 
			
		||||
	} `json:"result"`
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										12
									
								
								internal/models/balanceresponse.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								internal/models/balanceresponse.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,12 @@
 | 
			
		||||
package models
 | 
			
		||||
 | 
			
		||||
import "encoding/json"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
type BalanceResponse struct {
 | 
			
		||||
	Ok     bool `json:"ok"`
 | 
			
		||||
	Result struct {
 | 
			
		||||
		Balance string      `json:"balance"`
 | 
			
		||||
		Nonce   json.Number `json:"nonce"`
 | 
			
		||||
	} `json:"result"`
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										18
									
								
								internal/models/tokenresponse.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								internal/models/tokenresponse.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,18 @@
 | 
			
		||||
package models
 | 
			
		||||
 | 
			
		||||
type ApiResponse struct {
 | 
			
		||||
	OK          bool   `json:"ok"`
 | 
			
		||||
	Description string `json:"description"`
 | 
			
		||||
	Result      Result `json:"result"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Result struct {
 | 
			
		||||
	Holdings []Holding `json:"holdings"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Holding struct {
 | 
			
		||||
	ContractAddress string `json:"contractAddress"`
 | 
			
		||||
	TokenSymbol     string `json:"tokenSymbol"`
 | 
			
		||||
	TokenDecimals   string `json:"tokenDecimals"`
 | 
			
		||||
	Balance         string `json:"balance"`
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										26
									
								
								internal/models/trackstatusresponse.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								internal/models/trackstatusresponse.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,26 @@
 | 
			
		||||
package models
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
type Transaction struct {
 | 
			
		||||
	CreatedAt     time.Time   `json:"createdAt"`
 | 
			
		||||
	Status        string      `json:"status"`
 | 
			
		||||
	TransferValue json.Number `json:"transferValue"`
 | 
			
		||||
	TxHash        string      `json:"txHash"`
 | 
			
		||||
	TxType        string      `json:"txType"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type TrackStatusResponse struct {
 | 
			
		||||
	Ok     bool `json:"ok"`
 | 
			
		||||
	Result struct {
 | 
			
		||||
		Transaction struct {
 | 
			
		||||
			CreatedAt     time.Time   `json:"createdAt"`
 | 
			
		||||
			Status        string      `json:"status"`
 | 
			
		||||
			TransferValue json.Number `json:"transferValue"`
 | 
			
		||||
			TxHash        string      `json:"txHash"`
 | 
			
		||||
			TxType        string      `json:"txType"`
 | 
			
		||||
		}
 | 
			
		||||
	} `json:"result"`
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										15
									
								
								internal/models/vouchersresponse.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								internal/models/vouchersresponse.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,15 @@
 | 
			
		||||
package models
 | 
			
		||||
 | 
			
		||||
// VoucherHoldingResponse represents a single voucher holding
 | 
			
		||||
type VoucherHoldingResponse struct {
 | 
			
		||||
	Ok          bool   `json:"ok"`
 | 
			
		||||
	Description string `json:"description"`
 | 
			
		||||
	Result      struct {
 | 
			
		||||
		Holdings []struct {
 | 
			
		||||
			ContractAddress string `json:"contractAddress"`
 | 
			
		||||
			TokenSymbol     string `json:"tokenSymbol"`
 | 
			
		||||
			TokenDecimals   string `json:"tokenDecimals"`
 | 
			
		||||
			Balance         string `json:"balance"`
 | 
			
		||||
		} `json:"holdings"`
 | 
			
		||||
	} `json:"result"`
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										43
									
								
								internal/storage/db.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								internal/storage/db.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,43 @@
 | 
			
		||||
package storage
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/db"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	DATATYPE_USERSUB = 64
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type SubPrefixDb struct {
 | 
			
		||||
	store db.Db
 | 
			
		||||
	pfx   []byte
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewSubPrefixDb(store db.Db, pfx []byte) *SubPrefixDb {
 | 
			
		||||
	return &SubPrefixDb{
 | 
			
		||||
		store: store,
 | 
			
		||||
		pfx:   pfx,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *SubPrefixDb) toKey(k []byte) []byte {
 | 
			
		||||
	return append(s.pfx, k...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *SubPrefixDb) Get(ctx context.Context, key []byte) ([]byte, error) {
 | 
			
		||||
	s.store.SetPrefix(DATATYPE_USERSUB)
 | 
			
		||||
	key = s.toKey(key)
 | 
			
		||||
	v, err := s.store.Get(ctx, key)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return v, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (s *SubPrefixDb) Put(ctx context.Context, key []byte, val []byte) error {
 | 
			
		||||
	s.store.SetPrefix(DATATYPE_USERSUB)
 | 
			
		||||
	key = s.toKey(key)
 | 
			
		||||
	return s.store.Put(ctx, key, val)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										54
									
								
								internal/storage/db_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								internal/storage/db_test.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,54 @@
 | 
			
		||||
package storage
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"testing"
 | 
			
		||||
 | 
			
		||||
	memdb "git.defalsify.org/vise.git/db/mem"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func TestSubPrefix(t *testing.T) {
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	db := memdb.NewMemDb()
 | 
			
		||||
	err := db.Connect(ctx, "")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	sdba := NewSubPrefixDb(db, []byte("tinkywinky"))
 | 
			
		||||
	err = sdba.Put(ctx, []byte("foo"), []byte("dipsy"))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	r, err := sdba.Get(ctx, []byte("foo"))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	if !bytes.Equal(r, []byte("dipsy")) {
 | 
			
		||||
		t.Fatalf("expected 'dipsy', got %s", r)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	sdbb := NewSubPrefixDb(db, []byte("lala"))
 | 
			
		||||
	r, err = sdbb.Get(ctx, []byte("foo"))
 | 
			
		||||
	if err == nil {
 | 
			
		||||
		t.Fatal("expected not found")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err = sdbb.Put(ctx, []byte("foo"), []byte("pu"))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	r, err = sdbb.Get(ctx, []byte("foo"))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Fatal(err)
 | 
			
		||||
	}
 | 
			
		||||
	if !bytes.Equal(r, []byte("pu")) {
 | 
			
		||||
		t.Fatalf("expected 'pu', got %s", r)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	r, err = sdba.Get(ctx, []byte("foo"))
 | 
			
		||||
	if !bytes.Equal(r, []byte("dipsy")) {
 | 
			
		||||
		t.Fatalf("expected 'dipsy', got %s", r)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										116
									
								
								internal/storage/gdbm.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										116
									
								
								internal/storage/gdbm.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,116 @@
 | 
			
		||||
package storage
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/db"
 | 
			
		||||
	"git.defalsify.org/vise.git/lang"
 | 
			
		||||
	gdbmdb "git.defalsify.org/vise.git/db/gdbm"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	dbC map[string]chan db.Db
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type ThreadGdbmDb struct {
 | 
			
		||||
	db db.Db
 | 
			
		||||
	connStr string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewThreadGdbmDb() *ThreadGdbmDb {
 | 
			
		||||
	if dbC == nil {
 | 
			
		||||
		dbC = make(map[string]chan db.Db)
 | 
			
		||||
	}
 | 
			
		||||
	return &ThreadGdbmDb{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) Connect(ctx context.Context, connStr string) error {
 | 
			
		||||
	var ok bool
 | 
			
		||||
	_, ok = dbC[connStr]
 | 
			
		||||
	if ok {
 | 
			
		||||
		logg.WarnCtxf(ctx, "already registered thread gdbm, skipping", "connStr", connStr)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	gdb := gdbmdb.NewGdbmDb()
 | 
			
		||||
	err := gdb.Connect(ctx, connStr)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	dbC[connStr] = make(chan db.Db, 1)
 | 
			
		||||
	dbC[connStr]<- gdb
 | 
			
		||||
	tdb.connStr = connStr
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) reserve() {
 | 
			
		||||
	if tdb.db == nil {
 | 
			
		||||
		tdb.db = <-dbC[tdb.connStr]
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) release() {
 | 
			
		||||
	if tdb.db == nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	dbC[tdb.connStr] <- tdb.db
 | 
			
		||||
	tdb.db = nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) SetPrefix(pfx uint8) {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	tdb.db.SetPrefix(pfx)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) SetSession(sessionId string) {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	tdb.db.SetSession(sessionId)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) SetLanguage(lng *lang.Language) {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	tdb.db.SetLanguage(lng)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) Safe() bool {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	v := tdb.db.Safe()
 | 
			
		||||
	tdb.release()
 | 
			
		||||
	return v
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) Prefix() uint8 {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	v := tdb.db.Prefix()
 | 
			
		||||
	tdb.release()
 | 
			
		||||
	return v
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) SetLock(typ uint8, locked bool) error {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	err := tdb.db.SetLock(typ, locked)
 | 
			
		||||
	tdb.release()
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) Put(ctx context.Context, key []byte, val []byte) error {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	err := tdb.db.Put(ctx, key, val)
 | 
			
		||||
	tdb.release()
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) Get(ctx context.Context, key []byte) ([]byte, error) {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	v, err := tdb.db.Get(ctx, key)
 | 
			
		||||
	tdb.release()
 | 
			
		||||
	return v, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func(tdb *ThreadGdbmDb) Close() error {
 | 
			
		||||
	tdb.reserve()
 | 
			
		||||
	close(dbC[tdb.connStr])
 | 
			
		||||
	delete(dbC, tdb.connStr)
 | 
			
		||||
	err := tdb.db.Close()
 | 
			
		||||
	tdb.db = nil
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										44
									
								
								internal/storage/storage.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										44
									
								
								internal/storage/storage.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,44 @@
 | 
			
		||||
package storage
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"git.defalsify.org/vise.git/db"
 | 
			
		||||
	"git.defalsify.org/vise.git/persist"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Storage struct {
 | 
			
		||||
	Persister *persist.Persister
 | 
			
		||||
	UserdataDb db.Db	
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type StorageProvider interface {
 | 
			
		||||
	Get(sessionId string) (*Storage, error)
 | 
			
		||||
	Put(sessionId string, storage *Storage) error
 | 
			
		||||
	Close() error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type SimpleStorageProvider struct {
 | 
			
		||||
	*Storage
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewSimpleStorageProvider(stateStore db.Db, userdataStore db.Db) StorageProvider {
 | 
			
		||||
	pe := persist.NewPersister(stateStore)
 | 
			
		||||
	pe = pe.WithFlush()
 | 
			
		||||
	return &SimpleStorageProvider{
 | 
			
		||||
		Storage: &Storage{
 | 
			
		||||
			Persister: pe,
 | 
			
		||||
			UserdataDb: userdataStore,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *SimpleStorageProvider) Get(sessionId string) (*Storage, error) {
 | 
			
		||||
	return p.Storage, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *SimpleStorageProvider) Put(sessionId string, storage *Storage) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (p *SimpleStorageProvider) Close() error {
 | 
			
		||||
	return p.Storage.UserdataDb.Close()
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										152
									
								
								internal/storage/storageservice.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										152
									
								
								internal/storage/storageservice.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,152 @@
 | 
			
		||||
package storage
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/db"
 | 
			
		||||
	fsdb "git.defalsify.org/vise.git/db/fs"
 | 
			
		||||
	"git.defalsify.org/vise.git/db/postgres"
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
	"git.defalsify.org/vise.git/persist"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/initializers"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg = logging.NewVanilla().WithDomain("storage")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type StorageService interface {
 | 
			
		||||
	GetPersister(ctx context.Context) (*persist.Persister, error)
 | 
			
		||||
	GetUserdataDb(ctx context.Context) db.Db
 | 
			
		||||
	GetResource(ctx context.Context) (resource.Resource, error)
 | 
			
		||||
	EnsureDbDir() error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MenuStorageService struct {
 | 
			
		||||
	dbDir         string
 | 
			
		||||
	resourceDir   string
 | 
			
		||||
	resourceStore db.Db
 | 
			
		||||
	stateStore    db.Db
 | 
			
		||||
	userDataStore db.Db
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func buildConnStr() string {
 | 
			
		||||
	host := initializers.GetEnv("DB_HOST", "localhost")
 | 
			
		||||
	user := initializers.GetEnv("DB_USER", "postgres")
 | 
			
		||||
	password := initializers.GetEnv("DB_PASSWORD", "")
 | 
			
		||||
	dbName := initializers.GetEnv("DB_NAME", "")
 | 
			
		||||
	port := initializers.GetEnv("DB_PORT", "5432")
 | 
			
		||||
 | 
			
		||||
	return fmt.Sprintf(
 | 
			
		||||
		"postgres://%s:%s@%s:%s/%s",
 | 
			
		||||
		user, password, host, port, dbName,
 | 
			
		||||
	)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewMenuStorageService(dbDir string, resourceDir string) *MenuStorageService {
 | 
			
		||||
	return &MenuStorageService{
 | 
			
		||||
		dbDir:       dbDir,
 | 
			
		||||
		resourceDir: resourceDir,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ms *MenuStorageService) getOrCreateDb(ctx context.Context, existingDb db.Db, fileName string) (db.Db, error) {
 | 
			
		||||
	database, ok := ctx.Value("Database").(string)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, fmt.Errorf("failed to select the database")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if existingDb != nil {
 | 
			
		||||
		return existingDb, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var newDb db.Db
 | 
			
		||||
	var err error
 | 
			
		||||
 | 
			
		||||
	if database == "postgres" {
 | 
			
		||||
		newDb = postgres.NewPgDb()
 | 
			
		||||
		connStr := buildConnStr()
 | 
			
		||||
		err = newDb.Connect(ctx, connStr)
 | 
			
		||||
	} else {
 | 
			
		||||
		newDb = NewThreadGdbmDb()
 | 
			
		||||
		storeFile := path.Join(ms.dbDir, fileName)
 | 
			
		||||
		err = newDb.Connect(ctx, storeFile)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return newDb, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ms *MenuStorageService) GetPersister(ctx context.Context) (*persist.Persister, error) {
 | 
			
		||||
	stateStore, err := ms.GetStateStore(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	pr := persist.NewPersister(stateStore)
 | 
			
		||||
	logg.TraceCtxf(ctx, "menu storage service", "persist", pr, "store", stateStore)
 | 
			
		||||
	return pr, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ms *MenuStorageService) GetUserdataDb(ctx context.Context) (db.Db, error) {
 | 
			
		||||
	if ms.userDataStore != nil {
 | 
			
		||||
		return ms.userDataStore, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	userDataStore, err := ms.getOrCreateDb(ctx, ms.userDataStore, "userdata.gdbm")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	ms.userDataStore = userDataStore
 | 
			
		||||
	return ms.userDataStore, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ms *MenuStorageService) GetResource(ctx context.Context) (resource.Resource, error) {
 | 
			
		||||
	ms.resourceStore = fsdb.NewFsDb()
 | 
			
		||||
	err := ms.resourceStore.Connect(ctx, ms.resourceDir)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	rfs := resource.NewDbResource(ms.resourceStore)
 | 
			
		||||
	return rfs, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ms *MenuStorageService) GetStateStore(ctx context.Context) (db.Db, error) {
 | 
			
		||||
	if ms.stateStore != nil {
 | 
			
		||||
		return ms.stateStore, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	stateStore, err := ms.getOrCreateDb(ctx, ms.stateStore, "state.gdbm")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	ms.stateStore = stateStore
 | 
			
		||||
	return ms.stateStore, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ms *MenuStorageService) EnsureDbDir() error {
 | 
			
		||||
	err := os.MkdirAll(ms.dbDir, 0700)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return fmt.Errorf("state dir create exited with error: %v\n", err)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (ms *MenuStorageService) Close() error {
 | 
			
		||||
	errA := ms.stateStore.Close()
 | 
			
		||||
	errB := ms.userDataStore.Close()
 | 
			
		||||
	errC := ms.resourceStore.Close()
 | 
			
		||||
	if errA != nil || errB != nil || errC != nil {
 | 
			
		||||
		return fmt.Errorf("%v %v %v", errA, errB, errC)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										124
									
								
								internal/testutil/TestEngine.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										124
									
								
								internal/testutil/TestEngine.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,124 @@
 | 
			
		||||
package testutil
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/logging"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/storage"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/testutil/testservice"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/testutil/testtag"
 | 
			
		||||
	testdataloader "github.com/peteole/testdata-loader"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	baseDir   = testdataloader.GetBasePath()
 | 
			
		||||
	logg      = logging.NewVanilla()
 | 
			
		||||
	scriptDir = path.Join(baseDir, "services", "registration")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func TestEngine(sessionId string) (engine.Engine, func(), chan bool) {
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	ctx = context.WithValue(ctx, "SessionId", sessionId)
 | 
			
		||||
	ctx = context.WithValue(ctx, "Database", "gdbm")
 | 
			
		||||
	pfp := path.Join(scriptDir, "pp.csv")
 | 
			
		||||
 | 
			
		||||
	var eventChannel = make(chan bool)
 | 
			
		||||
 | 
			
		||||
	cfg := engine.Config{
 | 
			
		||||
		Root:       "root",
 | 
			
		||||
		SessionId:  sessionId,
 | 
			
		||||
		OutputSize: uint32(160),
 | 
			
		||||
		FlagCount:  uint32(128),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	dbDir := ".test_state"
 | 
			
		||||
	resourceDir := scriptDir
 | 
			
		||||
	menuStorageService := storage.NewMenuStorageService(dbDir, resourceDir)
 | 
			
		||||
 | 
			
		||||
	err := menuStorageService.EnsureDbDir()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rs, err := menuStorageService.GetResource(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	pe, err := menuStorageService.GetPersister(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	userDataStore, err := menuStorageService.GetUserdataDb(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	dbResource, ok := rs.(*resource.DbResource)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	lhs, err := handlers.NewLocalHandlerService(pfp, true, dbResource, cfg, rs)
 | 
			
		||||
	lhs.SetDataStore(&userDataStore)
 | 
			
		||||
	lhs.SetPersister(pe)
 | 
			
		||||
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if testtag.AccountService == nil {
 | 
			
		||||
		testtag.AccountService = &server.AccountService{}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	switch testtag.AccountService.(type) {
 | 
			
		||||
	case *testservice.TestAccountService:
 | 
			
		||||
		go func() {
 | 
			
		||||
			eventChannel <- false
 | 
			
		||||
		}()
 | 
			
		||||
	case *server.AccountService:
 | 
			
		||||
		go func() {
 | 
			
		||||
			time.Sleep(5 * time.Second) // Wait for 5 seconds
 | 
			
		||||
			eventChannel <- true
 | 
			
		||||
		}()
 | 
			
		||||
	default:
 | 
			
		||||
		panic("Unknown account service type")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	hl, err := lhs.GetHandler(testtag.AccountService)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		fmt.Fprintf(os.Stderr, err.Error())
 | 
			
		||||
		os.Exit(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	en := lhs.GetEngine()
 | 
			
		||||
	en = en.WithFirst(hl.Init)
 | 
			
		||||
	cleanFn := func() {
 | 
			
		||||
		err := en.Finish()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logg.Errorf(err.Error())
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		err = menuStorageService.Close()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logg.Errorf(err.Error())
 | 
			
		||||
		}
 | 
			
		||||
		logg.Infof("testengine storage closed")
 | 
			
		||||
	}
 | 
			
		||||
	return en, cleanFn, eventChannel
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										59
									
								
								internal/testutil/mocks/dbmock.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										59
									
								
								internal/testutil/mocks/dbmock.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,59 @@
 | 
			
		||||
package mocks
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/lang"
 | 
			
		||||
	"github.com/stretchr/testify/mock"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type MockDb struct {
 | 
			
		||||
	mock.Mock
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) SetPrefix(prefix uint8) {
 | 
			
		||||
	m.Called(prefix)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) Prefix() uint8 {
 | 
			
		||||
	args := m.Called()
 | 
			
		||||
	return args.Get(0).(uint8)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) Safe() bool {
 | 
			
		||||
	args := m.Called()
 | 
			
		||||
	return args.Get(0).(bool)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) SetLanguage(language *lang.Language) {
 | 
			
		||||
	m.Called(language)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) SetLock(uint8, bool) error {
 | 
			
		||||
	args := m.Called()
 | 
			
		||||
	return args.Error(0)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) Connect(ctx context.Context, connectionStr string) error {
 | 
			
		||||
	args := m.Called(ctx, connectionStr)
 | 
			
		||||
	return args.Error(0)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) SetSession(sessionId string) {
 | 
			
		||||
	m.Called(sessionId)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) Put(ctx context.Context, key, value []byte) error {
 | 
			
		||||
	args := m.Called(ctx, key, value)
 | 
			
		||||
	return args.Error(0)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) Get(ctx context.Context, key []byte) ([]byte, error) {
 | 
			
		||||
	args := m.Called(ctx, key)
 | 
			
		||||
	return nil, args.Error(0)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockDb) Close() error {
 | 
			
		||||
	args := m.Called(nil)
 | 
			
		||||
	return args.Error(0)
 | 
			
		||||
}
 | 
			
		||||
@ -10,7 +10,7 @@ type MockEngine struct {
 | 
			
		||||
	InitFunc   func(context.Context) (bool, error)
 | 
			
		||||
	ExecFunc   func(context.Context, []byte) (bool, error)
 | 
			
		||||
	FlushFunc  func(context.Context, io.Writer) (int, error)
 | 
			
		||||
	FinishFunc func(context.Context) error
 | 
			
		||||
	FinishFunc func() error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockEngine) Init(ctx context.Context) (bool, error) {
 | 
			
		||||
@ -25,6 +25,6 @@ func (m *MockEngine) Flush(ctx context.Context, w io.Writer) (int, error) {
 | 
			
		||||
	return m.FlushFunc(ctx, w)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockEngine) Finish(ctx context.Context) error {
 | 
			
		||||
	return m.FinishFunc(ctx)
 | 
			
		||||
func (m *MockEngine) Finish() error {
 | 
			
		||||
	return m.FinishFunc()
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										47
									
								
								internal/testutil/mocks/httpmocks/requesthandlermock.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								internal/testutil/mocks/httpmocks/requesthandlermock.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,47 @@
 | 
			
		||||
package httpmocks
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"git.defalsify.org/vise.git/engine"
 | 
			
		||||
	"git.defalsify.org/vise.git/persist"
 | 
			
		||||
	"git.defalsify.org/vise.git/resource"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// MockRequestHandler implements handlers.RequestHandler interface for testing
 | 
			
		||||
type MockRequestHandler struct {
 | 
			
		||||
	ProcessFunc          func(handlers.RequestSession) (handlers.RequestSession, error)
 | 
			
		||||
	GetConfigFunc        func() engine.Config
 | 
			
		||||
	GetEngineFunc        func(cfg engine.Config, rs resource.Resource, pe *persist.Persister) engine.Engine
 | 
			
		||||
	OutputFunc           func(rs handlers.RequestSession) (handlers.RequestSession, error)
 | 
			
		||||
	ResetFunc            func(rs handlers.RequestSession) (handlers.RequestSession, error)
 | 
			
		||||
	ShutdownFunc         func()
 | 
			
		||||
	GetRequestParserFunc func() handlers.RequestParser
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestHandler) Process(rqs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
	return m.ProcessFunc(rqs)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestHandler) GetConfig() engine.Config {
 | 
			
		||||
	return m.GetConfigFunc()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestHandler) GetEngine(cfg engine.Config, rs resource.Resource, pe *persist.Persister) engine.Engine {
 | 
			
		||||
	return m.GetEngineFunc(cfg, rs, pe)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestHandler) Output(rs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
	return m.OutputFunc(rs)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestHandler) Reset(rs handlers.RequestSession) (handlers.RequestSession, error) {
 | 
			
		||||
	return m.ResetFunc(rs)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestHandler) Shutdown() {
 | 
			
		||||
	m.ShutdownFunc()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestHandler) GetRequestParser() handlers.RequestParser {
 | 
			
		||||
	return m.GetRequestParserFunc()
 | 
			
		||||
}
 | 
			
		||||
@ -1,14 +1,12 @@
 | 
			
		||||
package httpmocks
 | 
			
		||||
 | 
			
		||||
import "context"
 | 
			
		||||
 | 
			
		||||
// MockRequestParser implements the handlers.RequestParser interface for testing
 | 
			
		||||
type MockRequestParser struct {
 | 
			
		||||
	GetSessionIdFunc func(any) (string, error)
 | 
			
		||||
	GetInputFunc     func(any) ([]byte, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockRequestParser) GetSessionId(ctx context.Context, rq any) (string, error) {
 | 
			
		||||
func (m *MockRequestParser) GetSessionId(rq any) (string, error) {
 | 
			
		||||
	return m.GetSessionIdFunc(rq)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@ -22,4 +22,4 @@ func (m *MockWriter) Header() http.Header {
 | 
			
		||||
	return http.Header{}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockWriter) WriteHeader(statusCode int) {}
 | 
			
		||||
func (m *MockWriter) WriteHeader(statusCode int) {}
 | 
			
		||||
							
								
								
									
										40
									
								
								internal/testutil/mocks/servicemock.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										40
									
								
								internal/testutil/mocks/servicemock.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,40 @@
 | 
			
		||||
package mocks
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/models"
 | 
			
		||||
	"github.com/grassrootseconomics/eth-custodial/pkg/api"
 | 
			
		||||
	"github.com/stretchr/testify/mock"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// MockAccountService implements AccountServiceInterface for testing
 | 
			
		||||
type MockAccountService struct {
 | 
			
		||||
	mock.Mock
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockAccountService) CreateAccount(ctx context.Context) (*api.OKResponse, error) {
 | 
			
		||||
	args := m.Called()
 | 
			
		||||
	return args.Get(0).(*api.OKResponse), args.Error(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockAccountService) CheckBalance(ctx context.Context, publicKey string) (*models.BalanceResponse, error) {
 | 
			
		||||
	args := m.Called(publicKey)
 | 
			
		||||
	return args.Get(0).(*models.BalanceResponse), args.Error(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockAccountService) CheckAccountStatus(ctx context.Context, trackingId string) (*models.TrackStatusResponse, error) {
 | 
			
		||||
	args := m.Called(trackingId)
 | 
			
		||||
	return args.Get(0).(*models.TrackStatusResponse), args.Error(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockAccountService) TrackAccountStatus(ctx context.Context,publicKey string) (*api.OKResponse, error) {
 | 
			
		||||
	args := m.Called(publicKey)
 | 
			
		||||
	return args.Get(0).(*api.OKResponse), args.Error(1)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
func (m *MockAccountService) FetchVouchers(ctx context.Context, publicKey string) (*models.VoucherHoldingResponse, error) {
 | 
			
		||||
	args := m.Called(publicKey)
 | 
			
		||||
	return args.Get(0).(*models.VoucherHoldingResponse), args.Error(1)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										24
									
								
								internal/testutil/mocks/userdbmock.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										24
									
								
								internal/testutil/mocks/userdbmock.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,24 @@
 | 
			
		||||
package mocks
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"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) 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)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *MockUserDataStore) WriteEntry(ctx context.Context, sessionId string, typ utils.DataTyp, value []byte) error {
 | 
			
		||||
	args := m.Called(ctx, sessionId, typ, value)
 | 
			
		||||
	return args.Error(0)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										101
									
								
								internal/testutil/testservice/TestAccountService.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										101
									
								
								internal/testutil/testservice/TestAccountService.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,101 @@
 | 
			
		||||
package testservice
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/models"
 | 
			
		||||
	"github.com/grassrootseconomics/eth-custodial/pkg/api"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type TestAccountService struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (tas *TestAccountService) CreateAccount(ctx context.Context) (*api.OKResponse, error) {
 | 
			
		||||
	return &api.OKResponse{
 | 
			
		||||
		Ok:          true,
 | 
			
		||||
		Description: "Account creation succeeded",
 | 
			
		||||
		Result: map[string]any{
 | 
			
		||||
			"trackingId": "075ccc86-f6ef-4d33-97d5-e91cfb37aa0d",
 | 
			
		||||
			"publicKey":  "0x623EFAFa8868df4B934dd12a8B26CB3Dd75A7AdD",
 | 
			
		||||
		},
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (tas *TestAccountService) CheckBalance(ctx context.Context, publicKey string) (*models.BalanceResponse, error) {
 | 
			
		||||
	balanceResponse := &models.BalanceResponse{
 | 
			
		||||
		Ok: true,
 | 
			
		||||
		Result: struct {
 | 
			
		||||
			Balance string      `json:"balance"`
 | 
			
		||||
			Nonce   json.Number `json:"nonce"`
 | 
			
		||||
		}{
 | 
			
		||||
			Balance: "0.003 CELO",
 | 
			
		||||
			Nonce:   json.Number("0"),
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return balanceResponse, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (tas *TestAccountService) CheckAccountStatus(ctx context.Context, trackingId string) (*models.TrackStatusResponse, error) {
 | 
			
		||||
	trackResponse := &models.TrackStatusResponse{
 | 
			
		||||
		Ok: true,
 | 
			
		||||
		Result: struct {
 | 
			
		||||
			Transaction struct {
 | 
			
		||||
				CreatedAt     time.Time   "json:\"createdAt\""
 | 
			
		||||
				Status        string      "json:\"status\""
 | 
			
		||||
				TransferValue json.Number "json:\"transferValue\""
 | 
			
		||||
				TxHash        string      "json:\"txHash\""
 | 
			
		||||
				TxType        string      "json:\"txType\""
 | 
			
		||||
			}
 | 
			
		||||
		}{
 | 
			
		||||
			Transaction: models.Transaction{
 | 
			
		||||
				CreatedAt:     time.Now(),
 | 
			
		||||
				Status:        "SUCCESS",
 | 
			
		||||
				TransferValue: json.Number("0.5"),
 | 
			
		||||
				TxHash:        "0x123abc456def",
 | 
			
		||||
				TxType:        "transfer",
 | 
			
		||||
			},
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
	return trackResponse, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (tas *TestAccountService) TrackAccountStatus(ctx context.Context, publicKey string) (*api.OKResponse, error) {
 | 
			
		||||
	return &api.OKResponse{
 | 
			
		||||
		Ok:          true,
 | 
			
		||||
		Description: "Account creation succeeded",
 | 
			
		||||
		Result: map[string]any{
 | 
			
		||||
			"active": true,
 | 
			
		||||
		},
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (tas *TestAccountService) FetchVouchers(ctx context.Context, publicKey string) (*models.VoucherHoldingResponse, error) {
 | 
			
		||||
	return &models.VoucherHoldingResponse{
 | 
			
		||||
		Ok: true,
 | 
			
		||||
		Result: struct {
 | 
			
		||||
			Holdings []struct {
 | 
			
		||||
				ContractAddress string `json:"contractAddress"`
 | 
			
		||||
				TokenSymbol     string `json:"tokenSymbol"`
 | 
			
		||||
				TokenDecimals   string `json:"tokenDecimals"`
 | 
			
		||||
				Balance         string `json:"balance"`
 | 
			
		||||
			} `json:"holdings"`
 | 
			
		||||
		}{
 | 
			
		||||
			Holdings: []struct {
 | 
			
		||||
				ContractAddress string `json:"contractAddress"`
 | 
			
		||||
				TokenSymbol     string `json:"tokenSymbol"`
 | 
			
		||||
				TokenDecimals   string `json:"tokenDecimals"`
 | 
			
		||||
				Balance         string `json:"balance"`
 | 
			
		||||
			}{
 | 
			
		||||
				{
 | 
			
		||||
					ContractAddress: "0x6CC75A06ac72eB4Db2eE22F781F5D100d8ec03ee",
 | 
			
		||||
					TokenSymbol:     "SRF",
 | 
			
		||||
					TokenDecimals:   "6",
 | 
			
		||||
					Balance:         "2745987",
 | 
			
		||||
				},
 | 
			
		||||
			},
 | 
			
		||||
		},
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										12
									
								
								internal/testutil/testtag/offlinetest.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								internal/testutil/testtag/offlinetest.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,12 @@
 | 
			
		||||
// +build !online
 | 
			
		||||
 | 
			
		||||
package testtag
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
	accountservice "git.grassecon.net/urdt/ussd/internal/testutil/testservice"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	AccountService server.AccountServiceInterface = &accountservice.TestAccountService{}
 | 
			
		||||
)
 | 
			
		||||
							
								
								
									
										9
									
								
								internal/testutil/testtag/onlinetest.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								internal/testutil/testtag/onlinetest.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,9 @@
 | 
			
		||||
// +build online
 | 
			
		||||
 | 
			
		||||
package testtag
 | 
			
		||||
 | 
			
		||||
import "git.grassecon.net/urdt/ussd/internal/handlers/server"
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	AccountService server.AccountServiceInterface
 | 
			
		||||
)
 | 
			
		||||
							
								
								
									
										35
									
								
								internal/utils/age.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								internal/utils/age.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,35 @@
 | 
			
		||||
package utils
 | 
			
		||||
 | 
			
		||||
import "time"
 | 
			
		||||
 | 
			
		||||
// CalculateAge calculates the age based on a given birthdate and the current date in the format dd/mm/yy
 | 
			
		||||
// It adjusts for cases where the current date is before the birthday in the current year.
 | 
			
		||||
func CalculateAge(birthdate, today time.Time) int {
 | 
			
		||||
	today = today.In(birthdate.Location())
 | 
			
		||||
	ty, tm, td := today.Date()
 | 
			
		||||
	today = time.Date(ty, tm, td, 0, 0, 0, 0, time.UTC)
 | 
			
		||||
	by, bm, bd := birthdate.Date()
 | 
			
		||||
	birthdate = time.Date(by, bm, bd, 0, 0, 0, 0, time.UTC)
 | 
			
		||||
	if today.Before(birthdate) {
 | 
			
		||||
		return 0
 | 
			
		||||
	}
 | 
			
		||||
	age := ty - by
 | 
			
		||||
	anniversary := birthdate.AddDate(age, 0, 0)
 | 
			
		||||
	if anniversary.After(today) {
 | 
			
		||||
		age--
 | 
			
		||||
	}
 | 
			
		||||
	return age
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// CalculateAgeWithYOB calculates the age based on the given year of birth (YOB).
 | 
			
		||||
// It subtracts the YOB from the current year to determine the age.
 | 
			
		||||
//
 | 
			
		||||
// Parameters:
 | 
			
		||||
//   yob: The year of birth as an integer.
 | 
			
		||||
//
 | 
			
		||||
// Returns:
 | 
			
		||||
//   The calculated age as an integer.
 | 
			
		||||
func CalculateAgeWithYOB(yob int) int {
 | 
			
		||||
    currentYear := time.Now().Year()
 | 
			
		||||
    return currentYear - yob
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										43
									
								
								internal/utils/db.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								internal/utils/db.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,43 @@
 | 
			
		||||
package utils
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/binary"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type DataTyp uint16
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	DATA_ACCOUNT DataTyp = iota
 | 
			
		||||
	DATA_ACCOUNT_CREATED
 | 
			
		||||
	DATA_TRACKING_ID
 | 
			
		||||
	DATA_PUBLIC_KEY
 | 
			
		||||
	DATA_CUSTODIAL_ID
 | 
			
		||||
	DATA_ACCOUNT_PIN
 | 
			
		||||
	DATA_ACCOUNT_STATUS
 | 
			
		||||
	DATA_FIRST_NAME
 | 
			
		||||
	DATA_FAMILY_NAME
 | 
			
		||||
	DATA_YOB
 | 
			
		||||
	DATA_LOCATION
 | 
			
		||||
	DATA_GENDER
 | 
			
		||||
	DATA_OFFERINGS
 | 
			
		||||
	DATA_RECIPIENT
 | 
			
		||||
	DATA_AMOUNT
 | 
			
		||||
	DATA_TEMPORARY_PIN
 | 
			
		||||
	DATA_VOUCHER_LIST
 | 
			
		||||
	DATA_TEMPORARY_SYM
 | 
			
		||||
	DATA_ACTIVE_SYM
 | 
			
		||||
	DATA_TEMPORARY_BAL
 | 
			
		||||
	DATA_ACTIVE_BAL
 | 
			
		||||
	DATA_PUBLIC_KEY_REVERSE
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func typToBytes(typ DataTyp) []byte {
 | 
			
		||||
	var b [2]byte
 | 
			
		||||
	binary.BigEndian.PutUint16(b[:], uint16(typ))
 | 
			
		||||
	return b[:]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func PackKey(typ DataTyp, data []byte) []byte {
 | 
			
		||||
	v := typToBytes(typ)
 | 
			
		||||
	return append(v, data...)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										11
									
								
								internal/utils/isocode.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								internal/utils/isocode.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,11 @@
 | 
			
		||||
package utils
 | 
			
		||||
 | 
			
		||||
var isoCodes = map[string]bool{
 | 
			
		||||
	"eng": true, // English
 | 
			
		||||
	"swa": true, // Swahili
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func IsValidISO639(code string) bool {
 | 
			
		||||
	return isoCodes[code]
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										32
									
								
								internal/utils/userStore.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								internal/utils/userStore.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,32 @@
 | 
			
		||||
package utils
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"git.defalsify.org/vise.git/db"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type DataStore interface {
 | 
			
		||||
	db.Db
 | 
			
		||||
	ReadEntry(ctx context.Context, sessionId string, typ DataTyp) ([]byte, error)
 | 
			
		||||
	WriteEntry(ctx context.Context, sessionId string, typ DataTyp, value []byte) error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type UserDataStore struct {
 | 
			
		||||
	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.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.SetPrefix(db.DATATYPE_USERDATA)
 | 
			
		||||
	store.SetSession(sessionId)
 | 
			
		||||
	k := PackKey(typ, []byte(sessionId))
 | 
			
		||||
	return store.Put(ctx, k, value)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										460
									
								
								menutraversal_test/group_test.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										460
									
								
								menutraversal_test/group_test.json
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,460 @@
 | 
			
		||||
{
 | 
			
		||||
    "groups": [
 | 
			
		||||
        {
 | 
			
		||||
            "name": "my_account_change_pin",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "5",
 | 
			
		||||
                    "expectedContent": "PIN Management\n1:Change PIN\n2:Reset other's PIN\n3:Guard my PIN\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "Enter your old PIN\n\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Enter a new four number PIN:\n\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Confirm your new PIN:\n\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Your PIN change request has been successful\n\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_language_change",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "2",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1235",
 | 
			
		||||
                    "expectedContent": "Incorrect pin\n1:retry\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Select language:\n0:english\n1:kiswahili"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "Your language change request was successful.\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_check_my_balance",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "Balances:\n1:My balance\n2:Community balance\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1235",
 | 
			
		||||
                    "expectedContent": "Incorrect pin\n1:retry\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Your balance is 0.003 CELO\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                     "expectedContent": "Balances:\n1:My balance\n2:Community balance\n0:Back"
 | 
			
		||||
                    
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_check_community_balance",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "Balances:\n1:My balance\n2:Community balance\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "2",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1235",
 | 
			
		||||
                    "expectedContent": "Incorrect pin\n1:retry\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Your community balance is 0.003 CELO\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                     "expectedContent": "Balances:\n1:My balance\n2:Community balance\n0:Back"
 | 
			
		||||
                    
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_edit_firstname",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "Enter your first names:\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "foo",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Profile updated successfully\n\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_edit_familyname",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "2",
 | 
			
		||||
                    "expectedContent": "Enter family name:\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "bar",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Profile updated successfully\n\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
               
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_edit_gender",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "Select gender: \n1:Male\n2:Female\n3:Unspecified\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Profile updated successfully\n\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_edit_yob",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "4",
 | 
			
		||||
                    "expectedContent": "Enter your year of birth\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1945",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Profile updated successfully\n\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_edit_location",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "5",
 | 
			
		||||
                    "expectedContent": "Enter your location:\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "Kilifi",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Profile updated successfully\n\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_edit_offerings",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "6",
 | 
			
		||||
                    "expectedContent": "Enter the services or goods you offer: \n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "Bananas",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "Profile updated successfully\n\n0:Back\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        },
 | 
			
		||||
        {
 | 
			
		||||
            "name": "menu_my_account_view_profile",
 | 
			
		||||
            "steps": [
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "3",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "7",
 | 
			
		||||
                    "expectedContent": "Please enter your PIN:"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "1234",
 | 
			
		||||
                    "expectedContent": "My profile:\nName: foo bar\nGender: male\nAge: 79\nLocation: Kilifi\nYou provide: Bananas\n\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My profile\n1:Edit name\n2:Edit family name\n3:Edit gender\n4:Edit year of birth\n5:Edit location\n6:Edit offerings\n7:View profile\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                },
 | 
			
		||||
                {
 | 
			
		||||
                    "input": "0",
 | 
			
		||||
                    "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                }
 | 
			
		||||
            ]
 | 
			
		||||
        }
 | 
			
		||||
    ]
 | 
			
		||||
}
 | 
			
		||||
      
 | 
			
		||||
        
 | 
			
		||||
       
 | 
			
		||||
        
 | 
			
		||||
    
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										380
									
								
								menutraversal_test/menu_traversal_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										380
									
								
								menutraversal_test/menu_traversal_test.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,380 @@
 | 
			
		||||
package menutraversaltest
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"log"
 | 
			
		||||
	"math/rand"
 | 
			
		||||
	"os"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"testing"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/testutil"
 | 
			
		||||
	"git.grassecon.net/urdt/ussd/internal/testutil/driver"
 | 
			
		||||
	"github.com/gofrs/uuid"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	testData      = driver.ReadData()
 | 
			
		||||
	testStore     = ".test_state"
 | 
			
		||||
	groupTestFile = "group_test.json"
 | 
			
		||||
	sessionID     string
 | 
			
		||||
	src           = rand.NewSource(42)
 | 
			
		||||
	g             = rand.New(src)
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func GenerateSessionId() string {
 | 
			
		||||
	uu := uuid.NewGenWithOptions(uuid.WithRandomReader(g))
 | 
			
		||||
	v, err := uu.NewV4()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		panic(err)
 | 
			
		||||
	}
 | 
			
		||||
	return v.String()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Extract the public key from the engine response
 | 
			
		||||
func extractPublicKey(response []byte) string {
 | 
			
		||||
	// Regex pattern to match the public key starting with 0x and 40 characters
 | 
			
		||||
	re := regexp.MustCompile(`0x[a-fA-F0-9]{40}`)
 | 
			
		||||
	match := re.Find(response)
 | 
			
		||||
	if match != nil {
 | 
			
		||||
		return string(match)
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Extracts the balance value from the engine response.
 | 
			
		||||
func extractBalance(response []byte) string {
 | 
			
		||||
	// Regex to match "Balance: <amount> <symbol>" followed by a newline
 | 
			
		||||
	re := regexp.MustCompile(`(?m)^Balance:\s+(\d+(\.\d+)?)\s+([A-Z]+)`)
 | 
			
		||||
	match := re.FindSubmatch(response)
 | 
			
		||||
	if match != nil {
 | 
			
		||||
		return string(match[1]) + " " + string(match[3]) // "<amount> <symbol>"
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Extracts the Maximum amount value from the engine response.
 | 
			
		||||
func extractMaxAmount(response []byte) string {
 | 
			
		||||
	// Regex to match "Maximum amount: <amount>" followed by a newline
 | 
			
		||||
	re := regexp.MustCompile(`(?m)^Maximum amount:\s+(\d+(\.\d+)?)`)
 | 
			
		||||
	match := re.FindSubmatch(response)
 | 
			
		||||
	if match != nil {
 | 
			
		||||
		return string(match[1]) // "<amount>"
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Extracts the send amount value from the engine response.
 | 
			
		||||
func extractSendAmount(response []byte) string {
 | 
			
		||||
	// Regex to match the pattern "will receive X.XX SYM from"
 | 
			
		||||
	re := regexp.MustCompile(`will receive (\d+\.\d{2}\s+[A-Z]+) from`)
 | 
			
		||||
	match := re.FindSubmatch(response)
 | 
			
		||||
	if match != nil {
 | 
			
		||||
		return string(match[1]) // Returns "X.XX SYM"
 | 
			
		||||
	}
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestMain(m *testing.M) {
 | 
			
		||||
	sessionID = GenerateSessionId()
 | 
			
		||||
	defer func() {
 | 
			
		||||
		if err := os.RemoveAll(testStore); err != nil {
 | 
			
		||||
			log.Fatalf("Failed to delete state store %s: %v", testStore, err)
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
	m.Run()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestAccountCreationSuccessful(t *testing.T) {
 | 
			
		||||
	en, fn, eventChannel := testutil.TestEngine(sessionID)
 | 
			
		||||
	defer fn()
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	sessions := testData
 | 
			
		||||
	for _, session := range sessions {
 | 
			
		||||
		groups := driver.FilterGroupsByName(session.Groups, "account_creation_successful")
 | 
			
		||||
		for _, group := range groups {
 | 
			
		||||
			for _, step := range group.Steps {
 | 
			
		||||
				cont, err := en.Exec(ctx, []byte(step.Input))
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
 | 
			
		||||
				}
 | 
			
		||||
				if !cont {
 | 
			
		||||
					break
 | 
			
		||||
				}
 | 
			
		||||
				w := bytes.NewBuffer(nil)
 | 
			
		||||
				_, err = en.Flush(ctx, w)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
 | 
			
		||||
				}
 | 
			
		||||
				b := w.Bytes()
 | 
			
		||||
				match, err := step.MatchesExpectedContent(b)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Error compiling regex for step '%s': %v", step.Input, err)
 | 
			
		||||
				}
 | 
			
		||||
				if !match {
 | 
			
		||||
					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	<-eventChannel
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestAccountRegistrationRejectTerms(t *testing.T) {
 | 
			
		||||
	// Generate a new UUID for this edge case test
 | 
			
		||||
	uu := uuid.NewGenWithOptions(uuid.WithRandomReader(g))
 | 
			
		||||
	v, err := uu.NewV4()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Fail()
 | 
			
		||||
	}
 | 
			
		||||
	edgeCaseSessionID := v.String()
 | 
			
		||||
	en, fn, _ := testutil.TestEngine(edgeCaseSessionID)
 | 
			
		||||
	defer fn()
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	sessions := testData
 | 
			
		||||
	for _, session := range sessions {
 | 
			
		||||
		groups := driver.FilterGroupsByName(session.Groups, "account_creation_reject_terms")
 | 
			
		||||
		for _, group := range groups {
 | 
			
		||||
			for _, step := range group.Steps {
 | 
			
		||||
				cont, err := en.Exec(ctx, []byte(step.Input))
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
				if !cont {
 | 
			
		||||
					break
 | 
			
		||||
				}
 | 
			
		||||
				w := bytes.NewBuffer(nil)
 | 
			
		||||
				if _, err := en.Flush(ctx, w); err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				b := w.Bytes()
 | 
			
		||||
				match, err := step.MatchesExpectedContent(b)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Error compiling regex for step '%s': %v", step.Input, err)
 | 
			
		||||
				}
 | 
			
		||||
				if !match {
 | 
			
		||||
					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestMainMenuHelp(t *testing.T) {
 | 
			
		||||
	en, fn, _ := testutil.TestEngine(sessionID)
 | 
			
		||||
	defer fn()
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	sessions := testData
 | 
			
		||||
	for _, session := range sessions {
 | 
			
		||||
		groups := driver.FilterGroupsByName(session.Groups, "main_menu_help")
 | 
			
		||||
		for _, group := range groups {
 | 
			
		||||
			for _, step := range group.Steps {
 | 
			
		||||
				cont, err := en.Exec(ctx, []byte(step.Input))
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
				if !cont {
 | 
			
		||||
					break
 | 
			
		||||
				}
 | 
			
		||||
				w := bytes.NewBuffer(nil)
 | 
			
		||||
				if _, err := en.Flush(ctx, w); err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				b := w.Bytes()
 | 
			
		||||
				balance := extractBalance(b)
 | 
			
		||||
 | 
			
		||||
				expectedContent := []byte(step.ExpectedContent)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{balance}"), []byte(balance), -1)
 | 
			
		||||
 | 
			
		||||
				step.ExpectedContent = string(expectedContent)
 | 
			
		||||
				match, err := step.MatchesExpectedContent(b)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Error compiling regex for step '%s': %v", step.Input, err)
 | 
			
		||||
				}
 | 
			
		||||
				if !match {
 | 
			
		||||
					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestMainMenuQuit(t *testing.T) {
 | 
			
		||||
	en, fn, _ := testutil.TestEngine(sessionID)
 | 
			
		||||
	defer fn()
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	sessions := testData
 | 
			
		||||
	for _, session := range sessions {
 | 
			
		||||
		groups := driver.FilterGroupsByName(session.Groups, "main_menu_quit")
 | 
			
		||||
		for _, group := range groups {
 | 
			
		||||
			for _, step := range group.Steps {
 | 
			
		||||
				cont, err := en.Exec(ctx, []byte(step.Input))
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
				if !cont {
 | 
			
		||||
					break
 | 
			
		||||
				}
 | 
			
		||||
				w := bytes.NewBuffer(nil)
 | 
			
		||||
				if _, err := en.Flush(ctx, w); err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				b := w.Bytes()
 | 
			
		||||
				balance := extractBalance(b)
 | 
			
		||||
 | 
			
		||||
				expectedContent := []byte(step.ExpectedContent)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{balance}"), []byte(balance), -1)
 | 
			
		||||
 | 
			
		||||
				step.ExpectedContent = string(expectedContent)
 | 
			
		||||
				match, err := step.MatchesExpectedContent(b)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Error compiling regex for step '%s': %v", step.Input, err)
 | 
			
		||||
				}
 | 
			
		||||
				if !match {
 | 
			
		||||
					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestMyAccount_MyAddress(t *testing.T) {
 | 
			
		||||
	en, fn, _ := testutil.TestEngine(sessionID)
 | 
			
		||||
	defer fn()
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	sessions := testData
 | 
			
		||||
	for _, session := range sessions {
 | 
			
		||||
		groups := driver.FilterGroupsByName(session.Groups, "menu_my_account_my_address")
 | 
			
		||||
		for _, group := range groups {
 | 
			
		||||
			for index, step := range group.Steps {
 | 
			
		||||
				t.Logf("step %v with input %v", index, step.Input)
 | 
			
		||||
				cont, err := en.Exec(ctx, []byte(step.Input))
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Errorf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
				if !cont {
 | 
			
		||||
					break
 | 
			
		||||
				}
 | 
			
		||||
				w := bytes.NewBuffer(nil)
 | 
			
		||||
				if _, err := en.Flush(ctx, w); err != nil {
 | 
			
		||||
					t.Errorf("Test case '%s' failed during Flush: %v", group.Name, err)
 | 
			
		||||
				}
 | 
			
		||||
				b := w.Bytes()
 | 
			
		||||
 | 
			
		||||
				balance := extractBalance(b)
 | 
			
		||||
				publicKey := extractPublicKey(b)
 | 
			
		||||
 | 
			
		||||
				expectedContent := []byte(step.ExpectedContent)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{balance}"), []byte(balance), -1)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{public_key}"), []byte(publicKey), -1)
 | 
			
		||||
 | 
			
		||||
				step.ExpectedContent = string(expectedContent)
 | 
			
		||||
				match, err := step.MatchesExpectedContent(b)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Error compiling regex for step '%s': %v", step.Input, err)
 | 
			
		||||
				}
 | 
			
		||||
				if !match {
 | 
			
		||||
					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", expectedContent, b)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestMainMenuSend(t *testing.T) {
 | 
			
		||||
	en, fn, _ := testutil.TestEngine(sessionID)
 | 
			
		||||
	defer fn()
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	sessions := testData
 | 
			
		||||
	for _, session := range sessions {
 | 
			
		||||
		groups := driver.FilterGroupsByName(session.Groups, "send_with_invalid_inputs")
 | 
			
		||||
		for _, group := range groups {
 | 
			
		||||
			for _, step := range group.Steps {
 | 
			
		||||
				cont, err := en.Exec(ctx, []byte(step.Input))
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
				if !cont {
 | 
			
		||||
					break
 | 
			
		||||
				}
 | 
			
		||||
				w := bytes.NewBuffer(nil)
 | 
			
		||||
				if _, err := en.Flush(ctx, w); err != nil {
 | 
			
		||||
					t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
 | 
			
		||||
				}
 | 
			
		||||
 | 
			
		||||
				b := w.Bytes()
 | 
			
		||||
				balance := extractBalance(b)
 | 
			
		||||
				max_amount := extractMaxAmount(b)
 | 
			
		||||
				send_amount := extractSendAmount(b)
 | 
			
		||||
 | 
			
		||||
				expectedContent := []byte(step.ExpectedContent)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{balance}"), []byte(balance), -1)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{max_amount}"), []byte(max_amount), -1)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{send_amount}"), []byte(send_amount), -1)
 | 
			
		||||
				expectedContent = bytes.Replace(expectedContent, []byte("{session_id}"), []byte(sessionID), -1)
 | 
			
		||||
 | 
			
		||||
				step.ExpectedContent = string(expectedContent)
 | 
			
		||||
				match, err := step.MatchesExpectedContent(b)
 | 
			
		||||
				if err != nil {
 | 
			
		||||
					t.Fatalf("Error compiling regex for step '%s': %v", step.Input, err)
 | 
			
		||||
				}
 | 
			
		||||
				if !match {
 | 
			
		||||
					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestGroups(t *testing.T) {
 | 
			
		||||
	groups, err := driver.LoadTestGroups(groupTestFile)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		log.Fatalf("Failed to load test groups: %v", err)
 | 
			
		||||
	}
 | 
			
		||||
	en, fn, _ := testutil.TestEngine(sessionID)
 | 
			
		||||
	defer fn()
 | 
			
		||||
	ctx := context.Background()
 | 
			
		||||
	// Create test cases from loaded groups
 | 
			
		||||
	tests := driver.CreateTestCases(groups)
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.Name, func(t *testing.T) {
 | 
			
		||||
			cont, err := en.Exec(ctx, []byte(tt.Input))
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				t.Errorf("Test case '%s' failed at input '%s': %v", tt.Name, tt.Input, err)
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			if !cont {
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			w := bytes.NewBuffer(nil)
 | 
			
		||||
			if _, err := en.Flush(ctx, w); err != nil {
 | 
			
		||||
				t.Errorf("Test case '%s' failed during Flush: %v", tt.Name, err)
 | 
			
		||||
			}
 | 
			
		||||
			b := w.Bytes()
 | 
			
		||||
			balance := extractBalance(b)
 | 
			
		||||
 | 
			
		||||
			expectedContent := []byte(tt.ExpectedContent)
 | 
			
		||||
			expectedContent = bytes.Replace(expectedContent, []byte("{balance}"), []byte(balance), -1)
 | 
			
		||||
 | 
			
		||||
			tt.ExpectedContent = string(expectedContent)
 | 
			
		||||
 | 
			
		||||
			match, err := tt.MatchesExpectedContent(b)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				t.Fatalf("Error compiling regex for step '%s': %v", tt.Input, err)
 | 
			
		||||
			}
 | 
			
		||||
			if !match {
 | 
			
		||||
				t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", tt.ExpectedContent, b)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										153
									
								
								menutraversal_test/test_setup.json
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										153
									
								
								menutraversal_test/test_setup.json
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,153 @@
 | 
			
		||||
[
 | 
			
		||||
    {
 | 
			
		||||
        "name": "session one",
 | 
			
		||||
        "groups": [
 | 
			
		||||
            {
 | 
			
		||||
                "name": "account_creation_successful",
 | 
			
		||||
                "steps": [
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "",
 | 
			
		||||
                        "expectedContent": "Welcome to Sarafu Network\nPlease select a language\n0:english\n1:kiswahili"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "0",
 | 
			
		||||
                        "expectedContent": "Do you agree to terms and conditions?\n0:yes\n1:no"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "0",
 | 
			
		||||
                        "expectedContent": "Please enter a new four number PIN for your account:\n0:Exit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1234",
 | 
			
		||||
                        "expectedContent": "Enter your four number PIN again:"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1111",
 | 
			
		||||
                        "expectedContent": "The PIN is not a match. Try again\n1:retry\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1",
 | 
			
		||||
                        "expectedContent": "Enter your four number PIN again:"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1234",
 | 
			
		||||
                        "expectedContent": "Your account is being created...Thank you for using Sarafu. Goodbye!"
 | 
			
		||||
                    }
 | 
			
		||||
                ]
 | 
			
		||||
            },
 | 
			
		||||
            {
 | 
			
		||||
                "name": "account_creation_reject_terms",
 | 
			
		||||
                "steps": [
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "",
 | 
			
		||||
                        "expectedContent": "Welcome to Sarafu Network\nPlease select a language\n0:english\n1:kiswahili"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "0",
 | 
			
		||||
                        "expectedContent": "Do you agree to terms and conditions?\n0:yes\n1:no"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1",
 | 
			
		||||
                        "expectedContent": "Thank you for using Sarafu. Goodbye!"
 | 
			
		||||
                    }
 | 
			
		||||
                ]
 | 
			
		||||
            },
 | 
			
		||||
            {
 | 
			
		||||
                "name": "send_with_invalid_inputs",
 | 
			
		||||
                "steps": [
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "",
 | 
			
		||||
                        "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1",
 | 
			
		||||
                        "expectedContent": "Enter recipient's phone number:\n0:Back"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "000",
 | 
			
		||||
                        "expectedContent": "000 is not registered or invalid, please try again:\n1:retry\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1",
 | 
			
		||||
                        "expectedContent": "Enter recipient's phone number:\n0:Back"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "065656",
 | 
			
		||||
                        "expectedContent": "{max_amount}\nEnter amount:\n0:Back"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "10000000",
 | 
			
		||||
                        "expectedContent": "Amount 10000000 is invalid, please try again:\n1:retry\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1",
 | 
			
		||||
                        "expectedContent": "{max_amount}\nEnter amount:\n0:Back"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1.00",
 | 
			
		||||
                        "expectedContent": "065656 will receive {send_amount} from {session_id}\nPlease enter your PIN to confirm:\n0:Back\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1222",
 | 
			
		||||
                        "expectedContent": "Incorrect pin\n1:retry\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1",
 | 
			
		||||
                        "expectedContent": "065656 will receive {send_amount} from {session_id}\nPlease enter your PIN to confirm:\n0:Back\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "1234",
 | 
			
		||||
                        "expectedContent": "Your request has been sent. 065656 will receive {send_amount} from {session_id}."
 | 
			
		||||
                    }
 | 
			
		||||
                ]
 | 
			
		||||
            },
 | 
			
		||||
            {
 | 
			
		||||
                "name": "main_menu_help",
 | 
			
		||||
                "steps": [
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "",
 | 
			
		||||
                        "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "4",
 | 
			
		||||
                        "expectedContent": "For more help,please call: 0757628885"
 | 
			
		||||
                    }
 | 
			
		||||
                ]
 | 
			
		||||
            },
 | 
			
		||||
            {
 | 
			
		||||
                "name": "main_menu_quit",
 | 
			
		||||
                "steps": [
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "",
 | 
			
		||||
                        "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "9",
 | 
			
		||||
                        "expectedContent": "Thank you for using Sarafu. Goodbye!"
 | 
			
		||||
                    }
 | 
			
		||||
                ]
 | 
			
		||||
            },
 | 
			
		||||
            {
 | 
			
		||||
                "name": "menu_my_account_my_address",
 | 
			
		||||
                "steps": [
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "",
 | 
			
		||||
                        "expectedContent": "{balance}\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "3",
 | 
			
		||||
                        "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "6",
 | 
			
		||||
                        "expectedContent": "Address: {public_key}\n9:Quit"
 | 
			
		||||
                    },
 | 
			
		||||
                    {
 | 
			
		||||
                        "input": "9",
 | 
			
		||||
                        "expectedContent": "Thank you for using Sarafu. Goodbye!"
 | 
			
		||||
                    }
 | 
			
		||||
                ]
 | 
			
		||||
            }
 | 
			
		||||
        ]
 | 
			
		||||
    }
 | 
			
		||||
]
 | 
			
		||||
							
								
								
									
										113
									
								
								request/base.go
									
									
									
									
									
								
							
							
						
						
									
										113
									
								
								request/base.go
									
									
									
									
									
								
							@ -1,113 +0,0 @@
 | 
			
		||||
package request
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/db"
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/engine"
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/persist"
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/resource"
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/entry"
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/errors"
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/storage"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type EngineFunc func(engine.Config, resource.Resource, *persist.Persister) engine.Engine
 | 
			
		||||
 | 
			
		||||
type BaseRequestHandler struct {
 | 
			
		||||
	cfgTemplate engine.Config
 | 
			
		||||
	rp          RequestParser
 | 
			
		||||
	rs          resource.Resource
 | 
			
		||||
	hn          entry.EntryHandler
 | 
			
		||||
	provider    storage.StorageProvider
 | 
			
		||||
	engineFunc	EngineFunc
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewBaseRequestHandler(cfg engine.Config, rs resource.Resource, stateDb db.Db, userdataDb db.Db, rp RequestParser, hn entry.EntryHandler) *BaseRequestHandler {
 | 
			
		||||
	h := &BaseRequestHandler{
 | 
			
		||||
		cfgTemplate: cfg,
 | 
			
		||||
		rs:          rs,
 | 
			
		||||
		hn:          hn,
 | 
			
		||||
		rp:          rp,
 | 
			
		||||
		provider:    storage.NewSimpleStorageProvider(stateDb, userdataDb),
 | 
			
		||||
	}
 | 
			
		||||
	h.engineFunc = h.getDefaultEngine
 | 
			
		||||
	return h
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) WithEngineFunc(fn EngineFunc) *BaseRequestHandler {
 | 
			
		||||
	f.engineFunc = fn
 | 
			
		||||
	return f
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) Shutdown(ctx context.Context) {
 | 
			
		||||
	err := f.provider.Close(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.Errorf("handler shutdown error", "err", err)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) GetEngine(cfg engine.Config, rs resource.Resource, pr *persist.Persister) engine.Engine {
 | 
			
		||||
	return f.engineFunc(cfg, rs, pr)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) getDefaultEngine(cfg engine.Config, rs resource.Resource, pr *persist.Persister) engine.Engine {
 | 
			
		||||
	en := engine.NewEngine(cfg, rs)
 | 
			
		||||
	en = en.WithPersister(pr)
 | 
			
		||||
	en = en.WithFirst(f.hn.Init)
 | 
			
		||||
	if f.cfgTemplate.EngineDebug {
 | 
			
		||||
		en = en.WithDebug(nil)
 | 
			
		||||
	}
 | 
			
		||||
	return en
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) Process(rqs RequestSession) (RequestSession, error) {
 | 
			
		||||
	var r bool
 | 
			
		||||
	var err error
 | 
			
		||||
 | 
			
		||||
	logg.InfoCtxf(rqs.Ctx, "new request", "data", rqs)
 | 
			
		||||
 | 
			
		||||
	rqs.Storage, err = f.provider.Get(rqs.Ctx, rqs.Config.SessionId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "storage get error", err)
 | 
			
		||||
		return rqs, errors.ErrStorage
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	f.hn.SetPersister(rqs.Storage.Persister)
 | 
			
		||||
	defer func() {
 | 
			
		||||
		f.hn.Exit()
 | 
			
		||||
	}()
 | 
			
		||||
 | 
			
		||||
	rqs.Engine = f.GetEngine(rqs.Config, f.rs, rqs.Storage.Persister)
 | 
			
		||||
	r, err = rqs.Engine.Exec(rqs.Ctx, rqs.Input)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		perr := f.provider.Put(rqs.Ctx, rqs.Config.SessionId, rqs.Storage)
 | 
			
		||||
		rqs.Storage = nil
 | 
			
		||||
		if perr != nil {
 | 
			
		||||
			logg.ErrorCtxf(rqs.Ctx, "", "storage put error", perr)
 | 
			
		||||
		}
 | 
			
		||||
		return rqs, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rqs.Continue = r
 | 
			
		||||
	return rqs, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) Output(rqs RequestSession) (RequestSession, error) {
 | 
			
		||||
	var err error
 | 
			
		||||
	_, err = rqs.Engine.Flush(rqs.Ctx, rqs.Writer)
 | 
			
		||||
	return rqs, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) Reset(ctx context.Context, rqs RequestSession) (RequestSession, error) {
 | 
			
		||||
	defer f.provider.Put(ctx, rqs.Config.SessionId, rqs.Storage)
 | 
			
		||||
	return rqs, rqs.Engine.Finish(ctx)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) GetConfig() engine.Config {
 | 
			
		||||
	return f.cfgTemplate
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *BaseRequestHandler) GetRequestParser() RequestParser {
 | 
			
		||||
	return f.rp
 | 
			
		||||
}
 | 
			
		||||
@ -1,37 +0,0 @@
 | 
			
		||||
package http
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"io/ioutil"
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/errors"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type DefaultRequestParser struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (rp *DefaultRequestParser) GetSessionId(ctx context.Context, rq any) (string, error) {
 | 
			
		||||
	rqv, ok := rq.(*http.Request)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return "", errors.ErrInvalidRequest
 | 
			
		||||
	}
 | 
			
		||||
	v := rqv.Header.Get("X-Vise-Session")
 | 
			
		||||
	if v == "" {
 | 
			
		||||
		return "", errors.ErrSessionMissing
 | 
			
		||||
	}
 | 
			
		||||
	return v, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (rp *DefaultRequestParser) GetInput(rq any) ([]byte, error) {
 | 
			
		||||
	rqv, ok := rq.(*http.Request)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.ErrInvalidRequest
 | 
			
		||||
	}
 | 
			
		||||
	defer rqv.Body.Close()
 | 
			
		||||
	v, err := ioutil.ReadAll(rqv.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return v, nil
 | 
			
		||||
}
 | 
			
		||||
@ -1,92 +0,0 @@
 | 
			
		||||
package http
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strconv"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/errors"
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/request"
 | 
			
		||||
	slogging "github.com/grassrootseconomics/go-vise/slog"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg = slogging.Get().With("component", "visedriver.http.session")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// HTTPRequestHandler implements the session handler for HTTP
 | 
			
		||||
type HTTPRequestHandler struct {
 | 
			
		||||
	request.RequestHandler
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (f *HTTPRequestHandler) WriteError(w http.ResponseWriter, code int, err error) {
 | 
			
		||||
	s := err.Error()
 | 
			
		||||
	w.Header().Set("Content-Length", strconv.Itoa(len(s)))
 | 
			
		||||
	w.WriteHeader(code)
 | 
			
		||||
	_, err = w.Write([]byte(s))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.Errorf("error writing error!!", "err", err, "olderr", s)
 | 
			
		||||
		w.WriteHeader(500)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func NewHTTPRequestHandler(h request.RequestHandler) *HTTPRequestHandler {
 | 
			
		||||
	return &HTTPRequestHandler{
 | 
			
		||||
		RequestHandler: h,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (hh *HTTPRequestHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 | 
			
		||||
	var code int
 | 
			
		||||
	var err error
 | 
			
		||||
	var perr error
 | 
			
		||||
 | 
			
		||||
	rqs := request.RequestSession{
 | 
			
		||||
		Ctx:    req.Context(),
 | 
			
		||||
		Writer: w,
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rp := hh.GetRequestParser()
 | 
			
		||||
	cfg := hh.GetConfig()
 | 
			
		||||
	cfg.SessionId, err = rp.GetSessionId(req.Context(), req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "header processing error", err)
 | 
			
		||||
		hh.WriteError(w, 400, err)
 | 
			
		||||
	}
 | 
			
		||||
	rqs.Config = cfg
 | 
			
		||||
	rqs.Input, err = rp.GetInput(req)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logg.ErrorCtxf(rqs.Ctx, "", "header processing error", err)
 | 
			
		||||
		hh.WriteError(w, 400, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rqs, err = hh.Process(rqs)
 | 
			
		||||
	switch err {
 | 
			
		||||
	case errors.ErrStorage:
 | 
			
		||||
		code = 500
 | 
			
		||||
	case errors.ErrEngineInit:
 | 
			
		||||
		code = 500
 | 
			
		||||
	case errors.ErrEngineExec:
 | 
			
		||||
		code = 500
 | 
			
		||||
	default:
 | 
			
		||||
		code = 200
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if code != 200 {
 | 
			
		||||
		hh.WriteError(w, 500, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	w.WriteHeader(200)
 | 
			
		||||
	w.Header().Set("Content-Type", "text/plain")
 | 
			
		||||
	rqs, err = hh.Output(rqs)
 | 
			
		||||
	rqs, perr = hh.Reset(rqs.Ctx, rqs)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		hh.WriteError(w, 500, err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	if perr != nil {
 | 
			
		||||
		hh.WriteError(w, 500, perr)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@ -1,233 +0,0 @@
 | 
			
		||||
package http
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/httptest"
 | 
			
		||||
	"testing"
 | 
			
		||||
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/engine"
 | 
			
		||||
	viseerrors "git.grassecon.net/grassrootseconomics/visedriver/errors"
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/request"
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/testutil/mocks/httpmocks"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// invalidRequestType is a custom type to test invalid request scenarios
 | 
			
		||||
type invalidRequestType struct{}
 | 
			
		||||
 | 
			
		||||
// errorReader is a helper type that always returns an error when Read is called
 | 
			
		||||
type errorReader struct{}
 | 
			
		||||
 | 
			
		||||
func (e *errorReader) Read(p []byte) (n int, err error) {
 | 
			
		||||
	return 0, errors.New("read error")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestRequestHandler_ServeHTTP(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name           string
 | 
			
		||||
		sessionID      string
 | 
			
		||||
		input          []byte
 | 
			
		||||
		parserErr      error
 | 
			
		||||
		processErr     error
 | 
			
		||||
		outputErr      error
 | 
			
		||||
		resetErr       error
 | 
			
		||||
		expectedStatus int
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Success",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			expectedStatus: http.StatusOK,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Missing Session ID",
 | 
			
		||||
			sessionID:      "",
 | 
			
		||||
			parserErr:      viseerrors.ErrSessionMissing,
 | 
			
		||||
			expectedStatus: http.StatusBadRequest,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Process Error",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			processErr:     viseerrors.ErrStorage,
 | 
			
		||||
			expectedStatus: http.StatusInternalServerError,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Output Error",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			outputErr:      errors.New("output error"),
 | 
			
		||||
			expectedStatus: http.StatusOK,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:           "Reset Error",
 | 
			
		||||
			sessionID:      "123",
 | 
			
		||||
			input:          []byte("test input"),
 | 
			
		||||
			resetErr:       errors.New("reset error"),
 | 
			
		||||
			expectedStatus: http.StatusOK,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			mockRequestParser := &httpmocks.MockRequestParser{
 | 
			
		||||
				GetSessionIdFunc: func(any) (string, error) {
 | 
			
		||||
					return tt.sessionID, tt.parserErr
 | 
			
		||||
				},
 | 
			
		||||
				GetInputFunc: func(any) ([]byte, error) {
 | 
			
		||||
					return tt.input, nil
 | 
			
		||||
				},
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			mockRequestHandler := &httpmocks.MockRequestHandler{
 | 
			
		||||
				ProcessFunc: func(rs request.RequestSession) (request.RequestSession, error) {
 | 
			
		||||
					return rs, tt.processErr
 | 
			
		||||
				},
 | 
			
		||||
				OutputFunc: func(rs request.RequestSession) (request.RequestSession, error) {
 | 
			
		||||
					return rs, tt.outputErr
 | 
			
		||||
				},
 | 
			
		||||
				ResetFunc: func(ctx context.Context, rs request.RequestSession) (request.RequestSession, error) {
 | 
			
		||||
					return rs, tt.resetErr
 | 
			
		||||
				},
 | 
			
		||||
				GetRequestParserFunc: func() request.RequestParser {
 | 
			
		||||
					return mockRequestParser
 | 
			
		||||
				},
 | 
			
		||||
				GetConfigFunc: func() engine.Config {
 | 
			
		||||
					return engine.Config{}
 | 
			
		||||
				},
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			sessionHandler := &HTTPRequestHandler{
 | 
			
		||||
				RequestHandler: mockRequestHandler,
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			req := httptest.NewRequest(http.MethodPost, "/", bytes.NewBuffer(tt.input))
 | 
			
		||||
			req.Header.Set("X-Vise-Session", tt.sessionID)
 | 
			
		||||
 | 
			
		||||
			rr := httptest.NewRecorder()
 | 
			
		||||
 | 
			
		||||
			sessionHandler.ServeHTTP(rr, req)
 | 
			
		||||
 | 
			
		||||
			if status := rr.Code; status != tt.expectedStatus {
 | 
			
		||||
				t.Errorf("handler returned wrong status code: got %v want %v",
 | 
			
		||||
					status, tt.expectedStatus)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestRequestHandler_WriteError(t *testing.T) {
 | 
			
		||||
	handler := &HTTPRequestHandler{}
 | 
			
		||||
	mockWriter := &httpmocks.MockWriter{}
 | 
			
		||||
	err := errors.New("test error")
 | 
			
		||||
 | 
			
		||||
	handler.WriteError(mockWriter, http.StatusBadRequest, err)
 | 
			
		||||
 | 
			
		||||
	if mockWriter.WrittenString != "" {
 | 
			
		||||
		t.Errorf("Expected empty body, got %s", mockWriter.WrittenString)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestDefaultRequestParser_GetSessionId(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name          string
 | 
			
		||||
		request       any
 | 
			
		||||
		expectedID    string
 | 
			
		||||
		expectedError error
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name: "Valid Session ID",
 | 
			
		||||
			request: func() *http.Request {
 | 
			
		||||
				req := httptest.NewRequest(http.MethodPost, "/", nil)
 | 
			
		||||
				req.Header.Set("X-Vise-Session", "123456")
 | 
			
		||||
				return req
 | 
			
		||||
			}(),
 | 
			
		||||
			expectedID:    "123456",
 | 
			
		||||
			expectedError: nil,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Missing Session ID",
 | 
			
		||||
			request:       httptest.NewRequest(http.MethodPost, "/", nil),
 | 
			
		||||
			expectedID:    "",
 | 
			
		||||
			expectedError: viseerrors.ErrSessionMissing,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Invalid Request Type",
 | 
			
		||||
			request:       invalidRequestType{},
 | 
			
		||||
			expectedID:    "",
 | 
			
		||||
			expectedError: viseerrors.ErrInvalidRequest,
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	parser := &DefaultRequestParser{}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			id, err := parser.GetSessionId(context.Background(), tt.request)
 | 
			
		||||
 | 
			
		||||
			if id != tt.expectedID {
 | 
			
		||||
				t.Errorf("Expected session ID %s, got %s", tt.expectedID, id)
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if err != tt.expectedError {
 | 
			
		||||
				t.Errorf("Expected error %v, got %v", tt.expectedError, err)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestDefaultRequestParser_GetInput(t *testing.T) {
 | 
			
		||||
	tests := []struct {
 | 
			
		||||
		name          string
 | 
			
		||||
		request       any
 | 
			
		||||
		expectedInput []byte
 | 
			
		||||
		expectedError error
 | 
			
		||||
	}{
 | 
			
		||||
		{
 | 
			
		||||
			name: "Valid Input",
 | 
			
		||||
			request: func() *http.Request {
 | 
			
		||||
				return httptest.NewRequest(http.MethodPost, "/", bytes.NewBufferString("test input"))
 | 
			
		||||
			}(),
 | 
			
		||||
			expectedInput: []byte("test input"),
 | 
			
		||||
			expectedError: nil,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Empty Input",
 | 
			
		||||
			request:       httptest.NewRequest(http.MethodPost, "/", nil),
 | 
			
		||||
			expectedInput: []byte{},
 | 
			
		||||
			expectedError: nil,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name:          "Invalid Request Type",
 | 
			
		||||
			request:       invalidRequestType{},
 | 
			
		||||
			expectedInput: nil,
 | 
			
		||||
			expectedError: viseerrors.ErrInvalidRequest,
 | 
			
		||||
		},
 | 
			
		||||
		{
 | 
			
		||||
			name: "Read Error",
 | 
			
		||||
			request: func() *http.Request {
 | 
			
		||||
				return httptest.NewRequest(http.MethodPost, "/", &errorReader{})
 | 
			
		||||
			}(),
 | 
			
		||||
			expectedInput: nil,
 | 
			
		||||
			expectedError: errors.New("read error"),
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	parser := &DefaultRequestParser{}
 | 
			
		||||
 | 
			
		||||
	for _, tt := range tests {
 | 
			
		||||
		t.Run(tt.name, func(t *testing.T) {
 | 
			
		||||
			input, err := parser.GetInput(tt.request)
 | 
			
		||||
 | 
			
		||||
			if !bytes.Equal(input, tt.expectedInput) {
 | 
			
		||||
				t.Errorf("Expected input %s, got %s", tt.expectedInput, input)
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if err != tt.expectedError && (err == nil || err.Error() != tt.expectedError.Error()) {
 | 
			
		||||
				t.Errorf("Expected error %v, got %v", tt.expectedError, err)
 | 
			
		||||
			}
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@ -1,42 +0,0 @@
 | 
			
		||||
package request
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"io"
 | 
			
		||||
 | 
			
		||||
	"git.grassecon.net/grassrootseconomics/visedriver/storage"
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/engine"
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/persist"
 | 
			
		||||
	"github.com/grassrootseconomics/go-vise/resource"
 | 
			
		||||
	slogging "github.com/grassrootseconomics/go-vise/slog"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	logg = slogging.Get().With("component", "visedriver.request")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type RequestSession struct {
 | 
			
		||||
	Ctx      context.Context
 | 
			
		||||
	Config   engine.Config
 | 
			
		||||
	Engine   engine.Engine
 | 
			
		||||
	Input    []byte
 | 
			
		||||
	Storage  *storage.Storage
 | 
			
		||||
	Writer   io.Writer
 | 
			
		||||
	Continue bool
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// TODO: seems like can remove this.
 | 
			
		||||
type RequestParser interface {
 | 
			
		||||
	GetSessionId(context.Context, any) (string, error)
 | 
			
		||||
	GetInput(any) ([]byte, error)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type RequestHandler interface {
 | 
			
		||||
	GetConfig() engine.Config
 | 
			
		||||
	GetRequestParser() RequestParser
 | 
			
		||||
	GetEngine(engine.Config, resource.Resource, *persist.Persister) engine.Engine
 | 
			
		||||
	Process(RequestSession) (RequestSession, error)
 | 
			
		||||
	Output(RequestSession) (RequestSession, error)
 | 
			
		||||
	Reset(context.Context, RequestSession) (RequestSession, error)
 | 
			
		||||
	Shutdown(ctx context.Context)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										18
									
								
								services/registration/Makefile
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								services/registration/Makefile
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,18 @@
 | 
			
		||||
# Variables to match files in the current directory
 | 
			
		||||
INPUTS = $(wildcard ./*.vis)
 | 
			
		||||
TXTS = $(wildcard ./*.txt.orig)
 | 
			
		||||
VISE_PATH := ../../go-vise
 | 
			
		||||
 | 
			
		||||
# Rule to build .bin files from .vis files
 | 
			
		||||
%.vis:
 | 
			
		||||
	go run  $(VISE_PATH)/dev/asm/main.go -f pp.csv $(basename $@).vis > $(basename $@).bin
 | 
			
		||||
	@echo "Built $(basename $@).bin from $(basename $@).vis"
 | 
			
		||||
 | 
			
		||||
# Rule to copy .orig files to .txt
 | 
			
		||||
%.txt.orig:
 | 
			
		||||
	cp -v $(basename $@).orig $(basename $@)
 | 
			
		||||
	@echo "Copied $(basename $@).orig to $(basename $@)"
 | 
			
		||||
 | 
			
		||||
# 'all' target depends on all .vis and .txt.orig files
 | 
			
		||||
all: $(INPUTS) $(TXTS)
 | 
			
		||||
	@echo "Running all: $(INPUTS) $(TXTS)"
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/_catch
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/_catch
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Something went wrong.Please try again
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/_catch.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/_catch.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
HALT
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_creation
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_creation
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Your account is being created...
 | 
			
		||||
							
								
								
									
										4
									
								
								services/registration/account_creation.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								services/registration/account_creation.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,4 @@
 | 
			
		||||
RELOAD verify_create_pin
 | 
			
		||||
CATCH create_pin_mismatch flag_pin_mismatch 1
 | 
			
		||||
LOAD quit 0
 | 
			
		||||
HALT
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_creation_failed
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_creation_failed
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Your account creation request failed. Please try again later.
 | 
			
		||||
							
								
								
									
										3
									
								
								services/registration/account_creation_failed.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								services/registration/account_creation_failed.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,3 @@
 | 
			
		||||
MOUT quit 9
 | 
			
		||||
HALT
 | 
			
		||||
INCMP quit 9
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_creation_failed_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_creation_failed_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Ombi lako la kusajiliwa haliwezi kukamilishwa. Tafadhali jaribu tena baadaye.
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_creation_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_creation_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Akaunti yako inatengenezwa...
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_menu
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_menu
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
My Account
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_menu_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_menu_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Akaunti yangu
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_pending
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_pending
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Your account is still being created.
 | 
			
		||||
							
								
								
									
										3
									
								
								services/registration/account_pending.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								services/registration/account_pending.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,3 @@
 | 
			
		||||
RELOAD check_account_status
 | 
			
		||||
CATCH main flag_account_success 1
 | 
			
		||||
HALT
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/account_pending_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/account_pending_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Akaunti yako bado inatengenezwa
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/address
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/address
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Address: {{.check_identifier}}
 | 
			
		||||
							
								
								
									
										6
									
								
								services/registration/address.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								services/registration/address.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,6 @@
 | 
			
		||||
LOAD check_identifier 0
 | 
			
		||||
RELOAD check_identifier
 | 
			
		||||
MAP check_identifier
 | 
			
		||||
MOUT quit 9
 | 
			
		||||
HALT
 | 
			
		||||
INCMP quit 9
 | 
			
		||||
							
								
								
									
										2
									
								
								services/registration/amount
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								services/registration/amount
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,2 @@
 | 
			
		||||
Maximum amount: {{.max_amount}}
 | 
			
		||||
Enter amount:
 | 
			
		||||
							
								
								
									
										15
									
								
								services/registration/amount.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								services/registration/amount.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,15 @@
 | 
			
		||||
LOAD reset_transaction_amount 0
 | 
			
		||||
LOAD max_amount 10
 | 
			
		||||
RELOAD max_amount
 | 
			
		||||
MAP max_amount
 | 
			
		||||
MOUT back 0
 | 
			
		||||
HALT
 | 
			
		||||
LOAD validate_amount 64
 | 
			
		||||
RELOAD validate_amount
 | 
			
		||||
CATCH api_failure  flag_api_call_error  1
 | 
			
		||||
CATCH invalid_amount flag_invalid_amount 1
 | 
			
		||||
INCMP _ 0
 | 
			
		||||
LOAD get_recipient 12
 | 
			
		||||
LOAD get_sender 64
 | 
			
		||||
LOAD get_amount 32
 | 
			
		||||
INCMP transaction_pin *
 | 
			
		||||
							
								
								
									
										2
									
								
								services/registration/amount_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								services/registration/amount_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,2 @@
 | 
			
		||||
Kiwango cha juu: {{.max_amount}}
 | 
			
		||||
Weka kiwango:
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/api_failure
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/api_failure
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Failed to connect to the custodial service.Please try again.
 | 
			
		||||
							
								
								
									
										5
									
								
								services/registration/api_failure.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								services/registration/api_failure.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,5 @@
 | 
			
		||||
MOUT retry 0
 | 
			
		||||
MOUT quit 9
 | 
			
		||||
HALT
 | 
			
		||||
INCMP _ 0
 | 
			
		||||
INCMP quit 9
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/back_menu
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/back_menu
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Back
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/back_menu_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/back_menu_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Rudi
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/balances
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/balances
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Balances:
 | 
			
		||||
							
								
								
									
										9
									
								
								services/registration/balances.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								services/registration/balances.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,9 @@
 | 
			
		||||
LOAD reset_account_authorized 0
 | 
			
		||||
RELOAD reset_account_authorized
 | 
			
		||||
MOUT my_balance 1
 | 
			
		||||
MOUT community_balance 2
 | 
			
		||||
MOUT back 0
 | 
			
		||||
HALT
 | 
			
		||||
INCMP _ 0
 | 
			
		||||
INCMP my_balance 1
 | 
			
		||||
INCMP community_balance 2
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/balances_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/balances_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Salio:
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/change_language
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/change_language
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Select language:
 | 
			
		||||
							
								
								
									
										10
									
								
								services/registration/change_language.vis
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								services/registration/change_language.vis
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,10 @@
 | 
			
		||||
LOAD reset_account_authorized 0
 | 
			
		||||
LOAD reset_incorrect 0
 | 
			
		||||
CATCH incorrect_pin flag_incorrect_pin 1
 | 
			
		||||
CATCH pin_entry flag_account_authorized 0
 | 
			
		||||
MOUT english 0
 | 
			
		||||
MOUT kiswahili 1
 | 
			
		||||
HALT
 | 
			
		||||
INCMP set_default 0
 | 
			
		||||
INCMP set_swa 1
 | 
			
		||||
INCMP . *
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/change_language_menu
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/change_language_menu
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Change language
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/change_language_menu_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/change_language_menu_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Badili lugha
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/change_language_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/change_language_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Chagua lugha:
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/change_pin_menu
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/change_pin_menu
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Change PIN
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/change_pin_menu_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/change_pin_menu_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Badili PIN
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/check_balance_menu
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/check_balance_menu
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Check balances
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/check_balance_menu_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/check_balance_menu_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Angalia salio
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/check_statement_menu
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/check_statement_menu
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Check statement
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/check_statement_menu_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/check_statement_menu_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Taarifa ya matumizi
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/comminity_balance_swa
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/comminity_balance_swa
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
Salio la kikundi
 | 
			
		||||
							
								
								
									
										1
									
								
								services/registration/community_balance
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								services/registration/community_balance
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1 @@
 | 
			
		||||
{{.fetch_custodial_balances}}
 | 
			
		||||
Some files were not shown because too many files have changed in this diff Show More
		Loading…
	
		Reference in New Issue
	
	Block a user