add headers

lbernardo · Jun 17, 2020 · ffd6be9 · ffd6be9
1 parent d50470a
commit ffd6be9
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 4 deletions.
diff --git a/bin/darwin/awslocal b/bin/darwin/awslocal
diff --git a/bin/linux/awslocal b/bin/linux/awslocal
diff --git a/internal/adapters/secondary/docker/docker.go b/internal/adapters/secondary/docker/docker.go
@@ -64,6 +64,7 @@ func ExecuteDockerLambda(content core.ExecuteLambdaRequest) (core.ResultLambdaRe
 	bodyStr = strings.ReplaceAll(bodyStr, "\n", "")
 	contentRequest.Body = bodyStr
 	contentRequest.PathParameters = content.Parameters
+	contentRequest.Headers = content.Headers
 
 	jsonRequest, _ := json.Marshal(contentRequest)
 

diff --git a/pkg/apigateway/apigateway.go b/pkg/apigateway/apigateway.go
@@ -2,12 +2,13 @@ package apigateway
 
 import (
 	"fmt"
-	"github.com/gorilla/mux"
-	"github.com/lbernardo/aws-local/internal/adapters/secondary/docker"
-	"github.com/lbernardo/aws-local/pkg/core"
 	"log"
 	"net/http"
 	"strings"
+
+	"github.com/gorilla/mux"
+	"github.com/lbernardo/aws-local/internal/adapters/secondary/docker"
+	"github.com/lbernardo/aws-local/pkg/core"
 )
 
 func StartApiGateway(params ParamsApiGateway) {
@@ -25,11 +26,18 @@ func StartApiGateway(params ParamsApiGateway) {
 
 		fff := func(w http.ResponseWriter, r *http.Request) {
 			parameters := mux.Vars(r)
+			headers := map[string]string{}
+
+			for key, _ := range r.Header {
+				headers[key] = r.Header.Get(key)
+			}
+
 			result, off := docker.ExecuteDockerLambda(core.ExecuteLambdaRequest{
 				Volume:      params.Volume,
 				Net:         params.Network,
 				Handler:     function,
 				Runtime:     params.Serverless.Provider.Runtime,
+				Headers:     headers,
 				Environment: params.Serverless.Provider.Environment,
 				Body:        r.Body,
 				Parameters:  parameters,

diff --git a/pkg/core/model.go b/pkg/core/model.go
@@ -9,6 +9,7 @@ type ExecuteLambdaRequest struct {
 	Runtime     string
 	Environment map[string]string
 	Body        io.ReadCloser
+	Headers     map[string]string
 	Parameters  map[string]string
 }
 
@@ -22,9 +23,9 @@ type ResultLambdaRequest struct {
 type ContentRequest struct {
 	Body           string            `json:"body"`
 	PathParameters map[string]string `json:"pathparameters"`
+	Headers        map[string]string `json:"headers"`
 }
 
-
 type Serverless struct {
 	Functions map[string]Functions `json:"functions"`
 	Provider  Provider             `json:"provider"`