import {
  BatchGetCommand,
  BatchWriteCommand,
  DynamoDBDocumentClient,
} from "../aws-sdk-js-v3/lib/lib-dynamodb";
import { DynamoDBClient } from "../aws-sdk-js-v3/clients/client-dynamodb";

(async () => {
  const region = "us-west-2";
  const client = new DynamoDBClient({ region });

  const TableName = `test-${Math.ceil(Math.random() * 10 ** 10)}`;

  const params = {
    TableName,
    AttributeDefinitions: [{ AttributeName: "id", AttributeType: "S" }],
    KeySchema: [{ AttributeName: "id", KeyType: "HASH" }],
    BillingMode: "PAY_PER_REQUEST",
  };

  await client.send(new CreateTableCommand({ ...params }));
  await waitForTableExists({ client }, { TableName });

  const translateConfig = {
    marshallOptions: { removeUndefinedValues: true },
  };
  const ddbDocClient = DynamoDBDocumentClient.from(
    client,
    translateConfig
  );

  console.log(
    await ddbDocClient.send(
      new BatchWriteCommand({
        RequestItems: {
          [TableName]: [
            {
              PutRequest: {
                Item: {
                  id: "1",
                  content: "content #1 from DynamoDBDocumentClient",
                  undefinedValue: undefined,
                },
              },
            },
            {
              PutRequest: {
                Item: {
                  id: "2",
                  content: "content #2 from DynamoDBDocumentClient",
                  undefinedValue: undefined,
                },
              },
            },
          ],
        },
      })
    )
  );

  console.log(
    await ddbDocClient.send(
      new BatchGetCommand({
        RequestItems: {
          [TableName]: {
            Keys: [{ id: "1" }, { id: "2" }],
          },
        },
      })
    )
  );

  console.log(
    await ddbDocClient.send(
      new BatchWriteCommand({
        RequestItems: {
          [TableName]: [
            { DeleteRequest: { Key: { id: "1" } } },
            { DeleteRequest: { Key: { id: "2" } } },
          ],
        },
      })
    )
  );

  await client.send(new DeleteTableCommand({ TableName }));
})();