user optimistic wallets writing to io json (#586)

acashmoney · web-flow · commit 24a2f370da2f · 2023-08-11T11:43:52.000-07:00
diff --git a/internal/ipwl/generate_io.go b/internal/ipwl/generate_io.go
@@ -1,11 +1,14 @@
 package ipwl
 
 import (
+	"fmt"
 	"io/ioutil"
 	"log"
+	"os"
 	"path/filepath"
 
 	"github.com/labdao/plex/internal/ipfs"
+	"github.com/labdao/plex/internal/web3"
 )
 
 func findMatchingFiles(inputDir string, tool Tool, layers int) (map[string][]string, error) {
@@ -88,12 +91,22 @@ func generateInputCombinations(inputFilepaths map[string][]string) []map[string]
 func createIOEntries(toolInfo ToolInfo, tool Tool, inputCombinations []map[string]string) []IO {
 	var ioData []IO
 
+	var userID string
+
+	if web3.IsValidEthereumAddress(os.Getenv("RECIPIENT_WALLET")) {
+		userID = os.Getenv("RECIPIENT_WALLET")
+	} else {
+		fmt.Println("RECIPIENT_WALLET is not a valid Ethereum address. Using empty string as user ID.")
+		userID = ""
+	}
+
 	for _, combination := range inputCombinations {
 		ioEntry := IO{
 			Tool:    toolInfo,
 			State:   "created",
 			Inputs:  map[string]FileInput{},
 			Outputs: map[string]Output{},
+			UserID:  userID,
 		}
 
 		for inputName, path := range combination {
diff --git a/internal/ipwl/io.go b/internal/ipwl/io.go
@@ -47,6 +47,7 @@ type IO struct {
 	Outputs       map[string]Output    `json:"outputs"`
 	State         string               `json:"state"`
 	ErrMsg        string               `json:"errMsg"`
+	UserID        string               `json:"userId"`
 	BacalhauJobId string               `json:"bacalhauJobId"`
 }
 
diff --git a/internal/web3/wallet.go b/internal/web3/wallet.go
@@ -0,0 +1,11 @@
+package web3
+
+import (
+	"regexp"
+)
+
+func IsValidEthereumAddress(address string) bool {
+	pattern := `^0x[0-9a-fA-F]{40}$`
+	match, _ := regexp.MatchString(pattern, address)
+	return match
+}

Original file line number	Diff line number	Diff line change
`@@ -47,6 +47,7 @@ type IO struct {`
`47`	`47`	Outputs map[string]Output `json:"outputs"`
`48`	`48`	State string `json:"state"`
`49`	`49`	ErrMsg string `json:"errMsg"`
	`50`	+ UserID string `json:"userId"`
`50`	`51`	BacalhauJobId string `json:"bacalhauJobId"`
`51`	`52`	`}`
`52`	`53`