// SciTE - Scintilla based Text Editor
// LexOthers.cxx - lexers for properties files, batch files, make files and error lists
// Copyright 1998-2000 by Neil Hodgson <neilh@scintilla.org>
// The License.txt file describes the conditions under which this software may be distributed.

#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <stdio.h>
#include <stdarg.h>

#include "Platform.h"

#include "PropSet.h"
#include "Accessor.h"
#include "KeyWords.h"
#include "Scintilla.h"
#include "SciLexer.h"

#ifdef NEW_BATCH
static void ColouriseBatchLine(char *lineBuffer, int startLine, int endLine, Accessor &styler) {
	for (unsigned int i = 0; i < 4; i++) {
		lineBuffer[i] = tolower(lineBuffer[i]);
	}
	if (0 == strncmp(lineBuffer, "rem", 3) && isspacechar(lineBuffer[3])) {
		styler.ColourTo(endLine, 1);
	} else if (0 == strncmp(lineBuffer, "set", 3) && isspacechar(lineBuffer[3])) {
		styler.ColourTo(endLine, 2);
	} else if (0 == strncmp(lineBuffer, "if", 2) && isspacechar(lineBuffer[2])) {
		styler.ColourTo(endLine, 2);
	} else if (lineBuffer[0] == ':') {
		if (lineBuffer[1] == ':') {	// Fake label, similar to REM, see http://www.winmag.com/columns/explorer/2000/21.htm
			styler.ColourTo(endLine, 1);
		} else {
			styler.ColourTo(endLine, 3);
		}
	} else if (lineBuffer[0] == '@') {	// Hide command (ECHO OFF)
		styler.ColourTo(startLine, 4);
		ColouriseBatchLine(lineBuffer + 1, startLine + 1, i, styler);
	} else {
		styler.ColourTo(endLine, 0);
	}
}
// ToDo: (not necessarily at beginning of line) GOTO, [IF] NOT, ERRORLEVEL
// if [no] (test) (command) -- test is EXIST (filename) | (string1)==(string2) | ERRORLEVEL (number)
// for %%(variable) IN (set) DO (command) -- variable is [a-zA-Z] -- eg for %%X in (*.txt) do type %%X
// ToDo: %n (parameters), %EnvironmentVariable% colourising
// ToDo: Colourise = > >> < | "
#else
static void ColouriseBatchLine(char *lineBuffer, int endLine, int, Accessor &styler) {
	if (0 == strncmp(lineBuffer, "rem", 3)) {
		styler.ColourTo(endLine, 1);
	} else if (0 == strncmp(lineBuffer, "set", 3)) {
		styler.ColourTo(endLine, 2);
	} else if (lineBuffer[0] == ':') {
		styler.ColourTo(endLine, 3);
	} else {
		styler.ColourTo(endLine, 0);
	}
}
#endif

static void ColouriseBatchDoc(unsigned int startPos, int length, int, WordList *[], Accessor &styler) {
	char lineBuffer[1024];
	styler.StartAt(startPos);
	styler.StartSegment(startPos);
	unsigned int linePos = 0, startLine = startPos;
	for (unsigned int i = startPos; i < startPos + length; i++) {
		if (linePos != 0 || !isspacechar(styler[i])) {	// Skip initial spaces
			lineBuffer[linePos++] = static_cast<char>(tolower(styler[i]));
		}
		if (styler[i] == '\r' || styler[i] == '\n' || (linePos >= sizeof(lineBuffer) - 1)) {
			lineBuffer[linePos] = '\0';
			if (lineBuffer[0] == '@') {	// Hide command (ECHO OFF)
				styler.ColourTo(startLine, 4);
				ColouriseBatchLine(lineBuffer + 1, i, startLine, styler);
			} else {
				ColouriseBatchLine(lineBuffer, i, startLine, styler);
			}
			linePos = 0;
			startLine = i+1;
		}
	}
	if (linePos > 0) {
		ColouriseBatchLine(lineBuffer, startPos + length, startLine, styler);
	}
}

