utils.ts 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. import fs from 'fs';
  2. import path from 'path';
  3. import dotenv from 'dotenv';
  4. export function isDevFn(mode: string): boolean {
  5. return mode === 'development';
  6. }
  7. export function isProdFn(mode: string): boolean {
  8. return mode === 'production';
  9. }
  10. /**
  11. * Whether to generate package preview
  12. */
  13. export function isReportMode(): boolean {
  14. return process.env.REPORT === 'true';
  15. }
  16. // Read all environment variable configuration files to process.env
  17. export function wrapperEnv(envConf: Recordable): ViteEnv {
  18. const ret: any = {};
  19. for (const envName of Object.keys(envConf)) {
  20. let realName = envConf[envName].replace(/\\n/g, '\n');
  21. realName = realName === 'true' ? true : realName === 'false' ? false : realName;
  22. if (envName === 'VITE_PORT') {
  23. realName = Number(realName);
  24. }
  25. if (envName === 'VITE_PROXY') {
  26. try {
  27. realName = JSON.parse(realName);
  28. } catch (error) {}
  29. }
  30. ret[envName] = realName;
  31. process.env[envName] = realName;
  32. }
  33. return ret;
  34. }
  35. /**
  36. * Get the environment variables starting with the specified prefix
  37. * @param match prefix
  38. * @param confFiles ext
  39. */
  40. export function getEnvConfig(match = 'VITE_GLOB_', confFiles = ['.env', '.env.production']) {
  41. let envConfig = {};
  42. confFiles.forEach((item) => {
  43. try {
  44. const env = dotenv.parse(fs.readFileSync(path.resolve(process.cwd(), item)));
  45. envConfig = { ...envConfig, ...env };
  46. } catch (error) {}
  47. });
  48. Object.keys(envConfig).forEach((key) => {
  49. const reg = new RegExp(`^(${match})`);
  50. if (!reg.test(key)) {
  51. Reflect.deleteProperty(envConfig, key);
  52. }
  53. });
  54. return envConfig;
  55. }
  56. /**
  57. * Get user root directory
  58. * @param dir file path
  59. */
  60. export function getRootPath(...dir: string[]) {
  61. return path.resolve(process.cwd(), ...dir);
  62. }