2078 lines
		
	
	
		
			68 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			2078 lines
		
	
	
		
			68 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package ussd
 | |
| 
 | |
| import (
 | |
| 	"bytes"
 | |
| 	"context"
 | |
| 	"fmt"
 | |
| 	"path"
 | |
| 	"strconv"
 | |
| 	"strings"
 | |
| 
 | |
| 	"git.defalsify.org/vise.git/asm"
 | |
| 
 | |
| 	"git.defalsify.org/vise.git/cache"
 | |
| 	"git.defalsify.org/vise.git/db"
 | |
| 	"git.defalsify.org/vise.git/lang"
 | |
| 	"git.defalsify.org/vise.git/logging"
 | |
| 	"git.defalsify.org/vise.git/persist"
 | |
| 	"git.defalsify.org/vise.git/resource"
 | |
| 	"git.defalsify.org/vise.git/state"
 | |
| 	"git.grassecon.net/urdt/ussd/common"
 | |
| 	"git.grassecon.net/urdt/ussd/internal/utils"
 | |
| 	"git.grassecon.net/urdt/ussd/models"
 | |
| 	"git.grassecon.net/urdt/ussd/remote"
 | |
| 	"gopkg.in/leonelquinteros/gotext.v1"
 | |
| 
 | |
| 	dbstorage "git.grassecon.net/urdt/ussd/internal/storage/db"
 | |
| 	dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api"
 | |
| )
 | |
| 
 | |
| var (
 | |
| 	logg           = logging.NewVanilla().WithDomain("ussdmenuhandler").WithContextKey("SessionId")
 | |
| 	scriptDir      = path.Join("services", "registration")
 | |
| 	translationDir = path.Join(scriptDir, "locale")
 | |
| )
 | |
| 
 | |
| // FlagManager handles centralized flag management
 | |
| type FlagManager struct {
 | |
| 	parser *asm.FlagParser
 | |
| }
 | |
| 
 | |
| // NewFlagManager creates a new FlagManager instance
 | |
| func NewFlagManager(csvPath string) (*FlagManager, error) {
 | |
| 	parser := asm.NewFlagParser()
 | |
| 	_, err := parser.Load(csvPath)
 | |
| 	if err != nil {
 | |
| 		return nil, fmt.Errorf("failed to load flag parser: %v", err)
 | |
| 	}
 | |
| 
 | |
| 	return &FlagManager{
 | |
| 		parser: parser,
 | |
| 	}, nil
 | |
| }
 | |
| 
 | |
| // GetFlag retrieves a flag value by its label
 | |
| func (fm *FlagManager) GetFlag(label string) (uint32, error) {
 | |
| 	return fm.parser.GetFlag(label)
 | |
| }
 | |
| 
 | |
| type Handlers struct {
 | |
| 	pe                   *persist.Persister
 | |
| 	st                   *state.State
 | |
| 	ca                   cache.Memory
 | |
| 	userdataStore        common.DataStore
 | |
| 	adminstore           *utils.AdminStore
 | |
| 	flagManager          *asm.FlagParser
 | |
| 	accountService       remote.AccountServiceInterface
 | |
| 	prefixDb             dbstorage.PrefixDb
 | |
| 	profile              *models.Profile
 | |
| 	ReplaceSeparatorFunc func(string) string
 | |
| }
 | |
| 
 | |
| // NewHandlers creates a new instance of the Handlers struct with the provided dependencies.
 | |
| func NewHandlers(appFlags *asm.FlagParser, userdataStore db.Db, adminstore *utils.AdminStore, accountService remote.AccountServiceInterface, replaceSeparatorFunc func(string) string) (*Handlers, error) {
 | |
| 	if userdataStore == nil {
 | |
| 		return nil, fmt.Errorf("cannot create handler with nil userdata store")
 | |
| 	}
 | |
| 	userDb := &common.UserDataStore{
 | |
| 		Db: userdataStore,
 | |
| 	}
 | |
| 
 | |
| 	// Instantiate the SubPrefixDb with "DATATYPE_USERDATA" prefix
 | |
| 	prefix := common.ToBytes(db.DATATYPE_USERDATA)
 | |
| 	prefixDb := dbstorage.NewSubPrefixDb(userdataStore, prefix)
 | |
| 
 | |
| 	h := &Handlers{
 | |
| 		userdataStore:        userDb,
 | |
| 		flagManager:          appFlags,
 | |
| 		adminstore:           adminstore,
 | |
| 		accountService:       accountService,
 | |
| 		prefixDb:             prefixDb,
 | |
| 		profile:              &models.Profile{Max: 6},
 | |
| 		ReplaceSeparatorFunc: replaceSeparatorFunc,
 | |
| 	}
 | |
| 	return h, nil
 | |
| }
 | |
| 
 | |
| // WithPersister sets persister instance to the handlers.
 | |
| func (h *Handlers) WithPersister(pe *persist.Persister) *Handlers {
 | |
| 	if h.pe != nil {
 | |
| 		panic("persister already set")
 | |
| 	}
 | |
| 	h.pe = pe
 | |
| 	return h
 | |
| }
 | |
| 
 | |
| // Init initializes the handler for a new session.
 | |
| func (h *Handlers) Init(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var r resource.Result
 | |
| 	if h.pe == nil {
 | |
| 		logg.WarnCtxf(ctx, "handler init called before it is ready or more than once", "state", h.st, "cache", h.ca)
 | |
| 		return r, nil
 | |
| 	}
 | |
| 	defer func() {
 | |
| 		h.Exit()
 | |
| 	}()
 | |
| 
 | |
| 	h.st = h.pe.GetState()
 | |
| 	h.ca = h.pe.GetMemory()
 | |
| 
 | |
| 	if len(input) == 0 {
 | |
| 		// move to the top node
 | |
| 		h.st.Code = []byte{}
 | |
| 	}
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if ok {
 | |
| 		ctx = context.WithValue(ctx, "SessionId", sessionId)
 | |
| 	}
 | |
| 
 | |
| 	flag_admin_privilege, _ := h.flagManager.GetFlag("flag_admin_privilege")
 | |
| 	isAdmin, _ := h.adminstore.IsAdmin(sessionId)
 | |
| 
 | |
| 	if isAdmin {
 | |
| 		r.FlagSet = append(r.FlagSet, flag_admin_privilege)
 | |
| 	} else {
 | |
| 		r.FlagReset = append(r.FlagReset, flag_admin_privilege)
 | |
| 	}
 | |
| 
 | |
| 	if h.st == nil || h.ca == nil {
 | |
| 		logg.ErrorCtxf(ctx, "perister fail in handler", "state", h.st, "cache", h.ca)
 | |
| 		return r, fmt.Errorf("cannot get state and memory for handler")
 | |
| 	}
 | |
| 
 | |
| 	logg.DebugCtxf(ctx, "handler has been initialized", "state", h.st, "cache", h.ca)
 | |
| 
 | |
| 	return r, nil
 | |
| }
 | |
| 
 | |
| func (h *Handlers) Exit() {
 | |
| 	h.pe = nil
 | |
| }
 | |
| 
 | |
| // SetLanguage sets the language across the menu.
 | |
