register single edit flag

cmd/main.go

@@ -12,8 +12,8 @@ import (
 	"git.defalsify.org/vise.git/persist"
 	"git.defalsify.org/vise.git/resource"
 	"git.defalsify.org/vise.git/state"
-	"git.grassecon.net/urdt/ussd/internal/models"
 	"git.grassecon.net/urdt/ussd/internal/handlers/ussd"
+	"git.grassecon.net/urdt/ussd/internal/models"
 )
 
 var (
@@ -33,7 +33,7 @@ func main() {
 	fmt.Fprintf(os.Stderr, "starting session at symbol '%s' using resource dir: %s\n", root, dir)
 
 	ctx := context.Background()
-	st := state.NewState(15)
+	st := state.NewState(16)
 	st.UseDebug()
 	state.FlagDebugger.Register(models.USERFLAG_LANGUAGE_SET, "LANGUAGE_CHANGE")
 	state.FlagDebugger.Register(models.USERFLAG_ACCOUNT_CREATED, "ACCOUNT_CREATED")
@@ -50,6 +50,7 @@ func main() {
 	state.FlagDebugger.Register(models.USERFLAG_VALIDPIN, "VALIDPIN")
 	state.FlagDebugger.Register(models.USERFLAG_VALIDPIN, "ACCOUNTUNLOCKED")
 	state.FlagDebugger.Register(models.USERFLAG_ACCOUNT_CREATION_FAILED, "ACCOUNT_CREATION_FAILED")
+	state.FlagDebugger.Register(models.USERFLAG_SINGLE_EDIT, "SINGLEEDIT")
 
 	rfs := resource.NewFsResource(scriptDir)
 	ca := cache.NewCache()
@@ -113,6 +114,7 @@ func main() {
 	rfs.AddLocalFunc("get_profile_info", ussdHandlers.GetProfileInfo)
 	rfs.AddLocalFunc("verify_yob", ussdHandlers.VerifyYob)
 	rfs.AddLocalFunc("reset_incorrect_date_format", ussdHandlers.ResetIncorrectYob)
+	rfs.AddLocalFunc("set_reset_single_edit", ussdHandlers.SetResetSingleEdit)
 
 	cont, err := en.Init(ctx)
 	en.SetDebugger(engine.NewSimpleDebug(nil))

internal/handlers/ussd/menuhandler.go

@@ -124,6 +124,24 @@ func (h *Handlers) SavePin(ctx context.Context, sym string, input []byte) (resou
 	return res, nil
 }
 
+func (h *Handlers) SetResetSingleEdit(ctx context.Context, sym string, input []byte) (resource.Result, error) {
+	res := resource.Result{}
+	menuOPtion := string(input)
+
+	switch menuOPtion {
+	case "2":
+		res.FlagSet = append(res.FlagSet, models.USERFLAG_SINGLE_EDIT)
+	case "3":
+		res.FlagSet = append(res.FlagSet, models.USERFLAG_SINGLE_EDIT)
+	case "4":
+		res.FlagSet = append(res.FlagSet, models.USERFLAG_SINGLE_EDIT)
+	default:
+		res.FlagReset = append(res.FlagReset, models.USERFLAG_SINGLE_EDIT)
+	}
+
+	return res, nil
+}
+
 func (h *Handlers) VerifyPin(ctx context.Context, sym string, input []byte) (resource.Result, error) {
 	res := resource.Result{}
 

internal/models/flags.go

@@ -18,4 +18,5 @@ const (
 	USERFLAG_PINMISMATCH
 	USERFLAG_INCORRECTDATEFORMAT
 	USERFLAG_ACCOUNT_CREATION_FAILED
+	USERFLAG_SINGLE_EDIT
 )