feat: track currentFilePaths in rollup plugin and do hot updates in vite

Author: joelgallant

app-config-rollup/src/index.ts

@@ -4,39 +4,46 @@ import { ConfigLoadingOptions, loadValidatedConfig } from '@app-config/config';
 import { asEnvOptions, currentEnvironment } from '@app-config/node';
 import type { SchemaLoadingOptions } from '@app-config/schema';
 
-interface Options {
+export interface Options {
   readGlobal?: boolean;
   injectValidationFunction?: boolean;
   loadingOptions?: ConfigLoadingOptions;
   schemaLoadingOptions?: SchemaLoadingOptions;
 }
 
 // vite resolves first before passing to the rollup plugin
-const manualImport = /(app-config|app-config-main)\/dist(\/es)?\/index\.js/;
+export const appConfigImportRegex = /(app-config|app-config-main)\/dist(\/es)?\/index\.js/;
 
 export default function appConfigRollup({
   readGlobal,
   injectValidationFunction,
   loadingOptions,
   schemaLoadingOptions,
-}: Options = {}): Plugin {
+}: Options = {}): Plugin & { currentFilePaths?: string[] } {
+  const currentFilePaths: string[] = [];
+
   return {
     name: '@app-config/rollup',
+    currentFilePaths,
     resolveId(source) {
-      if (packageNameRegex.exec(source) || manualImport.exec(source)) {
+      if (packageNameRegex.exec(source) || appConfigImportRegex.exec(source)) {
         return '.config-placeholder';
       }
 
       return null;
     },
     async load(id) {
-      if (packageNameRegex.exec(id) || manualImport.exec(id)) {
-        const { parsed: config, validationFunctionCode } = await loadValidatedConfig(
-          loadingOptions,
-          schemaLoadingOptions,
-        );
-
-        // TODO: alternative for addDependecy with filePaths
+      if (packageNameRegex.exec(id) || appConfigImportRegex.exec(id)) {
+        const {
+          parsed: config,
+          validationFunctionCode,
+          filePaths,
+        } = await loadValidatedConfig(loadingOptions, schemaLoadingOptions);
+
+        if (filePaths) {
+          currentFilePaths.length = 0;
+          currentFilePaths.push(...filePaths);
+        }
 
         let generatedText: string;
 

app-config-vite/src/index.ts

@@ -1,2 +1,24 @@
+import rollupPlugin, { Options } from '@app-config/rollup';
+import type { Plugin } from 'vite';
+
 export * from '@app-config/rollup';
-export { default } from '@app-config/rollup';
+
+export default function appConfigVite(options: Options = {}): Plugin {
+  const plugin = rollupPlugin(options);
+
+  return {
+    ...plugin,
+    name: '@app-config/vite',
+
+    async handleHotUpdate({ server, file }) {
+      if (plugin.currentFilePaths?.includes(file)) {
+        server.ws.send({
+          type: 'full-reload',
+          path: '@app-config/main',
+        });
+
+        return [];
+      }
+    },
+  };
+}