betterment to selection type of tuple.

mindler-olli · Mar 17, 2024 · 8791c75 · 8791c75
1 parent 3480f46
commit 8791c75
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 15 deletions.
diff --git a/src/queryBuilders/getItemQueryBuilder.integration.test.ts b/src/queryBuilders/getItemQueryBuilder.integration.test.ts
@@ -98,11 +98,25 @@ describe("GetItemQueryBuilder", () => {
         stringTimestamp: "123",
       })
       .consistentRead(true)
-      .attributes(["tuplez[0]"])
+      .attributes(["tuplez[1]"])
       .execute();
 
     expect(Object.keys(data!).length).toBe(1);
-    expect(data?.tuplez?.[0]).toEqual(TEST_DATA[6].tuplez[0]);
+    expect(data?.tuplez?.[0]).toEqual(TEST_DATA[6].tuplez[1]);
+  });
+  it("handles selecting multiple attributes from tuples", async () => {
+    const data = await tsynamoClient
+      .getItemFrom("myOtherTable")
+      .keys({
+        userId: TEST_DATA[6].userId,
+        stringTimestamp: "123",
+      })
+      .consistentRead(true)
+      .attributes(["tuplez[0]", "tuplez[1]"])
+      .execute();
+
+    expect(Object.keys(data!).length).toBe(1);
+    expect(data?.tuplez).toEqual(TEST_DATA[6].tuplez);
   });
 
   it("can't await instance directly", async () => {

diff --git a/src/typeHelpers.test-d.ts b/src/typeHelpers.test-d.ts
@@ -116,18 +116,41 @@ describe("typeHelpers typecheck", () => {
     type selectTuple1 = SelectAttributes<Table, ["key", "tuple[0]"]>;
     expectTypeOf<selectTuple1>().toEqualTypeOf<{
       key: number;
-      tuple: {
-        0: {
+      tuple: [
+        {
           kissa: "koira";
-        };
-      };
+        }
+      ];
     }>();
 
     type selectTuple2 = SelectAttributes<Table, ["key", "tuple[1]"]>;
-    // TODO Fix this =(
-    // expectTypeOf<selectTuple2>().toEqualTypeOf<{
-    //   key: number;
-    //   tuple: { 0: number };
-    // }>();
+    expectTypeOf<selectTuple2>().toEqualTypeOf<{
+      key: number;
+      tuple: [number];
+    }>();
+
+    type selectTuple3 = SelectAttributes<
+      Table,
+      ["key", "tuple[0]", "tuple[1]"]
+    >;
+    // TODO Fix this to be [{kissa: "koira"}, number]
+    expectTypeOf<selectTuple3>().toEqualTypeOf<{
+      key: number;
+      tuple: [
+        {
+          kissa: "koira";
+        }
+      ] &
+        [number];
+    }>();
+
+    type selectArray = SelectAttributes<Table, ["key", "cats[0]"]>;
+    expectTypeOf<selectArray>().toEqualTypeOf<{
+      key: number;
+      cats: {
+        name: string;
+        age: number;
+      }[];
+    }>();
   });
 });
diff --git a/src/typeHelpers.ts b/src/typeHelpers.ts
@@ -75,11 +75,14 @@ type RecursiveSelectAttributes<Table, Properties> = Properties extends [
   infer First,
   ...infer Rest
 ]
-  ? First extends keyof Table
-    ? { [key in First]: RecursiveSelectAttributes<Table[First], Rest> }
-    : // When "First" in path is a number, and Table is an Array, Recurse to the inner type of the array
-    [First, Table] extends [`${number}`, any[]]
+  ? [First, IsTuple<Table>] extends [`${number}`, true]
+    ? First extends keyof Table
+      ? [RecursiveSelectAttributes<Table[First], Rest>]
+      : Table
+    : [First, Table] extends [`${number}`, any[]]
     ? RecursiveSelectAttributes<As<Table, any[]>[number], Rest>[]
+    : First extends keyof Table
+    ? { [key in First]: RecursiveSelectAttributes<Table[First], Rest> }
     : never
   : Table;
 

diff --git a/test/testFixture.ts b/test/testFixture.ts
@@ -191,6 +191,10 @@ export const TEST_DATA = [
         name: "Sylvester J. Pussycat Sr.",
         age: 78,
       },
+      {
+        name: "Tommy the Cat",
+        age: 12,
+      },
     ],
   },
   {