feat!: Now use in resolvers options instead of imports option.

s3xysteak · May 3, 2024 · 0dace9b · 0dace9b
1 parent 7660d28
commit 0dace9b
Show file tree

Hide file tree

Showing 5 changed files with 134 additions and 32 deletions.
diff --git a/src/core/generator.ts b/src/core/generator.ts
@@ -93,11 +93,23 @@ ${content.substring(lastComment).trim()}
 const __UnExportList = ${JSON.stringify(exportList)} as const
 
 /**
- * @returns Use in \`imports\` option of unplugin-auto-import.
+ * @returns - Call in \`resolvers\` option of \`unplugin-auto-import\`.
  */
-export function ${rename}(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>): Record<string, (string | [string, string])[]> {
-  return {
-    '${pkgName}': __UnExportList.map(v => map && map[v] ? [v, map[v]] as [string, string] : v),
+export function ${rename}(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>) {
+  return (name: string) => {
+    if (!__UnExportList.includes(name as any))
+      return
+
+    return map && (map as any)[name]
+      ? {
+          name,
+          as: (map as any)[name],
+          from: '${pkgName}',
+        }
+      : {
+          name,
+          from: '${pkgName}',
+        }
   }
 }
 `
@@ -108,12 +120,26 @@ const __UnExportList = /** @type {const} */ (${JSON.stringify(exportList)})
 
 /**
  * @param {Partial<{ [K in typeof __UnExportList[number]]: string }>} [map]
- * @returns {Record<string, (string | [string, string])[]>} Use in \`imports\` option of \`unplugin-auto-import\`.
- */ 
+ * @returns - Call in \`resolvers\` option of \`unplugin-auto-import\`.
+ */
 export function ${rename}(map) {
-  return {
-    '${pkgName}': __UnExportList.map(v => map && map[v] ? [v, map[v]] : v),
+  /** @param {string} name */
+  const func = (name) => {
+    if (!__UnExportList.includes(name))
+      return
+
+    return map && map[name]
+      ? {
+          name,
+          as: map[name],
+          from: '${pkgName}',
+        }
+      : {
+          name,
+          from: '${pkgName}',
+        }
   }
+  return func
 }
 `
   const JS = getTemplate(rawJs)

diff --git a/test/generator.test.ts b/test/generator.test.ts
@@ -26,11 +26,23 @@ export * from '@s3xysteak/utils'
 const __UnExportList = ${JSON.stringify(exportList)} as const
 
 /**
- * @returns Use in \`imports\` option of unplugin-auto-import.
+ * @returns - Call in \`resolvers\` option of \`unplugin-auto-import\`.
  */
-export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>): Record<string, (string | [string, string])[]> {
-  return {
-    'unplugin-export-collector': __UnExportList.map(v => map && map[v] ? [v, map[v]] as [string, string] : v),
+export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>) {
+  return (name: string) => {
+    if (!__UnExportList.includes(name as any))
+      return
+
+    return map && (map as any)[name]
+      ? {
+          name,
+          as: (map as any)[name],
+          from: 'unplugin-export-collector',
+        }
+      : {
+          name,
+          from: 'unplugin-export-collector',
+        }
   }
 }
 
@@ -65,12 +77,26 @@ const __UnExportList = /** @type {const} */ (${JSON.stringify(exportList)})
 
 /**
  * @param {Partial<{ [K in typeof __UnExportList[number]]: string }>} [map]
- * @returns {Record<string, (string | [string, string])[]>} Use in \`imports\` option of \`unplugin-auto-import\`.
- */ 
+ * @returns - Call in \`resolvers\` option of \`unplugin-auto-import\`.
+ */
 export function autoImport(map) {
-  return {
-    'unplugin-export-collector': __UnExportList.map(v => map && map[v] ? [v, map[v]] : v),
+  /** @param {string} name */
+  const func = (name) => {
+    if (!__UnExportList.includes(name))
+      return
+
+    return map && map[name]
+      ? {
+          name,
+          as: map[name],
+          from: 'unplugin-export-collector',
+        }
+      : {
+          name,
+          from: 'unplugin-export-collector',
+        }
   }
+  return func
 }
 
 // --- Auto-Generated By Unplugin-Export-Collector ---
@@ -91,11 +117,23 @@ export function autoImport(map) {
 const __UnExportList = ${JSON.stringify(exportList)} as const
 
 /**
- * @returns Use in \`imports\` option of unplugin-auto-import.
+ * @returns - Call in \`resolvers\` option of \`unplugin-auto-import\`.
  */
-export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>): Record<string, (string | [string, string])[]> {
-  return {
-    'unplugin-export-collector': __UnExportList.map(v => map && map[v] ? [v, map[v]] as [string, string] : v),
+export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>) {
+  return (name: string) => {
+    if (!__UnExportList.includes(name as any))
+      return
+
+    return map && (map as any)[name]
+      ? {
+          name,
+          as: (map as any)[name],
+          from: 'unplugin-export-collector',
+        }
+      : {
+          name,
+          from: 'unplugin-export-collector',
+        }
   }
 }
 

diff --git a/test/parser-lab/generatorTest.js b/test/parser-lab/generatorTest.js
@@ -15,12 +15,26 @@ const __UnExportList = /** @type {const} */ (["ClassIndex","custom","fRe","func1
 
 /**
  * @param {Partial<{ [K in typeof __UnExportList[number]]: string }>} [map]
- * @returns {Record<string, (string | [string, string])[]>} Use in `imports` option of `unplugin-auto-import`.
- */ 
+ * @returns - Call in `resolvers` option of `unplugin-auto-import`.
+ */
 export function autoImport(map) {
-  return {
-    'unplugin-export-collector': __UnExportList.map(v => map && map[v] ? [v, map[v]] : v),
+  /** @param {string} name */
+  const func = (name) => {
+    if (!__UnExportList.includes(name))
+      return
+
+    return map && map[name]
+      ? {
+          name,
+          as: map[name],
+          from: 'unplugin-export-collector',
+        }
+      : {
+          name,
+          from: 'unplugin-export-collector',
+        }
   }
+  return func
 }
 
 // --- Auto-Generated By Unplugin-Export-Collector ---
diff --git a/test/parser-lab/generatorTest.ts b/test/parser-lab/generatorTest.ts
@@ -16,11 +16,23 @@ export * from '@s3xysteak/utils'
 const __UnExportList = ["ClassIndex","custom","fRe","func1","func2","func3","funcIndex","getThree","two"] as const
 
 /**
- * @returns Use in `imports` option of unplugin-auto-import.
+ * @returns - Call in `resolvers` option of `unplugin-auto-import`.
  */
-export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>): Record<string, (string | [string, string])[]> {
-  return {
-    'unplugin-export-collector': __UnExportList.map(v => map && map[v] ? [v, map[v]] as [string, string] : v),
+export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>) {
+  return (name: string) => {
+    if (!__UnExportList.includes(name as any))
+      return
+
+    return map && (map as any)[name]
+      ? {
+          name,
+          as: (map as any)[name],
+          from: 'unplugin-export-collector',
+        }
+      : {
+          name,
+          from: 'unplugin-export-collector',
+        }
   }
 }
 

diff --git a/test/parser-lab/generatorTestWithWriteTo.ts b/test/parser-lab/generatorTestWithWriteTo.ts
@@ -3,11 +3,23 @@
 const __UnExportList = ["ClassIndex","custom","fRe","func1","func2","func3","funcIndex","getThree","two"] as const
 
 /**
- * @returns Use in `imports` option of unplugin-auto-import.
+ * @returns - Call in `resolvers` option of `unplugin-auto-import`.
  */
-export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>): Record<string, (string | [string, string])[]> {
-  return {
-    'unplugin-export-collector': __UnExportList.map(v => map && map[v] ? [v, map[v]] as [string, string] : v),
+export function autoImport(map?: Partial<{ [K in typeof __UnExportList[number]]: string }>) {
+  return (name: string) => {
+    if (!__UnExportList.includes(name as any))
+      return
+
+    return map && (map as any)[name]
+      ? {
+          name,
+          as: (map as any)[name],
+          from: 'unplugin-export-collector',
+        }
+      : {
+          name,
+          from: 'unplugin-export-collector',
+        }
   }
 }