From 5a1bbddfc72b7d80200f1b7e018685c3b8e17ae1 Mon Sep 17 00:00:00 2001
From: absidue <48293849+absidue@users.noreply.github.com>
Date: Sat, 2 Dec 2023 16:48:24 +0100
Subject: [PATCH] feat(generator): Add support for generating view models

---
 src/parser/generator.ts | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/parser/generator.ts b/src/parser/generator.ts
index 931af51b4..23fb13a7f 100644
--- a/src/parser/generator.ts
+++ b/src/parser/generator.ts
@@ -176,8 +176,13 @@ export function isRenderer(value: unknown) {
   const is_object = typeof value === 'object';
   if (!is_object) return false;
   const keys = Reflect.ownKeys(value as object);
-  if (keys.length === 1 && keys[0].toString().includes('Renderer')) {
-    return Parser.sanitizeClassName(keys[0].toString());
+
+  if (keys.length === 1) {
+    const first_key = keys[0].toString();
+
+    if (first_key.endsWith('Renderer') || first_key.endsWith('Model')) {
+      return Parser.sanitizeClassName(first_key);
+    }
   }
   return false;
 }