From 1474e50f7253745ea632c2ef6e562609fe0d423c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joaqu=C3=ADn=20S=C3=A1nchez?= <userquin@gmail.com>
Date: Thu, 21 Mar 2024 20:15:45 +0100
Subject: [PATCH] feat(injectManifest): add sw build env options and change
 vite configuration (#690)

---
 src/options.ts    | 10 ++++++++++
 src/types.ts      | 34 +++++++++++++++++++++++++++++++++-
 src/vite-build.ts | 11 ++++++++++-
 3 files changed, 53 insertions(+), 2 deletions(-)

diff --git a/src/options.ts b/src/options.ts
index 23b61617..d50faea6 100644
--- a/src/options.ts
+++ b/src/options.ts
@@ -125,6 +125,7 @@ export async function resolveOptions(ctx: PWAPluginContext): Promise<ResolvedVit
     sourcemap = viteConfig.build.sourcemap,
     enableWorkboxModulesLogs,
     buildPlugins,
+    envOptions = {},
     ...userInjectManifest
   } = options.injectManifest || {}
   const injectManifest = Object.assign({}, defaultInjectManifest, userInjectManifest)
@@ -174,6 +175,11 @@ export async function resolveOptions(ctx: PWAPluginContext): Promise<ResolvedVit
     }
   }
 
+  const {
+    envDir = viteConfig.envDir,
+    envPrefix = viteConfig.envPrefix,
+  } = envOptions
+
   const resolvedVitePWAOptions: ResolvedVitePWAOptions = {
     base: basePath,
     mode,
@@ -213,6 +219,10 @@ export async function resolveOptions(ctx: PWAPluginContext): Promise<ResolvedVit
       sourcemap,
       enableWorkboxModulesLogs,
     },
+    injectManifestEnvOptions: {
+      envDir,
+      envPrefix,
+    },
     pwaAssets: resolvePWAAssetsOptions(pwaAssets),
   }
 
diff --git a/src/types.ts b/src/types.ts
index 52e7375d..a3ffbb74 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -1,4 +1,4 @@
-import type { BuildOptions, Plugin, ResolvedConfig, UserConfig } from 'vite'
+import type { BuildOptions, InlineConfig, Plugin, ResolvedConfig, UserConfig } from 'vite'
 import type { GenerateSWOptions, InjectManifestOptions, ManifestEntry } from 'workbox-build'
 import type { OutputBundle, RollupOptions } from 'rollup'
 import type { BuiltInPreset, Preset } from '@vite-pwa/assets-generator/config'
@@ -85,6 +85,26 @@ export type CustomInjectManifestOptions = InjectManifestOptions & {
    * Since `v0.15.0` you can add custom Rollup options to build your service worker: we expose the same configuration to build a worker using Vite.
    */
   rollupOptions?: Omit<RollupOptions, 'plugins' | 'output'>
+
+  /**
+   * Environment options.
+   *
+   * @since v0.19.6
+   */
+  envOptions?: {
+    /**
+     * Configure Vite `envDir` option.
+     *
+     * @default Vite `envDir`.
+     */
+    envDir?: UserConfig['envDir']
+    /**
+     * Configure Vite `envPrefix` option.
+     *
+     * @default Vite `envPrefix`.
+     */
+    envPrefix?: UserConfig['envPrefix']
+  }
 }
 
 export interface PWAIntegration {
@@ -94,6 +114,14 @@ export interface PWAIntegration {
     viteOptions: ResolvedConfig,
     options: Partial<VitePWAOptions>,
   ) => void | Promise<void>
+  /**
+   * Allow integrations to configure Vite options for custom service worker build.
+   *
+   * @param options Vite options for custom service worker build.
+   * @see src/vite-build.ts module
+   * @since v0.19.6
+   */
+  configureCustomSWViteBuild?: (options: InlineConfig) => void | Promise<void>
 }
 
 /**
@@ -388,6 +416,10 @@ export interface ResolvedVitePWAOptions extends Required<Omit<VitePWAOptions, 'p
     sourcemap?: BuildOptions['sourcemap']
     enableWorkboxModulesLogs?: true
   }
+  injectManifestEnvOptions: {
+    envDir: ResolvedConfig['envDir']
+    envPrefix: ResolvedConfig['envPrefix']
+  }
   pwaAssets: false | ResolvedPWAAssetsOptions
 }
 
diff --git a/src/vite-build.ts b/src/vite-build.ts
index 4d02e703..51b4f001 100644
--- a/src/vite-build.ts
+++ b/src/vite-build.ts
@@ -33,6 +33,9 @@ export async function buildSW(
   // log sw build
   logSWViteBuild(normalizePath(relative(viteOptions.root, options.swSrc)), viteOptions, format)
 
+  // allow integrations to modify the build config
+  await options.integration?.configureCustomSWViteBuild?.(inlineConfig)
+
   await build(inlineConfig)
 
   if (format !== 'iife') {
@@ -86,13 +89,19 @@ function prepareViteBuild(
   define['process.env.NODE_ENV'] = JSON.stringify(nodeEnv)
 
   const buildPlugins = options.buildPlugins
-  const { format, plugins, rollupOptions } = options.injectManifestRollupOptions
+  const {
+    format,
+    plugins,
+    rollupOptions,
+  } = options.injectManifestRollupOptions
 
   const inlineConfig = {
     root: viteOptions.root,
     base: viteOptions.base,
     resolve: viteOptions.resolve,
     mode: viteOptions.mode,
+    envDir: options.injectManifestEnvOptions.envDir,
+    envPrefix: options.injectManifestEnvOptions.envPrefix,
     // don't copy anything from public folder
     publicDir: false,
     build: {