opencode/yao
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #625 from trheyi/main

Browse source 
[add] support for retrieving cookies in sui Request struct
This commit is contained in:
Max 2024-06-25 19:48:40 +08:00 • committed by GitHub
commit b8d6c3a539
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.gitignore vendored
View file

@ -39,4 +39,5 @@ xgen/v1.0/*
!xgen/v1.0/layouts__index.async.js !xgen/v1.0/layouts__index.async.js
!pipe/ui !pipe/ui
*-unit-test *-unit-test
docker/build/test docker/build/test
db

1
sui/api/request.go
View file

@ -213,6 +213,7 @@ func (r *Request) Render() (string, int, error) {
data["$payload"] = r.Request.Payload data["$payload"] = r.Request.Payload
data["$query"] = r.Request.Query data["$query"] = r.Request.Query
data["$param"] = r.Request.Params data["$param"] = r.Request.Params
data["$cookie"] = r.Request.Cookies()
data["$url"] = r.Request.URL data["$url"] = r.Request.URL
printData := false printData := false

14
sui/core/request.go
View file

@ -40,6 +40,20 @@ func NewRequestMock(mock *PageMock) *Request {
} }
} }
// Cookies get the cookies
func (r *Request) Cookies() map[string]string {
cookies := map[string]string{}
cookie := r.Headers.Get("Cookie")
parts := strings.Split(cookie, ";")
for _, part := range parts {
kv := strings.Split(strings.TrimSpace(part), "=")
if len(kv) == 2 {
cookies[kv[0]] = kv[1]
}
}
return cookies
}
// ExecString get the data // ExecString get the data
func (r *Request) ExecString(data string) (Data, error) { func (r *Request) ExecString(data string) (Data, error) {
var res Data var res Data