Merge pull request #753 from trheyi/main

Fix PublicRoot bug and get Sid from request
YaoApp · Sep 25, 2024 · 6c6aaad · 6c6aaad
2 parents 0ea4009 + 896e52a
commit 6c6aaad
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/sui/api/render.go b/sui/api/render.go
@@ -194,7 +194,7 @@ func TemplateRender(process *process.Process) interface{} {
 		}
 	}
 
-	r := core.Request{Theme: opt["theme"], Locale: opt["locale"]}
+	r := core.Request{Theme: opt["theme"], Locale: opt["locale"], Sid: process.Sid}
 	if process.NumOfArgs() > 5 {
 
 		raw, err := jsoniter.Marshal(process.Args[5])

diff --git a/sui/api/request.go b/sui/api/request.go
@@ -53,10 +53,18 @@ func NewRequestContext(c *gin.Context) (*Request, int, error) {
 
 	path := strings.TrimSuffix(c.Request.URL.Path, ".sui")
 
+	sid := ""
+	if v, has := c.Get("__sid"); has {
+		if s, ok := v.(string); ok {
+			sid = s
+		}
+	}
+
 	return &Request{
 		File:    file,
 		context: c,
 		Request: &core.Request{
+			Sid:     sid,
 			Method:  c.Request.Method,
 			Query:   c.Request.URL.Query(),
 			Body:    body,

diff --git a/sui/core/sui.go b/sui/core/sui.go
@@ -96,7 +96,9 @@ func (sui *DSL) PublicRoot(data map[string]interface{}) (string, error) {
 
 	// Merge the data
 	for k, v := range sessionData {
-		data[k] = v
+		if _, ok := data[k]; !ok {
+			data[k] = v
+		}
 	}
 
 	vars := map[string]interface{}{"$session": data}