Netlify plugin to contextualize vite env vars.

index.js

module.exports = {
  onPreBuild: () => {
    const BRANCH = process.env.BRANCH.toUpperCase().replace(/-/g, '_')
    console.log('Vite env plugin starting for branch:', BRANCH)
    Object.keys(process.env).sort().forEach(key => {
      if (!key.startsWith('VITE_') || key.endsWith(`_${BRANCH}`)) {
        return
      }

      const contextualValue = process.env[`${key}_${BRANCH}`]
      if (contextualValue) {
        console.log(`Reassigning ${key} to ${contextualValue}`)
        process.env[key] = contextualValue
      } else {
        console.log('Keeping default value for', key)
      }
    })
    process.env.VITE_BRANCH = BRANCH
  }
}