static void ColouriseDiffLine(char *lineBuffer, int endLine, Accessor &styler) {
	// It is needed to remember the current state to recognize starting
	// comment lines before the first "diff " or "--- ". If a real
	// difference starts then each line starting with ' ' is a whitespace
	// otherwise it is considered a comment (Only in..., Binary file...)
	if (0 == strncmp(lineBuffer, "diff ", 3)) {
		styler.ColourTo(endLine, 2);
	} else if (0 == strncmp(lineBuffer, "--- ", 3)) {
		styler.ColourTo(endLine, 3);
	} else if (0 == strncmp(lineBuffer, "+++ ", 3)) {
		styler.ColourTo(endLine, 3);
	} else if (lineBuffer[0] == '@') {
		styler.ColourTo(endLine, 4);
	} else if (lineBuffer[0] == '-') {
		styler.ColourTo(endLine, 5);
	} else if (lineBuffer[0] == '+') {
		styler.ColourTo(endLine, 6);
	} else if (lineBuffer[0] != ' ') {
		styler.ColourTo(endLine, 1);
	} else {
		styler.ColourTo(endLine, 0);
	}
}

static void ColouriseDiffDoc(unsigned int startPos, int length, int, WordList *[], Accessor &styler) {
	char lineBuffer[1024];
	styler.StartAt(startPos);
	styler.StartSegment(startPos);
	unsigned int linePos = 0;
	for (unsigned int i = startPos; i < startPos + length; i++) {
		lineBuffer[linePos++] = styler[i];
		if (styler[i] == '\r' || styler[i] == '\n' || (linePos >= sizeof(lineBuffer) - 1)) {
			ColouriseDiffLine(lineBuffer, i, styler);
			linePos = 0;
		}
	}
	if (linePos > 0)
		ColouriseDiffLine(lineBuffer, startPos + length, styler);
}

static void ColourisePropsLine(char *lineBuffer,
							  unsigned int lengthLine, 
							  unsigned int startLine, 
							  unsigned int endPos, 
							  Accessor &styler) {
	unsigned int i = 0;
	while (isspacechar(lineBuffer[i]) && (i < lengthLine))	// Skip initial spaces
		i++;
	if (lineBuffer[i] == '#' || lineBuffer[i] == '!' || lineBuffer[i] == ';') {
		styler.ColourTo(endPos, 1);
	} else if (lineBuffer[i] == '[') {
		styler.ColourTo(endPos, 2);
	} else if (lineBuffer[i] == '@') {
		styler.ColourTo(startLine+i, 4);
		if (lineBuffer[++i] == '=')
			styler.ColourTo(startLine+i, 3);
		styler.ColourTo(endPos, 0);
	} else {
		while (lineBuffer[i] != '=' && (i < lengthLine))	// Search the '=' character
			i++;
		if (lineBuffer[i] == '=') {
			styler.ColourTo(startLine+i-1, 0);
			styler.ColourTo(startLine+i, 3);
			styler.ColourTo(endPos, 0);
		} else {
			styler.ColourTo(endPos, 0);
		}
	}
}

static void ColourisePropsDoc(unsigned int startPos, int length, int, WordList *[], Accessor &styler) {
	char lineBuffer[1024];
	styler.StartAt(startPos);
	styler.StartSegment(startPos);
	unsigned int linePos = 0, startLine = startPos;
	for (unsigned int i = startPos; i <= startPos + length; i++) {
		lineBuffer[linePos++] = styler[i];
		if ((styler[i] == '\r' && styler.SafeGetCharAt(i+1) != '\n') ||
			styler[i] == '\n' ||
			(linePos >= sizeof(lineBuffer) - 1)) {
				lineBuffer[linePos] = '\0';
				ColourisePropsLine(lineBuffer, linePos, startLine, i, styler);
				linePos = 0;
				startLine = i+1;
		}
	}
	if (linePos > 0)
		ColourisePropsLine(lineBuffer, linePos, startLine, startPos + length, styler);
}

static void ColouriseMakeLine(char *lineBuffer,
							  unsigned int lengthLine, 
							  unsigned int startLine, 
							  unsigned int endPos, 
							  Accessor &styler) {
	unsigned int i = 0, state = 0;
	bool bSpecial = false;
	// Skip initial spaces
	while (isspacechar(lineBuffer[i]) && (i < lengthLine))
		i++;
	if (lineBuffer[i] == '#') {
		styler.ColourTo(endPos, 1);
		return;
	}
	if (lineBuffer[i] == '!') {
		styler.ColourTo(endPos, 2);
		return;
	}
	while (i < lengthLine) {
		if (lineBuffer[i] == '$' && lineBuffer[i+1] == '(') {
			styler.ColourTo(startLine + i - 1, state);
			state = 3;
		} else if (state == 3 && lineBuffer[i] == ')') {
			styler.ColourTo(startLine + i, state);
			state = 0;
		}
		if (state == 0 && !bSpecial && (lineBuffer[i] == ':' || lineBuffer[i] == '=')) {
			styler.ColourTo(startLine + i - 1, state);
			styler.ColourTo(startLine + i, 4);
			bSpecial = true;	// Only react to the first '=' or ':' of the line
		}
		i++;
	}
	if (state == 3)
		styler.ColourTo(endPos, 9);
	else
		styler.ColourTo(endPos, 0);
}

