Skip to content

Commit f938c34

Browse files
authored
feat(generator): Add support for generating view models (#550)
1 parent bd487f8 commit f938c34

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/parser/generator.ts

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,13 @@ export function isRenderer(value: unknown) {
176176
const is_object = typeof value === 'object';
177177
if (!is_object) return false;
178178
const keys = Reflect.ownKeys(value as object);
179-
if (keys.length === 1 && keys[0].toString().includes('Renderer')) {
180-
return Parser.sanitizeClassName(keys[0].toString());
179+
180+
if (keys.length === 1) {
181+
const first_key = keys[0].toString();
182+
183+
if (first_key.endsWith('Renderer') || first_key.endsWith('Model')) {
184+
return Parser.sanitizeClassName(first_key);
185+
}
181186
}
182187
return false;
183188
}

0 commit comments

Comments
 (0)