fix delte and put item typing when no returnValues set

woltsu · Jun 6, 2024 · efe9e98 · efe9e98
1 parent fd6a3d4
commit efe9e98
Show file tree

Hide file tree

Showing 6 changed files with 77 additions and 42 deletions.
diff --git a/src/queryBuilders/deleteItemQueryBuilder.integration.test.ts b/src/queryBuilders/deleteItemQueryBuilder.integration.test.ts
@@ -86,4 +86,40 @@ describe("DeleteItemQueryBuilder", () => {
 
     expect(res).toBeDefined();
   });
+  it("doesn't return values if no returnValues is specified", async () => {
+    await tsynamoClient
+      .putItem("myTable")
+      .item({
+        userId: "1",
+        dataTimestamp: 2,
+      })
+      .execute();
+
+    const res = await tsynamoClient
+      .deleteItem("myTable")
+      .keys({ userId: "1", dataTimestamp: 2 })
+      .execute();
+
+    expectTypeOf(res).toBeNever();
+
+    expect(res).toBeUndefined();
+  });
+  it("does return values if returnValues is specified", async () => {
+    await tsynamoClient
+      .putItem("myTable")
+      .item({
+        userId: "1",
+        dataTimestamp: 2,
+      })
+      .execute();
+
+    const res = await tsynamoClient
+      .deleteItem("myTable")
+      .keys({ userId: "1", dataTimestamp: 2 })
+      .returnValues("ALL_OLD")
+      .execute();
+
+    expectTypeOf(res).not.toBeNever();
+    expect(res).toBeDefined();
+  });
 });
diff --git a/src/queryBuilders/deleteItemQueryBuilder.ts b/src/queryBuilders/deleteItemQueryBuilder.ts
@@ -24,7 +24,7 @@ import {
 export interface DeleteItemQueryBuilderInterface<
   DDB,
   Table extends keyof DDB,
-  O extends DDB[Table]
+  O
 > {
   // conditionExpression
   conditionExpression<Key extends ObjectKeyPaths<DDB[Table]>>(
@@ -86,28 +86,25 @@ export interface DeleteItemQueryBuilderInterface<
 
   returnValues(
     option: Extract<ReturnValuesOptions, "NONE" | "ALL_OLD">
-  ): DeleteItemQueryBuilder<DDB, Table, O>;
+  ): DeleteItemQueryBuilder<DDB, Table, ExecuteOutput<DDB[Table]>>;
 
   returnValuesOnConditionCheckFailure(
     option: Extract<ReturnValuesOptions, "NONE" | "ALL_OLD">
-  ): DeleteItemQueryBuilder<DDB, Table, O>;
+  ): DeleteItemQueryBuilder<DDB, Table, ExecuteOutput<DDB[Table]>>;
 
   keys<Keys extends PickPk<DDB[Table]> & PickSkRequired<DDB[Table]>>(
     pk: Keys
   ): DeleteItemQueryBuilder<DDB, Table, O>;
 
   compile(): DeleteCommand;
-  execute(): Promise<ExecuteOutput<O>[] | undefined>;
+  execute(): Promise<O[] | undefined>;
 }
 
 /**
  * @todo support ReturnValuesOnConditionCheckFailure
  */