static void ColouriseMakeDoc(unsigned int startPos, int length, int, WordList *[], Accessor &styler) {
	char lineBuffer[1024];
	styler.StartAt(startPos);
	styler.StartSegment(startPos);
	unsigned int linePos = 0, startLine = startPos;
	for (unsigned int i = startPos; i <= startPos + length; i++) {
		lineBuffer[linePos++] = styler[i];
		if (styler[i] == '\r' || styler[i] == '\n' || (linePos >= sizeof(lineBuffer) - 1)) {
			lineBuffer[linePos] = '\0';
			ColouriseMakeLine(lineBuffer, linePos, startLine, i, styler);
			linePos = 0;
			startLine = i+1;
		}
	}
	if (linePos > 0)
		ColouriseMakeLine(lineBuffer, linePos, startLine, startPos + length, styler);
}

static void ColouriseErrorListLine(char *lineBuffer,
								   unsigned int lengthLine,
								   unsigned int endPos,
								   Accessor &styler) {
	if (lineBuffer[0] == '>') {
		// Command or return status
		styler.ColourTo(endPos, SCE_ERR_CMD);
	} else if (strstr(lineBuffer, "File \"") && strstr(lineBuffer, ", line ")) {
		styler.ColourTo(endPos, SCE_ERR_PYTHON);
	} else if (0 == strncmp(lineBuffer, "Error ", strlen("Error "))) {
		// Borland error message
		styler.ColourTo(endPos, SCE_ERR_BORLAND);
	} else if (0 == strncmp(lineBuffer, "Warning ", strlen("Warning "))) {
		// Borland warning message
		styler.ColourTo(endPos, SCE_ERR_BORLAND);
	} else if (strstr(lineBuffer, " at "  ) &&
		strstr(lineBuffer, " at "  ) < lineBuffer+lengthLine &&
		strstr(lineBuffer, " line ") &&
		strstr(lineBuffer, " line ") < lineBuffer+lengthLine) {
		// perl error message
		styler.ColourTo(endPos, SCE_ERR_PERL);
	} else {
		// Look for <filename>:<line>:message
		// Look for <filename>(line)message
		// Look for <filename>(line,pos)message
		int state = 0;
		for (unsigned int i = 0; i < lengthLine; i++) {
			if (state == 0 && lineBuffer[i] == ':' && isdigit(lineBuffer[i + 1])) {
				state = 1;
			} else if (state == 0 && lineBuffer[i] == '(') {
				state = 10;
			} else if (state == 1 && isdigit(lineBuffer[i])) {
				state = 2;
			} else if (state == 2 && lineBuffer[i] == ':') {
				state = 3;
				break;
			} else if (state == 2 && !isdigit(lineBuffer[i])) {
				state = 99;
			} else if (state == 10 && isdigit(lineBuffer[i])) {
				state = 11;
			} else if (state == 11 && lineBuffer[i] == ',') {
				state = 14;
			} else if (state == 11 && lineBuffer[i] == ')') {
				state = 12;
			} else if (state == 12 && lineBuffer[i] == ':') {
				state = 13;
			} else if (state == 14 && lineBuffer[i] == ')') {
				state = 15;
				break;
			} else if (((state == 11) || (state == 14)) && !((lineBuffer[i] == ' ') || isdigit(lineBuffer[i]))) {
				state = 99;
			}
		}
		if (state == 3) {
			styler.ColourTo(endPos, SCE_ERR_GCC);
		} else if ((state == 13) || (state == 14) || (state == 15)) {
			styler.ColourTo(endPos, SCE_ERR_MS);
		} else {
			styler.ColourTo(endPos, SCE_ERR_DEFAULT);
		}
	}
}

