fix number return types

narthur · Apr 12, 2024 · a427c7c · a427c7c
1 parent e021b3b
commit a427c7c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/codegen/makeGetter.spec.ts b/src/codegen/makeGetter.spec.ts
@@ -69,6 +69,7 @@ describe("makeGetter", () => {
       },
       `<({ id: 1, value: "the_option_name", color: "red" } | { id: 2, value: "the_option_name_2", color: "blue" })>`,
     ],
+    [{ type: "number" }, `: number {`],
   ])("%s => `%s`", (field, expected) => {
     expect(run(field)).toContain(expected);
   });

diff --git a/src/codegen/makeGetter.ts b/src/codegen/makeGetter.ts
@@ -38,6 +38,10 @@ function getReturnType(field: FieldDefinition, tables: Table[]): string {
       .join(" | ");
   }
 
+  if (field.type === "number") {
+    return "number";
+  }
+
   return makeFieldType(field);
 }