import { DynamoDB } from "aws-sdk"

type DynamoDocType = DynamoDB.DocumentClient.DocumentClientOptions | DynamoDB.Types.ClientConfiguration

/**
 * Promised enabled DocumentClient
 * @class DocumentClient
 */
export class DocumentClient {
  private documentInstance: DynamoDB.DocumentClient

  constructor(ddbInstance: DynamoDocType) {
    this.documentInstance = new DynamoDB.DocumentClient(ddbInstance)
  }

  /**
   * DocumentClient get method
   * @param params documentclient parameters
   * @return Promise<GetItemOutput>
   */
  public get(params: DynamoDB.DocumentClient.GetItemInput): Promise<DynamoDB.DocumentClient.GetItemOutput> {
    return this.getClient().get(params).promise()
  }

  /**
   * DocumentClient put method
   * @param params documentclient parameters
   * @return Promise<PutItemOutput>
   */
  public put(params: DynamoDB.DocumentClient.PutItemInput): Promise<DynamoDB.DocumentClient.PutItemOutput> {
    return this.getClient().put(params).promise()
  }

  /**
   * DocumentClient update method
   * @param params documentclient parameters
   * @return Promise<UpdateItemOutput>
   */
  public update(params: DynamoDB.DocumentClient.UpdateItemInput): Promise<DynamoDB.DocumentClient.UpdateItemOutput> {
    return this.getClient().update(params).promise()
  }

  /**
   * DocumentClient scan method
   * @param params documentclient parameters
   * @return Promise<ScanOutput>
   */
  public scan(params: DynamoDB.DocumentClient.ScanInput): Promise<DynamoDB.DocumentClient.ScanOutput> {
    return this.getClient().scan(params).promise()
  }

  /**
   * DocumentClient query method
   * @param params documentclient parameters
   * @return Promise<QueryOutput>
   */
  public query(params: DynamoDB.DocumentClient.QueryInput): Promise<DynamoDB.DocumentClient.QueryOutput> {
    return this.getClient().query(params).promise()
  }

  /**
   * DocumentClient delete method
   * @param params documentclient parameters
   * @return Promise<DeleteItemOutput>
   */
  public delete(params: DynamoDB.DocumentClient.DeleteItemInput): Promise<DynamoDB.DocumentClient.DeleteItemOutput> {
    return this.getClient().delete(params).promise()
  }

  /**
   * Return DocumentClient instanse
   * @return DocumentClient instance
   */
  public getClient(): DynamoDB.DocumentClient {
    return this.documentInstance
  }
}