x

core/node_headers.go

@@ -1,7 +1,9 @@
 package core
 
 import (
+	"fmt"
 	"net/http"
+	"regexp"
 
 	"github.com/dop251/goja"
 )
@@ -12,6 +14,24 @@ func MakeHeadersObject(vm *goja.Runtime, header http.Header) *goja.Object {
 		return header.Get(name)
 	})
 
+	obj.Set("gets", func(name string) []string {
+		return header.Values(name)
+	})
+
+	obj.Set("values", func(name string) []string {
+		return header.Values(name)
+	})
+
+	obj.Set("cookie", func(name string) string {
+		for _, v := range header.Values("Set-Cookie") {
+			res := regexp.MustCompile(fmt.Sprintf("%s=([^;]+)", name)).FindStringSubmatch(v)
+			if len(res) > 0 {
+				return res[1]
+			}
+		}
+		return ""
+	})
+
 	obj.Set("has", func(name string) bool {
 		return header.Get(name) != ""
 	})