attempt to get literal type

souporserious · Jun 21, 2024 · dde0821 · dde0821
1 parent f8f1cf6
commit dde0821
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 3 deletions.
diff --git a/.changeset/stale-comics-count.md b/.changeset/stale-comics-count.md
@@ -0,0 +1,5 @@
+---
+"@tsxmod/utils": minor
+---
+
+Attempts to resolve literal type value if no resolved value found in `getDefaultValuesFromProperties`.
diff --git a/packages/utils/src/properties/getDefaultValuesFromProperties.test.ts b/packages/utils/src/properties/getDefaultValuesFromProperties.test.ts
@@ -4,7 +4,7 @@ import { getDefaultValuesFromProperties } from './getDefaultValuesFromProperties
 describe('getDefaultValuesFromProperties', () => {
   const project = new Project()
 
-  test('should parse a set of properties', () => {
+  test('parses a set of properties', () => {
     const sourceFile = project.createSourceFile(
       'test.ts',
       `const createCounter = (initialCount = 0, options: { incrementBy: number } = { incrementBy: 1 }) => {}`,
@@ -22,7 +22,7 @@ describe('getDefaultValuesFromProperties', () => {
     })
   })
 
-  test('handles renamed property default values', () => {
+  test('renamed property default values', () => {
     const sourceFile = project.createSourceFile(
       'test.ts',
       `function useCounter({ initialCount: renamedInitialCount = 0 }: { initialCount: number }) {}`,
@@ -39,4 +39,38 @@ describe('getDefaultValuesFromProperties', () => {
       initialCount: 0,
     })
   })
+
+  test('template string default values', () => {
+    const sourceFile = project.createSourceFile(
+      'test.ts',
+      'const a = 1; const b = 2; const createCounter = (initialCount = `${a + b}`) => {}',
+      { overwrite: true }
+    )
+    const functionDeclaration = sourceFile.getFirstDescendantByKindOrThrow(
+      SyntaxKind.ArrowFunction
+    )
+    const defaultValues = getDefaultValuesFromProperties(
+      functionDeclaration.getParameters()
+    )
+    expect(defaultValues).toEqual({
+      initialCount: '3',
+    })
+  })
+
+  test('function default values', () => {
+    const sourceFile = project.createSourceFile(
+      'test.ts',
+      `const createCounter = (initialCount = () => 0) => {}`,
+      { overwrite: true }
+    )
+    const functionDeclaration = sourceFile.getFirstDescendantByKindOrThrow(
+      SyntaxKind.ArrowFunction
+    )
+    const defaultValues = getDefaultValuesFromProperties(
+      functionDeclaration.getParameters()
+    )
+    expect(defaultValues).toEqual({
+      initialCount: '() => 0',
+    })
+  })
 })
diff --git a/packages/utils/src/properties/getDefaultValuesFromProperties.ts b/packages/utils/src/properties/getDefaultValuesFromProperties.ts
@@ -32,7 +32,7 @@ export function getDefaultValuesFromProperties(
 
       defaultValues[name] = isLiteralExpressionValue(resolvedValue)
         ? resolvedValue
-        : initializer.getText()
+        : initializer.getType().getLiteralValue() ?? initializer.getText()
     }
   })