evil-qemacs/nim.c
2020-10-02 21:32:29 +00:00

334 lines
11 KiB
C

/*
* Nim mode for QEmacs.
*
* Copyright (c) 2015-2017 Charlie Gordon.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include "qe.h"
#define MAX_KEYWORD_SIZE 16
/*---------------- Nim coloring ----------------*/
static char const nim_keywords[] = {
/* Nim keywords */
"addr|and|as|asm|atomic|bind|block|break|case|cast|concept|const|"
"continue|converter|defer|discard|distinct|div|do|elif|else|end|"
"enum|except|export|finally|for|from|func|generic|if|import|in|include|"
"interface|is|isnot|iterator|let|macro|method|mixin|mod|nil|not|notin|"
"object|of|or|out|proc|ptr|raise|ref|return|shl|shr|static|template|"
"try|tuple|type|using|var|when|while|with|without|xor|yield|"
/* predefined operators */
"inc|dec|"
/* predefined constants */
"true|false|"
};
static char const nim_types[] = {
"int|uint|cint|cuint|clong|cstring|string|char|byte|bool|"
"openArray|seq|array|void|pointer|float|csize|cdouble|"
"cchar|cschar|cshort|cu|nil|expr|stmt|typedesc|auto|any|"
"range|openarray|varargs|set|cfloat|"
"int8|int16|int32|int64|uint8|uint16|uint32|uint64|"
};
enum {
NIM_STYLE_TEXT = QE_STYLE_DEFAULT,
NIM_STYLE_PREPROCESS = QE_STYLE_PREPROCESS,
NIM_STYLE_COMMENT = QE_STYLE_COMMENT,
NIM_STYLE_STRING = QE_STYLE_STRING,
NIM_STYLE_NUMBER = QE_STYLE_NUMBER,
NIM_STYLE_KEYWORD = QE_STYLE_KEYWORD,
NIM_STYLE_TYPE = QE_STYLE_TYPE,
NIM_STYLE_FUNCTION = QE_STYLE_FUNCTION,
NIM_STYLE_PRAGMA = QE_STYLE_PREPROCESS,
};
/* nim-mode colorization states */
enum {
IN_NIM_COMMENT = 0x80,
IN_NIM_CHARLIT = 0x40,
IN_NIM_STRING = 0x20,
IN_NIM_LONG_STRING = 0x10,
IN_NIM_RAW_STRING = 0x08,
IN_NIM_STRING_BQ = 0x04,
IN_NIM_PRAGMA = 0x02,
};
static void nim_colorize_line(QEColorizeContext *cp,
unsigned int *str, int n, ModeDef *syn)
{
int i = 0, start = i, style = 0, c, sep = 0, klen;
int state = cp->colorize_state;
char kbuf[64];
if (state & IN_NIM_COMMENT) {
goto parse_comment;
}
if (state & IN_NIM_CHARLIT) {
sep = '\'';
goto parse_string;
}
if (state & IN_NIM_STRING) {
sep = '\"';
goto parse_string;
}
if (state & IN_NIM_LONG_STRING) {
sep = '\"';
goto parse_long_string;
}
if (state & IN_NIM_STRING_BQ) {
sep = '`';
goto parse_string;
}
while (i < n) {
start = i;
c = str[i++];
switch (c) {
case '#':
if (start == 0 && str[i] == '!') {
i = n;
style = NIM_STYLE_PREPROCESS;
break;
}
parse_comment:
state &= ~IN_NIM_COMMENT;
/* XXX: should handle trailing backslash more generically */
for (; i < n; i++) {
if (str[i] == '\\')
state |= IN_NIM_COMMENT;
else
if (!qe_isblank(str[i]))
state &= ~IN_NIM_COMMENT;
}
style = NIM_STYLE_COMMENT;
break;
#if 0
case 'r':
case 'R':
if (str[i] == '\"') {
state |= IN_NIM_RAW_STRING;
goto has_quote;
}
goto has_alpha;
#endif
case '`':
sep = c;
style = IN_NIM_STRING_BQ;
goto parse_string;
case '\'':
sep = c;
style = IN_NIM_CHARLIT;
goto parse_string;
case '\"':
/* parse string const */
i--;
has_quote:
sep = str[i++];
if (str[i] == (unsigned int)sep && str[i + 1] == (unsigned int)sep) {
/* long string */
state |= IN_NIM_LONG_STRING | IN_NIM_RAW_STRING;
i += 2;
parse_long_string:
while (i < n) {
c = str[i++];
if (!(state & IN_NIM_RAW_STRING) && c == '\\') {
if (i < n) {
i += 1;
}
} else
if (c == sep && str[i] == (unsigned int)sep
&& str[i + 1] == (unsigned int)sep && str[i + 2] != (unsigned int)sep) {
i += 2;
state &= ~(IN_NIM_LONG_STRING | IN_NIM_RAW_STRING);
break;
}
}
} else {
state = IN_NIM_STRING;
parse_string:
while (i < n) {
c = str[i++];
if (!(state & IN_NIM_RAW_STRING) && c == '\\') {
if (i < n) {
i += 1;
}
continue;
}
if (c == sep) {
if ((state & IN_NIM_RAW_STRING) && str[i] == '"') {
i += 1;
continue;
}
state &= ~(IN_NIM_STRING | IN_NIM_RAW_STRING);
break;
}
}
}
style = NIM_STYLE_STRING;
break;
case '.':
if (str[i] == '}') {
i += 1;
state &= ~IN_NIM_PRAGMA;
style = NIM_STYLE_PRAGMA;
break;
}
continue;
break;
// Handle ( ) { } [ ] (. .) {. .} [. .] : :: .. `
case '{':
if (str[i] == '.' && str[i + 1] != '.') {
/* Nim pragmas */
for (i++;; i++) {
if (qe_isalnum_(str[i]))
continue;
if (str[i] == '.' && str[i + 1] != '}')
continue;
break;
}
state |= IN_NIM_PRAGMA;
style = NIM_STYLE_PRAGMA;
break;
}
continue;
default:
if (qe_isdigit(c)) {
int j, k;
static const char * const suffixes[] = {
"i8", "i16", "i32", "i64", "u8", "u16", "u32", "u64",
"f32", "f64", "f128",
};
if (c == '0' && qe_match2(str[i], 'b', 'B')) {
/* binary numbers */
for (i += 1; qe_isbindigit_(str[i]); i++)
continue;
} else
if (c == '0' && (str[i] == 'o' || qe_match2(str[i], 'c', 'C'))) {
/* octal numbers */
for (i += 1; qe_isoctdigit_(str[i]); i++)
continue;
} else
if (c == '0' && qe_match2(str[i], 'x', 'X')) {
/* hexadecimal numbers */
for (i += 1; qe_isxdigit_(str[i]); i++)
continue;
} else {
/* decimal numbers */
for (; qe_isdigit_(str[i]); i++)
continue;
if (str[i] == '.' && qe_isdigit_(str[i + 1])) {
i++;
/* decimal floats require a digit after the '.' */
for (; qe_isdigit_(str[i]); i++)
continue;
}
if (qe_match2(str[i], 'e', 'E')) {
int k = i + 1;
if (qe_match2(str[i], '+', '-'))
k++;
if (qe_isdigit(str[k])) {
for (i = k + 1; qe_isdigit_(str[i]); i++)
continue;
}
}
}
/* handle optional ' and type suffix */
j = i;
if (str[j] == '\'')
j++;
if (qe_isalpha(str[j])) {
for (k = 0; k < countof(suffixes); k++) {
if (ustrstart(str + j, suffixes[k], &klen)
&& !qe_isalnum_(str[j + klen])) {
i = j + klen;
break;
}
}
}
/* XXX: should detect malformed number constants */
style = NIM_STYLE_NUMBER;
break;
}
//has_alpha:
if (qe_isalpha_(c)) {
i += ustr_get_identifier(kbuf, countof(kbuf), c, str, i, n);
if (str[i] == '"') {
/* generalized raw string literal */
state |= IN_NIM_RAW_STRING;
goto has_quote;
}
if (strfind(syn->keywords, kbuf)) {
style = NIM_STYLE_KEYWORD;
break;
}
if ((start == 0 || str[start - 1] != '.')
&& (str[i] != '.')) {
if (strfind(syn->types, kbuf)) {
//|| (qe_isupper(c) && haslower)
style = NIM_STYLE_TYPE;
break;
}
}
if (check_fcall(str, i)) {
style = NIM_STYLE_FUNCTION;
break;
}
continue;
}
continue;
}
if (style) {
SET_COLOR(str, start, i, style);
style = 0;
}
}
// XXX: should handle line continuation with trailing \\ followed
// by white space
SET_COLOR1(str, n, style); /* set style on eol char */
cp->colorize_state = state;
}
static ModeDef nim_mode = {
.name = "Nim",
.extensions = "nim",
.shell_handlers = "nim",
.keywords = nim_keywords,
.types = nim_types,
.colorize_func = nim_colorize_line,
};
static int nim_init(void)
{
qe_register_mode(&nim_mode, MODEF_SYNTAX);
return 0;
}
qe_module_init(nim_init);