package com.nincraft.modpackdownloader.processor;

import com.google.gson.GsonBuilder;
import com.nincraft.modpackdownloader.container.CurseFile;
import com.nincraft.modpackdownloader.container.Manifest;
import com.nincraft.modpackdownloader.container.Mod;
import com.nincraft.modpackdownloader.handler.ForgeHandler;
import com.nincraft.modpackdownloader.summary.UpdateCheckSummarizer;
import com.nincraft.modpackdownloader.util.Arguments;
import com.nincraft.modpackdownloader.util.DownloadHelper;
import com.nincraft.modpackdownloader.util.Reference;
import lombok.Getter;
import lombok.extern.log4j.Log4j2;
import lombok.val;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.concurrent.Executors;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@Log4j2
public class UpdateModsProcessor extends AbstractProcessor {

	private static Reference reference = Reference.getInstance();
	private static UpdateCheckSummarizer updateCheckSummarizer = UpdateCheckSummarizer.getInstance();
	@Getter
	private boolean checkUpdate;

	public UpdateModsProcessor(Arguments arguments, DownloadHelper downloadHelper) {
		super(arguments, downloadHelper);
		checkUpdate = !StringUtils.isBlank(arguments.getCheckMCUpdate());
	}

	private void backupCurseManifest(final File manifestFile) {
		try {
			FileUtils.copyFile(manifestFile, new File(manifestFile.getAbsolutePath() + ".bak"), true);
		} catch (IOException e) {
			log.error("Could not backup Curse manifest file", e);
		}
	}

	private void updateMods(final Manifest manifest, final List<Mod> modList) {
		if (!manifest.getBatchAddCurse().isEmpty()) {
			log.info("Found batch add for Curse");
			addBatch(manifest, modList);
		}

		Reference.updateTotal = modList.size();

		ForgeHandler forgeHandler = new ForgeHandler(arguments, downloadHelper);
		Runnable forgeThread = new Thread(() ->
				manifest.getMinecraft().setModLoaders(
						forgeHandler.updateForge(manifest.getMinecraftVersion(), manifest.getMinecraft().getModLoaders())));

		setExecutorService(Executors.newFixedThreadPool(Reference.updateTotal + 1));

		getExecutorService().execute(forgeThread);

		log.trace("Updating {} mods...", Reference.updateTotal);

		int updateCount = 1;
		for (val mod : modList) {
			log.info(reference.getUpdatingModXOfY(), mod.getName(), updateCount++, Reference.updateTotal);
			Runnable modUpdate = new Thread(() -> {
				MOD_HANDLERS.get(mod.getClass()).updateMod(mod);
				Reference.updateCount++;
				log.info("Finished updating {}", mod.getName());
			});
			getExecutorService().execute(modUpdate);
		}
		getExecutorService().shutdown();
		log.trace("Finished updating {} mods.", Reference.updateTotal);
	}

	private void addBatch(final Manifest manifestFile, final List<Mod> modList) {
		CurseFile curseFile;
		String projectIdPattern = "(\\d)+";
		String projectNamePattern = "(((?:[a-z][a-z]+))(-)?)+";
		for (String projectUrl : manifestFile.getBatchAddCurse()) {
			String projectIdName = projectUrl.substring(projectUrl.lastIndexOf('/') + 1);
			Pattern pId = Pattern.compile(projectIdPattern);
			Matcher m = pId.matcher(projectIdName);
			String projectId = null;
			if (m.find()) {
				projectId = m.group();
			}

			String projectName = null;

			pId = Pattern.compile(projectNamePattern);
			m = pId.matcher(projectIdName);
			if (m.find()) {
				projectName = m.group();
			}

			if (projectId != null && projectName != null) {
				curseFile = new CurseFile(projectId, projectName);
				curseFile.init();
				log.info("Adding {} from batch add", curseFile.getName());
				modList.add(curseFile);
				manifestFile.getCurseFiles().add(curseFile);
			} else {
				log.warn("Unable to add {} from batch add", projectUrl);
			}
		}
	}

	private void updateManifest(final File file, final Manifest manifest) {
		log.info("Updating Manifest File...");
		// Sort Mod Lists
		manifest.getCurseFiles().sort(MOD_COMPARATOR);
		manifest.getThirdParty().sort(MOD_COMPARATOR);

		// Clean up Empty Lists
		cleanupLists(manifest);

		// Dump Manifest to file
		val prettyGson = new GsonBuilder().setPrettyPrinting().excludeFieldsWithoutExposeAnnotation()
				.disableHtmlEscaping().create();
		try (val fileWriter = new FileWriter(file)) {
			fileWriter.write(prettyGson.toJson(manifest));
		} catch (final IOException e) {
			log.error(e);
		}
	}

	private void cleanupLists(final Manifest manifest) {

		// Clean up Empty Lists
		if (manifest.getCurseFiles().isEmpty()) {
			manifest.setCurseFiles(null);
		}
		if (manifest.getThirdParty().isEmpty()) {
			manifest.setThirdParty(null);
		}
		if (manifest.getMinecraft().getModLoaders().isEmpty()) {
			manifest.getMinecraft().setModLoaders(null);
		}

		// Always Clean up Batch Add
		manifest.setBatchAddCurse(null);
	}

	@Override
	protected void init(final Map<File, Manifest> manifestMap) {
		// no-op
	}

	@Override
	protected boolean preprocess(final Entry<File, Manifest> manifestEntry) {
		if (!isCheckUpdate()) {
			backupManifest(manifestEntry.getKey(), manifestEntry.getValue());
		} else {
			arguments.setMcVersion(arguments.getCheckMCUpdate());
		}
		return true;
	}

	@Override
	protected boolean process(final Entry<File, Manifest> manifestEntry) {
		updateMods(manifestEntry.getValue(), buildModList(manifestEntry.getKey(), manifestEntry.getValue()));
		return true;
	}

	@Override
	protected boolean postProcess(final Entry<File, Manifest> manifestEntry) {
		if (isCheckUpdate() || arguments.isUpdateMods()) {
			updateCheckSummarizer.summarize();
		} else {
			updateManifest(manifestEntry.getKey(), manifestEntry.getValue());
		}
		return true;
	}

	private void backupManifest(final File manifestFile, final Manifest manifest) {
		if (CollectionUtils.isNotEmpty(manifest.getCurseFiles())) {
			backupCurseManifest(manifestFile);
		}
	}
}

@@ -5644,62 +5644,8 @@
 ) %7B%0A
-%09%09if (isCheckUpdate() %7C%7C arguments.isUpdateMods()) %7B%0A%09
 %09%09up
@@ -5675,30 +5675,53 @@
 marize();%0A%09%09
-%7D else
+if (arguments.isUpdateMods())
  %7B%0A%09%09%09update

Fix updating