-export class DeleteItemQueryBuilder<
-  DDB,
-  Table extends keyof DDB,
-  O extends DDB[Table]
-> implements DeleteItemQueryBuilderInterface<DDB, Table, O>
+export class DeleteItemQueryBuilder<DDB, Table extends keyof DDB, O>
+  implements DeleteItemQueryBuilderInterface<DDB, Table, O>
 {
   readonly #props: DeleteItemQueryBuilderProps;
 
@@ -153,8 +150,8 @@ export class DeleteItemQueryBuilder<
 
   returnValues(
     option: Extract<ReturnValuesOptions, "NONE" | "ALL_OLD">
-  ): DeleteItemQueryBuilder<DDB, Table, O> {
-    return new DeleteItemQueryBuilder<DDB, Table, O>({
+  ): DeleteItemQueryBuilder<DDB, Table, ExecuteOutput<DDB[Table]>> {
+    return new DeleteItemQueryBuilder({
       ...this.#props,
       node: {
         ...this.#props.node,
@@ -168,8 +165,8 @@ export class DeleteItemQueryBuilder<
 
   returnValuesOnConditionCheckFailure(
     option: Extract<ReturnValuesOptions, "NONE" | "ALL_OLD">
-  ): DeleteItemQueryBuilder<DDB, Table, O> {
-    return new DeleteItemQueryBuilder<DDB, Table, O>({
+  ): DeleteItemQueryBuilder<DDB, Table, ExecuteOutput<DDB[Table]>> {
+    return new DeleteItemQueryBuilder({
       ...this.#props,
       node: {
         ...this.#props.node,
@@ -200,7 +197,7 @@ export class DeleteItemQueryBuilder<
     return this.#props.queryCompiler.compile(this.#props.node);
   };
 
-  execute = async (): Promise<ExecuteOutput<O>[] | undefined> => {
+  execute = async (): Promise<unknown extends O ? never : O[] | undefined> => {
     const deleteCommand = this.compile();
     const data = await this.#props.ddbClient.send(deleteCommand);
     return data.Attributes as any;

diff --git a/src/queryBuilders/expressionBuilder.ts b/src/queryBuilders/expressionBuilder.ts
@@ -210,11 +210,8 @@ export type ExprArgs<
   | AttributeBetweenExprArg<DDB, Table, Key>
   | BuilderExprArg<DDB, Table, O, AllowKeysInExpression>
   | NotExprArg<DDB, Table, O, AllowKeysInExpression>;
-export class ExpressionBuilder<
-  DDB,
-  Table extends keyof DDB,
-  O extends DDB[Table]
-> implements ExpressionBuilderInterface<DDB, Table, O>
+export class ExpressionBuilder<DDB, Table extends keyof DDB, O>
+  implements ExpressionBuilderInterface<DDB, Table, O>
 {
   readonly #props: ExpressionBuilderProps;
 

diff --git a/src/queryBuilders/putItemQueryBuilder.integration.test.ts b/src/queryBuilders/putItemQueryBuilder.integration.test.ts
@@ -44,13 +44,25 @@ describe("PutItemQueryBuilder", () => {
     expect(result).toEqual(itemToPut);
   });
 
+  it("doesnt return values without returnValues", async () => {
+    let result = await tsynamoClient
+      .putItem("myTable")
+      .item(itemToPut)
+      .execute();
+
+    expectTypeOf(result).toBeNever();
+    expect(result).toBeUndefined();
+  });
+
   it("handles ReturnValues option", async () => {
     let result = await tsynamoClient
       .putItem("myTable")
       .item(itemToPut)
       .returnValues("ALL_OLD")
       .execute();
 
+    expectTypeOf(result).not.toBeNever();
+
     expect(result).toBeUndefined();
 
     result = await tsynamoClient

diff --git a/src/queryBuilders/putItemQueryBuilder.ts b/src/queryBuilders/putItemQueryBuilder.ts
@@ -16,11 +16,7 @@ import {
   NotExprArg,
 } from "./expressionBuilder";
 
-export interface PutItemQueryBuilderInterface<
-  DDB,
-  Table extends keyof DDB,
-  O extends DDB[Table]
-> {
+export interface PutItemQueryBuilderInterface<DDB, Table extends keyof DDB, O> {
   // conditionExpression
   conditionExpression<Key extends ObjectKeyPaths<DDB[Table]>>(
     ...args: ComparatorExprArg<DDB, Table, Key>
@@ -81,21 +77,18 @@ export interface PutItemQueryBuilderInterface<
 
   returnValues(
     option: Extract<ReturnValuesOptions, "NONE" | "ALL_OLD">
-  ): PutItemQueryBuilder<DDB, Table, O>;
+  ): PutItemQueryBuilder<DDB, Table, ExecuteOutput<DDB[Table]>>;
 
-  item<Item extends ExecuteOutput<O>>(
+  item<Item extends ExecuteOutput<DDB[Table]>>(
     item: Item
   ): PutItemQueryBuilder<DDB, Table, O>;
 
   compile(): PutCommand;
-  execute(): Promise<ExecuteOutput<O>[] | undefined>;
+  execute(): Promise<O[] | undefined>;
 }
 
-export class PutItemQueryBuilder<
-  DDB,
-  Table extends keyof DDB,
-  O extends DDB[Table]
-> implements PutItemQueryBuilderInterface<DDB, Table, O>
+export class PutItemQueryBuilder<DDB, Table extends keyof DDB, O>
+  implements PutItemQueryBuilderInterface<DDB, Table, O>
 {
   readonly #props: PutItemQueryBuilderProps;
 
@@ -139,7 +132,7 @@ export class PutItemQueryBuilder<
     });
   }
 
-  item<Item extends ExecuteOutput<O>>(
+  item<Item extends ExecuteOutput<DDB[Table]>>(
     item: Item
   ): PutItemQueryBuilder<DDB, Table, O> {
     return new PutItemQueryBuilder<DDB, Table, O>({
@@ -156,8 +149,8 @@ export class PutItemQueryBuilder<
 
   returnValues(
     option: Extract<ReturnValuesOptions, "NONE" | "ALL_OLD">
-  ): PutItemQueryBuilder<DDB, Table, O> {
-    return new PutItemQueryBuilder<DDB, Table, O>({
+  ): PutItemQueryBuilder<DDB, Table, ExecuteOutput<DDB[Table]>> {
+    return new PutItemQueryBuilder({
       ...this.#props,
       node: {
         ...this.#props.node,
@@ -173,7 +166,7 @@ export class PutItemQueryBuilder<
     return this.#props.queryCompiler.compile(this.#props.node);
   };
 
-  execute = async (): Promise<ExecuteOutput<O>[] | undefined> => {
+  execute = async (): Promise<unknown extends O ? never : O[] | undefined> => {
     const putCommand = this.compile();
     const data = await this.#props.ddbClient.send(putCommand);
     return data.Attributes as any;

diff --git a/src/queryCreator.ts b/src/queryCreator.ts
@@ -71,10 +71,10 @@ export class QueryCreator<DDB> {
    *
    * @see https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/dynamodb/command/PutItemCommand/
    */
-  putItem<Table extends keyof DDB & string>(
+  putItem<Table extends keyof DDB & string, O>(
     table: Table
-  ): PutItemQueryBuilder<DDB, Table, DDB[Table]> {
-    return new PutItemQueryBuilder<DDB, Table, DDB[Table]>({
+  ): PutItemQueryBuilder<DDB, Table, O> {
+    return new PutItemQueryBuilder({
       node: {
         kind: "PutNode",
         table: {
@@ -97,10 +97,10 @@ export class QueryCreator<DDB> {
    *
    * @see https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/dynamodb/command/DeleteItemCommand/
    */
-  deleteItem<Table extends keyof DDB & string>(
+  deleteItem<Table extends keyof DDB & string, O>(
     table: Table
-  ): DeleteItemQueryBuilder<DDB, Table, DDB[Table]> {
-    return new DeleteItemQueryBuilder<DDB, Table, DDB[Table]>({
+  ): DeleteItemQueryBuilder<DDB, Table, O> {
+    return new DeleteItemQueryBuilder({
       node: {
         kind: "DeleteNode",
         table: {