response.go

package golamb

import (
	"encoding/json"
	"net/http"

	"github.com/aws/aws-lambda-go/events"
)

// Responder responds to the request.
type Responder interface {
	// Respond responds to the http request.
	Respond() (*events.APIGatewayV2HTTPResponse, error)

	// SetHeader sets a response header with the given key and value.
	SetHeader(key string, value string) Responder

	// SetCookie sets a response cookie with the given key and value.
	SetCookie(cookie *http.Cookie) Responder
}

type response struct {
	status  int
	body    any
	headers map[string]string
	cookies []string
}

func (r *response) Respond() (*events.APIGatewayV2HTTPResponse, error) {
	var body string
	if r.body != nil {
		b, err := json.Marshal(r.body)
		if err != nil {
			return nil, err
		}
		body = string(b)
	}
	return &events.APIGatewayV2HTTPResponse{
		StatusCode: r.status,
		Body:       body,
		Headers:    r.headers,
		Cookies:    r.cookies,
	}, nil
}

func (r *response) SetHeader(key string, value string) Responder {
	r.headers[key] = value
	return r
}

func (r *response) SetCookie(cookie *http.Cookie) Responder {
	r.cookies = append(r.cookies, cookie.String())
	return r
}