From f5308d746f3089ef6818b0139fe249827a47628b Mon Sep 17 00:00:00 2001
From: Evan You <yyx990803@gmail.com>
Date: Wed, 15 Dec 2021 15:24:15 +0800
Subject: [PATCH] feat: support array of patterns in data loaders

---
 src/node/staticDataPlugin.ts | 26 ++++++++++++++++++--------
 1 file changed, 18 insertions(+), 8 deletions(-)

diff --git a/src/node/staticDataPlugin.ts b/src/node/staticDataPlugin.ts
index 29cb522a..7ff9dc91 100644
--- a/src/node/staticDataPlugin.ts
+++ b/src/node/staticDataPlugin.ts
@@ -9,12 +9,17 @@ const loaderMatch = /\.data\.(j|t)s$/
 let server: ViteDevServer
 
 interface LoaderModule {
+  watch: string[] | string | undefined
+  load: () => any
+}
+
+interface CachedLoaderModule {
   base: string
-  pattern: string | undefined
+  pattern: string[] | undefined
   loader: () => any
 }
 
-const idToLoaderModulesMap: Record<string, LoaderModule | undefined> =
+const idToLoaderModulesMap: Record<string, CachedLoaderModule | undefined> =
   Object.create(null)
 
 // During build, the load hook will be called on the same file twice
@@ -50,7 +55,7 @@ export const staticDataPlugin: Plugin = {
       }
 
       const base = dirname(id)
-      let pattern: string | undefined
+      let pattern: string[] | undefined
       let loader: () => any
 
       const existing = idToLoaderModulesMap[id]
@@ -60,10 +65,15 @@ export const staticDataPlugin: Plugin = {
         // use vite's load config util as a away to load Node.js file with
         // TS & native ESM support
         const loaderModule = (await loadConfigFromFile({} as any, id))
-          ?.config as any
-        pattern = loaderModule.watch
-        if (pattern && pattern.startsWith('./')) {
-          pattern = pattern.slice(2)
+          ?.config as LoaderModule
+        pattern =
+          typeof loaderModule.watch === 'string'
+            ? [loaderModule.watch]
+            : loaderModule.watch
+        if (pattern) {
+          pattern = pattern.map((p) => {
+            return p.startsWith('./') ? p.slice(2) : p
+          })
         }
         loader = loaderModule.load
       }
@@ -91,7 +101,7 @@ export const staticDataPlugin: Plugin = {
       const { base, pattern } = idToLoaderModulesMap[id]!
       ;(server as any)._globImporters[id] = {
         module: server.moduleGraph.getModuleById(id),
-        importGlobs: [{ base, pattern }]
+        importGlobs: pattern?.map((pattern) => ({ base, pattern }))
       }
     }
     return null