Fix WebResource write typings

Change-type: patch
balena-io-modules · Mar 12, 2024 · c04b83a · c04b83a
1 parent 90916ad
commit c04b83a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -37,7 +37,11 @@ export interface WebResource {
 	size?: number;
 };
 `,
-	write: '',
+	write: `
+export interface WebResourceWrite extends Blob {
+	name: string;
+};
+`,
 };
 
 const trimNL = new TemplateTag(
@@ -116,7 +120,7 @@ const sqlTypeToTypescriptType = (
 		case 'JSON':
 			return 'object';
 		case 'WebResource':
-			return 'WebResource';
+			return opts.mode === 'read' ? 'WebResource' : 'WebResourceWrite';
 		default:
 			throw new Error(`Unknown data type: '${f.dataType}'`);
 	}

diff --git a/test/index.ts b/test/index.ts
@@ -44,7 +44,13 @@ const test = (
 			${expectation}
 		`);
 		} else {
-			expect(result).to.equal(expectation);
+			expect(result).to.equal(source`
+				export interface WebResourceWrite extends Blob {
+					name: string;
+				};
+
+				${expectation}
+			`);
 		}
 	});
 };
@@ -358,7 +364,7 @@ test(
 			modified_at: Date;
 			id: number;
 			a_date: Date;
-			a_file: WebResource;
+			a_file: WebResourceWrite;
 			parent: number;
 			references__other: number;
 		}