response.go

package response

import (
	"fmt"
	"net/http"
)

const (
	// Ok - ok
	Ok = "OK"
	// RequestBodyNil - Request body is nil
	RequestBodyNil = "Request body is nil"
	// InvalidRequestMethod - Request Method is invalid
	InvalidRequestMethod = "Invalid HTTP Request Method"
	// TooManyRequests - too many requests
	TooManyRequests = "Max Requests Limit reached"
	// NoWriteKeyInBasicAuth - Failed to read writeKey from header
	NoWriteKeyInBasicAuth = "Failed to read writeKey from header"
	// NoWriteKeyInQueryParams - Failed to read writeKey from Query Params
	NoWriteKeyInQueryParams = "Failed to read writeKey from Query Params"
	// RequestBodyReadFailed - Failed to read body from request
	RequestBodyReadFailed = "Failed to read body from request"
	// RequestBodyTooLarge - Request size exceeds max limit
	RequestBodyTooLarge = "Request size exceeds max limit"
	// InvalidWriteKey - Invalid Write Key
	InvalidWriteKey = "Invalid Write Key"
	// InvalidJSON - Invalid JSON
	InvalidJSON = "Invalid JSON"
	// InvalidWebhookSource - Source does not accept webhook events
	InvalidWebhookSource = "Source does not accept webhook events"
	// SourceTransformerResponseErrorReadFailed - Failed to read error from source transformer response
	SourceTransformerResponseErrorReadFailed = "Failed to read error from source transformer response"
	// SourceDisabled - write key is present, but the source for it is disabled.
	SourceDisabled = "Source is disabled"
	// SourceTransformerFailed - Internal server error in source transformer
	SourceTransformerFailed = "Internal server error in source transformer"
	// SourceTransformerFailedToReadOutput - Output not found in source transformer response
	SourceTransformerFailedToReadOutput = "Output not found in source transformer response"
	// SourceTransformerInvalidResponseFormat - Invalid format of source transformer response
	SourceTransformerInvalidResponseFormat = "Invalid format of source transformer response"
	// SourceTransformerInvalidOutputFormatInResponse - Invalid output format in source transformer response
	SourceTransformerInvalidOutputFormatInResponse = "Invalid output format in source transformer response"
	// SourceTransformerInvalidOutputJSON - Invalid output json in source transformer response
	SourceTransformerInvalidOutputJSON = "Invalid output json in source transformer response"
	// NonIdentifiableRequest - Request neither has anonymousId nor userId
	NonIdentifiableRequest = "Request neither has anonymousId nor userId"
	// ErrorInMarshal - Error while marshalling
	ErrorInMarshal = "Error while marshalling"
	// ErrorInParseForm - Error during parsing form
	ErrorInParseForm = "Error during parsing form"
	// ErrorInParseMultiform - Error during parsing multiform
	ErrorInParseMultiform = "Error during parsing multiform"
	// NotKassetteEvent = Event is not a Valid Kassette Event
	// NotKassetteEvent = Event is not a Valid Kassette Event
	NotKassetteEvent = "Event is not a valid kassette event"
	// ContextDeadlineExceeded - context deadline exceeded
	ContextDeadlineExceeded = "context deadline exceeded"
	// GatewayTimeout - Gateway timeout
	GatewayTimeout = "Gateway timeout"

	transPixelResponse = "\x47\x49\x46\x38\x39\x61\x01\x00\x01\x00\x80\x00\x00\x00\x00\x00\x00\x00\x00\x21\xF9\x04" +
		"\x01\x00\x00\x00\x00\x2C\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02\x44\x01\x00\x3B"
)

var statusMap = map[string]status{
	Ok:                      {message: Ok, code: http.StatusOK},
	RequestBodyNil:          {message: RequestBodyNil, code: http.StatusBadRequest},
	InvalidRequestMethod:    {message: InvalidRequestMethod, code: http.StatusBadRequest},
	TooManyRequests:         {message: TooManyRequests, code: http.StatusTooManyRequests},
	NoWriteKeyInBasicAuth:   {message: NoWriteKeyInBasicAuth, code: http.StatusUnauthorized},
	NoWriteKeyInQueryParams: {message: NoWriteKeyInQueryParams, code: http.StatusUnauthorized},
	RequestBodyReadFailed:   {message: RequestBodyReadFailed, code: http.StatusInternalServerError},
	RequestBodyTooLarge:     {message: RequestBodyTooLarge, code: http.StatusRequestEntityTooLarge},
	InvalidWriteKey:         {message: InvalidWriteKey, code: http.StatusUnauthorized},
	SourceDisabled:          {message: SourceDisabled, code: http.StatusNotFound},
	InvalidJSON:             {message: InvalidJSON, code: http.StatusBadRequest},
	// webhook specific status
	InvalidWebhookSource:                           {message: InvalidWebhookSource, code: http.StatusNotFound},
	SourceTransformerFailed:                        {message: SourceTransformerFailed, code: http.StatusBadRequest},
	SourceTransformerResponseErrorReadFailed:       {message: SourceTransformerResponseErrorReadFailed, code: http.StatusInternalServerError},
	SourceTransformerFailedToReadOutput:            {message: SourceTransformerFailedToReadOutput, code: http.StatusInternalServerError},
	SourceTransformerInvalidResponseFormat:         {message: SourceTransformerInvalidResponseFormat, code: http.StatusInternalServerError},
	SourceTransformerInvalidOutputFormatInResponse: {message: SourceTransformerInvalidOutputFormatInResponse, code: http.StatusInternalServerError},
	SourceTransformerInvalidOutputJSON:             {message: SourceTransformerInvalidOutputJSON, code: http.StatusInternalServerError},
	NonIdentifiableRequest:                         {message: NonIdentifiableRequest, code: http.StatusBadRequest},
	ErrorInMarshal:                                 {message: ErrorInMarshal, code: http.StatusBadRequest},
	ErrorInParseForm:                               {message: ErrorInParseForm, code: http.StatusBadRequest},
	ErrorInParseMultiform:                          {message: ErrorInParseMultiform, code: http.StatusBadRequest},
	NotKassetteEvent:                               {message: NotKassetteEvent, code: http.StatusBadRequest},
	ContextDeadlineExceeded:                        {message: GatewayTimeout, code: http.StatusGatewayTimeout},
}

// status holds the gateway response status message and code
type status struct {
	message string
	code    int
}

func GetStatus(key string) string {
	if status, ok := statusMap[key]; ok {
		return status.message
	}
	return key
}

func GetPixelResponse() string {
	return transPixelResponse
}

func GetErrorStatusCode(key string) int {
	if status, ok := statusMap[key]; ok {
		return status.code
	}
	return http.StatusInternalServerError
}

func MakeResponse(msg string) string {
	return fmt.Sprintf(`{"msg": %q}`, msg)
}