index.ts 1.82 KB
import { getRootPath } from "../../../utils";
import { writeFileSync } from "fs";
import { Plugin } from "vite";
import { GLOB_CONFIG_FILE_NAME } from "./const";
import { getGlobalConfigName } from "./getGlobConfigName";


const stringToJSONParse = (string: string) => {
  try {
    return JSON.parse(string);
  } catch (error) {
    return string;
  }
};

export function parseEnv(env: Record<string, string>) {
  const res: Record<string, string> = {};

  Object.keys(env).forEach((key) => {
    try {
      const value = env[key];
      res[key] = stringToJSONParse(value);
    } catch (err) {
      // eslint-disable-next-line no-console
      console.log(`env variable ${key} can't serialization!`);
    }
  });

  return res;
}

export function GenerateBuildConfig(viteEnv: Record<string, any>) {
  return {
    name: 'vite-plugin-generate-global-config',
    apply: 'build',
    enforce: 'post',
    closeBundle() {

      function createConfig({ config, configName, configFileName }: { configName: string, config: Record<string, any>, configFileName?: string } = { configName: '', config: {} }) {
        try {
          const windowConf = `window.${configName}`;
          // Ensure that the variable will not be modified 
          const configStr = `${windowConf}=${JSON.stringify(parseEnv(config))};
            Object.freeze(${windowConf});
            Object.defineProperty(window, "${configName}", {
              configurable: false,
              writable: false,
            });
          `.replace(/\s/g, '');

          writeFileSync(getRootPath(`dist/${configFileName}`), configStr, { encoding: 'utf-8' })
        } catch (error) {

        }
      }

      const configName = getGlobalConfigName(viteEnv as ImportMetaEnv)

      createConfig({ config: viteEnv, configFileName: GLOB_CONFIG_FILE_NAME, configName })

    }
  } as Plugin
}