accepts array of ids when setting link_row fields

narthur · Apr 9, 2024 · 50865f4 · 50865f4
1 parent be265a0
commit 50865f4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/codegen/makeClassMethods.spec.ts b/src/codegen/makeClassMethods.spec.ts
@@ -126,4 +126,15 @@ describe("makeClassMethods", () => {
       ]),
     ).not.toContain("setTheFieldName");
   });
+
+  it("accepts id array for link_row setters", () => {
+    expect(
+      makeClassMethods([
+        f({
+          name: "the_field_name",
+          type: "link_row",
+        }),
+      ]),
+    ).toContain("value: number[]");
+  });
 });
diff --git a/src/codegen/makeClassMethods.ts b/src/codegen/makeClassMethods.ts
@@ -30,6 +30,12 @@ function makeGetter(field: FieldDefinition): string {
 }
 
 function makeSetter(field: FieldDefinition): string {
+  if (field.type === "link_row") {
+    return `public ${toCamelCase(`set ${field.name}`)}(value: number[]): Promise<void> {
+        return this.setField("${field.name}", value);
+    }`;
+  }
+
   return `public ${toCamelCase(`set ${field.name}`)}(value: ${makeFieldType(
     field,
   )}): Promise<void> {