parameterize plugin configuration

promplate · Jul 1, 2024 · e1c912f · e1c912f
1 parent 38be6a9
commit e1c912f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/lib/plugins/dir2json.ts b/src/lib/plugins/dir2json.ts
@@ -2,7 +2,7 @@ import { type Dirent, existsSync, promises as fs } from "node:fs";
 import path from "node:path";
 import { type PluginOption, normalizePath } from "vite";
 
-export default (): PluginOption => {
+export default ({ includes }: { includes: string[] }): PluginOption => {
   return {
     name: "dir-as-json",
 
@@ -43,7 +43,7 @@ export default (): PluginOption => {
         const [parentPath, entries] = dirEntries.pop()!;
         const absoluteParentPath = path.join(dirPath, parentPath);
         promises.push(...entries.map(async (entry) => {
-          if (entry.isFile() && !entry.name.endsWith(".pyi") && !entry.name.endsWith(".d.ts")) {
+          if (entry.isFile() && includes.some(ext => entry.name.endsWith(ext))) {
             const filePath = path.join(absoluteParentPath, entry.name);
             this.addWatchFile(filePath);
             const relativePath = path.relative(dirPath, filePath);

diff --git a/vite.config.ts b/vite.config.ts
@@ -4,6 +4,6 @@ import Unocss from "unocss/vite";
 import { defineConfig } from "vite";
 
 export default defineConfig({
-  plugins: [Unocss(), sveltekit(), dir2json()],
+  plugins: [Unocss(), sveltekit(), dir2json({ includes: ["py", "j2"] })],
   assetsInclude: ["src/python/**/*"],
 });