import { execSync } from 'child_process'
import { copyFileSync, existsSync, readFileSync, writeFileSync } from 'fs'
import glob from 'glob'
import path from 'path'
import { pathToFileURL } from 'url'

/** Prepares the package for publishing. the tarball in case it will be written to disk. */
export async function preparePackage({ sourcePackageDir }: { sourcePackageDir: string }) {
	if (!existsSync(path.join(sourcePackageDir, 'src/index.ts'))) {
		throw new Error(`No src/index.ts file found in '${sourcePackageDir}'!`)
	}

	const manifest = JSON.parse(readFileSync(path.join(sourcePackageDir, 'package.json'), 'utf8'))

	execSync('yarn run -T lazy build-package', { cwd: sourcePackageDir, stdio: 'inherit' })

	// save package.json and reinstate it in postpack
	copyFileSync(
		path.join(sourcePackageDir, 'package.json'),
		path.join(sourcePackageDir, 'package.json.bak')
	)

	const cssFiles = glob.sync(path.join(sourcePackageDir, '*.css'))

	// construct the final package.json
	const newManifest = structuredClone({
		// filter out comments
		...Object.fromEntries(
			Object.entries(manifest).filter(([key]) => !key.startsWith('/*') && key !== 'types')
		),
		main: 'dist-cjs/index.js',
		module: 'dist-esm/index.mjs',
		source: 'src/index.ts',
		exports: {
			'.': {
				import: './dist-esm/index.mjs',
				require: './dist-cjs/index.js',
			},
			...Object.fromEntries(
				cssFiles.map((file) => [`./${path.basename(file)}`, `./${path.basename(file)}`])
			),
		},
		files: [...(manifest.files ?? []), 'dist-esm', 'dist-cjs', 'src'],
	})
	writeFileSync(
		path.join(sourcePackageDir, 'package.json'),
		JSON.stringify(newManifest, null, `\t`)
	)

	// GOTCHA: Yarn's pack command seems to have a race condition where it doesn't reliably pick up
	// files, adding a tiny delay seems to fix it, but we make the delay extra long here just to be
	// safe.
	await new Promise((resolve) => setTimeout(resolve, 1000))
}

if (import.meta.url === pathToFileURL(process.argv[1]).href) {
	// module was called directly
	;(async () => {
		await preparePackage({
			sourcePackageDir: path.resolve(path.normalize(process.argv[2] ?? process.cwd())),
		})
	})()
}