// SciTE - Scintilla based Text Editor
/** @file LexAVE.cxx
 ** Lexer for Avenue.
 **/
// Copyright 1998-2001 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"

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

	WordList &keywords = *keywordlists[0];

	styler.StartAt(startPos);

	bool fold = styler.GetPropertyInt("fold") != 0;
	int lineCurrent = styler.GetLine(startPos);
	int levelPrev = styler.LevelAt(lineCurrent) & SC_FOLDLEVELNUMBERMASK;
	int levelCurrent = levelPrev;

	int state = initStyle;
	if (state == SCE_AVE_STRINGEOL)	// Does not leak onto next line
		state = SCE_AVE_DEFAULT;
	char chNext = styler[startPos];
	unsigned int lengthDoc = startPos + length;
	int visibleChars = 0;
	styler.StartSegment(startPos);

	for (unsigned int i = startPos; i < lengthDoc; i++) {
		char ch = chNext;
		chNext = styler.SafeGetCharAt(i + 1);
		if ((ch == '\r' && chNext != '\n') || (ch == '\n')) {
			// Trigger on CR only (Mac style) or either on LF from CR+LF (Dos/Win) or on LF alone (Unix)
			// Avoid triggering two times on Dos/Win
			// End of line
			if (state == SCE_AVE_STRINGEOL) {
				styler.ColourTo(i, state);
				state = SCE_AVE_DEFAULT;
			}
			if (fold) {
				int lev = levelPrev;
				if (visibleChars == 0)
					lev |= SC_FOLDLEVELWHITEFLAG;
				if ((levelCurrent > levelPrev) && (visibleChars > 0))
					lev |= SC_FOLDLEVELHEADERFLAG;
				styler.SetLevel(lineCurrent, lev);
				lineCurrent++;
				levelPrev = levelCurrent;
			}
			visibleChars = 0;
		}
		if (!isspace(ch))
			visibleChars++;
		if (styler.IsLeadByte(ch)) {
			chNext = styler.SafeGetCharAt(i + 2);
			i += 1;
			continue;
		}

		if (state == SCE_AVE_DEFAULT) {
			if (iswordstart(ch) || (ch == '.') )  {
				styler.ColourTo(i-1, state);
				state = SCE_AVE_IDENTIFIER;
			} else if (ch == '\'') {
				styler.ColourTo(i-1, state);
				state = SCE_AVE_COMMENT;
			} else if (ch == '\"') {
				styler.ColourTo(i-1, state);
				state = SCE_AVE_STRING;
			} else if (ch == '#') {
				styler.ColourTo(i-1, state);
				state = SCE_AVE_ENUM;
			} else if (isoperator(ch) ) {
				styler.ColourTo(i-1, state);
				styler.ColourTo(i, SCE_AVE_OPERATOR);
			}
		}
	  	else if (state == SCE_AVE_COMMENT) {
			if (ch == '\r' || ch == '\n') {
				styler.ColourTo(i-1, state);
				state = SCE_AVE_DEFAULT;
			}
		}
	  	else if (state == SCE_AVE_ENUM) {
			if (isoperator(ch)  || ch == ' ' || ch == '\'' || ch == '\r' || ch == '\n') {
				styler.ColourTo(i-1, state);
				state = SCE_AVE_DEFAULT;
			}
		}
		else if (state == SCE_AVE_STRING) {
			if (ch == '\"') {
				if (chNext == '\"') {
					i++;
					ch = chNext;
					chNext = styler.SafeGetCharAt(i + 1);
				} else
				{
					styler.ColourTo(i, state);
					state = SCE_AVE_DEFAULT;
				}
			} else if (chNext == '\r' || chNext == '\n') {
				styler.ColourTo(i-1, SCE_AVE_STRINGEOL);
				state = SCE_AVE_STRINGEOL;
			}
		}
		if ((state == SCE_AVE_IDENTIFIER)) {
			if (!iswordchar(ch) || ch == '.' ) {
				char s[100];
				unsigned int start = styler.GetStartSegment();
				unsigned int end = i - 1;
				for (unsigned int ii = 0; ii < end - start + 1 && ii < 30; ii++) 	{
					s[ii] = static_cast<char>(tolower(styler[start + ii]));
					s[ii + 1] = '\0';
				}

				char chAttr = SCE_AVE_IDENTIFIER;

				if (isdigit(s[0]))
					chAttr = SCE_AVE_NUMBER;
				else {
					if ((strcmp(s, "for") == 0) || (strcmp(s, "if") == 0) || (strcmp(s, "while") == 0))
					{
						levelCurrent +=1;
						chAttr = SCE_AVE_STATEMENT;
					}

					if (strcmp(s, "end") == 0)
					{
						levelCurrent -=1;
						chAttr = SCE_AVE_STATEMENT;
					}

					if ( (strcmp(s, "then") == 0) ||  (strcmp(s, "else") == 0)       || (strcmp(s, "break") == 0) ||
						(strcmp(s, "each") == 0) ||
						(strcmp(s, "exit") == 0) ||  (strcmp(s, "continue") == 0) || (strcmp(s, "return") == 0) ||
						(strcmp(s, "by") == 0)   ||  (strcmp(s, "in") == 0)          || (strcmp(s, "elseif") == 0))
					{
						chAttr = SCE_AVE_STATEMENT;
					}

					if ((strcmp(s, "av") == 0) || (strcmp(s, "self") == 0))
					{
						chAttr = SCE_AVE_KEYWORD;
					}

					if (keywords.InList(s))
					{
						chAttr = SCE_AVE_WORD;
					}
				}
				styler.ColourTo(end, chAttr);
				state = SCE_AVE_DEFAULT;

				if (ch == '\'') {
					state = SCE_AVE_COMMENT;
				} else if (ch == '\"') {
					state = SCE_AVE_STRING;
				} else if (isoperator(ch)) {
					styler.ColourTo(i, SCE_AVE_OPERATOR);
				}
			}
		}

	}
	styler.ColourTo(lengthDoc - 1, state);

	// Fill in the real level of the next line, keeping the current flags as they will be filled in later
	if (fold) {
		int flagsNext = styler.LevelAt(lineCurrent) & ~SC_FOLDLEVELNUMBERMASK;
		styler.SetLevel(lineCurrent, levelPrev | flagsNext);

	}
}

LexerModule lmAVE(SCLEX_AVE, ColouriseAveDoc, "ave");