Allow returning binary response data

Author: kaklakariada

src/main/java/com/github/kaklakariada/aws/lambda/RequestHandlingService.java

@@ -93,7 +93,8 @@ private ApiGatewayRequest parseRequest(InputStream input) {
 	private ApiGatewayResponse handleRequest(final ApiGatewayRequest request, Context context) {
 		try {
 			final Object result = handler.handleRequest(request, context);
-			return new ApiGatewayResponse(serializeResult(result));
+			final String responseBody = serializeResult(result);
+			return ApiGatewayResponse.ok(responseBody);
 		} catch (final LambdaException e) {
 			LOG.error("Error processing request: " + e.getMessage());
 			return buildErrorResponse(e, context);

src/main/java/com/github/kaklakariada/aws/lambda/model/response/ApiGatewayResponse.java

@@ -17,22 +17,39 @@
  */
 package com.github.kaklakariada.aws.lambda.model.response;
 
+import java.util.Base64;
 import java.util.Collections;
 import java.util.Map;
 
+import com.fasterxml.jackson.annotation.JsonProperty;
+
 public class ApiGatewayResponse {
 	private final int statusCode;
 	private final Map<String, String> headers;
 	private final String body;
+	private final boolean base64Encoded;
 
 	public ApiGatewayResponse(int statusCode, Map<String, String> headers, String body) {
+		this(statusCode, headers, body, false);
+	}
+
+	private ApiGatewayResponse(int statusCode, Map<String, String> headers, String body, boolean base64Encoded) {
 		this.statusCode = statusCode;
 		this.headers = headers;
 		this.body = body;
+		this.base64Encoded = base64Encoded;
+	}
+
+	public static ApiGatewayResponse ok(String body) {
+		return new ApiGatewayResponse(200, Collections.emptyMap(), body);
 	}
 
-	public ApiGatewayResponse(String body) {
-		this(200, Collections.emptyMap(), body);
+	public static ApiGatewayResponse ok(byte[] body) {
+		return new ApiGatewayResponse(200, Collections.emptyMap(), base64Encode(body), true);
+	}
+
+	private static String base64Encode(byte[] body) {
+		return Base64.getEncoder().encodeToString(body);
 	}
 
 	public int getStatusCode() {
@@ -46,4 +63,9 @@ public Map<String, String> getHeaders() {
 	public String getBody() {
 		return body;
 	}
+
+	@JsonProperty("isBase64Encoded")
+	public boolean isBase64Encoded() {
+		return base64Encoded;
+	}
 }

src/test/java/com/github/kaklakariada/aws/lambda/RequestHandlingServiceTest.java

@@ -91,6 +91,7 @@ private ObjectNode response(int status, String body) {
 		response.set("statusCode", jsonFactory.numberNode(status));
 		response.set("headers", headers);
 		response.set("body", jsonFactory.textNode(body));
+		response.set("isBase64Encoded", jsonFactory.booleanNode(false));
 		return response;
 	}
 

src/test/java/com/github/kaklakariada/aws/lambda/model/response/ApiGatewayResponseTest.java

@@ -21,20 +21,34 @@
 import static org.hamcrest.core.IsEqual.equalTo;
 import static org.hamcrest.core.IsNull.nullValue;
 
+import java.util.Base64;
+
 import org.junit.jupiter.api.Test;
 
 class ApiGatewayResponseTest {
 
 	@Test
-	void testConstructorWithBody() {
-		final ApiGatewayResponse response = new ApiGatewayResponse("body");
+	void testConstructorWithStringBody() {
+		final ApiGatewayResponse response = ApiGatewayResponse.ok("body");
 		assertThat(response.getBody(), equalTo("body"));
 		assertThat(response.getStatusCode(), equalTo(200));
 		assertThat(response.getHeaders().size(), equalTo(0));
+		assertThat(response.isBase64Encoded(), equalTo(false));
+	}
+
+	@Test
+	void testNullStringBody() {
+		assertThat(ApiGatewayResponse.ok((String) null).getBody(), nullValue());
 	}
 
 	@Test
-	void testNullBody() {
-		assertThat(new ApiGatewayResponse(null).getBody(), nullValue());
+	void testConstructorWithBinaryBody() {
+		final byte[] body = "body".getBytes();
+		final ApiGatewayResponse response = ApiGatewayResponse.ok(body);
+		assertThat(response.getBody(), equalTo("Ym9keQ=="));
+		assertThat(Base64.getDecoder().decode(response.getBody()), equalTo(body));
+		assertThat(response.getStatusCode(), equalTo(200));
+		assertThat(response.getHeaders().size(), equalTo(0));
+		assertThat(response.isBase64Encoded(), equalTo(true));
 	}
 }