package com.kfn.study;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.HashMap;
import java.util.Map;

import software.amazon.awssdk.http.urlconnection.UrlConnectionHttpClient;
import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.dynamodb.DynamoDbClient;
import software.amazon.awssdk.services.dynamodb.model.AttributeValue;
import software.amazon.awssdk.services.dynamodb.model.PutItemRequest;

public class DynamoDBClient {
	private static final String LOCAL_DYNAMODB_ENDPOINT = "http://host.docker.internal:8000";

	public static DynamoDbClient createClient() throws URISyntaxException {

		// Set up the DynamoDB client to point to the local DynamoDB instance
		DynamoDbClient dynamoDbClient = DynamoDbClient.builder().region(Region.US_WEST_2) // Region is ignored in local
																							// dynamo
				.endpointOverride(new URI(LOCAL_DYNAMODB_ENDPOINT)) // Point to local DynamoDB
				.httpClient(UrlConnectionHttpClient.create()) // Use UrlConnectionHttpClient
				.build();
		System.out.println("established connection");
		return dynamoDbClient;

	}

	public void insertData(String name) {
		try {
			Map<String, AttributeValue> item = new HashMap<>();
			// primary key
			item.put("DBPK", AttributeValue.builder().s("ENTITY#" + name).build());
			// sort key
			item.put("DBSK", AttributeValue.builder().s("ENTITY#" + name).build());
			// attribute
			item.put("name", AttributeValue.builder().s("Kris Nunes").build());

			PutItemRequest request = PutItemRequest.builder().tableName("HelloDynamoDB").item(item).build();

			createClient().putItem(request);

		} catch (Exception e) {
			// TODO Auto-generated catch block

		}

	}

}

package com.kfn.study;

import java.util.Map;

import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.RequestHandler;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.ObjectMapper;

public class LambdaHandler implements RequestHandler<Map<String, Object>, String> {

	private final ObjectMapper objectMapper = new ObjectMapper(); // Jackson for JSON parsing

	@Override
	public String handleRequest(Map<String, Object> event, Context context) {
		// Get HTTP method and path
		String httpMethod = (String) event.get("httpMethod");
		String path = (String) event.get("path");
		System.out.println(httpMethod);
		System.out.println(path);

		// Dispatch based on path and method
		switch (path) {
		case "/Entity":
			return hanEntityRequest(httpMethod, event);
		case "/People":
			return "Unsupported path: " + path;
		default:
			return "Unsupported path: " + path;
		}
	}

	// Handler for /Customer
	private String hanEntityRequest(String method, Map<String, Object> event) {
		switch (method) {
		case "GET":

			return "Fetching customer details";
		case "POST":
			// Extract and process the request body for POST
			String body = (String) event.get("body");
			// Assuming the payload is a JSON object with fields

			try {
				Map<String, String> bodyMap = objectMapper.readValue(body, Map.class);
				String name = bodyMap.get("name");
				DynamoDBClient client = new DynamoDBClient();
				client.insertData(name);
			} catch (JsonMappingException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (JsonProcessingException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}

		default:
			return "Unsupported HTTP method for /Customer";
		}
	}
}