Function header() for setting HTTP header key pair.

nthnn · Nov 7, 2023 · 3d3a94d · 3d3a94d
1 parent 95d9fdc
commit 3d3a94d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/ankovm/installer.go b/ankovm/installer.go
@@ -19,6 +19,7 @@ func installWebFunc(vmEnv *env.Env, path string, buff *bytes.Buffer, w http.Resp
 	installDefinition(vmEnv, "echo", std.EchoFn(buff))
 	installDefinition(vmEnv, "include", std.IncludeFn(vmEnv, path))
 
+	installDefinition(vmEnv, "header", std.HeaderFn(w))
 	installDefinition(vmEnv, "httpHeaders", std.HttpHeaderFn(r))
 	installDefinition(vmEnv, "httpRemote", std.HttpRemoteFn(r))
 

diff --git a/ankovm/std/webfunc.go b/ankovm/std/webfunc.go
@@ -54,6 +54,12 @@ func IncludeFn(vmEnv *env.Env, path string) func(fileName string) {
 	}
 }
 
+func HeaderFn(w http.ResponseWriter) func(name, value string) {
+	return func(name, value string) {
+		w.Header().Add(name, value)
+	}
+}
+
 func HttpHeaderFn(r *http.Request) func() map[string][]string {
 	return func() map[string][]string {
 		return r.Header