import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.List;

import javax.swing.JTextArea;
import javax.swing.SwingWorker;

public class VmfEntityObf extends SwingWorker<List<String>, String>{
	public VmfEntityObf(JTextArea console, File vmf, List<String> targetnameExceptions) {
		this.console = console;
		this.vmf = vmf;
		this.targetnameExceptions = targetnameExceptions;
	}

	@Override
	protected List<String> doInBackground() throws IOException {
		publish("\n");
		publish("--------------------VMF Entity Targetname Obfuscator--------------------" + "\n");

		// Reading lines from file into List
		publish("Loading vmf... ");


// hier passieren komische Sachen


		List<String> vmfList = null;
		BufferedReader br = new BufferedReader(new InputStreamReader(new FileInputStream(this.vmf)));

		String line = br.readLine();
		while(line != null) { // for whatever fkn reason SwingWorker just stops here and doesn't do shit no more
			vmfList.add(line);
			line = br.readLine();
		}
		br.close();
		publish("DONE" + "\n");

		// Setting up for progress calculations
		int totallines = vmfList.size();
		int progress = 0;
		int localprogress = 0;

		// lets first skip all worldgeometry never containing any targetname information
		publish("Skipping WorldGeometry... ");
		for(int i = 0; vmfList.get(i).contains("entity"); i++) {
			totallines--; // removing worldgeometry-lines from the progress calculations
		}
		publish("DONE" + "\n");

		// Wildcards need to be excluded from the renaming progress (future update might include wildcards)
		publish("\n");
		publish("Finding Wildcards: ");
		int start;
		int end;
		for(int i = totallines; i < vmfList.size(); i++) {
			line = vmfList.get(i);
			if(line.contains("*")){
				start = line.indexOf("\"", line.indexOf("\"", line.indexOf("\"")+1)+1)+1;
				end = line.indexOf("*");
				if(!(this.targetnameExceptions.contains(line.substring(start, end)))){
					this.targetnameExceptions.add(line.substring(start, end));
				}
			}
			// Local Progress Calculations
			if((double)progress / (double)totallines >= 0.025 * localprogress) {
				if(localprogress % 4 == 0) {
					publish(localprogress + "");
				} else {
					publish(".");
				}
				localprogress++;
			}
			// Overall Progress Calculations
			progress++;
			setProgress((int)(100 * (double)progress / (double)(3 * totallines)));
		}
		publish("10" + "\n");
		localprogress = 0;

		// we run through the remaining lines and search for entities applicable for targetname change
		publish("\n");
		publish("Finding Targetnames: 0");
		List<String> applicableNames = null;
		boolean bol = false;
		String targetname;

		for(int i = totallines; i < vmfList.size(); i++) {
			// search for lines with "targetname"
			if(line.contains("targetname") && !(line.contains("none") || line.contains(","))){
				targetname = line.substring(line.indexOf("\"", line.indexOf("\"", line.indexOf("\"")+1)+1)+1, line.length()-1);
				// check if found targetname was already added
				if(!(applicableNames.contains(targetname))){
					// check if found targetname matches any Exceptions
					for(int j = 0; i < this.targetnameExceptions.size(); j++){
						if(targetname.startsWith(this.targetnameExceptions.get(j))){
							bol = true;
						}
					}
					// if it doesnt match any Exceptions add it to toBeRenamed
					if (!bol){
						applicableNames.add(targetname);
					}
					// reset bol for next line
					bol = false;
				}
			}
			// Local Progress Calculations
			if((double)progress / (double)totallines >= 0.025 * localprogress) {
				if(localprogress % 4 == 0) {
					publish(localprogress + "");
				} else {
					publish(".");
				}
				localprogress++;
			}
			// Overall Progress Calculations
			progress++;
			setProgress((int)(100 * (double)progress / (double)(3 * totallines)));
		}
		publish("10" + "\n");
		localprogress = 0;

		// now we replace found entities targetname keyvalue with an Integer
		publish("\n");
		publish("Renaming Targetnames: 0");

		String entry;
		for(int i = totallines; i < vmfList.size(); i++) { // run though all lines of vmf beginning from the first entity entry
			line = vmfList.get(i);
			for(int j = 0; j < applicableNames.size(); j++){ // run though all found targetnames for each line
				entry = applicableNames.get(j);
				if(line.contains(" " + entry + ",")  ||		// inside "connections"
				   line.contains("\"" + entry + "\"")||		// "targetname" keyvalue
				   line.contains("\"" + entry + ",") ||		//
				   line.contains("\"" + entry + ",") ||		//
				   line.contains("," + entry + ",")) {
					// replace the targetname with an Integer
					line = line.replace(entry,"" + j);
				// we have to do the same checks for lowerCase as Hammer ignores Case-sensitivity
				} else if(line.toLowerCase().contains(" " + entry.toLowerCase() + ",")  ||
						  line.toLowerCase().contains("\"" + entry.toLowerCase() + "\"")||
						  line.toLowerCase().contains("\"" + entry.toLowerCase() + ",") ||
						  line.toLowerCase().contains("\"" + entry.toLowerCase() + ",") ||
						  line.toLowerCase().contains("," + entry.toLowerCase() + ",")
						) {
					// replace the targetname with an Integer
					line = line.toLowerCase().replace(entry.toLowerCase(),"" + j);
				}
				// Local Progress Calculations
				if((double)progress / (double)totallines >= 0.025 * localprogress) { //problemo here
					if(localprogress % 4 == 0) {
						publish(localprogress + "");
					} else {
						publish(".");
					}
					localprogress++;
				}
				// Overall Progress Calculations
				progress++;
				setProgress((int)(100 * (double)progress / (double)(3 * totallines)));
			}
			vmfList.set(i, line);
		}
		publish("10" + "\n");

		// all steps are done
		publish("\n");
		publish("------------------------------------------------------------------------" + "\n");
		publish("\n");
		return vmfList;
	}

	@Override
    protected void process(List<String> chunks) {
		for(String line : chunks) {
			this.console.append(line + "\n");
		}
    }

	@Override
    protected void done() {
        System.out.println("test");
    }

	private JTextArea console;
	private File vmf;
	private List<String> targetnameExceptions;
}