feat: create resolver for json files (#237)

* feat: create resolver for json files

* chore: clear code

examples/ts/src/data.json

@@ -0,0 +1,9 @@
+{
+  "count": 1,
+  "name": "foo",
+  "obj": {
+    "a": false,
+    "b": ""
+  },
+  "arr": [1, 2, 4]
+}

examples/ts/src/index.ts

@@ -1,3 +1,5 @@
+import data from './data.json'
+
 import type { TestBase } from '@/test'
 
 export interface Test extends TestBase {
@@ -12,4 +14,6 @@ export { manualDts } from './manual-dts'
 
 export { ParametersTest, test, method } from './test'
 
+export { data }
+
 export type { User } from './types'

examples/ts/tsconfig.json

@@ -23,5 +23,5 @@
     },
     "lib": ["esnext", "dom"]
   },
-  "include": ["src", "*.d.ts"]
+  "include": ["src", "*.d.ts", "src/**/*.json"]
 }

src/plugin.ts

@@ -11,7 +11,7 @@ import { createProgram } from 'vue-tsc'
 import debug from 'debug'
 import { cyan, green, yellow } from 'kolorist'
 import { rollupDeclarationFiles } from './rollup'
-import { SvelteResolver, VueResolver, parseResolvers } from './resolvers'
+import { JsonResolver, SvelteResolver, VueResolver, parseResolvers } from './resolvers'
 import {
   normalizeGlob,
   removePureImport,
@@ -103,7 +103,12 @@ export function dtsPlugin(options: PluginOptions = {}): import('vite').Plugin {
   let bundled = false
   let timeRecord = 0
 
-  const resolvers = parseResolvers([VueResolver(), SvelteResolver(), ...(options.resolvers || [])])
+  const resolvers = parseResolvers([
+    JsonResolver(),
+    VueResolver(),
+    SvelteResolver(),
+    ...(options.resolvers || [])
+  ])
 
   const rootFiles = new Set<string>()
   const outputFiles = new Map<string, string>()

src/resolvers/index.ts

@@ -1,5 +1,6 @@
 import type { Resolver } from '../types'
 
+export * from './json'
 export * from './svelte'
 export * from './vue'
 

src/resolvers/json.ts

@@ -0,0 +1,24 @@
+import type { Resolver } from '../types'
+
+const jsonRE = /\.json$/
+
+export function JsonResolver(): Resolver {
+  return {
+    name: 'json',
+    supports(id) {
+      return jsonRE.test(id)
+    },
+    transform({ id, program }) {
+      const sourceFile = program.getSourceFile(id)
+
+      if (!sourceFile) return []
+
+      return [
+        {
+          path: `${id}.d.ts`,
+          content: `declare const _default: ${sourceFile.text};\n\nexport default _default;\n`
+        }
+      ]
+    }
+  }
+}

src/resolvers/svelte.ts

@@ -12,7 +12,7 @@ export function SvelteResolver(): Resolver {
       return [
         {
           path: `${id}.d.ts`,
-          content: "export { SvelteComponentTyped as default } from 'svelte';"
+          content: "export { SvelteComponentTyped as default } from 'svelte';\n"
         }
       ]
     }

src/resolvers/vue.ts

@@ -11,15 +11,12 @@ export function VueResolver(): Resolver {
       return vueRE.test(id)
     },
     transform({ id, root, program, service }) {
-      let sourceFile = program.getSourceFile(id)
-
-      if (!sourceFile && vueRE.test(id)) {
-        sourceFile =
-          program.getSourceFile(id + '.ts') ||
-          program.getSourceFile(id + '.js') ||
-          program.getSourceFile(id + '.tsx') ||
-          program.getSourceFile(id + '.jsx')
-      }
+      const sourceFile =
+        program.getSourceFile(id) ||
+        program.getSourceFile(id + '.ts') ||
+        program.getSourceFile(id + '.js') ||
+        program.getSourceFile(id + '.tsx') ||
+        program.getSourceFile(id + '.jsx')
 
       if (!sourceFile) return []