package tlIItools;

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * String pairs for replacements.
 *
 * @author Ben Culkin
 */
public class ReplPair implements Comparable<ReplPair> {
	/**
	 * Represents an error encountered parsing ReplPairs
	 *
	 * @author Ben Culkin
	 */
	public static class ReplError {
		/**
		 * The line the error occured on.
		 */
		public int line;
		/**
		 * The number of pairs we have processed so far.
		 */
		public int numPairs;

		/**
		 * The text of the line we errored on.
		 */
		public String txt;
		/**
		 * The message of the error.
		 */
		public String msg;

		/**
		 * Create a new ReplPair parse error.
		 *
		 * @param lne
		 * 	The line the error occured on.
		 * @param nPairs
		 * 	The number of pairs processed up to this point.
		 * @param msg
		 * 	The message detailing the error.
		 * @param txt
		 * 	The text that caused the error.
		 */
		public ReplError(int lne, int nPairs, String msg, String txt) {
			line = lne;
			numPairs = nPairs;

			this.txt = txt;
			this.msg = msg;
		}

		@Override
		public String toString() {
			String errString;
			if      (txt == null)    errString = "No associated line";
			else if (txt.equals("")) errString = "Text of line was empty";
			else                     errString = "Text of line was: " + txt;

			return String.format("line %d, pair %d:%s\n\t%s", line, numPairs, msg, errString);
		}
	}

	/**
	 * Possible statuses of pairs with respect to exporting.
	 * @author Ben Culkin
	 */
	public static enum StageStatus {
		/**
		 * Only use for staging pairs; don't export.
		 */
		INTERNAL,
		/**
		 * Don't use for staging pairs; do export.
		 */
		EXTERNAL,
		/**
		 * Use for staging pairs; do export.
		 */
		BOTH;
	}

	/**
	 * The priority for this replacement.
	 */
	public int priority;

	/**
	 * The name of this replacement.
	 *
	 * Defaults to the 'find' string.
	 */
	public String name;
	/**
	 * The string to look for.
	 */
	public String find;

	/**
	 * The string to replace it with.
	 */
	public String replace;

	private StageStatus stat = StageStatus.BOTH;

	/**
	 * Create a new blank replacement pair.
	 */
	public ReplPair() {
		this("", "", 1, null);
	}

	/**
	 * Create a new replacement pair with a priority of 1.
	 *
	 * @param f
	 * 	The string to find.
	 * @param r
	 * 	The string to replace.
	 */
	public ReplPair(String f, String r) {
		this(f, r, 1);
	}

	/**
	 * Create a new named replacement pair with a priority of 1.
	 *
	 * @param f
	 * 	The string to find.
	 * @param r
	 * 	The string to replace.
	 * @param n
	 * 	The name of the replacement pair.
	 */
	public ReplPair(String f, String r, String n) {
		this(f, r, 1, n);
	}

	/**
	 * Create a new replacement pair with a set priority.
	 *
	 * @param f
	 * 	The string to find.
	 * @param r
	 * 	The string to replace.
	 * @param p
	 * 	The priority for the replacement.
	 */
	public ReplPair(String f, String r, int p) {
		this(f, r, p, f);
	}

	/**
	 * Create a new replacement pair with a set priority and name.
	 *
	 * @param f
	 * 	The string to find.
	 * @param r
	 * 	The string to replace.
	 * @param n
	 * 	The name of the replacement pair.
	 * @param p
	 * 	The priority for the replacement.
	 */
	public ReplPair(String f, String r, int p, String n) {
		find    = f;
		replace = r;

		name = n;

		priority = p;
	}
	/**
	 * Read a list of replacement pairs from an input source.
	 *
	 * @param scn
	 * 	The source to read the replacements from.
	 * @return
	 * 	The list of replacements.
	 */
	public static List<ReplPair> readList(Scanner scn) {
		return ReplPair.readList(new ArrayList<>(), scn);
	}

	/**
	 * Read a list of replacement pairs from an input source, adding them to
	 * an existing list.
	 *
	 * @param detals
	 * 	The list to add the replacements to.
	 * @param scn
	 * 	The source to read the replacements from.
	 * @return
	 * 	The list of replacements.
	 */
	public static List<ReplPair> readList(List<ReplPair> detals, Scanner scn) {
		List<ReplError> errList = new ArrayList<>();

		List<ReplPair> rplPar = readList(detals, scn, errList);

		if (errList.size() != 0) {
			String errString;
			if (errList.size() == 0) errString = "An error";
			else                       errString = "Errors";

			throw new IllegalArgumentException(String.format("%s occured parsing replacement pairs:\n%s", errString, errList));
		}

		return rplPar;
	}

