Merge pull request #74 from webdoc-labs/feature/type-inference

Feature: Add type interference for strings, numbers, and booleans.

Author: ShukantPal

common/config/rush/pnpm-lock.yaml

@@ -12,6 +12,7 @@ import {
   type VariableDeclarator,
   isArrowFunctionExpression,
   isAssignmentExpression,
+  isBooleanLiteral,
   isCallExpression,
   isClassDeclaration,
   isClassExpression,
@@ -23,6 +24,7 @@ import {
   isInterfaceDeclaration,
   isLiteral,
   isMemberExpression,
+  isNumericLiteral,
   isObjectExpression,
   isObjectMethod,
   isObjectProperty,
@@ -50,6 +52,7 @@ import {
   extractReturns,
   extractType,
 } from "./extract-metadata";
+import {createSimpleKeywordType} from "@webdoc/model";
 
 // + Extract the symbol name, type from the Node
 // + Set the appopriate flags
@@ -113,8 +116,18 @@ export default function extractSymbol(
       if (isStringLiteral(node.value)) {
         // Quotes for strings
         nodeSymbol.meta.defaultValue = `"${node.value.value}"`;
+
+        if (!nodeSymbol.meta.dataType) {
+          nodeSymbol.meta.dataType = createSimpleKeywordType("string");
+        }
       } else {
         nodeSymbol.meta.defaultValue = node.value.value;
+
+        if (!nodeSymbol.meta.dataType && isNumericLiteral(node.value)) {
+          nodeSymbol.meta.dataType = createSimpleKeywordType("number");
+        } else if (!nodeSymbol.meta.dataType && isBooleanLiteral(node.value)) {
+          nodeSymbol.meta.dataType = createSimpleKeywordType("boolean");
+        }
       }
     }
   } else if (isClassDeclaration(node) || isClassExpression(node)) {

packages/webdoc-parser/src/symbols-babel/extract-symbol.js

@@ -27,10 +27,13 @@ describe("@webdoc/parser.LanguageIntegration{@lang js}", function() {
 
     expect(symClassName.members.length).to.equal(6);
 
+    const symInitProperty = findSymbol("ClassName.initProperty", symtree);
+
     expect(findSymbol("ClassName.constructor", symtree)).to.not.equal(undefined);
     expect(findSymbol("ClassName.classProperty", symtree)).to.not.equal(undefined);
     expect(findSymbol("ClassName.classMethod", symtree)).to.not.equal(undefined);
-    expect(findSymbol("ClassName.initProperty", symtree)).to.not.equal(undefined);
+    expect(symInitProperty).to.not.equal(undefined);
+    expect(symInitProperty.meta.defaultValue).to.equal(9);
   });
 
   it("should parse assigned members properly", function() {

packages/webdoc-parser/test/lang-js.js

@@ -107,4 +107,20 @@ describe("@webdoc/parser.LanguageIntegration{@lang ts}", function() {
 
     expect(symbolObjectName.meta.object).to.equal("this.objectName");
   });
+
+  it("should infer simple types", function() {
+    const symtree = buildSymbolTree(`
+      class ClassName {
+        bool = true;
+        num = 1;
+        str = "string";
+      }
+    `);
+
+    const [bool, num, str] = symtree.members[0].members;
+
+    expect(bool.meta.dataType[0]).to.equal("boolean");
+    expect(num.meta.dataType[0]).to.equal("number");
+    expect(str.meta.dataType[0]).to.equal("string");
+  });
 });