import fs from 'fs'
import path from 'path'
import { fileURLToPath } from 'url'

const __filename = fileURLToPath(import.meta.url)
const __dirname = path.dirname(__filename)

// 处理 manifest.json
function processManifest() {
	const manifestPath = path.join(__dirname, '../public/manifest.json')
	const manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf8'))

	// 移除本地调试相关的配置
	if (manifest.host_permissions) {
		manifest.host_permissions = manifest.host_permissions.filter(
			(permission) => !permission.includes('localhost'),
		)
	}

	if (
		manifest.content_security_policy &&
		manifest.content_security_policy.extension_pages
	) {
		// 移除 CSP 中的本地开发相关配置
		manifest.content_security_policy.extension_pages =
			manifest.content_security_policy.extension_pages
				.replace(/script-src 'self' http:\/\/localhost:5173;\s*/g, '')
				.replace(/\s*http:\/\/localhost:5173\s*/g, ' ')
				.replace(/\s*ws:\/\/localhost:5173\s*/g, ' ')
				.replace(/;\s+/g, '; ') // 标准化分号后的空格
				.replace(/\s+/g, ' ') // 合并多个空格为一个
				.trim()
	}

	// 移除 CSP 中的 sandbox 配置（Firefox 不支持）
	if (
		manifest.content_security_policy &&
		manifest.content_security_policy.sandbox
	) {
		delete manifest.content_security_policy.sandbox
	}

	// 移除 background.service_worker，替换为 background.scripts
	if (manifest.background && manifest.background.service_worker) {
		manifest.background.scripts = [manifest.background.service_worker]
		delete manifest.background.service_worker
	}

	// 添加 firefox 特有配置
	manifest.browser_specific_settings = {
		gecko: {
			id: 'msr-mod@firefox-addon.astrian.moe',
			strict_min_version: '115.0',
		},
	}

	fs.writeFileSync(manifestPath, JSON.stringify(manifest, null, 2))
	console.log('✅ Manifest.json processed')
}

// 处理 index.html
function processIndexHtml() {
	const indexPath = path.join(__dirname, '../index.html')
	let content = fs.readFileSync(indexPath, 'utf8')

	// 替换脚本地址
	content = content.replace(
		/src="[^"]*\/src\/main\.ts"/g,
		'src="./src/main.ts"',
	)

	// 移除 crossorigin 属性
	content = content.replace(/\s+crossorigin/g, '')

	fs.writeFileSync(indexPath, content)
	console.log('✅ Index.html processed')
}

// 执行处理
try {
	processManifest()
	processIndexHtml()
	console.log('🎉 Build preparation completed!')
} catch (error) {
	console.error('❌ Error during build preparation:', error)
	process.exit(1)
}