	/**
	 * Read a list of replacement pairs from an input source, adding them to
	 * an existing list.
	 *
	 * @param detals
	 * 	The list to add the replacements to.
	 * @param scn
	 * 	The source to read the replacements from.
	 * @param errs
	 * 	The list to stick errors in.
	 * @return
	 * 	The list of replacements.
	 */
	public static List<ReplPair> readList(List<ReplPair> detals, Scanner scn, List<ReplError> errs) {
		int lno = 0;
		int pno = 0;

		int defPrior = 1;
		int defStage = 1;

		boolean defMulti = false;

		StageStatus defStatus = StageStatus.BOTH;

		List<List<ReplPair>> stages = new ArrayList<>();
		stages.add(new ArrayList<>());

		// For every line in the source...
		while (scn.hasNextLine()) {
			String name = scn.nextLine().trim();
			lno += 1;

			// If its commented or blank, skip it
			if (name.equals(""))      continue;
			if (name.startsWith("#")) continue;

			// Global control. Process it.
			if (name.startsWith("|//")) {
				name = name.substring(3);

				// Split out each control
				String[] bits = name.split(";");

				for (String bit : bits) {
					String bitHead = bit.toUpperCase();
					String bitBody = bit;

					int idx = bit.indexOf('/');
					if (idx != -1) {
						bitHead = bit.substring(0, idx).toUpperCase();
						bitBody = bit.substring(idx + 1);
					}

					switch (bitHead) {
					case "P":
						try {
							defPrior = Integer.parseInt(bitBody);
						} catch (NumberFormatException nfex) {
							String errMsg = String.format("'%s' is not a valid priority (must be an integer)", bitBody);
							errs.add(new ReplError(lno, pno, errMsg, name));
						}
						break;
					case "S":
						try {
							int tmpStage = Integer.parseInt(bitBody);
							if (tmpStage < 0) {
								String errMsg = String.format("'%s' is not a valid stage (must be a positive integer)", bitBody);
								errs.add(new ReplError(lno, pno, errMsg, name));

								break;
							}
							defStage = tmpStage;
						} catch (NumberFormatException nfex) {
							String errMsg = String.format("'%s' is not a valid stage (must be a positive integer)", bitBody);
							errs.add(new ReplError(lno, pno, errMsg, name));
						}
						break;
					case "MT":
						defMulti = true;
						break;
					case "MF":
						defMulti = false;
						break;
					case "M":
						defMulti = Boolean.parseBoolean(bitBody);
						break;
					case "I":
						defStatus = StageStatus.INTERNAL;
						break;
					case "E":
						defStatus = StageStatus.EXTERNAL;
						break;
					case "B":
						defStatus = StageStatus.BOTH;
						break;
					default: 
						errs.add(new ReplError(lno, pno, String.format("Invalid control name '%s'", bitHead), name));
						break;
					}
				}

				continue;
			}

			ReplPair rp = new ReplPair();
			rp.priority = defPrior;
			rp.stat = defStatus;
			
			int stage = defStage;

			boolean isMulti = defMulti;

			// Name has attached controls, process them.
			if (name.startsWith("//")) {
				name = name.substring(2);

				int idx = name.indexOf("//");
				if (idx == -1) {
					String msg = "Did not find control terminator (//) in name where it should be";

					errs.add(new ReplError(lno, pno, msg, name));
					continue;
				}

				String contName = name.substring(0, idx);
				String actName  = name.substring(idx + 2);

				// Split out each control
				String[] bits = contName.split(";");

				for (String bit : bits) {
					String bitHead = bit.toUpperCase();
					String bitBody = bit;

					idx = bit.indexOf('/');
					if (idx != -1) {
						bitHead = bit.substring(0, idx).toUpperCase();
						bitBody = bit.substring(idx + 1);
					}
					
					switch (bitHead) {
					case "N":
						rp.name = bitBody;
						break;
					case "P":
						try {
							rp.priority = Integer.parseInt(bitBody);
						} catch (NumberFormatException nfex) {
							String errMsg = String.format("'%s' is not a valid priority (must be an integer)", bitBody);
							errs.add(new ReplError(lno, pno, errMsg, name));
						}
						break;
					case "S":
						try {
							int tmpStage = Integer.parseInt(bitBody);
							if (tmpStage < 0) {
								String errMsg = String.format("'%s' is not a valid stage (must be a positive integer)", bitBody);
								errs.add(new ReplError(lno, pno, errMsg, name));

								break;
							}
							stage = tmpStage;
						} catch (NumberFormatException nfex) {
							String errMsg = String.format("'%s' is not a valid stage (must be a positive integer)", bitBody);
							errs.add(new ReplError(lno, pno, errMsg, name));
						}
						break;
					case "MT":
						isMulti = true;
						break;
					case "MF":
						isMulti = false;
						break;
					case "M":
						isMulti = Boolean.parseBoolean(bitBody);
						break;
					case "I":
						rp.stat = StageStatus.INTERNAL;
						break;
					case "E":
						rp.stat = StageStatus.EXTERNAL;
						break;
					case "B":
						rp.stat = StageStatus.BOTH;
						break;
					default: 
						errs.add(new ReplError(lno, pno, String.format("Unknown control name '%s'", bitHead), name));
						break;
					}
				}
				
				// Multi-line name with a trailer
				if (isMulti) {
					String tmp = actName;

					while (tmp.endsWith("\\")) {
						boolean incNL = tmp.endsWith("|\\");

						if (!scn.hasNextLine()) break;

						tmp = scn.nextLine().trim();

						if (tmp.equals(""))      continue;
						if (tmp.startsWith("#")) continue;

						actName = String.format("%s%s%s", actName, incNL ? "\n" : "", tmp);
					}
				}
				
				name = actName;
			}

			rp.find = name;
			if (rp.name == null) rp.name = name;

			// We started to process the pair, mark it as being
			// started
			pno += 1;
			String body = null;

			// Read in the next uncommented line
			do {
				if (!scn.hasNextLine()) {
					String msg = 
						"Ran out of input looking for replacement body for raw name " + name;
						
					errs.add(new ReplError(lno, pno, msg, null));
					break;
				}

				body = scn.nextLine().trim();
				lno += 1;
			} while (body.startsWith("#"));

			isMulti = defMulti;

			// Body has attached controls, process them.
			if (body.startsWith("//")) {
				body = body.substring(2);

				int idx = body.indexOf("//");
				if (idx == -1) {
					String msg = "Did not find control terminator (//) in body where it should be";

					errs.add(new ReplError(lno, pno, msg, body));
					continue;
				}

				String contBody = body.substring(0, idx);
				String actBody  = body.substring(idx + 2);

				// Split out each control
				String[] bits = actBody.split(";");

				for (String bit : bits) {
					String bitHead = bit.toUpperCase();
					String bitBody = bit;

					idx = bit.indexOf('/');
					if (idx != -1) {
						bitHead = bit.substring(0, idx).toUpperCase();
						bitBody = bit.substring(idx + 1);
					}

					switch (bitHead) {
					case "MT":
						isMulti = true;
						break;
					case "MF":
						isMulti = false;
						break;
					case "M":
						isMulti = Boolean.parseBoolean(bitBody);
						break;
					default: 
						errs.add(new ReplError(lno, pno, String.format("Invalid control name '%s'", bitHead), body));
						break;
					}
				}

				// Multi-line name with a trailer
				if (isMulti) {
					String tmp = actBody;

					while (tmp.endsWith("\\")) {
						boolean incNL = tmp.endsWith("|\\");

						if (!scn.hasNextLine()) break;

						tmp = scn.nextLine().trim();

						if (tmp.startsWith("#")) continue;

						actBody = String.format("%s%s%s", actBody, incNL ? "\n" : "", tmp);
					}
				}
				
				body = actBody;
			}

			rp.replace = body;

			List<ReplPair> stageList;
			if (stages.size() < stage) {
				stageList = stages.get(stage);

				if (stageList == null) {
					stageList = new ArrayList<>();
					stages.set(stage, stageList);
				}
			} else {
				stageList = new ArrayList<>();
				stages.set(stage, stageList);
			}

			stageList.add(rp);
		}

		// Special-case one-stage processing.
		if (stages.size() == 1) {
			detals.addAll(stages.iterator().next());

			detals.sort(null);

			return detals;
		}

		// Handle stages
		List<ReplPair> tmpList = new ArrayList<>();
		tmpList.addAll(detals);

		for (List<ReplPair> stageList : stages) {
			List<ReplPair> curStage = new ArrayList<>();

			for (ReplPair rp : stageList) {
				// Process through every pair in the previous
				// stages
				for (ReplPair curPar : tmpList) {
					rp.replace = rp.replace.replaceAll(curPar.find, curPar.replace);
				}

				// If we're external; add straight to the output
				if (rp.stat == StageStatus.EXTERNAL) detals.add(rp);
				else                     curStage.add(rp);
			}
			
			tmpList.addAll(curStage);
			tmpList.sort(null);
		}

		// Copy over to output, excluding internals
		for (ReplPair rp : tmpList) {
			if (rp.stat == StageStatus.INTERNAL) continue;

			detals.add(rp);
		}

		detals.sort(null);

		return detals;
	}

	@Override
	public String toString() {
		return String.format("s/%s/%s/p%d", find, replace, priority);
	}
	
	@Override
	public int compareTo(ReplPair rp) {
		return this.priority - rp.priority;
	}
}