forked from grassrootseconomics/visedriver
		
	Compare commits
	
		
			48 Commits
		
	
	
		
			lash/purif
			...
			master
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 770fef68b9 | |||
| dd2751839d | |||
| 5a9c822075 | |||
|  | c8c0daac24 | ||
|  | 1bcffe2d23 | ||
| bf10e5357c | |||
|  | d91c96f541 | ||
|  | ec9032a42e | ||
|  | 6619afe62b | ||
|  | 1eb0b15552 | ||
|  | ef803e0ee2 | ||
|  | 03d19283f6 | ||
|  | 15ce29a1a4 | ||
|  | 6749c632b0 | ||
|  | 8530c45074 | ||
|  | d5e636fbd6 | ||
|  | f7d31e4e81 | ||
|  | 90ecec1798 | ||
|  | 874edb3da6 | ||
|  | 60ff1b0ab3 | ||
|  | 9b3dad579b | ||
|  | 348fff8936 | ||
|  | c5bb1c80a5 | ||
|  | b8a377befb | ||
|  | c9b92191f3 | ||
|  | ddd8d7cac0 | ||
|  | 37973a6c9b | ||
|  | 975720919c | ||
|  | c0534ede1b | ||
|  | 24e729d275 | ||
|  | ae6e2a99c5 | ||
|  | 7ca3974371 | ||
|  | adbfab3964 | ||
|  | 5228aef088 | ||
|  | 5bf0a0e858 | ||
|  | f13dab9a45 | ||
|  | 3f8e08151a | ||
|  | 9e4c65c8b4 | ||
|  | 611c5a8dfc | ||
|  | dcf777bf08 | ||
|  | ec4ad6e44b | ||
|  | 5b312f9569 | ||
|  | 2d89db3a37 | ||
|  | 6d8992fe38 | ||
|  | 5cffbe5cd8 | ||
|  | b28a047777 | ||
|  | 2ea51d88d8 | ||
|  | 2e0e854c4b | 
| @ -1,14 +0,0 @@ | ||||
| /** | ||||
| !/cmd/africastalking | ||||
| !/cmd/ssh | ||||
| !/common | ||||
| !/config | ||||
| !/initializers | ||||
| !/internal | ||||
| !/models | ||||
| !/remote | ||||
| !/services | ||||
| !/LICENSE | ||||
| !/README.md | ||||
| !/go.* | ||||
| !/.env.example | ||||
							
								
								
									
										56
									
								
								.github/workflows/docker.yaml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										56
									
								
								.github/workflows/docker.yaml
									
									
									
									
										vendored
									
									
								
							| @ -1,56 +0,0 @@ | ||||
| name: release | ||||
| 
 | ||||
| on: | ||||
|   push: | ||||
|     tags: | ||||
|       - "v*" | ||||
| 
 | ||||
| jobs: | ||||
|   docker: | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|       - name: Set up QEMU | ||||
|         uses: docker/setup-qemu-action@v2 | ||||
| 
 | ||||
|       - name: Set up Docker Buildx | ||||
|         uses: docker/setup-buildx-action@v2 | ||||
| 
 | ||||
|       - name: Check out repo | ||||
|         uses: actions/checkout@v3 | ||||
|         with: | ||||
|           fetch-depth: 0 | ||||
| 
 | ||||
|       - name: Cache Docker layers | ||||
|         uses: actions/cache@v3 | ||||
|         with: | ||||
|           path: /tmp/.buildx-cache | ||||
|           key: ${{ runner.os }}-buildx-${{ github.sha }} | ||||
|           restore-keys: | | ||||
|             ${{ runner.os }}-buildx- | ||||
| 
 | ||||
|       - name: Login to GHCR Docker registry | ||||
|         uses: docker/login-action@v1 | ||||
|         with: | ||||
|           registry: ghcr.io | ||||
|           username: ${{ github.repository_owner }} | ||||
|           password: ${{ secrets.GITHUB_TOKEN }} | ||||
| 
 | ||||
|       - name: Set outputs | ||||
|         run: | | ||||
|           echo "RELEASE_TAG=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV \ | ||||
|           && echo "RELEASE_SHORT_COMMIT=$(git rev-parse --short HEAD)" >> $GITHUB_ENV | ||||
| 
 | ||||
|       - name: Build and push image | ||||
|         uses: docker/build-push-action@v2 | ||||
|         with: | ||||
|           context: ./ | ||||
|           file: ./Dockerfile | ||||
|           platforms: linux/amd64 | ||||
|           push: true | ||||
|           build-args: | | ||||
|             BUILD=${{ env.RELEASE_SHORT_COMMIT }} | ||||
|           cache-from: type=local,src=/tmp/.buildx-cache | ||||
|           cache-to: type=local,dest=/tmp/.buildx-cache | ||||
|           tags: | | ||||
|             ghcr.io/grassrootseconomics/urdt-ussd:latest | ||||
|             ghcr.io/grassrootseconomics/urdt-ussd:${{ env.RELEASE_TAG }} | ||||
							
								
								
									
										44
									
								
								Dockerfile
									
									
									
									
									
								
							
							
						
						
									
										44
									
								
								Dockerfile
									
									
									
									
									
								
							| @ -1,44 +0,0 @@ | ||||
| FROM golang:1.23.0-bookworm AS build | ||||
| 
 | ||||
| ENV CGO_ENABLED=1 | ||||
| 
 | ||||
| ARG BUILDPLATFORM | ||||
| ARG TARGETPLATFORM | ||||
| ARG BUILD=dev | ||||
| 
 | ||||
| WORKDIR /build | ||||
| COPY . . | ||||
| RUN apt update && apt install libgdbm-dev | ||||
| RUN git clone https://git.defalsify.org/vise.git go-vise | ||||
| 
 | ||||
| WORKDIR /build/services/registration | ||||
| RUN echo "Compiling go-vise files" | ||||
| RUN make VISE_PATH=/build/go-vise -B | ||||
| 
 | ||||
| WORKDIR /build | ||||
| RUN echo "Building on $BUILDPLATFORM, building for $TARGETPLATFORM" | ||||
| RUN go mod download | ||||
| RUN go build -tags logtrace -o ussd-africastalking -ldflags="-X main.build=${BUILD} -s -w" cmd/africastalking/main.go | ||||
| RUN go build -tags logtrace -o ussd-ssh -ldflags="-X main.build=${BUILD} -s -w" cmd/ssh/main.go | ||||
| 
 | ||||
| FROM debian:bookworm-slim | ||||
| 
 | ||||
| ENV DEBIAN_FRONTEND=noninteractive | ||||
| 
 | ||||
| RUN apt update && apt install libgdbm-dev ca-certificates -y | ||||
| RUN apt-get clean && rm -rf /var/lib/apt/lists/* | ||||
| 
 | ||||
| WORKDIR /service | ||||
| 
 | ||||
| COPY --from=build /build/ussd-africastalking . | ||||
| COPY --from=build /build/ussd-ssh . | ||||
| COPY --from=build /build/LICENSE . | ||||
| COPY --from=build /build/README.md . | ||||
| COPY --from=build /build/services ./services | ||||
| COPY --from=build /build/.env.example . | ||||
| RUN mv .env.example .env | ||||
| 
 | ||||
| EXPOSE 7123 | ||||
| EXPOSE 7122 | ||||
| 
 | ||||
| CMD ["./ussd-africastalking"] | ||||
							
								
								
									
										92
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										92
									
								
								README.md
									
									
									
									
									
								
							| @ -1,92 +0,0 @@ | ||||
| # URDT USSD service | ||||
| 
 | ||||
| This is a USSD service built using the [go-vise](https://github.com/nolash/go-vise) engine. | ||||
| 
 | ||||
| ## Prerequisites | ||||
| ### 1. [go-vise](https://github.com/nolash/go-vise) | ||||
| 
 | ||||
| Set up `go-vise` by cloning the repository into a separate directory. The main upstream repository is hosted at: `https://git.defalsify.org/vise.git` | ||||
| ``` | ||||
| git clone https://git.defalsify.org/vise.git | ||||
| ``` | ||||
| 
 | ||||
| ## Setup | ||||
| 1. Clone the ussd repo in its own directory | ||||
| 
 | ||||
|     ``` | ||||
|     git clone https://git.grassecon.net/urdt/ussd.git | ||||
|     ``` | ||||
| 
 | ||||
| 2. Navigate to the project directory. | ||||
| 3. Enter the `services/registration` subfolder: | ||||
|     ``` | ||||
|     cd services/registration | ||||
|     ``` | ||||
| 4. make the .bin files from the .vis files | ||||
|     ``` | ||||
|     make VISE_PATH=/var/path/to/your/go-vise -B | ||||
|     ``` | ||||
| 5. Return to the project root (`cd ../..`) | ||||
| 6. Run the USSD menu  | ||||
|     ``` | ||||
|     go run cmd/main.go -session-id=0712345678 | ||||
|     ``` | ||||
| ## Running the different binaries | ||||
| 1. ### CLI:  | ||||
|     ``` | ||||
|     go run cmd/main.go -session-id=0712345678 | ||||
|     ``` | ||||
| 2. ### Africastalking:  | ||||
|     ``` | ||||
|     go run cmd/africastalking/main.go | ||||
|     ``` | ||||
| 3. ### Async:  | ||||
|     ``` | ||||
|     go run cmd/async/main.go | ||||
|     ``` | ||||
| 4. ### Http:  | ||||
|     ``` | ||||
|     go run cmd/http/main.go | ||||
|     ``` | ||||
|      | ||||
| ## Flags | ||||
| Below are the supported flags: | ||||
| 
 | ||||
| 1. `-session-id`:  | ||||
|      | ||||
|     Specifies the session ID. (CLI only).  | ||||
|      | ||||
|     Default: `075xx2123`. | ||||
| 
 | ||||
|     Example: | ||||
|     ``` | ||||
|     go run cmd/main.go -session-id=0712345678 | ||||
|     ``` | ||||
| 
 | ||||
| 2. `-d`:  | ||||
| 
 | ||||
|     Enables engine debug output.  | ||||
|      | ||||
|     Default: `false`. | ||||
| 
 | ||||
|     Example: | ||||
|     ``` | ||||
|     go run cmd/main.go -session-id=0712345678 -d | ||||
|     ``` | ||||
| 
 | ||||
| 3. `-db`:  | ||||
| 
 | ||||
|     Specifies the database type. | ||||
|      | ||||
|     Default: `gdbm`. | ||||
| 
 | ||||
|     Example: | ||||
|     ``` | ||||
|     go run cmd/main.go -session-id=0712345678 -d -db=postgres | ||||
|     ``` | ||||
| 
 | ||||
|     >Note: If using `-db=postgres`, ensure PostgreSQL is running with the connection details specified in your `.env` file. | ||||
| 
 | ||||
| ## License | ||||
| 
 | ||||
| [AGPL-3.0](LICENSE). | ||||
| @ -1,7 +0,0 @@ | ||||
| { | ||||
|   "admins": [ | ||||
|     { | ||||
|       "phonenumber" : "<replace with any admin number to test with >" | ||||
|     } | ||||
|    ] | ||||
| } | ||||
| @ -1,47 +0,0 @@ | ||||
| package commands | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"encoding/json" | ||||
| 	"os" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/utils" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla().WithDomain("adminstore") | ||||
| ) | ||||
| 
 | ||||
| type Admin struct { | ||||
| 	PhoneNumber string `json:"phonenumber"` | ||||
| } | ||||
| 
 | ||||
| type Config struct { | ||||
| 	Admins []Admin `json:"admins"` | ||||
| } | ||||
| 
 | ||||
| func Seed(ctx context.Context) error { | ||||
| 	var config Config | ||||
| 	adminstore, err := utils.NewAdminStore(ctx, "../admin_numbers") | ||||
| 	store := adminstore.FsStore | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	defer store.Close() | ||||
| 	data, err := os.ReadFile("admin_numbers.json") | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	if err := json.Unmarshal(data, &config); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	for _, admin := range config.Admins { | ||||
| 		err := store.Put(ctx, []byte(admin.PhoneNumber), []byte("1")) | ||||
| 		if err != nil { | ||||
| 			logg.Printf(logging.LVL_DEBUG, "Failed to insert admin number", admin.PhoneNumber) | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
| @ -1,17 +0,0 @@ | ||||
| package main | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"log" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/cmd/admin/commands" | ||||
| ) | ||||
| 
 | ||||
| func main() { | ||||
| 	ctx := context.Background() | ||||
| 	err := commands.Seed(ctx) | ||||
| 	if err != nil { | ||||
| 		log.Fatalf("Failed to initialize a list of admins with error %s", err) | ||||
| 	} | ||||
| 
 | ||||
| } | ||||
							
								
								
									
										126
									
								
								cmd/lang/main.go
									
									
									
									
									
								
							
							
						
						
									
										126
									
								
								cmd/lang/main.go
									
									
									
									
									
								
							| @ -1,126 +0,0 @@ | ||||
| // create language files from environment
 | ||||
| package main | ||||
| 
 | ||||
| import ( | ||||
| 	"flag" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"path" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| 	"git.defalsify.org/vise.git/lang" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/config" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/initializers" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	 | ||||
| 	changeHeadSrc = `LOAD reset_account_authorized 0 | ||||
| LOAD reset_incorrect 0 | ||||
| CATCH incorrect_pin flag_incorrect_pin 1 | ||||
| CATCH pin_entry flag_account_authorized 0 | ||||
| `  | ||||
| 
 | ||||
| 	selectSrc = `LOAD set_language 6 | ||||
| RELOAD set_language | ||||
| CATCH terms flag_account_created 0 | ||||
| MOVE language_changed | ||||
| ` | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla() | ||||
| 	mouts string | ||||
| 	incmps string | ||||
| ) | ||||
| 
 | ||||
| func init() { | ||||
| 	initializers.LoadEnvVariables() | ||||
| } | ||||
| 
 | ||||
| func toLanguageLabel(ln lang.Language) string { | ||||
| 	s := ln.Name | ||||
| 	v := strings.Split(s, " (") | ||||
| 	if len(v) > 1 { | ||||
| 		s = v[0] | ||||
| 	} | ||||
| 	return s | ||||
| } | ||||
| 
 | ||||
| func toLanguageKey(ln lang.Language) string { | ||||
| 	s := toLanguageLabel(ln) | ||||
| 	return strings.ToLower(s) | ||||
| } | ||||
| 
 | ||||
| func main() { | ||||
| 	var srcDir string | ||||
| 
 | ||||
| 	flag.StringVar(&srcDir, "o", ".", "resource dir write to") | ||||
| 	flag.Parse() | ||||
| 
 | ||||
| 	logg.Infof("start command", "dir", srcDir) | ||||
| 
 | ||||
| 	err := config.LoadConfig() | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "config load error: %v", err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 	logg.Tracef("using languages", "lang", config.Languages) | ||||
| 
 | ||||
| 	for i, v := range(config.Languages) { | ||||
| 		ln, err := lang.LanguageFromCode(v) | ||||
| 		if err != nil { | ||||
| 			fmt.Fprintf(os.Stderr, "error parsing language: %s\n", v) | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
| 		n := i + 1 | ||||
| 		s := toLanguageKey(ln) | ||||
| 		mouts += fmt.Sprintf("MOUT %s %v\n", s, n) | ||||
| 		v = "set_" + ln.Code | ||||
| 		incmps += fmt.Sprintf("INCMP %s %v\n", v, n) | ||||
| 
 | ||||
| 		p := path.Join(srcDir, v) | ||||
| 		w, err := os.OpenFile(p, os.O_WRONLY | os.O_CREATE | os.O_EXCL, 0600) | ||||
| 		if err != nil { | ||||
| 			fmt.Fprintf(os.Stderr, "failed open language set template output: %v\n", err) | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
| 		s = toLanguageLabel(ln) | ||||
| 		defer w.Close() | ||||
| 		_, err = w.Write([]byte(s)) | ||||
| 		if err != nil { | ||||
| 			fmt.Fprintf(os.Stderr, "failed write select language vis output: %v\n", err) | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
| 	} | ||||
| 	src := mouts + "HALT\n" + incmps | ||||
| 	src += "INCMP . *\n" | ||||
| 
 | ||||
| 	p := path.Join(srcDir, "select_language.vis") | ||||
| 	w, err := os.OpenFile(p, os.O_WRONLY | os.O_CREATE | os.O_EXCL, 0600) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "failed open select language vis output: %v\n", err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 	defer w.Close() | ||||
| 	_, err = w.Write([]byte(src)) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "failed write select language vis output: %v\n", err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 
 | ||||
| 	src = changeHeadSrc + src | ||||
| 	p = path.Join(srcDir, "change_language.vis") | ||||
| 	w, err = os.OpenFile(p, os.O_WRONLY | os.O_CREATE | os.O_EXCL, 0600) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "failed open select language vis output: %v\n", err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 	defer w.Close() | ||||
| 	_, err = w.Write([]byte(src)) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "failed write select language vis output: %v\n", err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| } | ||||
| @ -1,100 +0,0 @@ | ||||
| package main | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"flag" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"path" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/config" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/initializers" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/storage" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/debug" | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg      = logging.NewVanilla() | ||||
| 	scriptDir = path.Join("services", "registration") | ||||
| ) | ||||
| 
 | ||||
| func init() { | ||||
| 	initializers.LoadEnvVariables() | ||||
| } | ||||
| 
 | ||||
| 
 | ||||
| func formatItem(k []byte, v []byte) (string, error) { | ||||
| 	o, err := debug.FromKey(k) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	s := fmt.Sprintf("%vValue: %v\n\n", o, string(v)) | ||||
| 	return s, nil | ||||
| } | ||||
| 
 | ||||
| func main() { | ||||
| 	config.LoadConfig() | ||||
| 
 | ||||
| 	var connStr string | ||||
| 	var sessionId string | ||||
| 	var database string | ||||
| 	var engineDebug bool | ||||
| 	var err error | ||||
| 
 | ||||
| 	flag.StringVar(&sessionId, "session-id", "075xx2123", "session id") | ||||
| 	flag.StringVar(&connStr, "c", ".state", "connection string") | ||||
| 	flag.BoolVar(&engineDebug, "d", false, "use engine debug output") | ||||
| 	flag.Parse() | ||||
| 
 | ||||
| 	if connStr != "" { | ||||
| 		connStr = config.DbConn | ||||
| 	} | ||||
| 	connData, err := storage.ToConnData(config.DbConn) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "connstr err: %v", err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 
 | ||||
| 	logg.Infof("start command", "conn", connData) | ||||
| 
 | ||||
| 	ctx := context.Background() | ||||
| 	ctx = context.WithValue(ctx, "SessionId", sessionId) | ||||
| 	ctx = context.WithValue(ctx, "Database", database) | ||||
| 
 | ||||
| 	resourceDir := scriptDir | ||||
| 	menuStorageService := storage.NewMenuStorageService(connData, resourceDir) | ||||
| 
 | ||||
| 	store, err := menuStorageService.GetUserdataDb(ctx) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "get userdata db: %v\n", err.Error()) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 	store.SetPrefix(db.DATATYPE_USERDATA) | ||||
| 
 | ||||
| 	d, err := store.Dump(ctx, []byte(sessionId)) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "store dump fail: %v\n", err.Error()) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 
 | ||||
| 	for true { | ||||
| 		k, v := d.Next(ctx) | ||||
| 		if k == nil { | ||||
| 			break | ||||
| 		} | ||||
| 		r, err := formatItem(k, v) | ||||
| 		if err != nil { | ||||
| 			fmt.Fprintf(os.Stderr, "format db item error: %v", err) | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
| 		fmt.Printf(r) | ||||
| 	} | ||||
| 
 | ||||
| 	err = store.Close() | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, err.Error()) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| } | ||||
| @ -1,90 +0,0 @@ | ||||
| package main | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"crypto/sha1" | ||||
| 	"flag" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"path" | ||||
| 
 | ||||
| 	testdataloader "github.com/peteole/testdata-loader" | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/config" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/storage" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/initializers" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/common" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg      = logging.NewVanilla() | ||||
| 	baseDir   = testdataloader.GetBasePath() | ||||
| 	scriptDir = path.Join("services", "registration") | ||||
| ) | ||||
| 
 | ||||
| func init() { | ||||
| 	initializers.LoadEnvVariables() | ||||
| } | ||||
| 
 | ||||
| func main() { | ||||
| 	config.LoadConfig() | ||||
| 
 | ||||
| 	var connStr string | ||||
| 	var sessionId string | ||||
| 	var database string | ||||
| 	var engineDebug bool | ||||
| 	var err error | ||||
| 
 | ||||
| 	flag.StringVar(&sessionId, "session-id", "075xx2123", "session id") | ||||
| 	flag.StringVar(&connStr, "c", "", "connection string") | ||||
| 	flag.BoolVar(&engineDebug, "d", false, "use engine debug output") | ||||
| 	flag.Parse() | ||||
| 
 | ||||
| 	if connStr != "" { | ||||
| 		connStr = config.DbConn | ||||
| 	} | ||||
| 	connData, err := storage.ToConnData(config.DbConn) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "connstr err: %v", err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 
 | ||||
| 	logg.Infof("start command", "conn", connData) | ||||
| 
 | ||||
| 	ctx := context.Background() | ||||
| 	ctx = context.WithValue(ctx, "SessionId", sessionId) | ||||
| 	ctx = context.WithValue(ctx, "Database", database) | ||||
| 
 | ||||
| 	resourceDir := scriptDir | ||||
| 	menuStorageService := storage.NewMenuStorageService(connData, resourceDir) | ||||
| 	 | ||||
| 	store, err := menuStorageService.GetUserdataDb(ctx) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, err.Error()) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 	userStore := common.UserDataStore{store} | ||||
| 
 | ||||
| 	h := sha1.New() | ||||
| 	h.Write([]byte(sessionId)) | ||||
| 	address := h.Sum(nil) | ||||
| 	addressString := fmt.Sprintf("%x", address) | ||||
| 
 | ||||
| 	err = userStore.WriteEntry(ctx, sessionId, common.DATA_PUBLIC_KEY, []byte(addressString)) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, err.Error()) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 
 | ||||
| 	err = userStore.WriteEntry(ctx, addressString, common.DATA_PUBLIC_KEY_REVERSE, []byte(sessionId)) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, err.Error()) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| 
 | ||||
| 	err = store.Close() | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, err.Error()) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										134
									
								
								common/db.go
									
									
									
									
									
								
							
							
						
						
									
										134
									
								
								common/db.go
									
									
									
									
									
								
							| @ -1,134 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"encoding/binary" | ||||
| 	"errors" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| ) | ||||
| 
 | ||||
| // DataType is a subprefix value used in association with vise/db.DATATYPE_USERDATA.
 | ||||
| //
 | ||||
| // All keys are used only within the context of a single account. Unless otherwise specified, the user context is the session id.
 | ||||
| //
 | ||||
| // * The first byte is vise/db.DATATYPE_USERDATA
 | ||||
| // * The last 2 bytes are the DataTyp value, big-endian.
 | ||||
| // * The intermediate bytes are the id of the user context.
 | ||||
| //
 | ||||
| // All values are strings
 | ||||
| type DataTyp uint16 | ||||
| 
 | ||||
| const ( | ||||
| 	// API Tracking id to follow status of account creation
 | ||||
| 	DATA_TRACKING_ID = iota | ||||
| 	// EVM address returned from API on account creation
 | ||||
| 	DATA_PUBLIC_KEY | ||||
| 	// Currently active PIN used to authenticate ussd state change requests
 | ||||
| 	DATA_ACCOUNT_PIN | ||||
| 	// The first name of the user
 | ||||
| 	DATA_FIRST_NAME | ||||
| 	// The last name of the user
 | ||||
| 	DATA_FAMILY_NAME | ||||
| 	// The year-of-birth of the user
 | ||||
| 	DATA_YOB | ||||
| 	// The location of the user
 | ||||
| 	DATA_LOCATION | ||||
| 	// The gender of the user
 | ||||
| 	DATA_GENDER | ||||
| 	// The offerings description of the user
 | ||||
| 	DATA_OFFERINGS | ||||
| 	// The ethereum address of the recipient of an ongoing send request
 | ||||
| 	DATA_RECIPIENT | ||||
| 	// The voucher value amount of an ongoing send request
 | ||||
| 	DATA_AMOUNT | ||||
| 	// A general swap field for temporary values
 | ||||
| 	DATA_TEMPORARY_VALUE | ||||
| 	// Currently active voucher symbol of user
 | ||||
| 	DATA_ACTIVE_SYM | ||||
| 	// Voucher balance of user's currently active voucher
 | ||||
| 	DATA_ACTIVE_BAL | ||||
| 	// String boolean indicating whether use of PIN is blocked
 | ||||
| 	DATA_BLOCKED_NUMBER | ||||
| 	// Reverse mapping of a user's evm address to a session id.
 | ||||
| 	DATA_PUBLIC_KEY_REVERSE | ||||
| 	// Decimal count of the currently active voucher
 | ||||
| 	DATA_ACTIVE_DECIMAL | ||||
| 	// EVM address of the currently active voucher
 | ||||
| 	DATA_ACTIVE_ADDRESS | ||||
| 	//Holds count of the number of incorrect PIN attempts
 | ||||
| 	DATA_INCORRECT_PIN_ATTEMPTS | ||||
| 	//ISO 639 code for the selected language.
 | ||||
| 	DATA_SELECTED_LANGUAGE_CODE | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	// List of valid voucher symbols in the user context.
 | ||||
| 	DATA_VOUCHER_SYMBOLS DataTyp = 256 + iota | ||||
| 	// List of voucher balances for vouchers valid in the user context.
 | ||||
| 	DATA_VOUCHER_BALANCES | ||||
| 	// List of voucher decimal counts for vouchers valid in the user context.
 | ||||
| 	DATA_VOUCHER_DECIMALS | ||||
| 	// List of voucher EVM addresses for vouchers valid in the user context.
 | ||||
| 	DATA_VOUCHER_ADDRESSES | ||||
| 	// List of senders for valid transactions in the user context.
 | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	DATA_TX_SENDERS = 512 + iota | ||||
| 	// List of recipients for valid transactions in the user context.
 | ||||
| 	DATA_TX_RECIPIENTS | ||||
| 	// List of voucher values for valid transactions in the user context.
 | ||||
| 	DATA_TX_VALUES | ||||
| 	// List of voucher EVM addresses for valid transactions in the user context.
 | ||||
| 	DATA_TX_ADDRESSES | ||||
| 	// List of valid transaction hashes in the user context.
 | ||||
| 	DATA_TX_HASHES | ||||
| 	// List of transaction dates for valid transactions in the user context.
 | ||||
| 	DATA_TX_DATES | ||||
| 	// List of voucher symbols for valid transactions in the user context.
 | ||||
| 	DATA_TX_SYMBOLS | ||||
| 	// List of voucher decimal counts for valid transactions in the user context.
 | ||||
| 	DATA_TX_DECIMALS | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla().WithDomain("urdt-common") | ||||
| ) | ||||
| 
 | ||||
| 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...) | ||||
| } | ||||
| 
 | ||||
| func StringToDataTyp(str string) (DataTyp, error) { | ||||
| 	switch str { | ||||
| 	case "DATA_FIRST_NAME": | ||||
| 		return DATA_FIRST_NAME, nil | ||||
| 	case "DATA_FAMILY_NAME": | ||||
| 		return DATA_FAMILY_NAME, nil | ||||
| 	case "DATA_YOB": | ||||
| 		return DATA_YOB, nil | ||||
| 	case "DATA_LOCATION": | ||||
| 		return DATA_LOCATION, nil | ||||
| 	case "DATA_GENDER": | ||||
| 		return DATA_GENDER, nil | ||||
| 	case "DATA_OFFERINGS": | ||||
| 		return DATA_OFFERINGS, nil | ||||
| 
 | ||||
| 	default: | ||||
| 		return 0, errors.New("invalid DataTyp string") | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| // ToBytes converts DataTyp or int to a byte slice
 | ||||
| func ToBytes[T ~uint16 | int](value T) []byte { | ||||
| 	bytes := make([]byte, 2) | ||||
| 	binary.BigEndian.PutUint16(bytes, uint16(value)) | ||||
| 	return bytes | ||||
| } | ||||
| @ -1,31 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"encoding/hex" | ||||
| 	"strings" | ||||
| ) | ||||
| 
 | ||||
| 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 | ||||
| } | ||||
| 
 | ||||
| func IsSameHex(left string, right string) bool { | ||||
| 	bl, err := NormalizeHex(left) | ||||
| 	if err != nil { | ||||
| 		return false | ||||
| 	} | ||||
| 	br, err := NormalizeHex(left) | ||||
| 	if err != nil { | ||||
| 		return false | ||||
| 	} | ||||
| 	return strings.Compare(bl, br) == 0 | ||||
| } | ||||
| @ -1,37 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"regexp" | ||||
| 
 | ||||
| 	"golang.org/x/crypto/bcrypt" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	// Define the regex pattern as a constant
 | ||||
| 	pinPattern = `^\d{4}$` | ||||
| 
 | ||||
| 	//Allowed incorrect  PIN attempts
 | ||||
| 	AllowedPINAttempts = uint8(3) | ||||
| 	 | ||||
| ) | ||||
| 
 | ||||
| // checks whether the given input is a 4 digit number
 | ||||
| func IsValidPIN(pin string) bool { | ||||
| 	match, _ := regexp.MatchString(pinPattern, pin) | ||||
| 	return match | ||||
| } | ||||
| 
 | ||||
| // HashPIN uses bcrypt with 8 salt rounds to hash the PIN
 | ||||
| func HashPIN(pin string) (string, error) { | ||||
| 	hash, err := bcrypt.GenerateFromPassword([]byte(pin), 8) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return string(hash), nil | ||||
| } | ||||
| 
 | ||||
| // VerifyPIN compareS the hashed PIN with the plaintext PIN
 | ||||
| func VerifyPIN(hashedPIN, pin string) bool { | ||||
| 	err := bcrypt.CompareHashAndPassword([]byte(hashedPIN), []byte(pin)) | ||||
| 	return err == nil | ||||
| } | ||||
| @ -1,173 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"golang.org/x/crypto/bcrypt" | ||||
| ) | ||||
| 
 | ||||
| func TestIsValidPIN(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name     string | ||||
| 		pin      string | ||||
| 		expected bool | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:     "Valid PIN with 4 digits", | ||||
| 			pin:      "1234", | ||||
| 			expected: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:     "Valid PIN with leading zeros", | ||||
| 			pin:      "0001", | ||||
| 			expected: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:     "Invalid PIN with less than 4 digits", | ||||
| 			pin:      "123", | ||||
| 			expected: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:     "Invalid PIN with more than 4 digits", | ||||
| 			pin:      "12345", | ||||
| 			expected: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:     "Invalid PIN with letters", | ||||
| 			pin:      "abcd", | ||||
| 			expected: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:     "Invalid PIN with special characters", | ||||
| 			pin:      "12@#", | ||||
| 			expected: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:     "Empty PIN", | ||||
| 			pin:      "", | ||||
| 			expected: false, | ||||
| 		}, | ||||
| 	} | ||||
| 
 | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			actual := IsValidPIN(tt.pin) | ||||
| 			if actual != tt.expected { | ||||
| 				t.Errorf("IsValidPIN(%q) = %v; expected %v", tt.pin, actual, tt.expected) | ||||
| 			} | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestHashPIN(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name string | ||||
| 		pin  string | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name: "Valid PIN with 4 digits", | ||||
| 			pin:  "1234", | ||||
| 		}, | ||||
| 		{ | ||||
| 			name: "Valid PIN with leading zeros", | ||||
| 			pin:  "0001", | ||||
| 		}, | ||||
| 		{ | ||||
| 			name: "Empty PIN", | ||||
| 			pin:  "", | ||||
| 		}, | ||||
| 	} | ||||
| 
 | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			hashedPIN, err := HashPIN(tt.pin) | ||||
| 			if err != nil { | ||||
| 				t.Errorf("HashPIN(%q) returned an error: %v", tt.pin, err) | ||||
| 				return | ||||
| 			} | ||||
| 
 | ||||
| 			if hashedPIN == "" { | ||||
| 				t.Errorf("HashPIN(%q) returned an empty hash", tt.pin) | ||||
| 			} | ||||
| 
 | ||||
| 			// Ensure the hash can be verified with bcrypt
 | ||||
| 			err = bcrypt.CompareHashAndPassword([]byte(hashedPIN), []byte(tt.pin)) | ||||
| 			if tt.pin != "" && err != nil { | ||||
| 				t.Errorf("HashPIN(%q) produced a hash that does not match: %v", tt.pin, err) | ||||
| 			} | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestVerifyMigratedHashPin(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		pin  string | ||||
| 		hash string | ||||
| 	}{ | ||||
| 		{ | ||||
| 			pin:  "1234", | ||||
| 			hash: "$2b$08$dTvIGxCCysJtdvrSnaLStuylPoOS/ZLYYkxvTeR5QmTFY3TSvPQC6", | ||||
| 		}, | ||||
| 	} | ||||
| 
 | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.pin, func(t *testing.T) { | ||||
| 			ok := VerifyPIN(tt.hash, tt.pin) | ||||
| 			if !ok { | ||||
| 				t.Errorf("VerifyPIN could not verify migrated PIN: %v", tt.pin) | ||||
| 			} | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestVerifyPIN(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name       string | ||||
| 		pin        string | ||||
| 		hashedPIN  string | ||||
| 		shouldPass bool | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:       "Valid PIN verification", | ||||
| 			pin:        "1234", | ||||
| 			hashedPIN:  hashPINHelper("1234"), | ||||
| 			shouldPass: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:       "Invalid PIN verification with incorrect PIN", | ||||
| 			pin:        "5678", | ||||
| 			hashedPIN:  hashPINHelper("1234"), | ||||
| 			shouldPass: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:       "Invalid PIN verification with empty PIN", | ||||
| 			pin:        "", | ||||
| 			hashedPIN:  hashPINHelper("1234"), | ||||
| 			shouldPass: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:       "Invalid PIN verification with invalid hash", | ||||
| 			pin:        "1234", | ||||
| 			hashedPIN:  "invalidhash", | ||||
| 			shouldPass: false, | ||||
| 		}, | ||||
| 	} | ||||
| 
 | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			result := VerifyPIN(tt.hashedPIN, tt.pin) | ||||
| 			if result != tt.shouldPass { | ||||
| 				t.Errorf("VerifyPIN(%q, %q) = %v; expected %v", tt.hashedPIN, tt.pin, result, tt.shouldPass) | ||||
| 			} | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| // Helper function to hash a PIN for testing purposes
 | ||||
| func hashPINHelper(pin string) string { | ||||
| 	hashedPIN, err := HashPIN(pin) | ||||
| 	if err != nil { | ||||
| 		panic("Failed to hash PIN for test setup: " + err.Error()) | ||||
| 	} | ||||
| 	return hashedPIN | ||||
| } | ||||
| @ -1,73 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| ) | ||||
| 
 | ||||
| // Define the regex patterns as constants
 | ||||
| const ( | ||||
| 	phoneRegex   = `^(?:\+254|254|0)?((?:7[0-9]{8})|(?:1[01][0-9]{7}))$` | ||||
| 	addressRegex = `^0x[a-fA-F0-9]{40}$` | ||||
| 	aliasRegex   = `^[a-zA-Z0-9]+$` | ||||
| ) | ||||
| 
 | ||||
| // IsValidPhoneNumber checks if the given number is a valid phone number
 | ||||
| func IsValidPhoneNumber(phonenumber string) bool { | ||||
| 	match, _ := regexp.MatchString(phoneRegex, phonenumber) | ||||
| 	return match | ||||
| } | ||||
| 
 | ||||
| // IsValidAddress checks if the given address is a valid Ethereum address
 | ||||
| func IsValidAddress(address string) bool { | ||||
| 	match, _ := regexp.MatchString(addressRegex, address) | ||||
| 	return match | ||||
| } | ||||
| 
 | ||||
| // IsValidAlias checks if the alias is a valid alias format
 | ||||
| func IsValidAlias(alias string) bool { | ||||
| 	match, _ := regexp.MatchString(aliasRegex, alias) | ||||
| 	return match | ||||
| } | ||||
| 
 | ||||
| // CheckRecipient validates the recipient format based on the criteria
 | ||||
| func CheckRecipient(recipient string) (string, error) { | ||||
| 	if IsValidPhoneNumber(recipient) { | ||||
| 		return "phone number", nil | ||||
| 	} | ||||
| 
 | ||||
| 	if IsValidAddress(recipient) { | ||||
| 		return "address", nil | ||||
| 	} | ||||
| 
 | ||||
| 	if IsValidAlias(recipient) { | ||||
| 		return "alias", nil | ||||
| 	} | ||||
| 
 | ||||
| 	return "", fmt.Errorf("invalid recipient: must be a phone number, address or alias") | ||||
| } | ||||
| 
 | ||||
| // FormatPhoneNumber formats a Kenyan phone number to "+254xxxxxxxx".
 | ||||
| func FormatPhoneNumber(phone string) (string, error) { | ||||
| 	if !IsValidPhoneNumber(phone) { | ||||
| 		return "", errors.New("invalid phone number") | ||||
| 	} | ||||
| 
 | ||||
| 	// Remove any leading "+" and spaces
 | ||||
| 	phone = strings.TrimPrefix(phone, "+") | ||||
| 	phone = strings.ReplaceAll(phone, " ", "") | ||||
| 
 | ||||
| 	// Replace leading "0" with "254" if present
 | ||||
| 	if strings.HasPrefix(phone, "0") { | ||||
| 		phone = "254" + phone[1:] | ||||
| 	} | ||||
| 
 | ||||
| 	// Add "+" if not already present
 | ||||
| 	if !strings.HasPrefix(phone, "254") { | ||||
| 		return "", errors.New("unexpected format") | ||||
| 	} | ||||
| 
 | ||||
| 	return "+" + phone, nil | ||||
| } | ||||
| @ -1,69 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| 	"git.defalsify.org/vise.git/resource" | ||||
| 	"git.defalsify.org/vise.git/persist" | ||||
| 	"git.defalsify.org/vise.git/lang" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/storage" | ||||
| 	dbstorage "git.grassecon.net/grassrootseconomics/visedriver/storage/db" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	ToConnData = storage.ToConnData | ||||
| ) | ||||
| 
 | ||||
| func StoreToDb(store *UserDataStore) db.Db { | ||||
| 	return store.Db | ||||
| } | ||||
| 
 | ||||
| func StoreToPrefixDb(store *UserDataStore, pfx []byte) dbstorage.PrefixDb { | ||||
| 	return dbstorage.NewSubPrefixDb(store.Db, pfx)	 | ||||
| } | ||||
| 
 | ||||
| type StorageServices interface { | ||||
| 	GetPersister(ctx context.Context) (*persist.Persister, error) | ||||
| 	GetUserdataDb(ctx context.Context) (db.Db, error) | ||||
| 	GetResource(ctx context.Context) (resource.Resource, error) | ||||
| } | ||||
| 
 | ||||
| type StorageService struct { | ||||
| 	svc *storage.MenuStorageService | ||||
| } | ||||
| 
 | ||||
| func NewStorageService(conn storage.ConnData) (*StorageService, error) { | ||||
| 	svc := &StorageService{ | ||||
| 		svc: storage.NewMenuStorageService(conn, ""), | ||||
| 	} | ||||
| 	return svc, nil | ||||
| } | ||||
| 
 | ||||
| func (ss *StorageService) WithGettext(path string, lns []lang.Language) *StorageService { | ||||
| 	ss.svc = ss.svc.WithGettext(path, lns) | ||||
| 	return ss | ||||
| } | ||||
| 
 | ||||
| // TODO: simplify enable poresource, conndata instead
 | ||||
| func(ss *StorageService) SetResourceDir(resourceDir string) error { | ||||
| 	ss.svc = ss.svc.WithResourceDir(resourceDir) | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func(ss *StorageService) GetPersister(ctx context.Context) (*persist.Persister, error) { | ||||
| 	return ss.svc.GetPersister(ctx) | ||||
| } | ||||
| 	 | ||||
| func(ss *StorageService) GetUserdataDb(ctx context.Context) (db.Db, error) { | ||||
| 	return ss.svc.GetUserdataDb(ctx) | ||||
| } | ||||
| 
 | ||||
| func(ss *StorageService) GetResource(ctx context.Context) (resource.Resource, error) { | ||||
| 	return nil, errors.New("not implemented") | ||||
| } | ||||
| 
 | ||||
| func(ss *StorageService) GetStateStore(ctx context.Context) (db.Db, error) { | ||||
| 	return ss.svc.GetStateStore(ctx) | ||||
| } | ||||
| @ -1,81 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"math/big" | ||||
| 	"reflect" | ||||
| 	"strconv" | ||||
| ) | ||||
| 
 | ||||
| type TransactionData struct { | ||||
| 	TemporaryValue string | ||||
| 	ActiveSym      string | ||||
| 	Amount         string | ||||
| 	PublicKey      string | ||||
| 	Recipient      string | ||||
| 	ActiveDecimal  string | ||||
| 	ActiveAddress  string | ||||
| } | ||||
| 
 | ||||
| func ParseAndScaleAmount(storedAmount, activeDecimal string) (string, error) { | ||||
| 	// Parse token decimal
 | ||||
| 	tokenDecimal, err := strconv.Atoi(activeDecimal) | ||||
| 	if err != nil { | ||||
| 
 | ||||
| 		return "", err | ||||
| 	} | ||||
| 
 | ||||
| 	// Parse amount
 | ||||
| 	amount, _, err := big.ParseFloat(storedAmount, 10, 0, big.ToZero) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 
 | ||||
| 	// Scale the amount
 | ||||
| 	multiplier := new(big.Float).SetInt(new(big.Int).Exp(big.NewInt(10), big.NewInt(int64(tokenDecimal)), nil)) | ||||
| 	finalAmount := new(big.Float).Mul(amount, multiplier) | ||||
| 
 | ||||
| 	// Convert finalAmount to a string
 | ||||
| 	finalAmountStr := new(big.Int) | ||||
| 	finalAmount.Int(finalAmountStr) | ||||
| 
 | ||||
| 	return finalAmountStr.String(), nil | ||||
| } | ||||
| 
 | ||||
| func ReadTransactionData(ctx context.Context, store DataStore, sessionId string) (TransactionData, error) { | ||||
| 	data := TransactionData{} | ||||
| 	fieldToKey := map[string]DataTyp{ | ||||
| 		"TemporaryValue": DATA_TEMPORARY_VALUE, | ||||
| 		"ActiveSym":      DATA_ACTIVE_SYM, | ||||
| 		"Amount":         DATA_AMOUNT, | ||||
| 		"PublicKey":      DATA_PUBLIC_KEY, | ||||
| 		"Recipient":      DATA_RECIPIENT, | ||||
| 		"ActiveDecimal":  DATA_ACTIVE_DECIMAL, | ||||
| 		"ActiveAddress":  DATA_ACTIVE_ADDRESS, | ||||
| 	} | ||||
| 
 | ||||
| 	v := reflect.ValueOf(&data).Elem() | ||||
| 	for fieldName, key := range fieldToKey { | ||||
| 		field := v.FieldByName(fieldName) | ||||
| 		if !field.IsValid() || !field.CanSet() { | ||||
| 			return data, errors.New("invalid struct field: " + fieldName) | ||||
| 		} | ||||
| 
 | ||||
| 		value, err := readStringEntry(ctx, store, sessionId, key) | ||||
| 		if err != nil { | ||||
| 			return data, err | ||||
| 		} | ||||
| 		field.SetString(value) | ||||
| 	} | ||||
| 
 | ||||
| 	return data, nil | ||||
| } | ||||
| 
 | ||||
| func readStringEntry(ctx context.Context, store DataStore, sessionId string, key DataTyp) (string, error) { | ||||
| 	entry, err := store.ReadEntry(ctx, sessionId, key) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return string(entry), nil | ||||
| } | ||||
| @ -1,129 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/alecthomas/assert/v2" | ||||
| ) | ||||
| 
 | ||||
| func TestParseAndScaleAmount(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name        string | ||||
| 		amount      string | ||||
| 		decimals    string | ||||
| 		want        string | ||||
| 		expectError bool | ||||
| 	}{ | ||||
| 		{ | ||||
| 			name:        "whole number", | ||||
| 			amount:      "123", | ||||
| 			decimals:    "2", | ||||
| 			want:        "12300", | ||||
| 			expectError: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:        "decimal number", | ||||
| 			amount:      "123.45", | ||||
| 			decimals:    "2", | ||||
| 			want:        "12345", | ||||
| 			expectError: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:        "zero decimals", | ||||
| 			amount:      "123.45", | ||||
| 			decimals:    "0", | ||||
| 			want:        "123", | ||||
| 			expectError: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:        "large number", | ||||
| 			amount:      "1000000.01", | ||||
| 			decimals:    "6", | ||||
| 			want:        "1000000010000", | ||||
| 			expectError: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:        "invalid amount", | ||||
| 			amount:      "abc", | ||||
| 			decimals:    "2", | ||||
| 			want:        "", | ||||
| 			expectError: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:        "invalid decimals", | ||||
| 			amount:      "123.45", | ||||
| 			decimals:    "abc", | ||||
| 			want:        "", | ||||
| 			expectError: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:        "zero amount", | ||||
| 			amount:      "0", | ||||
| 			decimals:    "2", | ||||
| 			want:        "0", | ||||
| 			expectError: false, | ||||
| 		}, | ||||
| 	} | ||||
| 
 | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			got, err := ParseAndScaleAmount(tt.amount, tt.decimals) | ||||
| 
 | ||||
| 			// Check error cases
 | ||||
| 			if tt.expectError { | ||||
| 				if err == nil { | ||||
| 					t.Errorf("ParseAndScaleAmount(%q, %q) expected error, got nil", tt.amount, tt.decimals) | ||||
| 				} | ||||
| 				return | ||||
| 			} | ||||
| 
 | ||||
| 			if err != nil { | ||||
| 				t.Errorf("ParseAndScaleAmount(%q, %q) unexpected error: %v", tt.amount, tt.decimals, err) | ||||
| 				return | ||||
| 			} | ||||
| 
 | ||||
| 			if got != tt.want { | ||||
| 				t.Errorf("ParseAndScaleAmount(%q, %q) = %v, want %v", tt.amount, tt.decimals, got, tt.want) | ||||
| 			} | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestReadTransactionData(t *testing.T) { | ||||
| 	sessionId := "session123" | ||||
| 	publicKey := "0X13242618721" | ||||
| 	ctx, store := InitializeTestDb(t) | ||||
| 
 | ||||
| 	// Test transaction data
 | ||||
| 	transactionData := map[DataTyp]string{ | ||||
| 		DATA_TEMPORARY_VALUE: "0712345678", | ||||
| 		DATA_ACTIVE_SYM:      "SRF", | ||||
| 		DATA_AMOUNT:          "1000000", | ||||
| 		DATA_PUBLIC_KEY:      publicKey, | ||||
| 		DATA_RECIPIENT:       "0x41c188d63Qa", | ||||
| 		DATA_ACTIVE_DECIMAL:  "6", | ||||
| 		DATA_ACTIVE_ADDRESS:  "0xd4c288865Ce", | ||||
| 	} | ||||
| 
 | ||||
| 	// Store the data
 | ||||
| 	for key, value := range transactionData { | ||||
| 		if err := store.WriteEntry(ctx, sessionId, key, []byte(value)); err != nil { | ||||
| 			t.Fatal(err) | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
| 	expectedResult := TransactionData{ | ||||
| 		TemporaryValue: "0712345678", | ||||
| 		ActiveSym:      "SRF", | ||||
| 		Amount:         "1000000", | ||||
| 		PublicKey:      publicKey, | ||||
| 		Recipient:      "0x41c188d63Qa", | ||||
| 		ActiveDecimal:  "6", | ||||
| 		ActiveAddress:  "0xd4c288865Ce", | ||||
| 	} | ||||
| 
 | ||||
| 	data, err := ReadTransactionData(ctx, store, sessionId) | ||||
| 
 | ||||
| 	assert.NoError(t, err) | ||||
| 	assert.Equal(t, expectedResult, data) | ||||
| } | ||||
| @ -1,119 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| 
 | ||||
| 	dbstorage "git.grassecon.net/grassrootseconomics/visedriver/storage/db" | ||||
| 	dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api" | ||||
| ) | ||||
| 
 | ||||
| // TransferMetadata helps organize data fields
 | ||||
| type TransferMetadata struct { | ||||
| 	Senders        string | ||||
| 	Recipients     string | ||||
| 	TransferValues string | ||||
| 	Addresses      string | ||||
| 	TxHashes       string | ||||
| 	Dates          string | ||||
| 	Symbols        string | ||||
| 	Decimals       string | ||||
| } | ||||
| 
 | ||||
| // ProcessTransfers converts transfers into formatted strings
 | ||||
| func ProcessTransfers(transfers []dataserviceapi.Last10TxResponse) TransferMetadata { | ||||
| 	var data TransferMetadata | ||||
| 	var senders, recipients, transferValues, addresses, txHashes, dates, symbols, decimals []string | ||||
| 
 | ||||
| 	for _, t := range transfers { | ||||
| 		senders = append(senders, t.Sender) | ||||
| 		recipients = append(recipients, t.Recipient) | ||||
| 
 | ||||
| 		// Scale down the amount
 | ||||
| 		scaledBalance := ScaleDownBalance(t.TransferValue, t.TokenDecimals) | ||||
| 		transferValues = append(transferValues, scaledBalance) | ||||
| 
 | ||||
| 		addresses = append(addresses, t.ContractAddress) | ||||
| 		txHashes = append(txHashes, t.TxHash) | ||||
| 		dates = append(dates, fmt.Sprintf("%s", t.DateBlock)) | ||||
| 		symbols = append(symbols, t.TokenSymbol) | ||||
| 		decimals = append(decimals, t.TokenDecimals) | ||||
| 	} | ||||
| 
 | ||||
| 	data.Senders = strings.Join(senders, "\n") | ||||
| 	data.Recipients = strings.Join(recipients, "\n") | ||||
| 	data.TransferValues = strings.Join(transferValues, "\n") | ||||
| 	data.Addresses = strings.Join(addresses, "\n") | ||||
| 	data.TxHashes = strings.Join(txHashes, "\n") | ||||
| 	data.Dates = strings.Join(dates, "\n") | ||||
| 	data.Symbols = strings.Join(symbols, "\n") | ||||
| 	data.Decimals = strings.Join(decimals, "\n") | ||||
| 
 | ||||
| 	return data | ||||
| } | ||||
| 
 | ||||
| // GetTransferData retrieves and matches transfer data
 | ||||
| // returns a formatted string of the full transaction/statement
 | ||||
| func GetTransferData(ctx context.Context, db dbstorage.PrefixDb, publicKey string, index int) (string, error) { | ||||
| 	keys := []DataTyp{DATA_TX_SENDERS, DATA_TX_RECIPIENTS, DATA_TX_VALUES, DATA_TX_ADDRESSES, DATA_TX_HASHES, DATA_TX_DATES, DATA_TX_SYMBOLS} | ||||
| 	data := make(map[DataTyp]string) | ||||
| 
 | ||||
| 	for _, key := range keys { | ||||
| 		value, err := db.Get(ctx, ToBytes(key)) | ||||
| 		if err != nil { | ||||
| 			return "", fmt.Errorf("failed to get %s: %v", ToBytes(key), err) | ||||
| 		} | ||||
| 		data[key] = string(value) | ||||
| 	} | ||||
| 
 | ||||
| 	// Split the data
 | ||||
| 	senders := strings.Split(string(data[DATA_TX_SENDERS]), "\n") | ||||
| 	recipients := strings.Split(string(data[DATA_TX_RECIPIENTS]), "\n") | ||||
| 	values := strings.Split(string(data[DATA_TX_VALUES]), "\n") | ||||
| 	addresses := strings.Split(string(data[DATA_TX_ADDRESSES]), "\n") | ||||
| 	hashes := strings.Split(string(data[DATA_TX_HASHES]), "\n") | ||||
| 	dates := strings.Split(string(data[DATA_TX_DATES]), "\n") | ||||
| 	syms := strings.Split(string(data[DATA_TX_SYMBOLS]), "\n") | ||||
| 
 | ||||
| 	// Check if index is within range
 | ||||
| 	if index < 1 || index > len(senders) { | ||||
| 		return "", fmt.Errorf("transaction not found: index %d out of range", index) | ||||
| 	} | ||||
| 
 | ||||
| 	// Adjust for 0-based indexing
 | ||||
| 	i := index - 1 | ||||
| 	transactionType := "Received" | ||||
| 	party := fmt.Sprintf("From: %s", strings.TrimSpace(senders[i])) | ||||
| 	if strings.TrimSpace(senders[i]) == publicKey { | ||||
| 		transactionType = "Sent" | ||||
| 		party = fmt.Sprintf("To: %s", strings.TrimSpace(recipients[i])) | ||||
| 	} | ||||
| 
 | ||||
| 	formattedDate := formatDate(strings.TrimSpace(dates[i])) | ||||
| 
 | ||||
| 	// Build the full transaction detail
 | ||||
| 	detail := fmt.Sprintf( | ||||
| 		"%s %s %s\n%s\nContract address: %s\nTxhash: %s\nDate: %s", | ||||
| 		transactionType, | ||||
| 		strings.TrimSpace(values[i]), | ||||
| 		strings.TrimSpace(syms[i]), | ||||
| 		party, | ||||
| 		strings.TrimSpace(addresses[i]), | ||||
| 		strings.TrimSpace(hashes[i]), | ||||
| 		formattedDate, | ||||
| 	) | ||||
| 
 | ||||
| 	return detail, nil | ||||
| } | ||||
| 
 | ||||
| // Helper function to format date in desired output
 | ||||
| func formatDate(dateStr string) string { | ||||
| 	parsedDate, err := time.Parse("2006-01-02 15:04:05 -0700 MST", dateStr) | ||||
| 	if err != nil { | ||||
| 		fmt.Println("Error parsing date:", err) | ||||
| 		return "" | ||||
| 	} | ||||
| 	return parsedDate.Format("2006-01-02 03:04:05 PM") | ||||
| } | ||||
| @ -1,34 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| 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 to the userdata store.
 | ||||
| func (store *UserDataStore) ReadEntry(ctx context.Context, sessionId string, typ DataTyp) ([]byte, error) { | ||||
| 	store.SetPrefix(db.DATATYPE_USERDATA) | ||||
| 	store.SetSession(sessionId) | ||||
| 	k := ToBytes(typ) | ||||
| 	return store.Get(ctx, k) | ||||
| } | ||||
| 
 | ||||
| // WriteEntry adds an entry to the userdata store.
 | ||||
| // BUG: this uses sessionId twice
 | ||||
| func (store *UserDataStore) WriteEntry(ctx context.Context, sessionId string, typ DataTyp, value []byte) error { | ||||
| 	store.SetPrefix(db.DATATYPE_USERDATA) | ||||
| 	store.SetSession(sessionId) | ||||
| 	k := ToBytes(typ) | ||||
| 	return store.Put(ctx, k, value) | ||||
| } | ||||
| @ -1,174 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"math/big" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	dbstorage "git.grassecon.net/grassrootseconomics/visedriver/storage/db" | ||||
| 	dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api" | ||||
| ) | ||||
| 
 | ||||
| // VoucherMetadata helps organize data fields
 | ||||
| type VoucherMetadata struct { | ||||
| 	Symbols   string | ||||
| 	Balances  string | ||||
| 	Decimals  string | ||||
| 	Addresses string | ||||
| } | ||||
| 
 | ||||
| // ProcessVouchers converts holdings into formatted strings
 | ||||
| func ProcessVouchers(holdings []dataserviceapi.TokenHoldings) VoucherMetadata { | ||||
| 	var data VoucherMetadata | ||||
| 	var symbols, balances, decimals, addresses []string | ||||
| 
 | ||||
| 	for i, h := range holdings { | ||||
| 		symbols = append(symbols, fmt.Sprintf("%d:%s", i+1, h.TokenSymbol)) | ||||
| 
 | ||||
| 		// Scale down the balance
 | ||||
| 		scaledBalance := ScaleDownBalance(h.Balance, h.TokenDecimals) | ||||
| 
 | ||||
| 		balances = append(balances, fmt.Sprintf("%d:%s", i+1, scaledBalance)) | ||||
| 		decimals = append(decimals, fmt.Sprintf("%d:%s", i+1, h.TokenDecimals)) | ||||
| 		addresses = append(addresses, fmt.Sprintf("%d:%s", i+1, h.ContractAddress)) | ||||
| 	} | ||||
| 
 | ||||
| 	data.Symbols = strings.Join(symbols, "\n") | ||||
| 	data.Balances = strings.Join(balances, "\n") | ||||
| 	data.Decimals = strings.Join(decimals, "\n") | ||||
| 	data.Addresses = strings.Join(addresses, "\n") | ||||
| 
 | ||||
| 	return data | ||||
| } | ||||
| 
 | ||||
| func ScaleDownBalance(balance, decimals string) string { | ||||
| 	// Convert balance and decimals to big.Float
 | ||||
| 	bal := new(big.Float) | ||||
| 	bal.SetString(balance) | ||||
| 
 | ||||
| 	dec, ok := new(big.Int).SetString(decimals, 10) | ||||
| 	if !ok { | ||||
| 		dec = big.NewInt(0) // Default to 0 decimals in case of conversion failure
 | ||||
| 	} | ||||
| 
 | ||||
| 	divisor := new(big.Float).SetInt(new(big.Int).Exp(big.NewInt(10), dec, nil)) | ||||
| 	scaledBalance := new(big.Float).Quo(bal, divisor) | ||||
| 
 | ||||
| 	// Return the scaled balance without trailing decimals if it's an integer
 | ||||
| 	if scaledBalance.IsInt() { | ||||
| 		return scaledBalance.Text('f', 0) | ||||
| 	} | ||||
| 	return scaledBalance.Text('f', -1) | ||||
| } | ||||
| 
 | ||||
| // GetVoucherData retrieves and matches voucher data
 | ||||
| func GetVoucherData(ctx context.Context, db dbstorage.PrefixDb, input string) (*dataserviceapi.TokenHoldings, error) { | ||||
| 	keys := []DataTyp{DATA_VOUCHER_SYMBOLS, DATA_VOUCHER_BALANCES, DATA_VOUCHER_DECIMALS, DATA_VOUCHER_ADDRESSES} | ||||
| 	data := make(map[DataTyp]string) | ||||
| 
 | ||||
| 	for _, key := range keys { | ||||
| 		value, err := db.Get(ctx, ToBytes(key)) | ||||
| 		if err != nil { | ||||
| 			return nil, fmt.Errorf("failed to get %s: %v", ToBytes(key), err) | ||||
| 		} | ||||
| 		data[key] = string(value) | ||||
| 	} | ||||
| 
 | ||||
| 	symbol, balance, decimal, address := MatchVoucher(input, | ||||
| 		data[DATA_VOUCHER_SYMBOLS], | ||||
| 		data[DATA_VOUCHER_BALANCES], | ||||
| 		data[DATA_VOUCHER_DECIMALS], | ||||
| 		data[DATA_VOUCHER_ADDRESSES], | ||||
| 	) | ||||
| 
 | ||||
| 	if symbol == "" { | ||||
| 		return nil, nil | ||||
| 	} | ||||
| 
 | ||||
| 	return &dataserviceapi.TokenHoldings{ | ||||
| 		TokenSymbol:     string(symbol), | ||||
| 		Balance:         string(balance), | ||||
| 		TokenDecimals:   string(decimal), | ||||
| 		ContractAddress: string(address), | ||||
| 	}, nil | ||||
| } | ||||
| 
 | ||||
| // MatchVoucher finds the matching voucher symbol, balance, decimals and contract address based on the input.
 | ||||
| func MatchVoucher(input, symbols, balances, decimals, addresses string) (symbol, balance, decimal, address string) { | ||||
| 	symList := strings.Split(symbols, "\n") | ||||
| 	balList := strings.Split(balances, "\n") | ||||
| 	decList := strings.Split(decimals, "\n") | ||||
| 	addrList := strings.Split(addresses, "\n") | ||||
| 
 | ||||
| 	logg.Tracef("found", "symlist", symList, "syms", symbols, "input", input) | ||||
| 	for i, sym := range symList { | ||||
| 		parts := strings.SplitN(sym, ":", 2) | ||||
| 
 | ||||
| 		if input == parts[0] || strings.EqualFold(input, parts[1]) { | ||||
| 			symbol = parts[1] | ||||
| 			if i < len(balList) { | ||||
| 				balance = strings.SplitN(balList[i], ":", 2)[1] | ||||
| 			} | ||||
| 			if i < len(decList) { | ||||
| 				decimal = strings.SplitN(decList[i], ":", 2)[1] | ||||
| 			} | ||||
| 			if i < len(addrList) { | ||||
| 				address = strings.SplitN(addrList[i], ":", 2)[1] | ||||
| 			} | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
| 	return | ||||
| } | ||||
| 
 | ||||
| // StoreTemporaryVoucher saves voucher metadata as temporary entries in the DataStore.
 | ||||
| func StoreTemporaryVoucher(ctx context.Context, store DataStore, sessionId string, data *dataserviceapi.TokenHoldings) error { | ||||
| 	tempData := fmt.Sprintf("%s,%s,%s,%s", data.TokenSymbol, data.Balance, data.TokenDecimals, data.ContractAddress) | ||||
| 
 | ||||
| 	if err := store.WriteEntry(ctx, sessionId, DATA_TEMPORARY_VALUE, []byte(tempData)); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 
 | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| // GetTemporaryVoucherData retrieves temporary voucher metadata from the DataStore.
 | ||||
| func GetTemporaryVoucherData(ctx context.Context, store DataStore, sessionId string) (*dataserviceapi.TokenHoldings, error) { | ||||
| 	temp_data, err := store.ReadEntry(ctx, sessionId, DATA_TEMPORARY_VALUE) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 
 | ||||
| 	values := strings.SplitN(string(temp_data), ",", 4) | ||||
| 
 | ||||
| 	data := &dataserviceapi.TokenHoldings{} | ||||
| 
 | ||||
| 	data.TokenSymbol = values[0] | ||||
| 	data.Balance = values[1] | ||||
| 	data.TokenDecimals = values[2] | ||||
| 	data.ContractAddress = values[3] | ||||
| 
 | ||||
| 	return data, nil | ||||
| } | ||||
| 
 | ||||
| // UpdateVoucherData updates the active voucher data in the DataStore.
 | ||||
| func UpdateVoucherData(ctx context.Context, store DataStore, sessionId string, data *dataserviceapi.TokenHoldings) error { | ||||
| 	logg.TraceCtxf(ctx, "dtal", "data", data) | ||||
| 	// Active voucher data entries
 | ||||
| 	activeEntries := map[DataTyp][]byte{ | ||||
| 		DATA_ACTIVE_SYM:     []byte(data.TokenSymbol), | ||||
| 		DATA_ACTIVE_BAL:     []byte(data.Balance), | ||||
| 		DATA_ACTIVE_DECIMAL: []byte(data.TokenDecimals), | ||||
| 		DATA_ACTIVE_ADDRESS: []byte(data.ContractAddress), | ||||
| 	} | ||||
| 
 | ||||
| 	// Write active data
 | ||||
| 	for key, value := range activeEntries { | ||||
| 		if err := store.WriteEntry(ctx, sessionId, key, value); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
| 	return nil | ||||
| } | ||||
| @ -1,200 +0,0 @@ | ||||
| package common | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/alecthomas/assert/v2" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	visedb "git.defalsify.org/vise.git/db" | ||||
| 	memdb "git.defalsify.org/vise.git/db/mem" | ||||
| 	dbstorage "git.grassecon.net/grassrootseconomics/visedriver/storage/db" | ||||
| 	dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api" | ||||
| ) | ||||
| 
 | ||||
| // InitializeTestDb sets up and returns an in-memory database and store.
 | ||||
| func InitializeTestDb(t *testing.T) (context.Context, *UserDataStore) { | ||||
| 	ctx := context.Background() | ||||
| 
 | ||||
| 	// Initialize memDb
 | ||||
| 	db := memdb.NewMemDb() | ||||
| 	err := db.Connect(ctx, "") | ||||
| 	require.NoError(t, err, "Failed to connect to memDb") | ||||
| 
 | ||||
| 	// Create UserDataStore with memDb
 | ||||
| 	store := &UserDataStore{Db: db} | ||||
| 
 | ||||
| 	t.Cleanup(func() { | ||||
| 		db.Close() // Ensure the DB is closed after each test
 | ||||
| 	}) | ||||
| 
 | ||||
| 	return ctx, store | ||||
| } | ||||
| 
 | ||||
| func TestMatchVoucher(t *testing.T) { | ||||
| 	symbols := "1:SRF\n2:MILO" | ||||
| 	balances := "1:100\n2:200" | ||||
| 	decimals := "1:6\n2:4" | ||||
| 	addresses := "1:0xd4c288865Ce\n2:0x41c188d63Qa" | ||||
| 
 | ||||
| 	// Test for valid voucher
 | ||||
| 	symbol, balance, decimal, address := MatchVoucher("2", symbols, balances, decimals, addresses) | ||||
| 
 | ||||
| 	// Assertions for valid voucher
 | ||||
| 	assert.Equal(t, "MILO", symbol) | ||||
| 	assert.Equal(t, "200", balance) | ||||
| 	assert.Equal(t, "4", decimal) | ||||
| 	assert.Equal(t, "0x41c188d63Qa", address) | ||||
| 
 | ||||
| 	// Test for non-existent voucher
 | ||||
| 	symbol, balance, decimal, address = MatchVoucher("3", symbols, balances, decimals, addresses) | ||||
| 
 | ||||
| 	// Assertions for non-match
 | ||||
| 	assert.Equal(t, "", symbol) | ||||
| 	assert.Equal(t, "", balance) | ||||
| 	assert.Equal(t, "", decimal) | ||||
| 	assert.Equal(t, "", address) | ||||
| } | ||||
| 
 | ||||
| func TestProcessVouchers(t *testing.T) { | ||||
| 	holdings := []dataserviceapi.TokenHoldings{ | ||||
| 		{ContractAddress: "0xd4c288865Ce", TokenSymbol: "SRF", TokenDecimals: "6", Balance: "100000000"}, | ||||
| 		{ContractAddress: "0x41c188d63Qa", TokenSymbol: "MILO", TokenDecimals: "4", Balance: "200000000"}, | ||||
| 	} | ||||
| 
 | ||||
| 	expectedResult := VoucherMetadata{ | ||||
| 		Symbols:   "1:SRF\n2:MILO", | ||||
| 		Balances:  "1:100\n2:20000", | ||||
| 		Decimals:  "1:6\n2:4", | ||||
| 		Addresses: "1:0xd4c288865Ce\n2:0x41c188d63Qa", | ||||
| 	} | ||||
| 
 | ||||
| 	result := ProcessVouchers(holdings) | ||||
| 
 | ||||
| 	assert.Equal(t, expectedResult, result) | ||||
| } | ||||
| 
 | ||||
| func TestGetVoucherData(t *testing.T) { | ||||
| 	ctx := context.Background() | ||||
| 
 | ||||
| 	db := memdb.NewMemDb() | ||||
| 	err := db.Connect(ctx, "") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 
 | ||||
| 	prefix := ToBytes(visedb.DATATYPE_USERDATA) | ||||
| 	spdb := dbstorage.NewSubPrefixDb(db, prefix) | ||||
| 
 | ||||
| 	// Test voucher data
 | ||||
| 	mockData := map[DataTyp][]byte{ | ||||
| 		DATA_VOUCHER_SYMBOLS:   []byte("1:SRF\n2:MILO"), | ||||
| 		DATA_VOUCHER_BALANCES:  []byte("1:100\n2:200"), | ||||
| 		DATA_VOUCHER_DECIMALS:  []byte("1:6\n2:4"), | ||||
| 		DATA_VOUCHER_ADDRESSES: []byte("1:0xd4c288865Ce\n2:0x41c188d63Qa"), | ||||
| 	} | ||||
| 
 | ||||
| 	// Put the data
 | ||||
| 	for key, value := range mockData { | ||||
| 		err = spdb.Put(ctx, []byte(ToBytes(key)), []byte(value)) | ||||
| 		if err != nil { | ||||
| 			t.Fatal(err) | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
| 	result, err := GetVoucherData(ctx, spdb, "1") | ||||
| 
 | ||||
| 	assert.NoError(t, err) | ||||
| 	assert.Equal(t, "SRF", result.TokenSymbol) | ||||
| 	assert.Equal(t, "100", result.Balance) | ||||
| 	assert.Equal(t, "6", result.TokenDecimals) | ||||
| 	assert.Equal(t, "0xd4c288865Ce", result.ContractAddress) | ||||
| } | ||||
| 
 | ||||
| func TestStoreTemporaryVoucher(t *testing.T) { | ||||
| 	ctx, store := InitializeTestDb(t) | ||||
| 	sessionId := "session123" | ||||
| 
 | ||||
| 	// Test data
 | ||||
| 	voucherData := &dataserviceapi.TokenHoldings{ | ||||
| 		TokenSymbol:     "SRF", | ||||
| 		Balance:         "200", | ||||
| 		TokenDecimals:   "6", | ||||
| 		ContractAddress: "0xd4c288865Ce0985a481Eef3be02443dF5E2e4Ea9", | ||||
| 	} | ||||
| 
 | ||||
| 	// Execute the function being tested
 | ||||
| 	err := StoreTemporaryVoucher(ctx, store, sessionId, voucherData) | ||||
| 	require.NoError(t, err) | ||||
| 
 | ||||
| 	// Verify stored data
 | ||||
| 	expectedData := fmt.Sprintf("%s,%s,%s,%s", "SRF", "200", "6", "0xd4c288865Ce0985a481Eef3be02443dF5E2e4Ea9") | ||||
| 
 | ||||
| 	storedValue, err := store.ReadEntry(ctx, sessionId, DATA_TEMPORARY_VALUE) | ||||
| 	require.NoError(t, err) | ||||
| 	require.Equal(t, expectedData, string(storedValue), "Mismatch for key %v", DATA_TEMPORARY_VALUE) | ||||
| } | ||||
| 
 | ||||
| func TestGetTemporaryVoucherData(t *testing.T) { | ||||
| 	ctx, store := InitializeTestDb(t) | ||||
| 	sessionId := "session123" | ||||
| 
 | ||||
| 	// Test voucher data
 | ||||
| 	tempData := &dataserviceapi.TokenHoldings{ | ||||
| 		TokenSymbol:     "SRF", | ||||
| 		Balance:         "200", | ||||
| 		TokenDecimals:   "6", | ||||
| 		ContractAddress: "0xd4c288865Ce0985a481Eef3be02443dF5E2e4Ea9", | ||||
| 	} | ||||
| 
 | ||||
| 	// Store the data
 | ||||
| 	err := StoreTemporaryVoucher(ctx, store, sessionId, tempData) | ||||
| 	require.NoError(t, err) | ||||
| 
 | ||||
| 	// Execute the function being tested
 | ||||
| 	data, err := GetTemporaryVoucherData(ctx, store, sessionId) | ||||
| 	require.NoError(t, err) | ||||
| 	require.Equal(t, tempData, data) | ||||
| } | ||||
| 
 | ||||
| func TestUpdateVoucherData(t *testing.T) { | ||||
| 	ctx, store := InitializeTestDb(t) | ||||
| 	sessionId := "session123" | ||||
| 
 | ||||
| 	// New voucher data
 | ||||
| 	newData := &dataserviceapi.TokenHoldings{ | ||||
| 		TokenSymbol:     "SRF", | ||||
| 		Balance:         "200", | ||||
| 		TokenDecimals:   "6", | ||||
| 		ContractAddress: "0xd4c288865Ce0985a481Eef3be02443dF5E2e4Ea9", | ||||
| 	} | ||||
| 
 | ||||
| 	// Old temporary data
 | ||||
| 	tempData := &dataserviceapi.TokenHoldings{ | ||||
| 		TokenSymbol:     "OLD", | ||||
| 		Balance:         "100", | ||||
| 		TokenDecimals:   "8", | ||||
| 		ContractAddress: "0xold", | ||||
| 	} | ||||
| 	require.NoError(t, StoreTemporaryVoucher(ctx, store, sessionId, tempData)) | ||||
| 
 | ||||
| 	// Execute update
 | ||||
| 	err := UpdateVoucherData(ctx, store, sessionId, newData) | ||||
| 	require.NoError(t, err) | ||||
| 
 | ||||
| 	// Verify active data was stored correctly
 | ||||
| 	activeEntries := map[DataTyp][]byte{ | ||||
| 		DATA_ACTIVE_SYM:     []byte(newData.TokenSymbol), | ||||
| 		DATA_ACTIVE_BAL:     []byte(newData.Balance), | ||||
| 		DATA_ACTIVE_DECIMAL: []byte(newData.TokenDecimals), | ||||
| 		DATA_ACTIVE_ADDRESS: []byte(newData.ContractAddress), | ||||
| 	} | ||||
| 
 | ||||
| 	for key, expectedValue := range activeEntries { | ||||
| 		storedValue, err := store.ReadEntry(ctx, sessionId, key) | ||||
| 		require.NoError(t, err) | ||||
| 		require.Equal(t, expectedValue, storedValue, "Active data mismatch for key %v", key) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										189
									
								
								config/config.go
									
									
									
									
									
								
							
							
						
						
									
										189
									
								
								config/config.go
									
									
									
									
									
								
							| @ -1,55 +1,48 @@ | ||||
| package config | ||||
| 
 | ||||
| import ( | ||||
| 	"net/url" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/initializers" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	createAccountPath          = "/api/v2/account/create" | ||||
| 	trackStatusPath            = "/api/track" | ||||
| 	balancePathPrefix          = "/api/account" | ||||
| 	trackPath                  = "/api/v2/account/status" | ||||
| 	tokenTransferPrefix        = "/api/v2/token/transfer" | ||||
| 	voucherHoldingsPathPrefix  = "/api/v1/holdings" | ||||
| 	voucherTransfersPathPrefix = "/api/v1/transfers/last10" | ||||
| 	voucherDataPathPrefix      = "/api/v1/token" | ||||
| 	AliasPrefix                = "api/v1/alias" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/env" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/storage" | ||||
| 	slogging "github.com/grassrootseconomics/go-vise/slog" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	defaultLanguage		   = "eng" | ||||
| 	languages []string | ||||
| 	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 | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	custodialURLBase string | ||||
| 	dataURLBase      string | ||||
| 	BearerToken      string | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	CreateAccountURL    string | ||||
| 	TrackStatusURL      string | ||||
| 	BalanceURL          string | ||||
| 	TrackURL            string | ||||
| 	TokenTransferURL    string | ||||
| 	VoucherHoldingsURL  string | ||||
| 	VoucherTransfersURL string | ||||
| 	VoucherDataURL      string | ||||
| 	CheckAliasURL       string | ||||
| 	DbConn		string | ||||
| 	DefaultLanguage	    string | ||||
| 	Languages	[]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 = initializers.GetEnv("DEFAULT_LANGUAGE", defaultLanguage) | ||||
| 	languages = strings.Split(initializers.GetEnv("LANGUAGES", defaultLanguage), ",") | ||||
| 	defaultLanguage = env.GetEnv("DEFAULT_LANGUAGE", defaultLanguage) | ||||
| 	languages = strings.Split(env.GetEnv("LANGUAGES", defaultLanguage), ",") | ||||
| 	haveDefaultLanguage := false | ||||
| 	for i, v := range(languages) { | ||||
| 	for i, v := range languages { | ||||
| 		languages[i] = strings.ReplaceAll(v, " ", "") | ||||
| 		if languages[i] == defaultLanguage { | ||||
| 			haveDefaultLanguage = true | ||||
| @ -63,37 +56,98 @@ func setLanguage() error { | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func setBase() error { | ||||
| 	var err error | ||||
| 
 | ||||
| 	custodialURLBase = initializers.GetEnv("CUSTODIAL_URL_BASE", "http://localhost:5003") | ||||
| 	dataURLBase = initializers.GetEnv("DATA_URL_BASE", "http://localhost:5006") | ||||
| 	BearerToken = initializers.GetEnv("BEARER_TOKEN", "") | ||||
| 
 | ||||
| 	_, err = url.Parse(custodialURLBase) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	_, err = url.Parse(dataURLBase) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 
 | ||||
| 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 setConn() error { | ||||
| 	DbConn = initializers.GetEnv("DB_CONN", "") | ||||
| 	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 { | ||||
| 	err := setBase() | ||||
| 	if err != nil { | ||||
| 	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 | ||||
| 	} | ||||
| 	err = setConn() | ||||
| 	}) | ||||
| 	configManager.AddKey("DB_CONN", "", true, nil) | ||||
| 	// ... add other keys ?  or is enough :/ ...
 | ||||
| 
 | ||||
| 	err := setConn() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| @ -101,17 +155,10 @@ func LoadConfig() error { | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	CreateAccountURL, _ = url.JoinPath(custodialURLBase, createAccountPath) | ||||
| 	TrackStatusURL, _ = url.JoinPath(custodialURLBase, trackStatusPath) | ||||
| 	BalanceURL, _ = url.JoinPath(custodialURLBase, balancePathPrefix) | ||||
| 	TrackURL, _ = url.JoinPath(custodialURLBase, trackPath) | ||||
| 	TokenTransferURL, _ = url.JoinPath(custodialURLBase, tokenTransferPrefix) | ||||
| 	VoucherHoldingsURL, _ = url.JoinPath(dataURLBase, voucherHoldingsPathPrefix) | ||||
| 	VoucherTransfersURL, _ = url.JoinPath(dataURLBase, voucherTransfersPathPrefix) | ||||
| 	VoucherDataURL, _ = url.JoinPath(dataURLBase, voucherDataPathPrefix) | ||||
| 	CheckAliasURL, _ = url.JoinPath(dataURLBase, AliasPrefix) | ||||
| 	DefaultLanguage = defaultLanguage | ||||
| 	Languages = languages | ||||
| 
 | ||||
| 	// Report configuration
 | ||||
| 	configManager.Report("INFO") | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
							
								
								
									
										64
									
								
								config/config_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										64
									
								
								config/config_test.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,64 @@ | ||||
| //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") | ||||
| 		} | ||||
| 	}) | ||||
| } | ||||
							
								
								
									
										84
									
								
								config/reporter.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										84
									
								
								config/reporter.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,84 @@ | ||||
| //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) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										18
									
								
								config/reporter_noop.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								config/reporter_noop.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,18 @@ | ||||
| //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) {} | ||||
| @ -1,5 +0,0 @@ | ||||
| package debug | ||||
| 
 | ||||
| var ( | ||||
| 	DebugCap uint32 | ||||
| ) | ||||
							
								
								
									
										84
									
								
								debug/db.go
									
									
									
									
									
								
							
							
						
						
									
										84
									
								
								debug/db.go
									
									
									
									
									
								
							| @ -1,84 +0,0 @@ | ||||
| package debug | ||||
| 
 | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"encoding/binary" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/common" | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	dbTypStr map[common.DataTyp]string = make(map[common.DataTyp]string) | ||||
| ) | ||||
| 
 | ||||
| type KeyInfo struct { | ||||
| 	SessionId string | ||||
| 	Typ uint8 | ||||
| 	SubTyp common.DataTyp | ||||
| 	Label string | ||||
| 	Description string | ||||
| } | ||||
| 
 | ||||
| func (k KeyInfo) String() string { | ||||
| 	v := uint16(k.SubTyp) | ||||
| 	s := subTypToString(k.SubTyp) | ||||
| 	if s == "" { | ||||
| 		v = uint16(k.Typ) | ||||
| 		s = typToString(k.Typ) | ||||
| 	} | ||||
| 	return fmt.Sprintf("Session Id: %s\nTyp: %s (%d)\n", k.SessionId, s, v) | ||||
| } | ||||
| 
 | ||||
| func ToKeyInfo(k []byte, sessionId string) (KeyInfo, error) { | ||||
| 	o := KeyInfo{} | ||||
| 	b := []byte(sessionId) | ||||
| 
 | ||||
| 	if len(k) <= len(b) { | ||||
| 		return o, fmt.Errorf("storage key missing") | ||||
| 	} | ||||
| 
 | ||||
| 	o.SessionId = sessionId | ||||
| 
 | ||||
| 	o.Typ = uint8(k[0]) | ||||
| 	k = k[1:] | ||||
| 	o.SessionId = string(k[:len(b)]) | ||||
| 	k = k[len(b):] | ||||
| 
 | ||||
| 	if o.Typ == db.DATATYPE_USERDATA { | ||||
| 		if len(k) == 0 { | ||||
| 			return o, fmt.Errorf("missing subtype key") | ||||
| 		} | ||||
| 		v := binary.BigEndian.Uint16(k[:2]) | ||||
| 		o.SubTyp = common.DataTyp(v) | ||||
| 		o.Label = subTypToString(o.SubTyp) | ||||
| 		k = k[2:] | ||||
| 	} else { | ||||
| 		o.Label = typToString(o.Typ) | ||||
| 	} | ||||
| 
 | ||||
| 	if len(k) != 0 { | ||||
| 		return o, fmt.Errorf("excess key information") | ||||
| 	} | ||||
| 
 | ||||
| 	return o, nil | ||||
| } | ||||
| 
 | ||||
| func FromKey(k []byte) (KeyInfo, error) { | ||||
| 	o := KeyInfo{} | ||||
| 
 | ||||
| 	if len(k) < 4 { | ||||
| 		return o, fmt.Errorf("insufficient key length") | ||||
| 	} | ||||
| 
 | ||||
| 	sessionIdBytes := k[1:len(k)-2] | ||||
| 	return ToKeyInfo(k, string(sessionIdBytes)) | ||||
| } | ||||
| 
 | ||||
| func subTypToString(v common.DataTyp) string { | ||||
| 	return dbTypStr[v + db.DATATYPE_USERDATA + 1] | ||||
| } | ||||
| 
 | ||||
| func typToString(v uint8) string { | ||||
| 	return dbTypStr[common.DataTyp(uint16(v))] | ||||
| } | ||||
| @ -1,44 +0,0 @@ | ||||
| // +build debugdb
 | ||||
| 
 | ||||
| package debug | ||||
| 
 | ||||
| import ( | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/common" | ||||
| ) | ||||
| 
 | ||||
| func init() { | ||||
| 	DebugCap |= 1 | ||||
| 	dbTypStr[db.DATATYPE_STATE] = "internal state" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TRACKING_ID] = "tracking id" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_PUBLIC_KEY] = "public key" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_ACCOUNT_PIN] = "account pin" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_FIRST_NAME] = "first name" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_FAMILY_NAME] = "family name" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_YOB] = "year of birth" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_LOCATION] = "location" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_GENDER] = "gender" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_OFFERINGS] = "offerings" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_RECIPIENT] = "recipient" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_AMOUNT] = "amount" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TEMPORARY_VALUE] = "temporary value" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_ACTIVE_SYM] = "active sym" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_ACTIVE_BAL] = "active bal" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_BLOCKED_NUMBER] = "blocked number" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_PUBLIC_KEY_REVERSE] = "public_key_reverse" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_ACTIVE_DECIMAL] = "active decimal" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_ACTIVE_ADDRESS] = "active address" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_VOUCHER_SYMBOLS] = "voucher symbols" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_VOUCHER_BALANCES] = "voucher balances" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_VOUCHER_DECIMALS] = "voucher decimals" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_VOUCHER_ADDRESSES] = "voucher addresses" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_SENDERS] = "tx senders" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_RECIPIENTS] = "tx recipients" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_VALUES] = "tx values" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_ADDRESSES] = "tx addresses" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_HASHES] = "tx hashes" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_DATES] = "tx dates" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_SYMBOLS] = "tx symbols" | ||||
| 	dbTypStr[db.DATATYPE_USERDATA + 1 + common.DATA_TX_DECIMALS] = "tx decimals" | ||||
| } | ||||
| @ -1,78 +0,0 @@ | ||||
| package debug | ||||
| 
 | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/common" | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| ) | ||||
| 
 | ||||
| func TestDebugDbSubKeyInfo(t *testing.T) { | ||||
| 	s := "foo" | ||||
| 	b := []byte{0x20} | ||||
| 	b = append(b, []byte(s)...) | ||||
| 	b = append(b, []byte{0x00, 0x02}...) | ||||
| 	r, err := ToKeyInfo(b, s) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if r.SessionId != s { | ||||
| 		t.Fatalf("expected %s, got %s", s, r.SessionId) | ||||
| 	} | ||||
| 	if r.Typ != 32 { | ||||
| 		t.Fatalf("expected 64, got %d", r.Typ) | ||||
| 	} | ||||
| 	if r.SubTyp != 2 { | ||||
| 		t.Fatalf("expected 2, got %d", r.SubTyp) | ||||
| 	} | ||||
| 	if DebugCap & 1 > 0 { | ||||
| 		if r.Label != "tracking id" { | ||||
| 			t.Fatalf("expected 'tracking id', got '%s'", r.Label) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestDebugDbKeyInfo(t *testing.T) { | ||||
| 	s := "bar" | ||||
| 	b := []byte{0x10} | ||||
| 	b = append(b, []byte(s)...) | ||||
| 	r, err := ToKeyInfo(b, s) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if r.SessionId != s { | ||||
| 		t.Fatalf("expected %s, got %s", s, r.SessionId) | ||||
| 	} | ||||
| 	if r.Typ != 16 { | ||||
| 		t.Fatalf("expected 16, got %d", r.Typ) | ||||
| 	} | ||||
| 	if DebugCap & 1 > 0 { | ||||
| 		if r.Label != "internal state" { | ||||
| 			t.Fatalf("expected 'internal_state', got '%s'", r.Label) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestDebugDbKeyInfoRestore(t *testing.T) { | ||||
| 	s := "bar" | ||||
| 	b := []byte{db.DATATYPE_USERDATA} | ||||
| 	b = append(b, []byte(s)...) | ||||
| 	k := common.ToBytes(common.DATA_ACTIVE_SYM) | ||||
| 	b = append(b, k...) | ||||
| 
 | ||||
| 	r, err := ToKeyInfo(b, s) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if r.SessionId != s { | ||||
| 		t.Fatalf("expected %s, got %s", s, r.SessionId) | ||||
| 	} | ||||
| 	if r.Typ != 32 { | ||||
| 		t.Fatalf("expected 32, got %d", r.Typ) | ||||
| 	} | ||||
| 	if DebugCap & 1 > 0 { | ||||
| 		if r.Label != "active sym" { | ||||
| 			t.Fatalf("expected 'active sym', got '%s'", r.Label) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @ -1,3 +0,0 @@ | ||||
| url: http://localhost:7123 | ||||
| dial: "*384*96#" | ||||
| phoneNumber: +254722123456 | ||||
| @ -1,21 +0,0 @@ | ||||
| services: | ||||
|   ussd-pg-store: | ||||
|     image: postgres:17-alpine | ||||
|     restart: unless-stopped | ||||
|     user: postgres | ||||
|     environment: | ||||
|       - POSTGRES_PASSWORD=postgres | ||||
|       - POSTGRES_USER=postgres | ||||
|     volumes: | ||||
|       - ./init_db.sql:/docker-entrypoint-initdb.d/init_db.sql | ||||
|       - ussd-pg:/var/lib/postgresql/data | ||||
|     ports: | ||||
|       - "127.0.0.1:5432:5432" | ||||
|     healthcheck: | ||||
|       test: ["CMD-SHELL", "pg_isready"] | ||||
|       interval: 10s | ||||
|       timeout: 5s | ||||
|       retries: 5 | ||||
| volumes: | ||||
|   ussd-pg: | ||||
|     driver: local | ||||
| @ -1 +0,0 @@ | ||||
| CREATE DATABASE urdt_ussd; | ||||
| @ -10,7 +10,7 @@ This document describes component versions: | ||||
| 
 | ||||
| ## User profile data | ||||
| 
 | ||||
| All user profile items are stored under keys matching the user's session id, prefixed with the 8-bit value `git.defalsify.org/vise.git/db.DATATYPE_USERDATA` (32), and followed with a 16-big big-endian value subprefix. | ||||
| 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: | ||||
| 
 | ||||
|  | ||||
| @ -2,8 +2,9 @@ package entry | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"git.defalsify.org/vise.git/resource" | ||||
| 	"git.defalsify.org/vise.git/persist" | ||||
| 
 | ||||
| 	"github.com/grassrootseconomics/go-vise/persist" | ||||
| 	"github.com/grassrootseconomics/go-vise/resource" | ||||
| ) | ||||
| 
 | ||||
| type EntryHandler interface { | ||||
|  | ||||
							
								
								
									
										2
									
								
								initializers/load.go → env/load.go
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								initializers/load.go → env/load.go
									
									
									
									
										vendored
									
									
								
							| @ -1,4 +1,4 @@ | ||||
| package initializers | ||||
| package env | ||||
| 
 | ||||
| import ( | ||||
| 	"log" | ||||
| @ -1,15 +1,15 @@ | ||||
| package errors | ||||
| 
 | ||||
| import ( | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/internal/handlers" | ||||
| 	"errors" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	ErrInvalidRequest = handlers.ErrInvalidRequest | ||||
| 	ErrSessionMissing = handlers.ErrSessionMissing | ||||
| 	ErrInvalidInput = handlers.ErrInvalidInput | ||||
| 	ErrStorage = handlers.ErrStorage | ||||
| 	ErrEngineType = handlers.ErrEngineType | ||||
| 	ErrEngineInit = handlers.ErrEngineInit | ||||
| 	ErrEngineExec = handlers.ErrEngineExec | ||||
| 	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") | ||||
| ) | ||||
|  | ||||
							
								
								
									
										34
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										34
									
								
								go.mod
									
									
									
									
									
								
							| @ -1,39 +1,27 @@ | ||||
| module git.grassecon.net/grassrootseconomics/visedriver | ||||
| 
 | ||||
| go 1.23.0 | ||||
| go 1.24 | ||||
| 
 | ||||
| toolchain go1.24.6 | ||||
| 
 | ||||
| require ( | ||||
| 	git.defalsify.org/vise.git v0.2.3-0.20250103172917-3e190a44568d | ||||
| 	github.com/alecthomas/assert/v2 v2.2.2 | ||||
| 	github.com/gofrs/uuid v4.4.0+incompatible | ||||
| 	github.com/grassrootseconomics/eth-custodial v1.3.0-beta | ||||
| 	github.com/grassrootseconomics/ussd-data-service v1.2.0-beta | ||||
| 	github.com/jackc/pgx/v5 v5.7.1 | ||||
| 	github.com/grassrootseconomics/go-vise v0.5.0 | ||||
| 	github.com/jackc/pgx/v5 v5.7.5 | ||||
| 	github.com/joho/godotenv v1.5.1 | ||||
| 	github.com/peteole/testdata-loader v0.3.0 | ||||
| 	github.com/stretchr/testify v1.9.0 | ||||
| 	golang.org/x/crypto v0.27.0 | ||||
| ) | ||||
| 
 | ||||
| require ( | ||||
| 	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/graygnuorg/go-gdbm v0.0.0-20220711140707-71387d66dce4 // indirect | ||||
| 	github.com/hexops/gotextdiff v1.0.3 // 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/puddle/v2 v2.2.2 // indirect | ||||
| 	github.com/kr/text v0.2.0 // indirect | ||||
| 	github.com/mattn/kinako v0.0.0-20170717041458-332c0a7e205a // indirect | ||||
| 	github.com/leonelquinteros/gotext v1.7.2 // indirect | ||||
| 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect | ||||
| 	github.com/rogpeppe/go-internal v1.13.1 // indirect | ||||
| 	github.com/stretchr/objx v0.5.2 // indirect | ||||
| 	github.com/stretchr/testify v1.9.0 // indirect | ||||
| 	github.com/x448/float16 v0.8.4 // indirect | ||||
| 	golang.org/x/sync v0.8.0 // indirect | ||||
| 	golang.org/x/sys v0.25.0 // indirect | ||||
| 	golang.org/x/text v0.18.0 // indirect | ||||
| 	gopkg.in/leonelquinteros/gotext.v1 v1.3.1 // indirect | ||||
| 	gopkg.in/yaml.v3 v3.0.1 // 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 | ||||
| ) | ||||
|  | ||||
							
								
								
									
										65
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										65
									
								
								go.sum
									
									
									
									
									
								
							| @ -1,76 +1,47 @@ | ||||
| git.defalsify.org/vise.git v0.2.3-0.20250103172917-3e190a44568d h1:bPAOVZOX4frSGhfOdcj7kc555f8dc9DmMd2YAyC2AMw= | ||||
| git.defalsify.org/vise.git v0.2.3-0.20250103172917-3e190a44568d/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/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.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/grassrootseconomics/ussd-data-service v1.2.0-beta h1:fn1gwbWIwHVEBtUC2zi5OqTlfI/5gU1SMk0fgGixIXk= | ||||
| github.com/grassrootseconomics/ussd-data-service v1.2.0-beta/go.mod h1:omfI0QtUwIdpu9gMcUqLMCG8O1XWjqJGBx1qUMiGWC0= | ||||
| 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/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/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.1 h1:x7SYsPBYDkHDksogeSmZZ5xzThcTgRz++I5E+ePFUcs= | ||||
| github.com/jackc/pgx/v5 v5.7.1/go.mod h1:e7O26IywZZ+naJtWWos6i6fvWK+29etgITqrqHLfoZA= | ||||
| 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/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/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/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/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.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/sys v0.25.0 h1:r+8e+loiHxRqhXVl6ML1nO3l1+oFoWbnlu2Ehimmi34= | ||||
| golang.org/x/sys v0.25.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= | ||||
| golang.org/x/term v0.24.0 h1:Mh5cbb+Zk2hqqXNO7S1iTjEphVL+jb8ZWaqh/g+JWkM= | ||||
| golang.org/x/term v0.24.0/go.mod h1:lOBK/LVxemqiMij05LGJ0tzNr8xlmwBRJ81PX6wVLH8= | ||||
| golang.org/x/text v0.18.0 h1:XvMDiNzPAl0jr17s6W9lcaIhGUfUORdGCNsuLmPG224= | ||||
| golang.org/x/text v0.18.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY= | ||||
| 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= | ||||
| 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,54 +0,0 @@ | ||||
| package handlers | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"io" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/engine" | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| 	"git.defalsify.org/vise.git/persist" | ||||
| 	"git.defalsify.org/vise.git/resource" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/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(context context.Context, 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() | ||||
| } | ||||
							
								
								
									
										113
									
								
								request/base.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										113
									
								
								request/base.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,113 @@ | ||||
| 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 | ||||
| } | ||||
| @ -5,7 +5,7 @@ import ( | ||||
| 	"io/ioutil" | ||||
| 	"net/http" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/internal/handlers" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/errors" | ||||
| ) | ||||
| 
 | ||||
| type DefaultRequestParser struct { | ||||
| @ -14,11 +14,11 @@ type DefaultRequestParser struct { | ||||
| func (rp *DefaultRequestParser) GetSessionId(ctx context.Context, rq any) (string, error) { | ||||
| 	rqv, ok := rq.(*http.Request) | ||||
| 	if !ok { | ||||
| 		return "", handlers.ErrInvalidRequest | ||||
| 		return "", errors.ErrInvalidRequest | ||||
| 	} | ||||
| 	v := rqv.Header.Get("X-Vise-Session") | ||||
| 	if v == "" { | ||||
| 		return "", handlers.ErrSessionMissing | ||||
| 		return "", errors.ErrSessionMissing | ||||
| 	} | ||||
| 	return v, nil | ||||
| } | ||||
| @ -26,7 +26,7 @@ func (rp *DefaultRequestParser) GetSessionId(ctx context.Context, rq any) (strin | ||||
| func (rp *DefaultRequestParser) GetInput(rq any) ([]byte, error) { | ||||
| 	rqv, ok := rq.(*http.Request) | ||||
| 	if !ok { | ||||
| 		return nil, handlers.ErrInvalidRequest | ||||
| 		return nil, errors.ErrInvalidRequest | ||||
| 	} | ||||
| 	defer rqv.Body.Close() | ||||
| 	v, err := ioutil.ReadAll(rqv.Body) | ||||
| @ -4,21 +4,21 @@ import ( | ||||
| 	"net/http" | ||||
| 	"strconv" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/request" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/errors" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/request" | ||||
| 	slogging "github.com/grassrootseconomics/go-vise/slog" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla().WithDomain("visedriver.http.session") | ||||
| 	logg = slogging.Get().With("component", "visedriver.http.session") | ||||
| ) | ||||
| 
 | ||||
| // HTTPSessionHandler implements the session handler for HTTP
 | ||||
| type HTTPSessionHandler struct { | ||||
| // HTTPRequestHandler implements the session handler for HTTP
 | ||||
| type HTTPRequestHandler struct { | ||||
| 	request.RequestHandler | ||||
| } | ||||
| 
 | ||||
| func (f *HTTPSessionHandler) WriteError(w http.ResponseWriter, code int, err error) { | ||||
| 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) | ||||
| @ -29,13 +29,13 @@ func (f *HTTPSessionHandler) WriteError(w http.ResponseWriter, code int, err err | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func NewHTTPSessionHandler(h request.RequestHandler) *HTTPSessionHandler { | ||||
| 	return &HTTPSessionHandler{ | ||||
| func NewHTTPRequestHandler(h request.RequestHandler) *HTTPRequestHandler { | ||||
| 	return &HTTPRequestHandler{ | ||||
| 		RequestHandler: h, | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func (hh *HTTPSessionHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { | ||||
| func (hh *HTTPRequestHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) { | ||||
| 	var code int | ||||
| 	var err error | ||||
| 	var perr error | ||||
| @ -80,7 +80,7 @@ func (hh *HTTPSessionHandler) ServeHTTP(w http.ResponseWriter, req *http.Request | ||||
| 	w.WriteHeader(200) | ||||
| 	w.Header().Set("Content-Type", "text/plain") | ||||
| 	rqs, err = hh.Output(rqs) | ||||
| 	rqs, perr = hh.Reset(rqs) | ||||
| 	rqs, perr = hh.Reset(rqs.Ctx, rqs) | ||||
| 	if err != nil { | ||||
| 		hh.WriteError(w, 500, err) | ||||
| 		return | ||||
| @ -8,10 +8,10 @@ import ( | ||||
| 	"net/http/httptest" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/engine" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/internal/handlers" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/testutil/mocks/httpmocks" | ||||
| 	"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
 | ||||
| @ -24,7 +24,7 @@ func (e *errorReader) Read(p []byte) (n int, err error) { | ||||
| 	return 0, errors.New("read error") | ||||
| } | ||||
| 
 | ||||
| func TestSessionHandler_ServeHTTP(t *testing.T) { | ||||
| func TestRequestHandler_ServeHTTP(t *testing.T) { | ||||
| 	tests := []struct { | ||||
| 		name           string | ||||
| 		sessionID      string | ||||
| @ -44,14 +44,14 @@ func TestSessionHandler_ServeHTTP(t *testing.T) { | ||||
| 		{ | ||||
| 			name:           "Missing Session ID", | ||||
| 			sessionID:      "", | ||||
| 			parserErr:      handlers.ErrSessionMissing, | ||||
| 			parserErr:      viseerrors.ErrSessionMissing, | ||||
| 			expectedStatus: http.StatusBadRequest, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:           "Process Error", | ||||
| 			sessionID:      "123", | ||||
| 			input:          []byte("test input"), | ||||
| 			processErr:     handlers.ErrStorage, | ||||
| 			processErr:     viseerrors.ErrStorage, | ||||
| 			expectedStatus: http.StatusInternalServerError, | ||||
| 		}, | ||||
| 		{ | ||||
| @ -88,7 +88,7 @@ func TestSessionHandler_ServeHTTP(t *testing.T) { | ||||
| 				OutputFunc: func(rs request.RequestSession) (request.RequestSession, error) { | ||||
| 					return rs, tt.outputErr | ||||
| 				}, | ||||
| 				ResetFunc: func(rs request.RequestSession) (request.RequestSession, error) { | ||||
| 				ResetFunc: func(ctx context.Context, rs request.RequestSession) (request.RequestSession, error) { | ||||
| 					return rs, tt.resetErr | ||||
| 				}, | ||||
| 				GetRequestParserFunc: func() request.RequestParser { | ||||
| @ -99,7 +99,7 @@ func TestSessionHandler_ServeHTTP(t *testing.T) { | ||||
| 				}, | ||||
| 			} | ||||
| 
 | ||||
| 			sessionHandler := &HTTPSessionHandler{ | ||||
| 			sessionHandler := &HTTPRequestHandler{ | ||||
| 				RequestHandler: mockRequestHandler, | ||||
| 			} | ||||
| 
 | ||||
| @ -118,8 +118,8 @@ func TestSessionHandler_ServeHTTP(t *testing.T) { | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestSessionHandler_WriteError(t *testing.T) { | ||||
| 	handler := &HTTPSessionHandler{} | ||||
| func TestRequestHandler_WriteError(t *testing.T) { | ||||
| 	handler := &HTTPRequestHandler{} | ||||
| 	mockWriter := &httpmocks.MockWriter{} | ||||
| 	err := errors.New("test error") | ||||
| 
 | ||||
| @ -151,13 +151,13 @@ func TestDefaultRequestParser_GetSessionId(t *testing.T) { | ||||
| 			name:          "Missing Session ID", | ||||
| 			request:       httptest.NewRequest(http.MethodPost, "/", nil), | ||||
| 			expectedID:    "", | ||||
| 			expectedError: handlers.ErrSessionMissing, | ||||
| 			expectedError: viseerrors.ErrSessionMissing, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name:          "Invalid Request Type", | ||||
| 			request:       invalidRequestType{}, | ||||
| 			expectedID:    "", | ||||
| 			expectedError: handlers.ErrInvalidRequest, | ||||
| 			expectedError: viseerrors.ErrInvalidRequest, | ||||
| 		}, | ||||
| 	} | ||||
| 
 | ||||
| @ -165,7 +165,7 @@ func TestDefaultRequestParser_GetSessionId(t *testing.T) { | ||||
| 
 | ||||
| 	for _, tt := range tests { | ||||
| 		t.Run(tt.name, func(t *testing.T) { | ||||
| 			id, err := parser.GetSessionId(context.Background(),tt.request) | ||||
| 			id, err := parser.GetSessionId(context.Background(), tt.request) | ||||
| 
 | ||||
| 			if id != tt.expectedID { | ||||
| 				t.Errorf("Expected session ID %s, got %s", tt.expectedID, id) | ||||
| @ -203,7 +203,7 @@ func TestDefaultRequestParser_GetInput(t *testing.T) { | ||||
| 			name:          "Invalid Request Type", | ||||
| 			request:       invalidRequestType{}, | ||||
| 			expectedInput: nil, | ||||
| 			expectedError: handlers.ErrInvalidRequest, | ||||
| 			expectedError: viseerrors.ErrInvalidRequest, | ||||
| 		}, | ||||
| 		{ | ||||
| 			name: "Read Error", | ||||
| @ -4,39 +4,39 @@ import ( | ||||
| 	"context" | ||||
| 	"io" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/resource" | ||||
| 	"git.defalsify.org/vise.git/persist" | ||||
| 	"git.defalsify.org/vise.git/engine" | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| 	"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 = logging.NewVanilla().WithDomain("visedriver.request") | ||||
| 	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 | ||||
| 	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(ctx context.Context, rq any) (string, error) | ||||
| 	GetInput(rq any) ([]byte, error) | ||||
| 	GetSessionId(context.Context, any) (string, error) | ||||
| 	GetInput(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() | ||||
| 	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) | ||||
| } | ||||
|  | ||||
							
								
								
									
										116
									
								
								session/base.go
									
									
									
									
									
								
							
							
						
						
									
										116
									
								
								session/base.go
									
									
									
									
									
								
							| @ -1,116 +0,0 @@ | ||||
| package session | ||||
| 
 | ||||
| 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.defalsify.org/vise.git/logging" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/request" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/storage" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/errors" | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/entry" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla().WithDomain("visedriver.session") | ||||
| ) | ||||
| 
 | ||||
| type BaseSessionHandler struct { | ||||
| 	cfgTemplate engine.Config | ||||
| 	rp request.RequestParser | ||||
| 	rs resource.Resource | ||||
| 	hn entry.EntryHandler | ||||
| 	provider storage.StorageProvider | ||||
| } | ||||
| 
 | ||||
| //func NewBaseSessionHandler(cfg engine.Config, rs resource.Resource, stateDb db.Db, userdataDb db.Db, rp request.RequestParser, hn *handlers.Handlers) *BaseSessionHandler {
 | ||||
| func NewBaseSessionHandler(cfg engine.Config, rs resource.Resource, stateDb db.Db, userdataDb db.Db, rp request.RequestParser, hn entry.EntryHandler) *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 request.RequestSession) (request.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, errors.ErrStorage | ||||
| 	} | ||||
| 
 | ||||
| 	//f.hn = f.hn.WithPersister(rqs.Storage.Persister)
 | ||||
| 	f.hn.SetPersister(rqs.Storage.Persister) | ||||
| 	defer func() { | ||||
| 		f.hn.Exit() | ||||
| 	}() | ||||
| 	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, errors.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 request.RequestSession) (request.RequestSession,  error) { | ||||
| 	var err error | ||||
| 	_, err = rqs.Engine.Flush(rqs.Ctx, rqs.Writer) | ||||
| 	return rqs, err | ||||
| } | ||||
| 
 | ||||
| func(f *BaseSessionHandler) Reset(rqs request.RequestSession) (request.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() request.RequestParser { | ||||
| 	return f.rp | ||||
| } | ||||
							
								
								
									
										98
									
								
								storage/conn.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										98
									
								
								storage/conn.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,98 @@ | ||||
| package storage | ||||
| 
 | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"net/url" | ||||
| ) | ||||
| 
 | ||||
| type DbMode uint8 | ||||
| 
 | ||||
| const ( | ||||
| 	DBTYPE_NONE = iota | ||||
| 	DBTYPE_MEM | ||||
| 	DBTYPE_FS | ||||
| 	DBTYPE_GDBM | ||||
| 	DBTYPE_POSTGRES | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	DBMODE_ANY DbMode = iota | ||||
| 	DBMODE_BINARY | ||||
| 	DBMODE_TEXT | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	STORETYPE_STATE = iota | ||||
| 	STORETYPE_RESOURCE | ||||
| 	STORETYPE_USER | ||||
| 	_STORETYPE_MAX | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	DbModeDebug  = []string{"ANY", "BIN", "TXT"} | ||||
| 	DbTypeDebug  = []string{"NONE", "MEM", "FS", "GDBM", "POSTGRES"} | ||||
| 	DbStoreDebug = []string{"STATE", "RESOURCE", "USER"} | ||||
| ) | ||||
| 
 | ||||
| type Conns map[int8]ConnData | ||||
| 
 | ||||
| func NewConns() Conns { | ||||
| 	c := make(Conns) | ||||
| 	return c | ||||
| } | ||||
| 
 | ||||
| func (c Conns) Set(conn ConnData, typ int8) { | ||||
| 	if typ < 0 || typ >= _STORETYPE_MAX { | ||||
| 		panic(fmt.Errorf("invalid store type: %d", typ)) | ||||
| 	} | ||||
| 	c[typ] = conn | ||||
| } | ||||
| 
 | ||||
| func (c Conns) Have(conn *ConnData) int8 { | ||||
| 	for i := range _STORETYPE_MAX { | ||||
| 		ii := int8(i) | ||||
| 		v, ok := c[ii] | ||||
| 		if !ok { | ||||
| 			continue | ||||
| 		} | ||||
| 		if v.Raw() == conn.Raw() { | ||||
| 			if v.Mode() == DBMODE_ANY || v.Mode() == conn.Mode() { | ||||
| 				return ii | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return -1 | ||||
| } | ||||
| 
 | ||||
| type ConnData struct { | ||||
| 	typ    int | ||||
| 	str    string | ||||
| 	domain string | ||||
| 	mode   DbMode | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) DbType() int { | ||||
| 	return cd.typ | ||||
| } | ||||
| 
 | ||||
| func (cd ConnData) String() string { | ||||
| 	return fmt.Sprintf("conn: %s, mod %s, typ %s", cd.str, DbModeDebug[uint8(cd.mode)], DbTypeDebug[uint8(cd.typ)]) | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) Domain() string { | ||||
| 	return cd.domain | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) Mode() DbMode { | ||||
| 	return cd.mode | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) Path() string { | ||||
| 	v, _ := url.Parse(cd.str) | ||||
| 	v.RawQuery = "" | ||||
| 	return v.String() | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) Raw() string { | ||||
| 	return cd.str | ||||
| } | ||||
| @ -1,127 +0,0 @@ | ||||
| package storage | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| 	gdbmdb "git.defalsify.org/vise.git/db/gdbm" | ||||
| 	"git.defalsify.org/vise.git/lang" | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla().WithDomain("gdbmstorage") | ||||
| ) | ||||
| 
 | ||||
| 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 | ||||
| } | ||||
| 
 | ||||
| func(tdb *ThreadGdbmDb) Dump(ctx context.Context, key []byte) (*db.Dumper, error) { | ||||
| 	tdb.reserve() | ||||
| 	defer tdb.release() | ||||
| 	return tdb.db.Dump(ctx, key) | ||||
| } | ||||
| @ -1,43 +0,0 @@ | ||||
| package storage | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| ) | ||||
| 
 | ||||
| // PrefixDb interface abstracts the database operations.
 | ||||
| type PrefixDb interface { | ||||
| 	Get(ctx context.Context, key []byte) ([]byte, error) | ||||
| 	Put(ctx context.Context, key []byte, val []byte) error | ||||
| } | ||||
| 
 | ||||
| var _ PrefixDb = (*SubPrefixDb)(nil) | ||||
| 
 | ||||
| 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(db.DATATYPE_USERDATA) | ||||
| 	key = s.toKey(key) | ||||
| 	return s.store.Get(ctx, key) | ||||
| } | ||||
| 
 | ||||
| func (s *SubPrefixDb) Put(ctx context.Context, key []byte, val []byte) error { | ||||
| 	s.store.SetPrefix(db.DATATYPE_USERDATA) | ||||
| 	key = s.toKey(key) | ||||
| 	return s.store.Put(ctx, key, val) | ||||
| } | ||||
| @ -1,54 +0,0 @@ | ||||
| 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) | ||||
| 	} | ||||
| } | ||||
| @ -4,38 +4,9 @@ import ( | ||||
| 	"fmt" | ||||
| 	"net/url" | ||||
| 	"path" | ||||
| 	"path/filepath" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| 	DBTYPE_MEM = iota | ||||
| 	DBTYPE_GDBM | ||||
| 	DBTYPE_POSTGRES | ||||
| ) | ||||
| 
 | ||||
| type ConnData struct { | ||||
| 	typ int | ||||
| 	str string | ||||
| 	domain string | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) DbType() int { | ||||
| 	return cd.typ | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) String() string { | ||||
| 	return cd.str | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) Domain() string { | ||||
| 	return cd.domain | ||||
| } | ||||
| 
 | ||||
| func (cd *ConnData) Path() string { | ||||
| 	v, _ := url.Parse(cd.str) | ||||
| 	v.RawQuery = "" | ||||
| 	return v.String() | ||||
| } | ||||
| 
 | ||||
| func probePostgres(s string) (string, string, bool) { | ||||
| 	domain := "public" | ||||
| 	v, err := url.Parse(s) | ||||
| @ -53,21 +24,62 @@ func probePostgres(s string) (string, string, bool) { | ||||
| } | ||||
| 
 | ||||
| func probeGdbm(s string) (string, string, bool) { | ||||
| 	if !path.IsAbs(s) { | ||||
| 	domain := "public" | ||||
| 	v, err := url.Parse(s) | ||||
| 	if err != nil { | ||||
| 		return "", "", false | ||||
| 	} | ||||
| 	if v.Scheme != "gdbm" { | ||||
| 		return "", "", false | ||||
| 	} | ||||
| 	s = v.Path | ||||
| 	return s, domain, true | ||||
| } | ||||
| 
 | ||||
| func probeFs(s string) (string, string, bool) { | ||||
| 	var err error | ||||
| 
 | ||||
| 	v, _ := url.Parse(s) | ||||
| 	if v.Scheme != "" && v.Scheme != "file://" { | ||||
| 		return "", "", false | ||||
| 	} | ||||
| 
 | ||||
| 	if !path.IsAbs(s) { | ||||
| 		s, err = filepath.Abs(s) | ||||
| 		if err != nil { | ||||
| 			panic(err) | ||||
| 		} | ||||
| 	} | ||||
| 	s = path.Clean(s) | ||||
| 	return s, "", true | ||||
| } | ||||
| 
 | ||||
| func probeMem(s string) (string, string, bool) { | ||||
| 	if s != "" { | ||||
| 		return "", "", false | ||||
| 	} | ||||
| 	return "", "", true | ||||
| } | ||||
| 
 | ||||
| func ToConnDataMode(connStr string, mode DbMode) (ConnData, error) { | ||||
| 	o, err := ToConnData(connStr) | ||||
| 	if err != nil { | ||||
| 		return o, err | ||||
| 	} | ||||
| 	o.mode = mode | ||||
| 	return o, nil | ||||
| } | ||||
| 
 | ||||
| func ToConnData(connStr string) (ConnData, error) { | ||||
| 	var o ConnData | ||||
| 
 | ||||
| 	if connStr == "" { | ||||
| 	v, domain, ok := probeMem(connStr) | ||||
| 	if ok { | ||||
| 		o.typ = DBTYPE_MEM | ||||
| 		return o, nil | ||||
| 	} | ||||
| 
 | ||||
| 	v, domain, ok := probePostgres(connStr) | ||||
| 	v, domain, ok = probePostgres(connStr) | ||||
| 	if ok { | ||||
| 		o.typ = DBTYPE_POSTGRES | ||||
| 		o.str = v | ||||
| @ -82,5 +94,12 @@ func ToConnData(connStr string) (ConnData, error) { | ||||
| 		return o, nil | ||||
| 	} | ||||
| 
 | ||||
| 	v, _, ok = probeFs(connStr) | ||||
| 	if ok { | ||||
| 		o.typ = DBTYPE_FS | ||||
| 		o.str = v | ||||
| 		return o, nil | ||||
| 	} | ||||
| 
 | ||||
| 	return o, fmt.Errorf("invalid connection string: %s", connStr) | ||||
| } | ||||
|  | ||||
| @ -5,24 +5,53 @@ import ( | ||||
| ) | ||||
| 
 | ||||
| func TestParseConnStr(t *testing.T) { | ||||
| 	_, err := ToConnData("postgres://foo:bar@localhost:5432/baz") | ||||
| 	v, err := ToConnData("postgres://foo:bar@localhost:5432/baz") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err)	 | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	_, err = ToConnData("/foo/bar") | ||||
| 	if v.DbType() != DBTYPE_POSTGRES { | ||||
| 		t.Fatalf("expected type %v, got %v", DBTYPE_POSTGRES, v.DbType()) | ||||
| 	} | ||||
| 	v, err = ToConnData("gdbm:///foo/bar") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err)	 | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	_, err = ToConnData("/foo/bar/") | ||||
| 	if v.DbType() != DBTYPE_GDBM { | ||||
| 		t.Fatalf("expected type %v, got %v", DBTYPE_GDBM, v.DbType()) | ||||
| 	} | ||||
| 	v, err = ToConnData("/foo/bar") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err)	 | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	_, err = ToConnData("foo/bar") | ||||
| 	if v.DbType() != DBTYPE_FS { | ||||
| 		t.Fatalf("expected type %v, got %v", DBTYPE_FS, v.DbType()) | ||||
| 	} | ||||
| 	v, err = ToConnData("/foo/bar/") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if v.DbType() != DBTYPE_FS { | ||||
| 		t.Fatalf("expected type %v, got %v", DBTYPE_FS, v.DbType()) | ||||
| 	} | ||||
| 	v, err = ToConnData("foo/bar") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if v.DbType() != DBTYPE_FS { | ||||
| 		t.Fatalf("expected type %v, got %v", DBTYPE_FS, v.DbType()) | ||||
| 	} | ||||
| 	v, err = ToConnData("") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if v.DbType() != DBTYPE_MEM { | ||||
| 		t.Fatalf("expected type %v, got %v", DBTYPE_MEM, v.DbType()) | ||||
| 	} | ||||
| 	v, err = ToConnData("http://foo/bar") | ||||
| 	if err == nil { | ||||
| 		t.Fatalf("expected error") | ||||
| 	} | ||||
| 	_, err = ToConnData("http://foo/bar") | ||||
| 	if err == nil { | ||||
| 		t.Fatalf("expected error") | ||||
| 	if v.DbType() != DBTYPE_NONE { | ||||
| 		t.Fatalf("expected type %v, got %v", DBTYPE_NONE, v.DbType()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| @ -1,8 +1,10 @@ | ||||
| package storage | ||||
| 
 | ||||
| import ( | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| 	"git.defalsify.org/vise.git/persist" | ||||
| 	"context" | ||||
| 
 | ||||
| 	"github.com/grassrootseconomics/go-vise/db" | ||||
| 	"github.com/grassrootseconomics/go-vise/persist" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
| @ -10,14 +12,18 @@ const ( | ||||
| ) | ||||
| 
 | ||||
| type Storage struct { | ||||
| 	Persister *persist.Persister | ||||
| 	UserdataDb db.Db	 | ||||
| 	Persister  *persist.Persister | ||||
| 	UserdataDb db.Db | ||||
| } | ||||
| 
 | ||||
| func (s *Storage) Close(ctx context.Context) error { | ||||
| 	return s.UserdataDb.Close(ctx) | ||||
| } | ||||
| 
 | ||||
| type StorageProvider interface { | ||||
| 	Get(sessionId string) (*Storage, error) | ||||
| 	Put(sessionId string, storage *Storage) error | ||||
| 	Close() error | ||||
| 	Get(ctx context.Context, sessionId string) (*Storage, error) | ||||
| 	Put(ctx context.Context, sessionId string, storage *Storage) error | ||||
| 	Close(ctx context.Context) error | ||||
| } | ||||
| 
 | ||||
| type SimpleStorageProvider struct { | ||||
| @ -29,20 +35,22 @@ func NewSimpleStorageProvider(stateStore db.Db, userdataStore db.Db) StorageProv | ||||
| 	pe = pe.WithFlush() | ||||
| 	return &SimpleStorageProvider{ | ||||
| 		Storage: &Storage{ | ||||
| 			Persister: pe, | ||||
| 			Persister:  pe, | ||||
| 			UserdataDb: userdataStore, | ||||
| 		}, | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func (p *SimpleStorageProvider) Get(sessionId string) (*Storage, error) { | ||||
| func (p *SimpleStorageProvider) Get(ctx context.Context, sessionId string) (*Storage, error) { | ||||
| 	p.Storage.UserdataDb.Start(ctx) | ||||
| 	return p.Storage, nil | ||||
| } | ||||
| 
 | ||||
| func (p *SimpleStorageProvider) Put(sessionId string, storage *Storage) error { | ||||
| func (p *SimpleStorageProvider) Put(ctx context.Context, sessionId string, storage *Storage) error { | ||||
| 	storage.UserdataDb.Stop(ctx) | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func (p *SimpleStorageProvider) Close() error { | ||||
| 	return p.Storage.UserdataDb.Close() | ||||
| func (p *SimpleStorageProvider) Close(ctx context.Context) error { | ||||
| 	return p.Storage.Close(ctx) | ||||
| } | ||||
|  | ||||
| @ -2,92 +2,126 @@ package storage | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"path" | ||||
| 	 | ||||
| 
 | ||||
| 	"github.com/grassrootseconomics/go-vise/db" | ||||
| 	fsdb "github.com/grassrootseconomics/go-vise/db/fs" | ||||
| 	memdb "github.com/grassrootseconomics/go-vise/db/mem" | ||||
| 	"github.com/grassrootseconomics/go-vise/db/postgres" | ||||
| 	"github.com/grassrootseconomics/go-vise/lang" | ||||
| 	"github.com/grassrootseconomics/go-vise/persist" | ||||
| 	"github.com/grassrootseconomics/go-vise/resource" | ||||
| 	slogging "github.com/grassrootseconomics/go-vise/slog" | ||||
| 	"github.com/jackc/pgx/v5/pgxpool" | ||||
| 	"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/lang" | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| 	"git.defalsify.org/vise.git/persist" | ||||
| 	"git.defalsify.org/vise.git/resource" | ||||
| 	gdbmstorage "git.grassecon.net/grassrootseconomics/visedriver/storage/db/gdbm" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla().WithDomain("storage") | ||||
| 	logg = slogging.Get().With("component", "storage") | ||||
| ) | ||||
| 
 | ||||
| type StorageService interface { | ||||
| 	GetPersister(ctx context.Context) (*persist.Persister, error) | ||||
| 	GetUserdataDb(ctx context.Context) db.Db | ||||
| 	GetUserdataDb(ctx context.Context) (db.Db, error) | ||||
| 	GetResource(ctx context.Context) (resource.Resource, error) | ||||
| } | ||||
| 
 | ||||
| type MenuStorageService struct { | ||||
| 	conn ConnData | ||||
| 	resourceDir   string | ||||
| 	poResource    resource.Resource | ||||
| 	resourceStore db.Db | ||||
| 	stateStore    db.Db | ||||
| 	userDataStore db.Db | ||||
| 	conns      Conns | ||||
| 	poResource resource.Resource | ||||
| 	store      map[int8]db.Db | ||||
| } | ||||
| 
 | ||||
| func NewMenuStorageService(conn ConnData, resourceDir string) *MenuStorageService { | ||||
| func NewMenuStorageService(conn Conns) *MenuStorageService { | ||||
| 	return &MenuStorageService{ | ||||
| 		conn: conn, | ||||
| 		resourceDir: resourceDir, | ||||
| 		conns: conn, | ||||
| 		store: make(map[int8]db.Db), | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func (ms *MenuStorageService) WithResourceDir(resourceDir string) *MenuStorageService { | ||||
| 	ms.resourceDir = resourceDir | ||||
| func (ms *MenuStorageService) WithDb(store db.Db, typ int8) *MenuStorageService { | ||||
| 	var err error | ||||
| 	if ms.store[typ] != nil { | ||||
| 		panic(fmt.Errorf("db already set for typ: %d", typ)) | ||||
| 	} | ||||
| 	ms.store[typ] = store | ||||
| 	ms.conns[typ], err = ToConnData(store.Connection()) | ||||
| 	if err != nil { | ||||
| 		panic(err) | ||||
| 	} | ||||
| 	return ms | ||||
| } | ||||
| 
 | ||||
| func (ms *MenuStorageService) getOrCreateDb(ctx context.Context, existingDb db.Db, section string) (db.Db, error) { | ||||
| 	var newDb db.Db | ||||
| func (ms *MenuStorageService) checkDb(ctx context.Context, typ int8) db.Db { | ||||
| 	store := ms.store[typ] | ||||
| 	if store != nil { | ||||
| 		return store | ||||
| 	} | ||||
| 	connData := ms.conns[typ] | ||||
| 	logg.DebugCtxf(ctx, "db check", "conn", connData, "store", DbStoreDebug[typ]) | ||||
| 	v := ms.conns.Have(&connData) | ||||
| 	if v == -1 { | ||||
| 		return nil | ||||
| 	} | ||||
| 	src := ms.store[v] | ||||
| 	if src == nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	ms.store[typ] = ms.store[v] | ||||
| 	logg.DebugCtxf(ctx, "found existing db", "typ", typ, "srctyp", v, "store", ms.store[typ], "srcstore", ms.store[v], "conn", connData) | ||||
| 	return ms.store[typ] | ||||
| } | ||||
| 
 | ||||
| func (ms *MenuStorageService) getOrCreateDb(ctx context.Context, section string, typ int8) (db.Db, error) { | ||||
| 	var err error | ||||
| 
 | ||||
| 	if existingDb != nil { | ||||
| 		return existingDb, nil | ||||
| 	newDb := ms.checkDb(ctx, typ) | ||||
| 	if newDb != nil { | ||||
| 		logg.InfoCtxf(ctx, "using existing db", "typ", typ, "db", newDb) | ||||
| 		return newDb, nil | ||||
| 	} | ||||
| 
 | ||||
| 
 | ||||
| 	connStr := ms.conn.String() | ||||
| 	dbTyp := ms.conn.DbType() | ||||
| 	connData := ms.conns[typ] | ||||
| 	connStr := connData.Raw() | ||||
| 	dbTyp := connData.DbType() | ||||
| 	if dbTyp == DBTYPE_POSTGRES { | ||||
| 		// TODO: move to vise
 | ||||
| 		err = ensureSchemaExists(ctx, ms.conn) | ||||
| 		err = ensureSchemaExists(ctx, connData) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 		newDb = postgres.NewPgDb().WithSchema(ms.conn.Domain()) | ||||
| 	} else if dbTyp == DBTYPE_GDBM { | ||||
| 		err = ms.ensureDbDir() | ||||
| 		newDb = postgres.NewPgDb().WithSchema(connData.Domain()) | ||||
| 
 | ||||
| 	} else if dbTyp == DBTYPE_FS { | ||||
| 		err = ms.ensureDbDir(connStr) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
| 		connStr = path.Join(connStr, section) | ||||
| 		newDb = gdbmstorage.NewThreadGdbmDb() | ||||
| 		fsdbInstance := fsdb.NewFsDb() | ||||
| 		if connData.Mode() == DBMODE_BINARY { | ||||
| 			fsdbInstance = fsdbInstance.WithBinary() | ||||
| 		} | ||||
| 		newDb = fsdbInstance | ||||
| 	} else if dbTyp == DBTYPE_MEM { | ||||
| 		logg.WarnCtxf(ctx, "using volatile storage (memdb)") | ||||
| 		newDb = memdb.NewMemDb() | ||||
| 	} else { | ||||
| 		return nil, fmt.Errorf("unsupported connection string: '%s'\n", ms.conn.String()) | ||||
| 		return nil, fmt.Errorf("unsupported connection string: '%s'\n", connData.Raw()) | ||||
| 	} | ||||
| 	logg.DebugCtxf(ctx, "connecting to db", "conn", connStr, "conndata", ms.conn) | ||||
| 	logg.InfoCtxf(ctx, "connecting to db", "conn", connData, "typ", typ) | ||||
| 	err = newDb.Connect(ctx, connStr) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	ms.store[typ] = newDb | ||||
| 
 | ||||
| 	return newDb, nil | ||||
| } | ||||
| 
 | ||||
| // WithGettext triggers use of gettext for translation of templates and menus.
 | ||||
| //
 | ||||
| // The first language in `lns` will be used as default language, to resolve node keys to 
 | ||||
| // The first language in `lns` will be used as default language, to resolve node keys to
 | ||||
| // language strings.
 | ||||
| //
 | ||||
| // If `lns` is an empty array, gettext will not be used.
 | ||||
| @ -98,7 +132,7 @@ func (ms *MenuStorageService) WithGettext(path string, lns []lang.Language) *Men | ||||
| 	} | ||||
| 	rs := resource.NewPoResource(lns[0], path) | ||||
| 
 | ||||
| 	for _, ln := range(lns) { | ||||
| 	for _, ln := range lns { | ||||
| 		rs = rs.WithLanguage(ln) | ||||
| 	} | ||||
| 
 | ||||
| @ -124,38 +158,49 @@ func ensureSchemaExists(ctx context.Context, conn ConnData) error { | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func applySession(ctx context.Context, store db.Db) error { | ||||
| 	sessionId, ok := ctx.Value("SessionId").(string) | ||||
| 	if !ok { | ||||
| 		logg.DebugCtxf(ctx, "missing session to apply", "store", store) | ||||
| 		return nil | ||||
| 		//return fmt.Errorf("missing session to apply to store: %v", store)
 | ||||
| 	} | ||||
| 	store.SetSession(sessionId) | ||||
| 	return nil | ||||
| } | ||||
| 
 | ||||
| func (ms *MenuStorageService) GetPersister(ctx context.Context) (*persist.Persister, error) { | ||||
| 	stateStore, err := ms.GetStateStore(ctx) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 
 | ||||
| 	err = applySession(ctx, stateStore) | ||||
| 	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") | ||||
| 	userStore, err := ms.getOrCreateDb(ctx, "userdata.gdbm", STORETYPE_USER) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 
 | ||||
| 	ms.userDataStore = userDataStore | ||||
| 	return ms.userDataStore, nil | ||||
| 	err = applySession(ctx, userStore) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return userStore, nil | ||||
| } | ||||
| 
 | ||||
| func (ms *MenuStorageService) GetResource(ctx context.Context) (resource.Resource, error) { | ||||
| 	ms.resourceStore = fsdb.NewFsDb() | ||||
| 	err := ms.resourceStore.Connect(ctx, ms.resourceDir) | ||||
| 	store, err := ms.getOrCreateDb(ctx, "resource.gdbm", STORETYPE_RESOURCE) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	rfs := resource.NewDbResource(ms.resourceStore) | ||||
| 	rfs := resource.NewDbResource(store) | ||||
| 	if ms.poResource != nil { | ||||
| 		logg.InfoCtxf(ctx, "using poresource for menu and template") | ||||
| 		rfs.WithMenuGetter(ms.poResource.GetMenu) | ||||
| @ -165,33 +210,34 @@ func (ms *MenuStorageService) GetResource(ctx context.Context) (resource.Resourc | ||||
| } | ||||
| 
 | ||||
| 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 | ||||
| 	return ms.getOrCreateDb(ctx, "state.gdbm", STORETYPE_STATE) | ||||
| } | ||||
| 
 | ||||
| func (ms *MenuStorageService) ensureDbDir() error { | ||||
| 	err := os.MkdirAll(ms.conn.String(), 0700) | ||||
| func (ms *MenuStorageService) ensureDbDir(path string) error { | ||||
| 	err := os.MkdirAll(path, 0700) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("state dir create exited with error: %v\n", err) | ||||
| 		return fmt.Errorf("store 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) | ||||
| // TODO: how to handle persister here?
 | ||||
| func (ms *MenuStorageService) Close(ctx context.Context) error { | ||||
| 	var errs []error | ||||
| 	var haveErr bool | ||||
| 	for i := range _STORETYPE_MAX { | ||||
| 		err := ms.store[int8(i)].Close(ctx) | ||||
| 		if err != nil { | ||||
| 			haveErr = true | ||||
| 		} | ||||
| 		errs = append(errs, err) | ||||
| 	} | ||||
| 	if haveErr { | ||||
| 		errStr := "" | ||||
| 		for i, err := range errs { | ||||
| 			errStr += fmt.Sprintf("(%d: %v)", i, err) | ||||
| 		} | ||||
| 		return errors.New(errStr) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
							
								
								
									
										114
									
								
								storage/storage_service_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										114
									
								
								storage/storage_service_test.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,114 @@ | ||||
| package storage | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"os" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	fsdb "github.com/grassrootseconomics/go-vise/db/fs" | ||||
| ) | ||||
| 
 | ||||
| func TestMenuStorageServiceOneSet(t *testing.T) { | ||||
| 	d, err := os.MkdirTemp("", "visedriver-menustorageservice") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	defer os.RemoveAll(d) | ||||
| 	conns := NewConns() | ||||
| 	connData, err := ToConnData(d) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	conns.Set(connData, STORETYPE_STATE) | ||||
| 
 | ||||
| 	ctx := context.Background() | ||||
| 	ms := NewMenuStorageService(conns) | ||||
| 	_, err = ms.GetStateStore(ctx) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	_, err = ms.GetResource(ctx) | ||||
| 	if err == nil { | ||||
| 		t.Fatalf("expected error getting resource") | ||||
| 	} | ||||
| 	_, err = ms.GetUserdataDb(ctx) | ||||
| 	if err == nil { | ||||
| 		t.Fatalf("expected error getting userdata") | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
| func TestMenuStorageServiceExplicit(t *testing.T) { | ||||
| 	d, err := os.MkdirTemp("", "visedriver-menustorageservice") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	defer os.RemoveAll(d) | ||||
| 	conns := NewConns() | ||||
| 	connData, err := ToConnData(d) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	conns.Set(connData, STORETYPE_STATE) | ||||
| 
 | ||||
| 	ctx := context.Background() | ||||
| 	d, err = os.MkdirTemp("", "visedriver-menustorageservice") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	defer os.RemoveAll(d) | ||||
| 	store := fsdb.NewFsDb() | ||||
| 	err = store.Connect(ctx, d) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 
 | ||||
| 	ms := NewMenuStorageService(conns) | ||||
| 	ms = ms.WithDb(store, STORETYPE_RESOURCE) | ||||
| 	_, err = ms.GetStateStore(ctx) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	_, err = ms.GetResource(ctx) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	_, err = ms.GetUserdataDb(ctx) | ||||
| 	if err == nil { | ||||
| 		t.Fatalf("expected error getting userdata") | ||||
| 	} | ||||
| 
 | ||||
| } | ||||
| 
 | ||||
| func TestMenuStorageServiceReuse(t *testing.T) { | ||||
| 	d, err := os.MkdirTemp("", "visedriver-menustorageservice") | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	defer os.RemoveAll(d) | ||||
| 	conns := NewConns() | ||||
| 	connData, err := ToConnData(d) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	conns.Set(connData, STORETYPE_STATE) | ||||
| 	conns.Set(connData, STORETYPE_USER) | ||||
| 
 | ||||
| 	ctx := context.Background() | ||||
| 	ctx = context.WithValue(ctx, "SessionId", "foo") | ||||
| 	ms := NewMenuStorageService(conns) | ||||
| 	stateStore, err := ms.GetStateStore(ctx) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	_, err = ms.GetResource(ctx) | ||||
| 	if err == nil { | ||||
| 		t.Fatalf("expected error getting resource") | ||||
| 	} | ||||
| 	userStore, err := ms.GetUserdataDb(ctx) | ||||
| 	if err != nil { | ||||
| 		t.Fatal(err) | ||||
| 	} | ||||
| 	if userStore != stateStore { | ||||
| 		t.Fatalf("expected same store, but they are %p and %p", userStore, stateStore) | ||||
| 	} | ||||
| } | ||||
| @ -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() error | ||||
| 	FinishFunc func(context.Context) 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() error { | ||||
| 	return m.FinishFunc() | ||||
| func (m *MockEngine) Finish(ctx context.Context) error { | ||||
| 	return m.FinishFunc(ctx) | ||||
| } | ||||
|  | ||||
| @ -1,10 +1,12 @@ | ||||
| package httpmocks | ||||
| 
 | ||||
| import ( | ||||
| 	"git.defalsify.org/vise.git/engine" | ||||
| 	"git.defalsify.org/vise.git/persist" | ||||
| 	"git.defalsify.org/vise.git/resource" | ||||
| 	"context" | ||||
| 
 | ||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/request" | ||||
| 	"github.com/grassrootseconomics/go-vise/engine" | ||||
| 	"github.com/grassrootseconomics/go-vise/persist" | ||||
| 	"github.com/grassrootseconomics/go-vise/resource" | ||||
| ) | ||||
| 
 | ||||
| // MockRequestHandler implements request.RequestHandler interface for testing
 | ||||
| @ -13,8 +15,8 @@ type MockRequestHandler struct { | ||||
| 	GetConfigFunc        func() engine.Config | ||||
| 	GetEngineFunc        func(cfg engine.Config, rs resource.Resource, pe *persist.Persister) engine.Engine | ||||
| 	OutputFunc           func(rs request.RequestSession) (request.RequestSession, error) | ||||
| 	ResetFunc            func(rs request.RequestSession) (request.RequestSession, error) | ||||
| 	ShutdownFunc         func() | ||||
| 	ResetFunc            func(ctx context.Context, rs request.RequestSession) (request.RequestSession, error) | ||||
| 	ShutdownFunc         func(ctx context.Context) | ||||
| 	GetRequestParserFunc func() request.RequestParser | ||||
| } | ||||
| 
 | ||||
| @ -34,12 +36,12 @@ func (m *MockRequestHandler) Output(rs request.RequestSession) (request.RequestS | ||||
| 	return m.OutputFunc(rs) | ||||
| } | ||||
| 
 | ||||
| func (m *MockRequestHandler) Reset(rs request.RequestSession) (request.RequestSession, error) { | ||||
| 	return m.ResetFunc(rs) | ||||
| func (m *MockRequestHandler) Reset(ctx context.Context, rs request.RequestSession) (request.RequestSession, error) { | ||||
| 	return m.ResetFunc(ctx, rs) | ||||
| } | ||||
| 
 | ||||
| func (m *MockRequestHandler) Shutdown() { | ||||
| 	m.ShutdownFunc() | ||||
| func (m *MockRequestHandler) Shutdown(ctx context.Context) { | ||||
| 	m.ShutdownFunc(ctx) | ||||
| } | ||||
| 
 | ||||
| func (m *MockRequestHandler) GetRequestParser() request.RequestParser { | ||||
|  | ||||
| @ -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) {} | ||||
|  | ||||
							
								
								
									
										41
									
								
								testutil/mocks/storage.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								testutil/mocks/storage.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,41 @@ | ||||
| package mocks | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 
 | ||||
| 	"github.com/grassrootseconomics/go-vise/db" | ||||
| 	memdb "github.com/grassrootseconomics/go-vise/db/mem" | ||||
| 	"github.com/grassrootseconomics/go-vise/persist" | ||||
| 	"github.com/grassrootseconomics/go-vise/resource" | ||||
| ) | ||||
| 
 | ||||
| type MemStorageService struct { | ||||
| 	Db db.Db | ||||
| 	pe *persist.Persister | ||||
| 	rs resource.Resource | ||||
| } | ||||
| 
 | ||||
| func NewMemStorageService(ctx context.Context) *MemStorageService { | ||||
| 	svc := &MemStorageService{ | ||||
| 		Db: memdb.NewMemDb(), | ||||
| 	} | ||||
| 	err := svc.Db.Connect(ctx, "") | ||||
| 	if err != nil { | ||||
| 		panic(err) | ||||
| 	} | ||||
| 	svc.pe = persist.NewPersister(svc.Db) | ||||
| 	svc.rs = resource.NewMenuResource() | ||||
| 	return svc | ||||
| } | ||||
| 
 | ||||
| func (mss *MemStorageService) GetPersister(ctx context.Context) (*persist.Persister, error) { | ||||
| 	return mss.pe, nil | ||||
| } | ||||
| 
 | ||||
| func (mss *MemStorageService) GetUserdataDb(ctx context.Context) (db.Db, error) { | ||||
| 	return mss.Db, nil | ||||
| } | ||||
| 
 | ||||
| func (mss *MemStorageService) GetResource(ctx context.Context) (resource.Resource, error) { | ||||
| 	return mss.rs, nil | ||||
| } | ||||
| @ -1,51 +0,0 @@ | ||||
| package utils | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 
 | ||||
| 	"git.defalsify.org/vise.git/db" | ||||
| 	fsdb "git.defalsify.org/vise.git/db/fs" | ||||
| 	"git.defalsify.org/vise.git/logging" | ||||
| ) | ||||
| 
 | ||||
| var ( | ||||
| 	logg = logging.NewVanilla().WithDomain("adminstore") | ||||
| ) | ||||
| 
 | ||||
| type AdminStore struct { | ||||
| 	ctx     context.Context | ||||
| 	FsStore db.Db | ||||
| } | ||||
| 
 | ||||
| func NewAdminStore(ctx context.Context, fileName string) (*AdminStore, error) { | ||||
| 	fsStore, err := getFsStore(ctx, fileName) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return &AdminStore{ctx: ctx, FsStore: fsStore}, nil | ||||
| } | ||||
| 
 | ||||
| func getFsStore(ctx context.Context, connectStr string) (db.Db, error) { | ||||
| 	fsStore := fsdb.NewFsDb() | ||||
| 	err := fsStore.Connect(ctx, connectStr) | ||||
| 	fsStore.SetPrefix(db.DATATYPE_USERDATA) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return fsStore, nil | ||||
| } | ||||
| 
 | ||||
| // Checks if the given sessionId is listed as an admin.
 | ||||
| func (as *AdminStore) IsAdmin(sessionId string) (bool, error) { | ||||
| 	_, err := as.FsStore.Get(as.ctx, []byte(sessionId)) | ||||
| 	if err != nil { | ||||
| 		if db.IsNotFound(err) { | ||||
| 			logg.Printf(logging.LVL_INFO, "Returning false because session id was not found") | ||||
| 			return false, nil | ||||
| 		} else { | ||||
| 			return false, err | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
| 	return true, nil | ||||
| } | ||||
							
								
								
									
										56
									
								
								utils/age.go
									
									
									
									
									
								
							
							
						
						
									
										56
									
								
								utils/age.go
									
									
									
									
									
								
							| @ -1,56 +0,0 @@ | ||||
| package utils | ||||
| 
 | ||||
| import ( | ||||
| 	"strconv" | ||||
| 	"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 | ||||
| } | ||||
| 
 | ||||
| 
 | ||||
| //IsValidYob checks if the provided yob can be considered valid 
 | ||||
| func IsValidYOb(yob string) bool { | ||||
| 	currentYear := time.Now().Year() | ||||
| 	yearOfBirth, err := strconv.ParseInt(yob, 10, 64) | ||||
| 	if err != nil { | ||||
| 		return false | ||||
| 	} | ||||
| 	if yearOfBirth >= 1900 && int(yearOfBirth) <= currentYear { | ||||
| 		return true | ||||
| 	} else { | ||||
| 		return false | ||||
| 	} | ||||
| 
 | ||||
| } | ||||
| @ -1,11 +0,0 @@ | ||||
| package utils | ||||
| 
 | ||||
| var isoCodes = map[string]bool{ | ||||
| 	"eng":     true, // English
 | ||||
| 	"swa":     true, // Swahili
 | ||||
| 	"default": true, // Default language: English
 | ||||
| } | ||||
| 
 | ||||
| func IsValidISO639(code string) bool { | ||||
| 	return isoCodes[code] | ||||
| } | ||||
| @ -1,17 +0,0 @@ | ||||
| package utils | ||||
| 
 | ||||
| func ConstructName(firstName, familyName, defaultValue string) string { | ||||
| 	name := defaultValue | ||||
| 	if familyName != defaultValue { | ||||
| 		if firstName != defaultValue { | ||||
| 			name = firstName + " " + familyName | ||||
| 		} else { | ||||
| 			name = familyName | ||||
| 		} | ||||
| 	} else { | ||||
| 		if firstName != defaultValue { | ||||
| 			name = firstName | ||||
| 		} | ||||
| 	} | ||||
| 	return name | ||||
| } | ||||
		Loading…
	
		Reference in New Issue
	
	Block a user