package launchserver.command.hash;

import java.io.BufferedReader;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Collections;

import launcher.helper.IOHelper;
import launcher.helper.LogHelper;
import launchserver.LaunchServer;
import launchserver.command.Command;
import launchserver.command.CommandException;
import org.json.JSONObject;
import org.json.JSONTokener;

public final class UnindexAssetCommand extends Command {
	public UnindexAssetCommand(LaunchServer server) {
		super(server);
	}

	@Override
	public String getArgsDescription() {
		return "<dir> <index> <output-dir>";
	}

	@Override
	public String getUsageDescription() {
		return "Unindex asset dir (1.7.10+)";
	}

	@Override
	public void invoke(String... args) throws Exception {
		verifyArgs(args, 3);
		String inputAssetDirName = IOHelper.verifyFileName(args[0]);
		String indexFileName = IOHelper.verifyFileName(args[1]);
		String outputAssetDirName = IOHelper.verifyFileName(args[2]);
		Path inputAssetDir = LaunchServer.UPDATES_DIR.resolve(inputAssetDirName);
		Path outputAssetDir = LaunchServer.UPDATES_DIR.resolve(outputAssetDirName);
		if (outputAssetDir.equals(inputAssetDir)) {
			throw new CommandException("Indexed and unindexed asset dirs can't be same");
		}

		// Create new asset dir
		LogHelper.subInfo("Creating unindexed asset dir: '%s'", outputAssetDirName);
		Files.createDirectory(outputAssetDir);

		// Read JSON file
		LogHelper.subInfo("Reading asset index file: '%s'", inputAssetDirName);
		JSONObject objects;
		try (BufferedReader reader = IOHelper.newReader(IndexAssetCommand.resolveIndexFile(inputAssetDir, indexFileName))) {
			objects = new JSONObject(new JSONTokener(reader)).getJSONObject(IndexAssetCommand.OBJECTS_DIR);
		}

		// Restore objects
		LogHelper.subInfo("Unindexing %d objects", objects.length());
		for (String name : objects.keySet()) {
			LogHelper.subInfo("Unindexing: '%s'", name);

			// Get JSON object
			JSONObject object = objects.getJSONObject(name);
			String hash = object.getString("hash");

			// Copy hashed file to target
			Path source = IndexAssetCommand.resolveObjectFile(inputAssetDir, hash);
			IOHelper.copy(source, outputAssetDir.resolve(name));
		}

		// Finished
		server.syncUpdatesDir(Collections.singleton(outputAssetDirName));
		LogHelper.subInfo("Asset successfully unindexed: '%s'", inputAssetDirName);
	}
}