const { exec } = require('child_process');
const fs = require('fs');

const getGitBranch = () => {
  return new Promise((resolve, reject) => {
    exec('git symbolic-ref --quiet --short HEAD || git describe --all --exact-match HEAD', (error, stdout, stderr) => {
      if (error) {
        reject(error);
      } else {
        const branch = stdout.trim();
        resolve(branch);
      }
    });
  });
};

const extractVersion = (branch) => {
  const versionRegex = /(v.*?)$/;
  const matches = branch.match(versionRegex);
  return matches ? matches[0] : '';
};

const updateEnvFile = (branch) => {
  try {
    const envFilePath = '.env.production';
    const existingContent = fs.readFileSync(envFilePath, 'utf8');
    const updatedBranch = extractVersion(branch);
    const updatedContent = existingContent.replace(/(VITE_APP_GIT_BRANCH=).*/, `$1'${updatedBranch}'`);
    fs.writeFileSync(envFilePath, updatedContent);
    console.log('Git 分支信息已更新到 .env.production 文件', branch, updatedBranch);
  } catch (error) {
    console.error('更新 .env.production 文件失败:', error);
  }
};

getGitBranch()
  .then((branch) => {
    updateEnvFile(branch);
  })
  .catch((error) => {
    console.error('获取 Git 分支失败:', error);
  });