import { defineConfig } from 'vite'
import react from '@vitejs/plugin-react'
import path from 'path';
import fs from 'fs';
import copy from 'rollup-plugin-copy'
const args = process.argv.slice(2);
const outDirArg = args.find(arg => arg.startsWith('--outDir='));
const output = outDirArg ? outDirArg.split('=')[1] : './dist';

// 读取 package.json 获取版本号
const packageJson = JSON.parse(fs.readFileSync(path.join(__dirname, 'package.json'), 'utf-8'));
const versionName = `v${packageJson.version}`;

// 版本注入插件
function injectVersionPlugin() {
	return {
	  name: 'inject-version',
	  transformIndexHtml(html:any) {
		// 在 head 标签结束前插入版本号脚本
    if(output.endsWith('web-dist')){
      return html.replace(
        '</head>',
        `		<script>const versionName='${versionName}';</script>\n<script src="./browser-adap.js"></script>\n</head>`
      );
    }
      
		return html.replace(
		  '</head>',
		  `		<script>const versionName='${versionName}';</script>\n</head>`
		);
	  },
	};
}
/**
 * 获取目录下所有HTML文件的绝对路径，并组织成对象
 * @param {string} dirPath - 要扫描的目录路径，默认为当前目录
 * @returns {Object} - 包含HTML文件路径的对象
 */
function getHtmlFiles(dirPath = path.resolve(__dirname)) {
  // 确保目录存在
  if (!fs.existsSync(dirPath)) {
    console.error(`目录不存在: ${dirPath}`);
    return {};
  }

  // 读取目录中的所有文件
  const files = fs.readdirSync(dirPath);
  
  // 过滤出HTML文件并创建结果对象
  const result:any = {};
  
  files.forEach((file:any) => {
    // 检查文件是否以.html结尾
    if (file.endsWith('.html')) {
      // 获取文件的绝对路径
      const absolutePath = path.resolve(dirPath, file);
      
      // 如果是index.html，键名为main，否则使用不带扩展名的文件名
      if (file === 'index.html') {
        result.main = absolutePath;
      } else {
        // 去除.html扩展名
        const fileName = file.slice(0, -5); // 移除'.html'
        result[fileName] = absolutePath;
      }
    }
  });
  return result;
}
export default defineConfig({
	base: './',
  plugins: [react(),injectVersionPlugin(),
    copy({
      verbose: false,
      hook: 'closeBundle',
      targets: [
        { src: 'browser-adap.js', dest: output } // 将文件从源复制到目标目录
      ]
    })
  ],
  build: {
		target: 'es2015',
		rollupOptions: {
			input: getHtmlFiles(),
			output: {
				dir: output
			},
		},
	},
})