handle shorthand property assignment

fliskdata · Sep 1, 2024 · 924d2d4 · 924d2d4
1 parent 3c3ae93
commit 924d2d4
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@flisk/analyze-tracking",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Analyzes tracking code in a project and generates data schemas",
   "main": "src/index.js",
   "bin": {

diff --git a/src/analyze/helpers.js b/src/analyze/helpers.js
@@ -143,7 +143,13 @@ function extractTsProperties(checker, node) {
     const key = prop.name ? prop.name.text : prop.key.text || prop.key.value;
     let valueType = 'any';
 
-    if (prop.initializer) {
+    if (ts.isShorthandPropertyAssignment(prop)) {
+      const symbol = checker.getSymbolAtLocation(prop.name);
+      if (symbol) {
+        valueType = getTypeOfNode(checker, symbol.valueDeclaration);
+        properties[key] = { type: valueType };
+      }
+    } else if (prop.initializer) {
       if (ts.isObjectLiteralExpression(prop.initializer)) {
         properties[key] = {
           type: 'object',