coforma/swift-tech-challenge

import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
import {
  GetCommand,
  GetCommandInput,
  DynamoDBDocumentClient,
  ScanCommand,
  ScanCommandInput,
  PutCommandInput,
  PutCommand,
} from "@aws-sdk/lib-dynamodb";

const awsConfig = {
  region: "us-east-1",
};

const client = DynamoDBDocumentClient.from(new DynamoDBClient(awsConfig));

const dynamoClient = {
  get: async function (params: GetCommandInput) {
    return await client.send(new GetCommand(params));
  },
  put: async (params: PutCommandInput) =>
    await client.send(new PutCommand(params)),
  singleScan: async function (params: ScanCommandInput) {
    return await client.send(new ScanCommand(params));
  },
  scanAll: async function (params: ScanCommandInput) {
    let items: any[] = [];
    let ExclusiveStartKey: any;

    do {
      const command = new ScanCommand({ ...params, ExclusiveStartKey });
      const result = await client.send(command);
      items = items.concat(result.Items ?? []);
      ExclusiveStartKey = result.LastEvaluatedKey;
    } while (ExclusiveStartKey);

    return items;
  },
};

export default dynamoClient;