| func (h *Handlers) SetLanguage(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	symbol, _ := h.st.Where()
 | |
| 	code := strings.Split(symbol, "_")[1]
 | |
| 
 | |
| 	if !utils.IsValidISO639(code) {
 | |
| 		//Fallback to english instead?
 | |
| 		code = "eng"
 | |
| 	}
 | |
| 	res.FlagSet = append(res.FlagSet, state.FLAG_LANG)
 | |
| 	res.Content = code
 | |
| 
 | |
| 	languageSetFlag, err := h.flagManager.GetFlag("flag_language_set")
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "Error setting the languageSetFlag", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	res.FlagSet = append(res.FlagSet, languageSetFlag)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // handles the account creation when no existing account is present for the session and stores associated data in the user data store.
 | |
| func (h *Handlers) createAccountNoExist(ctx context.Context, sessionId string, res *resource.Result) error {
 | |
| 	flag_account_created, _ := h.flagManager.GetFlag("flag_account_created")
 | |
| 	r, err := h.accountService.CreateAccount(ctx)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	trackingId := r.TrackingId
 | |
| 	publicKey := r.PublicKey
 | |
| 
 | |
| 	data := map[common.DataTyp]string{
 | |
| 		common.DATA_TRACKING_ID: trackingId,
 | |
| 		common.DATA_PUBLIC_KEY:  publicKey,
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	for key, value := range data {
 | |
| 		err = store.WriteEntry(ctx, sessionId, key, []byte(value))
 | |
| 		if err != nil {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	publicKeyNormalized, err := common.NormalizeHex(publicKey)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	err = store.WriteEntry(ctx, publicKeyNormalized, common.DATA_PUBLIC_KEY_REVERSE, []byte(sessionId))
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	res.FlagSet = append(res.FlagSet, flag_account_created)
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // CreateAccount checks if any account exists on the JSON data file, and if not,
 | |
| // creates an account on the API,
 | |
| // sets the default values and flags.
 | |
| func (h *Handlers) CreateAccount(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	_, err = store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 	if err != nil {
 | |
| 		if db.IsNotFound(err) {
 | |
| 			logg.InfoCtxf(ctx, "Creating an account because it doesn't exist")
 | |
| 			err = h.createAccountNoExist(ctx, sessionId, &res)
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed on createAccountNoExist", "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // CheckBlockedNumPinMisMatch checks if the provided PIN matches a temporary PIN stored for a blocked number.
 | |
| func (h *Handlers) CheckBlockedNumPinMisMatch(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	res := resource.Result{}
 | |
| 	flag_pin_mismatch, _ := h.flagManager.GetFlag("flag_pin_mismatch")
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	// Get blocked number from storage.
 | |
| 	store := h.userdataStore
 | |
| 	blockedNumber, err := store.ReadEntry(ctx, sessionId, common.DATA_BLOCKED_NUMBER)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read blockedNumber entry with", "key", common.DATA_BLOCKED_NUMBER, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	// Get temporary PIN for the blocked number.
 | |
| 	temporaryPin, err := store.ReadEntry(ctx, string(blockedNumber), common.DATA_TEMPORARY_VALUE)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read temporaryPin entry with", "key", common.DATA_TEMPORARY_VALUE, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	if bytes.Equal(temporaryPin, input) {
 | |
| 		res.FlagReset = append(res.FlagReset, flag_pin_mismatch)
 | |
| 	} else {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_pin_mismatch)
 | |
| 	}
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // VerifyNewPin checks if a new PIN meets the required format criteria.
 | |
| func (h *Handlers) VerifyNewPin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	res := resource.Result{}
 | |
| 	_, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	flag_valid_pin, _ := h.flagManager.GetFlag("flag_valid_pin")
 | |
| 	pinInput := string(input)
 | |
| 	// Validate that the PIN is a 4-digit number.
 | |
| 	if common.IsValidPIN(pinInput) {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_valid_pin)
 | |
| 	} else {
 | |
| 		res.FlagReset = append(res.FlagReset, flag_valid_pin)
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SaveTemporaryPin saves the valid PIN input to the DATA_TEMPORARY_VALUE,
 | |
| // during the account creation process
 | |
| // and during the change PIN process.
 | |
| func (h *Handlers) SaveTemporaryPin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_incorrect_pin, _ := h.flagManager.GetFlag("flag_incorrect_pin")
 | |
| 	accountPIN := string(input)
 | |
| 
 | |
| 	// Validate that the PIN is a 4-digit number.
 | |
| 	if !common.IsValidPIN(accountPIN) {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_incorrect_pin)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 	res.FlagReset = append(res.FlagReset, flag_incorrect_pin)
 | |
| 	store := h.userdataStore
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(accountPIN))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to write temporaryAccountPIN entry with", "key", common.DATA_TEMPORARY_VALUE, "value", accountPIN, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SaveOthersTemporaryPin allows authorized users to set temporary PINs for blocked numbers.
 | |
| func (h *Handlers) SaveOthersTemporaryPin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	temporaryPin := string(input)
 | |
| 	// First, we retrieve the blocked number associated with this session
 | |
| 	blockedNumber, err := store.ReadEntry(ctx, sessionId, common.DATA_BLOCKED_NUMBER)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read blockedNumber entry with", "key", common.DATA_BLOCKED_NUMBER, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Then we save the temporary PIN for that blocked number
 | |
| 	err = store.WriteEntry(ctx, string(blockedNumber), common.DATA_TEMPORARY_VALUE, []byte(temporaryPin))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to write temporaryPin entry with", "key", common.DATA_TEMPORARY_VALUE, "value", temporaryPin, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ConfirmPinChange validates user's new PIN. If input matches the temporary PIN, saves it as the new account PIN.
 | |
| func (h *Handlers) ConfirmPinChange(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	flag_pin_mismatch, _ := h.flagManager.GetFlag("flag_pin_mismatch")
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	temporaryPin, err := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read temporaryPin entry with", "key", common.DATA_TEMPORARY_VALUE, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	if bytes.Equal(temporaryPin, input) {
 | |
| 		res.FlagReset = append(res.FlagReset, flag_pin_mismatch)
 | |
| 	} else {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_pin_mismatch)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	// Hash the PIN
 | |
| 	hashedPIN, err := common.HashPIN(string(temporaryPin))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to hash temporaryPin", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// save the hashed PIN as the new account PIN
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_ACCOUNT_PIN, []byte(hashedPIN))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to write DATA_ACCOUNT_PIN entry with", "key", common.DATA_ACCOUNT_PIN, "hashedPIN value", hashedPIN, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // VerifyCreatePin checks whether the confirmation PIN is similar to the temporary PIN
 | |
| // If similar, it sets the USERFLAG_PIN_SET flag and writes the account PIN allowing the user
 | |
| // to access the main menu.
 | |
| func (h *Handlers) VerifyCreatePin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	flag_valid_pin, _ := h.flagManager.GetFlag("flag_valid_pin")
 | |
| 	flag_pin_mismatch, _ := h.flagManager.GetFlag("flag_pin_mismatch")
 | |
| 	flag_pin_set, _ := h.flagManager.GetFlag("flag_pin_set")
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	temporaryPin, err := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read temporaryPin entry with", "key", common.DATA_TEMPORARY_VALUE, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	if bytes.Equal(input, temporaryPin) {
 | |
| 		res.FlagSet = []uint32{flag_valid_pin}
 | |
| 		res.FlagReset = []uint32{flag_pin_mismatch}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_pin_set)
 | |
| 	} else {
 | |
| 		res.FlagSet = []uint32{flag_pin_mismatch}
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	// Hash the PIN
 | |
| 	hashedPIN, err := common.HashPIN(string(temporaryPin))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to hash temporaryPin", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_ACCOUNT_PIN, []byte(hashedPIN))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to write DATA_ACCOUNT_PIN entry with", "key", common.DATA_ACCOUNT_PIN, "value", hashedPIN, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // retrieves language codes from the context that can be used for handling translations.
 | |
| func codeFromCtx(ctx context.Context) string {
 | |
| 	var code string
 | |
| 	if ctx.Value("Language") != nil {
 | |
| 		lang := ctx.Value("Language").(lang.Language)
 | |
| 		code = lang.Code
 | |
| 	}
 | |
| 	return code
 | |
| }
 | |
| 
 | |
| // SaveFirstname updates the first name in the gdbm with the provided input.
 | |
| func (h *Handlers) SaveFirstname(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	firstName := string(input)
 | |
| 	store := h.userdataStore
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 	flag_firstname_set, _ := h.flagManager.GetFlag("flag_firstname_set")
 | |
| 
 | |
| 	allowUpdate := h.st.MatchFlag(flag_allow_update, true)
 | |
| 	firstNameSet := h.st.MatchFlag(flag_firstname_set, true)
 | |
| 	if allowUpdate {
 | |
| 		temporaryFirstName, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 		err = store.WriteEntry(ctx, sessionId, common.DATA_FIRST_NAME, []byte(temporaryFirstName))
 | |
| 		if err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write firstName entry with", "key", common.DATA_FIRST_NAME, "value", temporaryFirstName, "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_firstname_set)
 | |
| 	} else {
 | |
| 		if firstNameSet {
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(firstName))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write temporaryFirstName entry with", "key", common.DATA_TEMPORARY_VALUE, "value", firstName, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 		} else {
 | |
| 			h.profile.InsertOrShift(0, firstName)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SaveFamilyname updates the family name in the gdbm with the provided input.
 | |
| func (h *Handlers) SaveFamilyname(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	familyName := string(input)
 | |
| 
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 	flag_familyname_set, _ := h.flagManager.GetFlag("flag_familyname_set")
 | |
| 	allowUpdate := h.st.MatchFlag(flag_allow_update, true)
 | |
| 	familyNameSet := h.st.MatchFlag(flag_familyname_set, true)
 | |
| 
 | |
| 	if allowUpdate {
 | |
| 		temporaryFamilyName, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 		err = store.WriteEntry(ctx, sessionId, common.DATA_FAMILY_NAME, []byte(temporaryFamilyName))
 | |
| 		if err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write familyName entry with", "key", common.DATA_FAMILY_NAME, "value", temporaryFamilyName, "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_familyname_set)
 | |
| 	} else {
 | |
| 		if familyNameSet {
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(familyName))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write temporaryFamilyName entry with", "key", common.DATA_TEMPORARY_VALUE, "value", familyName, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 		} else {
 | |
| 			h.profile.InsertOrShift(1, familyName)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SaveYOB updates the Year of Birth(YOB) in the gdbm with the provided input.
 | |
| func (h *Handlers) SaveYob(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	yob := string(input)
 | |
| 	store := h.userdataStore
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 	flag_yob_set, _ := h.flagManager.GetFlag("flag_yob_set")
 | |
| 
 | |
| 	allowUpdate := h.st.MatchFlag(flag_allow_update, true)
 | |
| 	yobSet := h.st.MatchFlag(flag_yob_set, true)
 | |
| 
 | |
| 	if allowUpdate {
 | |
| 		temporaryYob, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 		err = store.WriteEntry(ctx, sessionId, common.DATA_YOB, []byte(temporaryYob))
 | |
| 		if err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write yob entry with", "key", common.DATA_TEMPORARY_VALUE, "value", temporaryYob, "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_yob_set)
 | |
| 	} else {
 | |
| 		if yobSet {
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(yob))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write temporaryYob entry with", "key", common.DATA_TEMPORARY_VALUE, "value", yob, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 		} else {
 | |
| 			h.profile.InsertOrShift(3, yob)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SaveLocation updates the location in the gdbm with the provided input.
 | |
| func (h *Handlers) SaveLocation(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	location := string(input)
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 	flag_location_set, _ := h.flagManager.GetFlag("flag_location_set")
 | |
| 	allowUpdate := h.st.MatchFlag(flag_allow_update, true)
 | |
| 	locationSet := h.st.MatchFlag(flag_location_set, true)
 | |
| 
 | |
| 	if allowUpdate {
 | |
| 		temporaryLocation, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 		err = store.WriteEntry(ctx, sessionId, common.DATA_LOCATION, []byte(temporaryLocation))
 | |
| 		if err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write location entry with", "key", common.DATA_LOCATION, "value", temporaryLocation, "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_location_set)
 | |
| 	} else {
 | |
| 		if locationSet {
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(location))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write temporaryLocation entry with", "key", common.DATA_TEMPORARY_VALUE, "value", location, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 			res.FlagSet = append(res.FlagSet, flag_location_set)
 | |
| 		} else {
 | |
| 			h.profile.InsertOrShift(4, location)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SaveGender updates the gender in the gdbm with the provided input.
 | |
| func (h *Handlers) SaveGender(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	symbol, _ := h.st.Where()
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	gender := strings.Split(symbol, "_")[1]
 | |
| 	store := h.userdataStore
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 	flag_gender_set, _ := h.flagManager.GetFlag("flag_gender_set")
 | |
| 
 | |
| 	allowUpdate := h.st.MatchFlag(flag_allow_update, true)
 | |
| 	genderSet := h.st.MatchFlag(flag_gender_set, true)
 | |
| 
 | |
| 	if allowUpdate {
 | |
| 		temporaryGender, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 		err = store.WriteEntry(ctx, sessionId, common.DATA_GENDER, []byte(temporaryGender))
 | |
| 		if err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write gender entry with", "key", common.DATA_GENDER, "value", gender, "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_gender_set)
 | |
| 	} else {
 | |
| 		if genderSet {
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(gender))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write temporaryGender entry with", "key", common.DATA_TEMPORARY_VALUE, "value", gender, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 		} else {
 | |
| 			h.profile.InsertOrShift(2, gender)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SaveOfferings updates the offerings(goods and services provided by the user) in the gdbm with the provided input.
 | |
| func (h *Handlers) SaveOfferings(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	offerings := string(input)
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 	flag_offerings_set, _ := h.flagManager.GetFlag("flag_offerings_set")
 | |
| 
 | |
| 	allowUpdate := h.st.MatchFlag(flag_allow_update, true)
 | |
| 	offeringsSet := h.st.MatchFlag(flag_offerings_set, true)
 | |
| 
 | |
| 	if allowUpdate {
 | |
| 		temporaryOfferings, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 		err = store.WriteEntry(ctx, sessionId, common.DATA_OFFERINGS, []byte(temporaryOfferings))
 | |
| 		if err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write offerings entry with", "key", common.DATA_TEMPORARY_VALUE, "value", offerings, "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_offerings_set)
 | |
| 	} else {
 | |
| 		if offeringsSet {
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(offerings))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write temporaryOfferings entry with", "key", common.DATA_TEMPORARY_VALUE, "value", offerings, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 		} else {
 | |
| 			h.profile.InsertOrShift(5, offerings)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetAllowUpdate resets the allowupdate flag that allows a user to update  profile data.
 | |
| func (h *Handlers) ResetAllowUpdate(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_allow_update)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetAllowUpdate resets the allowupdate flag that allows a user to update  profile data.
 | |
| func (h *Handlers) ResetValidPin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	flag_valid_pin, _ := h.flagManager.GetFlag("flag_valid_pin")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_valid_pin)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetAccountAuthorized resets the account authorization flag after a successful PIN entry.
 | |
| func (h *Handlers) ResetAccountAuthorized(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	flag_account_authorized, _ := h.flagManager.GetFlag("flag_account_authorized")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_account_authorized)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // CheckIdentifier retrieves the PublicKey from the JSON data file.
 | |
| func (h *Handlers) CheckIdentifier(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	publicKey, _ := store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 
 | |
| 	res.Content = string(publicKey)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // Authorize attempts to unlock the next sequential nodes by verifying the provided PIN against the already set PIN.
 | |
| // It sets the required flags that control the flow.
 | |
| func (h *Handlers) Authorize(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	flag_incorrect_pin, _ := h.flagManager.GetFlag("flag_incorrect_pin")
 | |
| 	flag_account_authorized, _ := h.flagManager.GetFlag("flag_account_authorized")
 | |
| 	flag_allow_update, _ := h.flagManager.GetFlag("flag_allow_update")
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	AccountPin, err := store.ReadEntry(ctx, sessionId, common.DATA_ACCOUNT_PIN)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read AccountPin entry with", "key", common.DATA_ACCOUNT_PIN, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	if len(input) == 4 {
 | |
| 		if common.VerifyPIN(string(AccountPin), string(input)) {
 | |
| 			if h.st.MatchFlag(flag_account_authorized, false) {
 | |
| 				res.FlagReset = append(res.FlagReset, flag_incorrect_pin)
 | |
| 				res.FlagSet = append(res.FlagSet, flag_allow_update, flag_account_authorized)
 | |
| 			} else {
 | |
| 				res.FlagSet = append(res.FlagSet, flag_allow_update)
 | |
| 				res.FlagReset = append(res.FlagReset, flag_account_authorized)
 | |
| 			}
 | |
| 		} else {
 | |
| 			res.FlagSet = append(res.FlagSet, flag_incorrect_pin)
 | |
| 			res.FlagReset = append(res.FlagReset, flag_account_authorized)
 | |
| 			return res, nil
 | |
| 		}
 | |
| 	} else {
 | |
| 		return res, nil
 | |
| 	}
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetIncorrectPin resets the incorrect pin flag  after a new PIN attempt.
 | |
| func (h *Handlers) ResetIncorrectPin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	flag_incorrect_pin, _ := h.flagManager.GetFlag("flag_incorrect_pin")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_incorrect_pin)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // Setback sets the flag_back_set flag when the navigation is back.
 | |
| func (h *Handlers) SetBack(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	//TODO:
 | |
| 	//Add check if the navigation is lateral nav instead of checking the input.
 | |
| 	if string(input) == "0" {
 | |
| 		flag_back_set, _ := h.flagManager.GetFlag("flag_back_set")
 | |
| 		res.FlagSet = append(res.FlagSet, flag_back_set)
 | |
| 	}
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // CheckAccountStatus queries the API using the TrackingId and sets flags
 | |
| // based on the account status.
 | |
| func (h *Handlers) CheckAccountStatus(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	flag_account_success, _ := h.flagManager.GetFlag("flag_account_success")
 | |
| 	flag_account_pending, _ := h.flagManager.GetFlag("flag_account_pending")
 | |
| 	flag_api_error, _ := h.flagManager.GetFlag("flag_api_call_error")
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	publicKey, err := store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read publicKey entry with", "key", common.DATA_PUBLIC_KEY, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	r, err := h.accountService.TrackAccountStatus(ctx, string(publicKey))
 | |
| 	if err != nil {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_api_error)
 | |
| 		logg.ErrorCtxf(ctx, "failed on TrackAccountStatus", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_api_error)
 | |
| 
 | |
| 	if r.Active {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_account_success)
 | |
| 		res.FlagReset = append(res.FlagReset, flag_account_pending)
 | |
| 	} else {
 | |
| 		res.FlagReset = append(res.FlagReset, flag_account_success)
 | |
| 		res.FlagSet = append(res.FlagSet, flag_account_pending)
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // Quit displays the Thank you message and exits the menu.
 | |
| func (h *Handlers) Quit(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	flag_account_authorized, _ := h.flagManager.GetFlag("flag_account_authorized")
 | |
| 
 | |
| 	code := codeFromCtx(ctx)
 | |
| 	l := gotext.NewLocale(translationDir, code)
 | |
| 	l.AddDomain("default")
 | |
| 
 | |
| 	res.Content = l.Get("Thank you for using Sarafu. Goodbye!")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_account_authorized)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // QuitWithHelp displays helpline information then exits the menu.
 | |
| func (h *Handlers) QuitWithHelp(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	flag_account_authorized, _ := h.flagManager.GetFlag("flag_account_authorized")
 | |
| 
 | |
| 	code := codeFromCtx(ctx)
 | |
| 	l := gotext.NewLocale(translationDir, code)
 | |
| 	l.AddDomain("default")
 | |
| 
 | |
| 	res.Content = l.Get("For more help,please call: 0757628885")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_account_authorized)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // VerifyYob verifies the length of the given input.
 | |
| func (h *Handlers) VerifyYob(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	flag_incorrect_date_format, _ := h.flagManager.GetFlag("flag_incorrect_date_format")
 | |
| 	date := string(input)
 | |
| 	_, err = strconv.Atoi(date)
 | |
| 	if err != nil {
 | |
| 		// If conversion fails, input is not numeric
 | |
| 		res.FlagSet = append(res.FlagSet, flag_incorrect_date_format)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	if utils.IsValidYOb(date) {
 | |
| 		res.FlagReset = append(res.FlagReset, flag_incorrect_date_format)
 | |
| 	} else {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_incorrect_date_format)
 | |
| 	}
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetIncorrectYob resets the incorrect date format flag after a new attempt.
 | |
| func (h *Handlers) ResetIncorrectYob(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	flag_incorrect_date_format, _ := h.flagManager.GetFlag("flag_incorrect_date_format")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_incorrect_date_format)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // CheckBalance retrieves the balance of the active voucher and sets
 | |
| // the balance as the result content.
 | |
| func (h *Handlers) CheckBalance(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	code := codeFromCtx(ctx)
 | |
| 	l := gotext.NewLocale(translationDir, code)
 | |
| 	l.AddDomain("default")
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	// get the active sym and active balance
 | |
| 	activeSym, err := store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_SYM)
 | |
| 	if err != nil {
 | |
| 		if db.IsNotFound(err) {
 | |
| 			balance := "0.00"
 | |
| 			res.Content = l.Get("Balance: %s\n", balance)
 | |
| 			return res, nil
 | |
| 		}
 | |
| 
 | |
| 		logg.ErrorCtxf(ctx, "failed to read activeSym entry with", "key", common.DATA_ACTIVE_SYM, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	activeBal, err := store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_BAL)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read activeBal entry with", "key", common.DATA_ACTIVE_BAL, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Convert activeBal from []byte to float64
 | |
| 	balFloat, err := strconv.ParseFloat(string(activeBal), 64)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to parse activeBal as float", "value", string(activeBal), "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Format to 2 decimal places
 | |
| 	balStr := fmt.Sprintf("%.2f %s", balFloat, activeSym)
 | |
| 
 | |
| 	res.Content = l.Get("Balance: %s\n", balStr)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // FetchCommunityBalance retrieves and displays the balance for community accounts in user's preferred language.
 | |
| func (h *Handlers) FetchCommunityBalance(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	// retrieve the language code from the context
 | |
| 	code := codeFromCtx(ctx)
 | |
| 	// Initialize the localization system with the appropriate translation directory
 | |
| 	l := gotext.NewLocale(translationDir, code)
 | |
| 	l.AddDomain("default")
 | |
| 	//TODO:
 | |
| 	//Check if the address is a community account,if then,get the actual balance
 | |
| 	res.Content = l.Get("Community Balance: 0.00")
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetOthersPin handles the PIN reset process for other users' accounts by:
 | |
| // 1. Retrieving the blocked phone number from the session
 | |
| // 2. Fetching the temporary PIN associated with that number
 | |
| // 3. Updating the account PIN with the temporary PIN
 | |
| func (h *Handlers) ResetOthersPin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	store := h.userdataStore
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	blockedPhonenumber, err := store.ReadEntry(ctx, sessionId, common.DATA_BLOCKED_NUMBER)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read blockedPhonenumber entry with", "key", common.DATA_BLOCKED_NUMBER, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	temporaryPin, err := store.ReadEntry(ctx, string(blockedPhonenumber), common.DATA_TEMPORARY_VALUE)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read temporaryPin entry with", "key", common.DATA_TEMPORARY_VALUE, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Hash the PIN
 | |
| 	hashedPIN, err := common.HashPIN(string(temporaryPin))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to hash temporaryPin", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	err = store.WriteEntry(ctx, string(blockedPhonenumber), common.DATA_ACCOUNT_PIN, []byte(hashedPIN))
 | |
| 	if err != nil {
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetUnregisteredNumber clears the unregistered number flag in the system,
 | |
| // indicating that a number's registration status should no longer be marked as unregistered.
 | |
| func (h *Handlers) ResetUnregisteredNumber(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	flag_unregistered_number, _ := h.flagManager.GetFlag("flag_unregistered_number")
 | |
| 	res.FlagReset = append(res.FlagReset, flag_unregistered_number)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ValidateBlockedNumber performs validation of phone numbers, specifically for blocked numbers in the system.
 | |
| // It checks phone number format and verifies registration status.
 | |
| func (h *Handlers) ValidateBlockedNumber(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	flag_unregistered_number, _ := h.flagManager.GetFlag("flag_unregistered_number")
 | |
| 	store := h.userdataStore
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	blockedNumber := string(input)
 | |
| 	_, err = store.ReadEntry(ctx, blockedNumber, common.DATA_PUBLIC_KEY)
 | |
| 	if !common.IsValidPhoneNumber(blockedNumber) {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_unregistered_number)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 	if err != nil {
 | |
| 		if db.IsNotFound(err) {
 | |
| 			logg.InfoCtxf(ctx, "Invalid or unregistered number")
 | |
| 			res.FlagSet = append(res.FlagSet, flag_unregistered_number)
 | |
| 			return res, nil
 | |
| 		} else {
 | |
| 			logg.ErrorCtxf(ctx, "Error on ValidateBlockedNumber", "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 	}
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_BLOCKED_NUMBER, []byte(blockedNumber))
 | |
| 	if err != nil {
 | |
| 		return res, nil
 | |
| 	}
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ValidateRecipient validates that the given input is valid.
 | |
| func (h *Handlers) ValidateRecipient(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_invalid_recipient, _ := h.flagManager.GetFlag("flag_invalid_recipient")
 | |
| 	flag_invalid_recipient_with_invite, _ := h.flagManager.GetFlag("flag_invalid_recipient_with_invite")
 | |
| 	flag_api_error, _ := h.flagManager.GetFlag("flag_api_call_error")
 | |
| 
 | |
| 	recipient := string(input)
 | |
| 
 | |
| 	if recipient != "0" {
 | |
| 		recipientType, err := common.CheckRecipient(recipient)
 | |
| 		if err != nil {
 | |
| 			// Invalid recipient format (not a phone number, address, or valid alias format)
 | |
| 			res.FlagSet = append(res.FlagSet, flag_invalid_recipient)
 | |
| 			res.Content = recipient
 | |
| 
 | |
| 			return res, nil
 | |
| 		}
 | |
| 
 | |
| 		// save the recipient as the temporaryRecipient
 | |
| 		err = store.WriteEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE, []byte(recipient))
 | |
| 		if err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write temporaryRecipient entry with", "key", common.DATA_TEMPORARY_VALUE, "value", recipient, "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 
 | |
| 		switch recipientType {
 | |
| 		case "phone number":
 | |
| 			// format the phone number
 | |
| 			formattedNumber, err := common.FormatPhoneNumber(recipient)
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "Failed to format the phone number: %s", recipient, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 
 | |
| 			// Check if the phone number is registered
 | |
| 			publicKey, err := store.ReadEntry(ctx, formattedNumber, common.DATA_PUBLIC_KEY)
 | |
| 			if err != nil {
 | |
| 				if db.IsNotFound(err) {
 | |
| 					logg.InfoCtxf(ctx, "Unregistered phone number: %s", recipient)
 | |
| 					res.FlagSet = append(res.FlagSet, flag_invalid_recipient_with_invite)
 | |
| 					res.Content = recipient
 | |
| 					return res, nil
 | |
| 				}
 | |
| 
 | |
| 				logg.ErrorCtxf(ctx, "failed to read publicKey entry with", "key", common.DATA_PUBLIC_KEY, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 
 | |
| 			// Save the publicKey as the recipient
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_RECIPIENT, publicKey)
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write recipient entry with", "key", common.DATA_RECIPIENT, "value", string(publicKey), "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 
 | |
| 		case "address":
 | |
| 			// Save the valid Ethereum address as the recipient
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_RECIPIENT, []byte(recipient))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write recipient entry with", "key", common.DATA_RECIPIENT, "value", recipient, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 
 | |
| 		case "alias":
 | |
| 			// Call the API to validate and retrieve the address for the alias
 | |
| 			r, aliasErr := h.accountService.CheckAliasAddress(ctx, recipient)
 | |
| 			if aliasErr != nil {
 | |
| 				res.FlagSet = append(res.FlagSet, flag_api_error)
 | |
| 				res.Content = recipient
 | |
| 
 | |
| 				logg.ErrorCtxf(ctx, "failed on CheckAliasAddress", "error", aliasErr)
 | |
| 				return res, err
 | |
| 			}
 | |
| 
 | |
| 			// Alias validation succeeded, save the Ethereum address
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_RECIPIENT, []byte(r.Address))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write recipient entry with", "key", common.DATA_RECIPIENT, "value", r.Address, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // TransactionReset resets the previous transaction data (Recipient and Amount)
 | |
| // as well as the invalid flags.
 | |
| func (h *Handlers) TransactionReset(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_invalid_recipient, _ := h.flagManager.GetFlag("flag_invalid_recipient")
 | |
| 	flag_invalid_recipient_with_invite, _ := h.flagManager.GetFlag("flag_invalid_recipient_with_invite")
 | |
| 	store := h.userdataStore
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_AMOUNT, []byte(""))
 | |
| 	if err != nil {
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_RECIPIENT, []byte(""))
 | |
| 	if err != nil {
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_invalid_recipient, flag_invalid_recipient_with_invite)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // InviteValidRecipient sends an invitation to the valid phone number.
 | |
| func (h *Handlers) InviteValidRecipient(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	code := codeFromCtx(ctx)
 | |
| 	l := gotext.NewLocale(translationDir, code)
 | |
| 	l.AddDomain("default")
 | |
| 
 | |
| 	recipient, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 
 | |
| 	// TODO
 | |
| 	// send an invitation SMS
 | |
| 	// if successful
 | |
| 	// res.Content = l.Get("Your invitation to %s to join Sarafu Network has been sent.",  string(recipient))
 | |
| 
 | |
| 	res.Content = l.Get("Your invite request for %s to Sarafu Network failed. Please try again later.", string(recipient))
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ResetTransactionAmount resets the transaction amount and invalid flag.
 | |
| func (h *Handlers) ResetTransactionAmount(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_invalid_amount, _ := h.flagManager.GetFlag("flag_invalid_amount")
 | |
| 	store := h.userdataStore
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_AMOUNT, []byte(""))
 | |
| 	if err != nil {
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_invalid_amount)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // MaxAmount gets the current balance from the API and sets it as
 | |
| // the result content.
 | |
| func (h *Handlers) MaxAmount(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	activeBal, err := store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_BAL)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read activeBal entry with", "key", common.DATA_ACTIVE_BAL, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	res.Content = string(activeBal)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ValidateAmount ensures that the given input is a valid amount and that
 | |
| // it is not more than the current balance.
 | |
| func (h *Handlers) ValidateAmount(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	flag_invalid_amount, _ := h.flagManager.GetFlag("flag_invalid_amount")
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	var balanceValue float64
 | |
| 
 | |
| 	// retrieve the active balance
 | |
| 	activeBal, err := store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_BAL)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read activeBal entry with", "key", common.DATA_ACTIVE_BAL, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	balanceValue, err = strconv.ParseFloat(string(activeBal), 64)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "Failed to convert the activeBal to a float", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Extract numeric part from the input amount
 | |
| 	amountStr := strings.TrimSpace(string(input))
 | |
| 	inputAmount, err := strconv.ParseFloat(amountStr, 64)
 | |
| 	if err != nil {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_invalid_amount)
 | |
| 		res.Content = amountStr
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	if inputAmount > balanceValue {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_invalid_amount)
 | |
| 		res.Content = amountStr
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	// Format the amount with 2 decimal places before saving
 | |
| 	formattedAmount := fmt.Sprintf("%.2f", inputAmount)
 | |
| 	err = store.WriteEntry(ctx, sessionId, common.DATA_AMOUNT, []byte(formattedAmount))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to write amount entry with", "key", common.DATA_AMOUNT, "value", formattedAmount, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	res.Content = formattedAmount
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetRecipient returns the transaction recipient phone number from the gdbm.
 | |
| func (h *Handlers) GetRecipient(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	recipient, _ := store.ReadEntry(ctx, sessionId, common.DATA_TEMPORARY_VALUE)
 | |
| 
 | |
| 	res.Content = string(recipient)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // RetrieveBlockedNumber gets the current number during the pin reset for other's is in progress.
 | |
| func (h *Handlers) RetrieveBlockedNumber(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	blockedNumber, _ := store.ReadEntry(ctx, sessionId, common.DATA_BLOCKED_NUMBER)
 | |
| 
 | |
| 	res.Content = string(blockedNumber)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetSender returns the sessionId (phoneNumber).
 | |
| func (h *Handlers) GetSender(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	res.Content = sessionId
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetAmount retrieves the amount from teh Gdbm Db.
 | |
| func (h *Handlers) GetAmount(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	// retrieve the active symbol
 | |
| 	activeSym, err := store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_SYM)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read activeSym entry with", "key", common.DATA_ACTIVE_SYM, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	amount, _ := store.ReadEntry(ctx, sessionId, common.DATA_AMOUNT)
 | |
| 
 | |
| 	res.Content = fmt.Sprintf("%s %s", string(amount), string(activeSym))
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // InitiateTransaction calls the TokenTransfer and returns a confirmation based on the result.
 | |
| func (h *Handlers) InitiateTransaction(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_account_authorized, _ := h.flagManager.GetFlag("flag_account_authorized")
 | |
| 
 | |
| 	code := codeFromCtx(ctx)
 | |
| 	l := gotext.NewLocale(translationDir, code)
 | |
| 	l.AddDomain("default")
 | |
| 
 | |
| 	data, err := common.ReadTransactionData(ctx, h.userdataStore, sessionId)
 | |
| 	if err != nil {
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	finalAmountStr, err := common.ParseAndScaleAmount(data.Amount, data.ActiveDecimal)
 | |
| 	if err != nil {
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Call TokenTransfer
 | |
| 	r, err := h.accountService.TokenTransfer(ctx, finalAmountStr, data.PublicKey, data.Recipient, data.ActiveAddress)
 | |
| 	if err != nil {
 | |
| 		flag_api_error, _ := h.flagManager.GetFlag("flag_api_call_error")
 | |
| 		res.FlagSet = append(res.FlagSet, flag_api_error)
 | |
| 		res.Content = l.Get("Your request failed. Please try again later.")
 | |
| 		logg.ErrorCtxf(ctx, "failed on TokenTransfer", "error", err)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	trackingId := r.TrackingId
 | |
| 	logg.InfoCtxf(ctx, "TokenTransfer", "trackingId", trackingId)
 | |
| 
 | |
| 	res.Content = l.Get(
 | |
| 		"Your request has been sent. %s will receive %s %s from %s.",
 | |
| 		data.TemporaryValue,
 | |
| 		data.Amount,
 | |
| 		data.ActiveSym,
 | |
| 		sessionId,
 | |
| 	)
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_account_authorized)
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetCurrentProfileInfo retrieves specific profile fields based on the current state of the USSD session.
 | |
| // Uses flag management system to track profile field status and handle menu navigation.
 | |
| func (h *Handlers) GetCurrentProfileInfo(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var profileInfo []byte
 | |
| 	var defaultValue string
 | |
| 	var err error
 | |
| 
 | |
| 	flag_firstname_set, _ := h.flagManager.GetFlag("flag_firstname_set")
 | |
| 	flag_familyname_set, _ := h.flagManager.GetFlag("flag_familyname_set")
 | |
| 	flag_yob_set, _ := h.flagManager.GetFlag("flag_yob_set")
 | |
| 	flag_gender_set, _ := h.flagManager.GetFlag("flag_gender_set")
 | |
| 	flag_location_set, _ := h.flagManager.GetFlag("flag_location_set")
 | |
| 	flag_offerings_set, _ := h.flagManager.GetFlag("flag_offerings_set")
 | |
| 	flag_back_set, _ := h.flagManager.GetFlag("flag_back_set")
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_back_set)
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	language, ok := ctx.Value("Language").(lang.Language)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("value for 'Language' is not of type lang.Language")
 | |
| 	}
 | |
| 	code := language.Code
 | |
| 	if code == "swa" {
 | |
| 		defaultValue = "Haipo"
 | |
| 	} else {
 | |
| 		defaultValue = "Not Provided"
 | |
| 	}
 | |
| 
 | |
| 	sm, _ := h.st.Where()
 | |
| 	parts := strings.SplitN(sm, "_", 2)
 | |
| 	filename := parts[1]
 | |
| 	dbKeyStr := "DATA_" + strings.ToUpper(filename)
 | |
| 	dbKey, err := common.StringToDataTyp(dbKeyStr)
 | |
| 
 | |
| 	if err != nil {
 | |
| 		return res, err
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	switch dbKey {
 | |
| 	case common.DATA_FIRST_NAME:
 | |
| 		profileInfo, err = store.ReadEntry(ctx, sessionId, common.DATA_FIRST_NAME)
 | |
| 		if err != nil {
 | |
| 			if db.IsNotFound(err) {
 | |
| 				res.Content = defaultValue
 | |
| 				break
 | |
| 			}
 | |
| 			logg.ErrorCtxf(ctx, "Failed to read first name entry with", "key", "error", common.DATA_FIRST_NAME, err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_firstname_set)
 | |
| 		res.Content = string(profileInfo)
 | |
| 	case common.DATA_FAMILY_NAME:
 | |
| 		profileInfo, err = store.ReadEntry(ctx, sessionId, common.DATA_FAMILY_NAME)
 | |
| 		if err != nil {
 | |
| 			if db.IsNotFound(err) {
 | |
| 				res.Content = defaultValue
 | |
| 				break
 | |
| 			}
 | |
| 			logg.ErrorCtxf(ctx, "Failed to read family name entry with", "key", "error", common.DATA_FAMILY_NAME, err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_familyname_set)
 | |
| 		res.Content = string(profileInfo)
 | |
| 
 | |
| 	case common.DATA_GENDER:
 | |
| 		profileInfo, err = store.ReadEntry(ctx, sessionId, common.DATA_GENDER)
 | |
| 		if err != nil {
 | |
| 			if db.IsNotFound(err) {
 | |
| 				res.Content = defaultValue
 | |
| 				break
 | |
| 			}
 | |
| 			logg.ErrorCtxf(ctx, "Failed to read gender entry with", "key", "error", common.DATA_GENDER, err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_gender_set)
 | |
| 		res.Content = string(profileInfo)
 | |
| 	case common.DATA_YOB:
 | |
| 		profileInfo, err = store.ReadEntry(ctx, sessionId, common.DATA_YOB)
 | |
| 		if err != nil {
 | |
| 			if db.IsNotFound(err) {
 | |
| 				res.Content = defaultValue
 | |
| 				break
 | |
| 			}
 | |
| 			logg.ErrorCtxf(ctx, "Failed to read year of birth(yob) entry with", "key", "error", common.DATA_YOB, err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_yob_set)
 | |
| 		res.Content = string(profileInfo)
 | |
| 	case common.DATA_LOCATION:
 | |
| 		profileInfo, err = store.ReadEntry(ctx, sessionId, common.DATA_LOCATION)
 | |
| 		if err != nil {
 | |
| 			if db.IsNotFound(err) {
 | |
| 				res.Content = defaultValue
 | |
| 				break
 | |
| 			}
 | |
| 			logg.ErrorCtxf(ctx, "Failed to read location entry with", "key", "error", common.DATA_LOCATION, err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_location_set)
 | |
| 		res.Content = string(profileInfo)
 | |
| 	case common.DATA_OFFERINGS:
 | |
| 		profileInfo, err = store.ReadEntry(ctx, sessionId, common.DATA_OFFERINGS)
 | |
| 		if err != nil {
 | |
| 			if db.IsNotFound(err) {
 | |
| 				res.Content = defaultValue
 | |
| 				break
 | |
| 			}
 | |
| 			logg.ErrorCtxf(ctx, "Failed to read offerings entry with", "key", "error", common.DATA_OFFERINGS, err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 		res.FlagSet = append(res.FlagSet, flag_offerings_set)
 | |
| 		res.Content = string(profileInfo)
 | |
| 	default:
 | |
| 		break
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetProfileInfo provides a comprehensive view of a user's profile.
 | |
| func (h *Handlers) GetProfileInfo(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var defaultValue string
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	language, ok := ctx.Value("Language").(lang.Language)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("value for 'Language' is not of type lang.Language")
 | |
| 	}
 | |
| 	code := language.Code
 | |
| 	if code == "swa" {
 | |
| 		defaultValue = "Haipo"
 | |
| 	} else {
 | |
| 		defaultValue = "Not Provided"
 | |
| 	}
 | |
| 
 | |
| 	// Helper function to handle nil byte slices and convert them to string
 | |
| 	getEntryOrDefault := func(entry []byte, err error) string {
 | |
| 		if err != nil || entry == nil {
 | |
| 			return defaultValue
 | |
| 		}
 | |
| 		return string(entry)
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	// Retrieve user data as strings with fallback to defaultValue
 | |
| 	firstName := getEntryOrDefault(store.ReadEntry(ctx, sessionId, common.DATA_FIRST_NAME))
 | |
| 	familyName := getEntryOrDefault(store.ReadEntry(ctx, sessionId, common.DATA_FAMILY_NAME))
 | |
| 	yob := getEntryOrDefault(store.ReadEntry(ctx, sessionId, common.DATA_YOB))
 | |
| 	gender := getEntryOrDefault(store.ReadEntry(ctx, sessionId, common.DATA_GENDER))
 | |
| 	location := getEntryOrDefault(store.ReadEntry(ctx, sessionId, common.DATA_LOCATION))
 | |
| 	offerings := getEntryOrDefault(store.ReadEntry(ctx, sessionId, common.DATA_OFFERINGS))
 | |
| 
 | |
| 	// Construct the full name
 | |
| 	name := utils.ConstructName(firstName, familyName, defaultValue)
 | |
| 
 | |
| 	// Calculate age from year of birth
 | |
| 	age := defaultValue
 | |
| 	if yob != defaultValue {
 | |
| 		if yobInt, err := strconv.Atoi(yob); err == nil {
 | |
| 			age = strconv.Itoa(utils.CalculateAgeWithYOB(yobInt))
 | |
| 		} else {
 | |
| 			return res, fmt.Errorf("invalid year of birth: %v", err)
 | |
| 		}
 | |
| 	}
 | |
| 	switch language.Code {
 | |
| 	case "eng":
 | |
| 		res.Content = fmt.Sprintf(
 | |
| 			"Name: %s\nGender: %s\nAge: %s\nLocation: %s\nYou provide: %s\n",
 | |
| 			name, gender, age, location, offerings,
 | |
| 		)
 | |
| 	case "swa":
 | |
| 		res.Content = fmt.Sprintf(
 | |
| 			"Jina: %s\nJinsia: %s\nUmri: %s\nEneo: %s\nUnauza: %s\n",
 | |
| 			name, gender, age, location, offerings,
 | |
| 		)
 | |
| 	default:
 | |
| 		res.Content = fmt.Sprintf(
 | |
| 			"Name: %s\nGender: %s\nAge: %s\nLocation: %s\nYou provide: %s\n",
 | |
| 			name, gender, age, location, offerings,
 | |
| 		)
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SetDefaultVoucher retrieves the current vouchers
 | |
| // and sets the first as the default voucher, if no active voucher is set.
 | |
| func (h *Handlers) SetDefaultVoucher(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	var err error
 | |
| 	store := h.userdataStore
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_no_active_voucher, _ := h.flagManager.GetFlag("flag_no_active_voucher")
 | |
| 
 | |
| 	// check if the user has an active sym
 | |
| 	_, err = store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_SYM)
 | |
| 
 | |
| 	if err != nil {
 | |
| 		if db.IsNotFound(err) {
 | |
| 			publicKey, err := store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to read publicKey entry with", "key", common.DATA_PUBLIC_KEY, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 
 | |
| 			// Fetch vouchers from the API using the public key
 | |
| 			vouchersResp, err := h.accountService.FetchVouchers(ctx, string(publicKey))
 | |
| 			if err != nil {
 | |
| 				res.FlagSet = append(res.FlagSet, flag_no_active_voucher)
 | |
| 				return res, nil
 | |
| 			}
 | |
| 
 | |
| 			// Return if there is no voucher
 | |
| 			if len(vouchersResp) == 0 {
 | |
| 				res.FlagSet = append(res.FlagSet, flag_no_active_voucher)
 | |
| 				return res, nil
 | |
| 			}
 | |
| 
 | |
| 			// Use only the first voucher
 | |
| 			firstVoucher := vouchersResp[0]
 | |
| 			defaultSym := firstVoucher.TokenSymbol
 | |
| 			defaultBal := firstVoucher.Balance
 | |
| 			defaultDec := firstVoucher.TokenDecimals
 | |
| 			defaultAddr := firstVoucher.ContractAddress
 | |
| 
 | |
| 			// Scale down the balance
 | |
| 			scaledBalance := common.ScaleDownBalance(defaultBal, defaultDec)
 | |
| 
 | |
| 			// set the active symbol
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_ACTIVE_SYM, []byte(defaultSym))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write defaultSym entry with", "key", common.DATA_ACTIVE_SYM, "value", defaultSym, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 			// set the active balance
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_ACTIVE_BAL, []byte(scaledBalance))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write defaultBal entry with", "key", common.DATA_ACTIVE_BAL, "value", scaledBalance, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 			// set the active decimals
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_ACTIVE_DECIMAL, []byte(defaultDec))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write defaultDec entry with", "key", common.DATA_ACTIVE_DECIMAL, "value", defaultDec, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 			// set the active contract address
 | |
| 			err = store.WriteEntry(ctx, sessionId, common.DATA_ACTIVE_ADDRESS, []byte(defaultAddr))
 | |
| 			if err != nil {
 | |
| 				logg.ErrorCtxf(ctx, "failed to write defaultAddr entry with", "key", common.DATA_ACTIVE_ADDRESS, "value", defaultAddr, "error", err)
 | |
| 				return res, err
 | |
| 			}
 | |
| 
 | |
| 			return res, nil
 | |
| 		}
 | |
| 
 | |
| 		logg.ErrorCtxf(ctx, "failed to read activeSym entry with", "key", common.DATA_ACTIVE_SYM, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_no_active_voucher)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // CheckVouchers retrieves the token holdings from the API using the "PublicKey" and stores
 | |
| // them to gdbm.
 | |
| func (h *Handlers) CheckVouchers(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	publicKey, err := store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read publicKey entry with", "key", common.DATA_PUBLIC_KEY, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Fetch vouchers from the API using the public key
 | |
| 	vouchersResp, err := h.accountService.FetchVouchers(ctx, string(publicKey))
 | |
| 	if err != nil {
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	// check the current active sym and update the data
 | |
| 	activeSym, _ := store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_SYM)
 | |
| 	if activeSym != nil {
 | |
| 		activeSymStr := string(activeSym)
 | |
| 
 | |
| 		// Find the matching voucher data
 | |
| 		var activeData *dataserviceapi.TokenHoldings
 | |
| 		for _, voucher := range vouchersResp {
 | |
| 			if voucher.TokenSymbol == activeSymStr {
 | |
| 				activeData = &voucher
 | |
| 				break
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		if activeData == nil {
 | |
| 			logg.ErrorCtxf(ctx, "activeSym not found in vouchers", "activeSym", activeSymStr)
 | |
| 			return res, fmt.Errorf("activeSym %s not found in vouchers", activeSymStr)
 | |
| 		}
 | |
| 
 | |
| 		// Scale down the balance
 | |
| 		scaledBalance := common.ScaleDownBalance(activeData.Balance, activeData.TokenDecimals)
 | |
| 
 | |
| 		// Update the balance field with the scaled value
 | |
| 		activeData.Balance = scaledBalance
 | |
| 
 | |
| 		// Pass the matching voucher data to UpdateVoucherData
 | |
| 		if err := common.UpdateVoucherData(ctx, h.userdataStore, sessionId, activeData); err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed on UpdateVoucherData", "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	data := common.ProcessVouchers(vouchersResp)
 | |
| 
 | |
| 	// Store all voucher data
 | |
| 	dataMap := map[common.DataTyp]string{
 | |
| 		common.DATA_VOUCHER_SYMBOLS:   data.Symbols,
 | |
| 		common.DATA_VOUCHER_BALANCES:  data.Balances,
 | |
| 		common.DATA_VOUCHER_DECIMALS:  data.Decimals,
 | |
| 		common.DATA_VOUCHER_ADDRESSES: data.Addresses,
 | |
| 	}
 | |
| 
 | |
| 	for key, value := range dataMap {
 | |
| 		if err := h.prefixDb.Put(ctx, []byte(common.ToBytes(key)), []byte(value)); err != nil {
 | |
| 			return res, nil
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetVoucherList fetches the list of vouchers and formats them.
 | |
| func (h *Handlers) GetVoucherList(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	// Read vouchers from the store
 | |
| 	voucherData, err := h.prefixDb.Get(ctx, common.ToBytes(common.DATA_VOUCHER_SYMBOLS))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "Failed to read the voucherData from prefixDb", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	formattedData := h.ReplaceSeparatorFunc(string(voucherData))
 | |
| 
 | |
| 	res.Content = string(formattedData)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ViewVoucher retrieves the token holding and balance from the subprefixDB
 | |
| // and displays it to the user for them to select it.
 | |
| func (h *Handlers) ViewVoucher(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	code := codeFromCtx(ctx)
 | |
| 	l := gotext.NewLocale(translationDir, code)
 | |
| 	l.AddDomain("default")
 | |
| 
 | |
| 	flag_incorrect_voucher, _ := h.flagManager.GetFlag("flag_incorrect_voucher")
 | |
| 
 | |
| 	inputStr := string(input)
 | |
| 	if inputStr == "0" || inputStr == "99" {
 | |
| 		res.FlagReset = append(res.FlagReset, flag_incorrect_voucher)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	metadata, err := common.GetVoucherData(ctx, h.prefixDb, inputStr)
 | |
| 	if err != nil {
 | |
| 		return res, fmt.Errorf("failed to retrieve voucher data: %v", err)
 | |
| 	}
 | |
| 
 | |
| 	if metadata == nil {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_incorrect_voucher)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	if err := common.StoreTemporaryVoucher(ctx, h.userdataStore, sessionId, metadata); err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed on StoreTemporaryVoucher", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_incorrect_voucher)
 | |
| 	res.Content = l.Get("Symbol: %s\nBalance: %s", metadata.TokenSymbol, metadata.Balance)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // SetVoucher retrieves the temp voucher data and sets it as the active data.
 | |
| func (h *Handlers) SetVoucher(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	// Get temporary data
 | |
| 	tempData, err := common.GetTemporaryVoucherData(ctx, h.userdataStore, sessionId)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed on GetTemporaryVoucherData", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Set as active and clear temporary data
 | |
| 	if err := common.UpdateVoucherData(ctx, h.userdataStore, sessionId, tempData); err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed on UpdateVoucherData", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	res.Content = tempData.TokenSymbol
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetVoucherDetails retrieves the voucher details.
 | |
| func (h *Handlers) GetVoucherDetails(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	store := h.userdataStore
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_api_error, _ := h.flagManager.GetFlag("flag_api_call_error")
 | |
| 
 | |
| 	// get the active address
 | |
| 	activeAddress, err := store.ReadEntry(ctx, sessionId, common.DATA_ACTIVE_ADDRESS)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read activeAddress entry with", "key", common.DATA_ACTIVE_ADDRESS, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// use the voucher contract address to get the data from the API
 | |
| 	voucherData, err := h.accountService.VoucherData(ctx, string(activeAddress))
 | |
| 	if err != nil {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_api_error)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	res.Content = fmt.Sprintf(
 | |
| 		"Name: %s\nSymbol: %s\nCommodity: %s\nLocation: %s", voucherData.TokenName, voucherData.TokenSymbol, voucherData.TokenCommodity, voucherData.TokenLocation,
 | |
| 	)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // CheckTransactions retrieves the transactions from the API using the "PublicKey" and stores to prefixDb.
 | |
| func (h *Handlers) CheckTransactions(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	flag_no_transfers, _ := h.flagManager.GetFlag("flag_no_transfers")
 | |
| 	flag_api_error, _ := h.flagManager.GetFlag("flag_api_error")
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	publicKey, err := store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read publicKey entry with", "key", common.DATA_PUBLIC_KEY, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Fetch transactions from the API using the public key
 | |
| 	transactionsResp, err := h.accountService.FetchTransactions(ctx, string(publicKey))
 | |
| 	if err != nil {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_api_error)
 | |
| 		logg.ErrorCtxf(ctx, "failed on FetchTransactions", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Return if there are no transactions
 | |
| 	if len(transactionsResp) == 0 {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_no_transfers)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	data := common.ProcessTransfers(transactionsResp)
 | |
| 
 | |
| 	// Store all transaction data
 | |
| 	dataMap := map[common.DataTyp]string{
 | |
| 		common.DATA_TX_SENDERS:    data.Senders,
 | |
| 		common.DATA_TX_RECIPIENTS: data.Recipients,
 | |
| 		common.DATA_TX_VALUES:     data.TransferValues,
 | |
| 		common.DATA_TX_ADDRESSES:  data.Addresses,
 | |
| 		common.DATA_TX_HASHES:     data.TxHashes,
 | |
| 		common.DATA_TX_DATES:      data.Dates,
 | |
| 		common.DATA_TX_SYMBOLS:    data.Symbols,
 | |
| 		common.DATA_TX_DECIMALS:   data.Decimals,
 | |
| 	}
 | |
| 
 | |
| 	for key, value := range dataMap {
 | |
| 		if err := h.prefixDb.Put(ctx, []byte(common.ToBytes(key)), []byte(value)); err != nil {
 | |
| 			logg.ErrorCtxf(ctx, "failed to write to prefixDb", "error", err)
 | |
| 			return res, err
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_no_transfers)
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // GetTransactionsList fetches the list of transactions and formats them.
 | |
| func (h *Handlers) GetTransactionsList(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 
 | |
| 	store := h.userdataStore
 | |
| 	publicKey, err := store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read publicKey entry with", "key", common.DATA_PUBLIC_KEY, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Read transactions from the store and format them
 | |
| 	TransactionSenders, err := h.prefixDb.Get(ctx, common.ToBytes(common.DATA_TX_SENDERS))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "Failed to read the TransactionSenders from prefixDb", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	TransactionSyms, err := h.prefixDb.Get(ctx, common.ToBytes(common.DATA_TX_SYMBOLS))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "Failed to read the TransactionSyms from prefixDb", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	TransactionValues, err := h.prefixDb.Get(ctx, common.ToBytes(common.DATA_TX_VALUES))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "Failed to read the TransactionValues from prefixDb", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 	TransactionDates, err := h.prefixDb.Get(ctx, common.ToBytes(common.DATA_TX_DATES))
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "Failed to read the TransactionDates from prefixDb", "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	// Parse the data
 | |
| 	senders := strings.Split(string(TransactionSenders), "\n")
 | |
| 	syms := strings.Split(string(TransactionSyms), "\n")
 | |
| 	values := strings.Split(string(TransactionValues), "\n")
 | |
| 	dates := strings.Split(string(TransactionDates), "\n")
 | |
| 
 | |
| 	var formattedTransactions []string
 | |
| 	for i := 0; i < len(senders); i++ {
 | |
| 		sender := strings.TrimSpace(senders[i])
 | |
| 		sym := strings.TrimSpace(syms[i])
 | |
| 		value := strings.TrimSpace(values[i])
 | |
| 		date := strings.Split(strings.TrimSpace(dates[i]), " ")[0]
 | |
| 
 | |
| 		status := "Received"
 | |
| 		if sender == string(publicKey) {
 | |
| 			status = "Sent"
 | |
| 		}
 | |
| 
 | |
| 		// Use the ReplaceSeparator function for the menu separator
 | |
| 		transactionLine := fmt.Sprintf("%d%s%s %s %s %s", i+1, h.ReplaceSeparatorFunc(":"), status, value, sym, date)
 | |
| 		formattedTransactions = append(formattedTransactions, transactionLine)
 | |
| 	}
 | |
| 
 | |
| 	res.Content = strings.Join(formattedTransactions, "\n")
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // ViewTransactionStatement retrieves the transaction statement
 | |
| // and displays it to the user.
 | |
| func (h *Handlers) ViewTransactionStatement(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	store := h.userdataStore
 | |
| 	publicKey, err := store.ReadEntry(ctx, sessionId, common.DATA_PUBLIC_KEY)
 | |
| 	if err != nil {
 | |
| 		logg.ErrorCtxf(ctx, "failed to read publicKey entry with", "key", common.DATA_PUBLIC_KEY, "error", err)
 | |
| 		return res, err
 | |
| 	}
 | |
| 
 | |
| 	flag_incorrect_statement, _ := h.flagManager.GetFlag("flag_incorrect_statement")
 | |
| 
 | |
| 	inputStr := string(input)
 | |
| 	if inputStr == "0" || inputStr == "99" || inputStr == "11" || inputStr == "22" {
 | |
| 		res.FlagReset = append(res.FlagReset, flag_incorrect_statement)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	// Convert input string to integer
 | |
| 	index, err := strconv.Atoi(strings.TrimSpace(inputStr))
 | |
| 	if err != nil {
 | |
| 		return res, fmt.Errorf("invalid input: must be a number between 1 and 10")
 | |
| 	}
 | |
| 
 | |
| 	if index < 1 || index > 10 {
 | |
| 		return res, fmt.Errorf("invalid input: index must be between 1 and 10")
 | |
| 	}
 | |
| 
 | |
| 	statement, err := common.GetTransferData(ctx, h.prefixDb, string(publicKey), index)
 | |
| 	if err != nil {
 | |
| 		return res, fmt.Errorf("failed to retrieve transfer data: %v", err)
 | |
| 	}
 | |
| 
 | |
| 	if statement == "" {
 | |
| 		res.FlagSet = append(res.FlagSet, flag_incorrect_statement)
 | |
| 		return res, nil
 | |
| 	}
 | |
| 
 | |
| 	res.FlagReset = append(res.FlagReset, flag_incorrect_statement)
 | |
| 	res.Content = statement
 | |
| 
 | |
| 	return res, nil
 | |
| }
 | |
| 
 | |
| // handles bulk updates of profile information.
 | |
| func (h *Handlers) insertProfileItems(ctx context.Context, sessionId string, res *resource.Result) error {
 | |
| 	var err error
 | |
| 	store := h.userdataStore
 | |
| 	profileFlagNames := []string{
 | |
| 		"flag_firstname_set",
 | |
| 		"flag_familyname_set",
 | |
| 		"flag_yob_set",
 | |
| 		"flag_gender_set",
 | |
| 		"flag_location_set",
 | |
| 		"flag_offerings_set",
 | |
| 	}
 | |
| 	profileDataKeys := []common.DataTyp{
 | |
| 		common.DATA_FIRST_NAME,
 | |
| 		common.DATA_FAMILY_NAME,
 | |
| 		common.DATA_GENDER,
 | |
| 		common.DATA_YOB,
 | |
| 		common.DATA_LOCATION,
 | |
| 		common.DATA_OFFERINGS,
 | |
| 	}
 | |
| 	for index, profileItem := range h.profile.ProfileItems {
 | |
| 		// Ensure the profileItem is not "0"(is set)
 | |
| 		if profileItem != "0" {
 | |
| 			flag, _ := h.flagManager.GetFlag(profileFlagNames[index])
 | |
| 			isProfileItemSet := h.st.MatchFlag(flag, true)
 | |
| 			if !isProfileItemSet {
 | |
| 				err = store.WriteEntry(ctx, sessionId, profileDataKeys[index], []byte(profileItem))
 | |
| 				if err != nil {
 | |
| 					logg.ErrorCtxf(ctx, "failed to write profile entry with", "key", profileDataKeys[index], "value", profileItem, "error", err)
 | |
| 					return err
 | |
| 				}
 | |
| 				res.FlagSet = append(res.FlagSet, flag)
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // UpdateAllProfileItems  is used to persist all the  new profile information and setup  the required profile flags.
 | |
| func (h *Handlers) UpdateAllProfileItems(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 | |
| 	var res resource.Result
 | |
| 	sessionId, ok := ctx.Value("SessionId").(string)
 | |
| 	if !ok {
 | |
| 		return res, fmt.Errorf("missing session")
 | |
| 	}
 | |
| 	err := h.insertProfileItems(ctx, sessionId, &res)
 | |
| 	if err != nil {
 | |
| 		return res, err
 | |
| 	}
 | |
| 	return res, nil
 | |
| }
 |