vite.js

const path = require('path')
const fs = require('fs')

module.exports = function proxyOptions({
  port = 8080,
  source = '^/(app|login|api|assets|files)',
 } = {}) {
  const config = getCommonSiteConfig()
  const webserver_port = config ? config.webserver_port : 8000
  if (!config) {
    console.log('No common_site_config.json found, using default port 8000')
  }
  let proxy = {}
  proxy[source] = {
    target: `http://127.0.0.1:${webserver_port}`,
    ws: true,
    router: function (req) {
      const site_name = req.headers.host.split(':')[0]
      return `http://${site_name}:${webserver_port}`
    }
  }
  return {
    name: 'frappeui-vite-plugin',
    config: () => ({
      server: {
        port: port,
        proxy: proxy,
      },
    }),
  }
}

function getCommonSiteConfig() {
  let currentDir = path.resolve('.')
  // traverse up till we find frappe-bench with sites directory
  while (currentDir !== '/') {
    if (
      fs.existsSync(path.join(currentDir, 'sites')) &&
      fs.existsSync(path.join(currentDir, 'apps'))
    ) {
      let configPath = path.join(currentDir, 'sites', 'common_site_config.json')
      if (fs.existsSync(configPath)) {
        return JSON.parse(fs.readFileSync(configPath))
      }
      return null
    }
    currentDir = path.resolve(currentDir, '..')
  }
  return null
}