static void ColouriseErrorListDoc(unsigned int startPos, int length, int, WordList *[], Accessor &styler) {
	char lineBuffer[1024];
	styler.StartAt(startPos);
	styler.StartSegment(startPos);
	unsigned int linePos = 0;
	for (unsigned int i = startPos; i <= startPos + length; i++) {
		lineBuffer[linePos++] = styler[i];
		if (styler[i] == '\r' || styler[i] == '\n' || (linePos >= sizeof(lineBuffer) - 1)) {
			ColouriseErrorListLine(lineBuffer, linePos, i, styler);
			linePos = 0;
		}
	}
	if (linePos > 0)
		ColouriseErrorListLine(lineBuffer, linePos, startPos + length, styler);
}

static int isSpecial(char s) {

	return (s == '\\') || (s == ',') || (s == ';') || (s == '\'') || (s == ' ') ||
	       (s == '\"') || (s == '`') || (s == '^') || (s == '~');
}

static int isTag(int start, Accessor &styler) {

	char s[6];
	unsigned int i = 0, e=1;
	while (i < 5 && e) {
		s[i] = styler[start + i];
		i++;
		e = styler[start + i] != '{';
	}
	s[i] = '\0';
	return (strcmp(s, "begin") == 0) || (strcmp(s, "end") == 0);
}

static void ColouriseLatexDoc(unsigned int startPos, int length, int initStyle,
			   WordList *[], Accessor &styler) {

	styler.StartAt(startPos);

	int state = initStyle;
	char chNext = styler[startPos];
	styler.StartSegment(startPos);
	int lengthDoc = startPos + length;

	for (int i = startPos; i < lengthDoc; i++) {
		char ch = chNext;
		chNext = styler.SafeGetCharAt(i + 1);

		if (styler.IsLeadByte(ch)) {
			chNext = styler.SafeGetCharAt(i + 2);
			i++;
			continue;
		}
		switch(state) {
			case SCE_L_DEFAULT :
				switch(ch) {
					case '\\' :
						styler.ColourTo(i - 1, state);
						if (isSpecial(styler[i + 1])) {
							styler.ColourTo(i + 1, SCE_L_COMMAND);
							i++;
							chNext = styler.SafeGetCharAt(i + 1);
						}
						else {
						    if (isTag(i+1, styler))
							state = SCE_L_TAG;
						    else
						    	state = SCE_L_COMMAND;
						}
						break;
					case '$' :
						styler.ColourTo(i - 1, state);
						state = SCE_L_MATH;
						if (chNext == '$') {
							i++;
							chNext = styler.SafeGetCharAt(i + 1);
						}
						break;
					case '%' :
						styler.ColourTo(i - 1, state);
						state = SCE_L_COMMENT;
						break;
				}
				break;
			case SCE_L_COMMAND :
				if (chNext == '[' || chNext == '{' || chNext == '}' ||
				    chNext == ' ' || chNext == '\r' || chNext == '\n') {
					styler.ColourTo(i, state);
					state = SCE_L_DEFAULT;
					i++;
					chNext = styler.SafeGetCharAt(i + 1);
				}
				break;
			case SCE_L_TAG :
				if (ch == '}') {
					styler.ColourTo(i, state);
					state = SCE_L_DEFAULT;
				}
				break;
			case SCE_L_MATH :
				if (ch == '$') {
					if (chNext == '$') {
						i++;
						chNext = styler.SafeGetCharAt(i + 1);
					}
					styler.ColourTo(i, state);
					state = SCE_L_DEFAULT;
				}
				break;
			case SCE_L_COMMENT :
				if (ch == '\r' || ch == '\n') {
					styler.ColourTo(i - 1, state);
					state = SCE_L_DEFAULT;
				}
		}
	}
	styler.ColourTo(lengthDoc, state);
}

LexerModule lmBatch(SCLEX_BATCH, ColouriseBatchDoc);
LexerModule lmDiff(SCLEX_DIFF, ColouriseDiffDoc);
LexerModule lmProps(SCLEX_PROPERTIES, ColourisePropsDoc);
LexerModule lmMake(SCLEX_MAKEFILE, ColouriseMakeDoc);
LexerModule lmErrorList(SCLEX_ERRORLIST, ColouriseErrorListDoc);
LexerModule lmLatex(SCLEX_LATEX, ColouriseLatexDoc);