Add async option which requires the client to poll for status and retrieve result upon completion

Author: sullrich

go.mod

@@ -13,6 +13,7 @@ require (
 	github.com/go-playground/universal-translator v0.18.1 // indirect
 	github.com/go-playground/validator/v10 v10.14.0 // indirect
 	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/google/uuid v1.3.0 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
 	github.com/klauspost/cpuid/v2 v2.2.4 // indirect
 	github.com/leodido/go-urn v1.2.4 // indirect

go.sum

@@ -25,6 +25,8 @@ github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MG
 github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
 github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
+github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
 github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
 github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=

main.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/gin-gonic/contrib/static"
 	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
 	"github.com/yuin/gopher-lua"
 	"golang.org/x/time/rate"
 
@@ -14,6 +15,14 @@ import (
 	"sync"
 )
 
+type ScriptStatus struct {
+	Finished bool
+	Result   string
+	Error    error
+}
+
+var scriptStatuses = make(map[string]*ScriptStatus)
+
 // Cache for Lua scripts
 var scriptCache = make(map[string]string)
 var scriptCacheMutex = &sync.RWMutex{}
@@ -126,6 +135,47 @@ func main() {
 		c.Next()
 	})
 
+	router.POST("/runLuaFileAsync/:filename", func(c *gin.Context) {
+		// Get filename from the URL
+		filename := c.Param("filename")
+
+		// Check if filename is safe
+		if filepath.Base(filename) != filename {
+			c.String(http.StatusBadRequest, "Invalid filename")
+			return
+		}
+
+		// Parse JSON from request body
+		var jsonData map[string]interface{}
+		err := c.ShouldBindJSON(&jsonData)
+		if err != nil {
+			c.String(http.StatusBadRequest, "Invalid JSON")
+			return
+		}
+
+		// Generate a unique ID for this script execution
+		id := uuid.New().String()
+
+		// Create a new ScriptStatus
+		scriptStatuses[id] = &ScriptStatus{
+			Finished: false,
+		}
+
+		// Start a goroutine to run the script
+		go func() {
+			result, err := runLuaScript(filename, jsonData)
+			// Update the status when the script is done
+			scriptStatuses[id].Finished = true
+			scriptStatuses[id].Result = result
+			if err != nil {
+				scriptStatuses[id].Error = err
+			}
+		}()
+
+		// Return the ID to the client
+		c.String(http.StatusOK, id)
+	})
+
 	router.POST("/runLuaFile/:filename", func(c *gin.Context) {
 		// Get filename from the URL
 		filename := c.Param("filename")