const { existsSync, readFileSync } = require('fs');
const { join, basename } = require('path');
const rimraf = require('rimraf');
const assert = require('assert');
const { merge } = require('lodash');
const signale = require('signale');
const chalk = require('chalk');
const babel = require('father-build/lib/babel');
const rollup = require('./rollup');
const registerBabel = require('father-build/lib/registerBabel').default;
const { getExistFile, getLernaPackages } = require('father-build/lib/utils');
const UserConfig = require('father-build/lib/getUserConfig');
const randomColor = require('father-build/lib/randomColor').default;
const getUserConfig = UserConfig.default;
const { CONFIG_FILES } = UserConfig;

function getBundleOpts(opts) {
	const { cwd, buildArgs = {}, rootConfig = {} } = opts;
	const entry = getExistFile({
		cwd,
		files: [
			'src/index.tsx',
			'src/index.ts',
			'src/index.jsx',
			'src/index.js',
		],
		returnRelative: true,
	});
	const userConfig = getUserConfig({ cwd });
	const userConfigs = Array.isArray(userConfig) ? userConfig : [userConfig];
	return userConfigs.map((userConfig) => {
		const bundleOpts = merge(
			{
				entry,
			},
			rootConfig,
			userConfig,
			buildArgs,
		);

		// Support config esm: 'rollup' and cjs: 'rollup'
		if (typeof bundleOpts.esm === 'string') {
			bundleOpts.esm = { type: bundleOpts.esm };
		}
		if (typeof bundleOpts.cjs === 'string') {
			bundleOpts.cjs = { type: bundleOpts.cjs };
		}

		return bundleOpts;
	});
}

function validateBundleOpts(bundleOpts, { cwd, rootPath }) {
	if (bundleOpts.runtimeHelpers) {
		const pkgPath = join(cwd, 'package.json');
		assert.ok(
			existsSync(pkgPath),
			`@babel/runtime dependency is required to use runtimeHelpers`,
		);
		const pkg = JSON.parse(readFileSync(pkgPath, 'utf-8'));
		assert.ok(
			(pkg.dependencies || {})['@babel/runtime'],
			`@babel/runtime dependency is required to use runtimeHelpers`,
		);
	}
	if (
		bundleOpts.cjs &&
		bundleOpts.cjs.lazy &&
		bundleOpts.cjs.type === 'rollup'
	) {
		throw new Error(
			`
cjs.lazy don't support rollup.
    `.trim(),
		);
	}
	if (!bundleOpts.esm && !bundleOpts.cjs && !bundleOpts.umd) {
		throw new Error(
			`
None format of ${chalk.cyan(
				'cjs | esm | umd',
			)} is configured, checkout https://github.com/umijs/father for usage details.
`.trim(),
		);
	}
	if (bundleOpts.entry) {
		const tsConfigPath = join(cwd, 'tsconfig.json');
		const tsConfig =
			existsSync(tsConfigPath) ||
			(rootPath && existsSync(join(rootPath, 'tsconfig.json')));
		if (
			!tsConfig &&
			((Array.isArray(bundleOpts.entry) &&
				bundleOpts.entry.some(isTypescriptFile)) ||
				(!Array.isArray(bundleOpts.entry) &&
					isTypescriptFile(bundleOpts.entry)))
		) {
			signale.info(
				`Project using ${chalk.cyan(
					'typescript',
				)} but tsconfig.json not exists. Use default config.`,
			);
		}
	}
}

function isTypescriptFile(filePath) {
	return filePath.endsWith('.ts') || filePath.endsWith('.tsx');
}

async function build(opts, extraOpts = {}) {
	const { cwd, rootPath, watch } = opts;
	const { pkg } = extraOpts;

	const dispose = [];

	// register babel for config files
	registerBabel({
		cwd,
		only: CONFIG_FILES,
	});

	const pkgName = (typeof pkg === 'string' ? pkg : pkg.name) || 'unknown';

	function log(msg) {
		console.log(`${pkg ? `${randomColor(`${pkgName}`)}: ` : ''}${msg}`);
	}

	// Get user config
	const bundleOptsArray = getBundleOpts(opts);

	for (const bundleOpts of bundleOptsArray) {
		validateBundleOpts(bundleOpts, { cwd, rootPath });

		// Clean dist
		log(chalk.gray(`Clean dist directory`));
		rimraf.sync(join(cwd, 'dist'));

		// Build umd
		if (bundleOpts.umd) {
			log(`Build umd`);
			await rollup({
				cwd,
				rootPath,
				log,
				type: 'umd',
				entry: bundleOpts.entry,
				watch,
				dispose,
				bundleOpts,
			});
		}

		// Build cjs
		if (bundleOpts.cjs) {
			const cjs = bundleOpts.cjs;
			log(`Build cjs with ${cjs.type}`);
			if (cjs.type === 'babel') {
				await babel({
					cwd,
					rootPath,
					watch,
					dispose,
					type: 'cjs',
					log,
					bundleOpts,
				});
			} else {
				await rollup({
					cwd,
					rootPath,
					log,
					type: 'cjs',
					entry: bundleOpts.entry,
					watch,
					dispose,
					bundleOpts,
				});
			}
		}

		// Build esm
		if (bundleOpts.esm) {
			const esm = bundleOpts.esm;
			log(`Build esm with ${esm.type}`);
			const importLibToEs = esm && esm.importLibToEs;
			if (esm && esm.type === 'babel') {
				await babel({
					cwd,
					rootPath,
					watch,
					dispose,
					type: 'esm',
					importLibToEs,
					log,
					bundleOpts,
				});
			} else {
				await rollup({
					cwd,
					rootPath,
					log,
					type: 'esm',
					entry: bundleOpts.entry,
					importLibToEs,
					watch,
					dispose,
					bundleOpts,
				});
			}
		}
	}

	return dispose;
}

async function buildForLerna(opts) {
	const { cwd, rootConfig = {}, buildArgs = {} } = opts;

	// register babel for config files
	registerBabel({
		cwd,
		only: CONFIG_FILES,
	});

	const userConfig = merge(getUserConfig({ cwd }), rootConfig, buildArgs);

	let pkgs = await getLernaPackages(cwd, userConfig.pkgFilter);

	// support define pkgs in lerna
	// TODO: 使用lerna包解决依赖编译问题
	if (userConfig.pkgs) {
		pkgs = userConfig.pkgs
			.map((item) => {
				return pkgs.find((pkg) => basename(pkg.contents) === item);
			})
			.filter(Boolean);
	}

	const dispose = [];
	for (const pkg of pkgs) {
		if (
			process.env.PACKAGE &&
			basename(pkg.contents) !== process.env.PACKAGE
		)
			continue;
		// build error when .DS_Store includes in packages root
		const pkgPath = pkg.contents;
		assert.ok(
			existsSync(join(pkgPath, 'package.json')),
			`package.json not found in packages/${pkg}`,
		);
		process.chdir(pkgPath);
		dispose.push(
			...(await build(
				{
					// eslint-disable-line
					...opts,
					buildArgs: opts.buildArgs,
					rootConfig: userConfig,
					cwd: pkgPath,
					rootPath: cwd,
				},
				{
					pkg,
				},
			)),
		);
	}
	return dispose;
}

module.exports = async function (opts) {
	const useLerna = existsSync(join(opts.cwd, 'lerna.json'));
	const isLerna = useLerna && process.env.LERNA !== 'none';

	const dispose = isLerna ? await buildForLerna(opts) : await build(opts);
	return () => dispose.forEach((e) => e());
};