evil-qemacs/script.c
2020-10-02 20:06:44 +00:00

359 lines
11 KiB
C

/*
* Shell script mode for QEmacs.
*
* Copyright (c) 2000-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"
/*---------------- Shell script colors ----------------*/
enum {
SHELL_SCRIPT_STYLE_TEXT = QE_STYLE_DEFAULT,
SHELL_SCRIPT_STYLE_COMMENT = QE_STYLE_COMMENT,
SHELL_SCRIPT_STYLE_PREPROCESS = QE_STYLE_PREPROCESS,
SHELL_SCRIPT_STYLE_COMMAND = QE_STYLE_FUNCTION,
SHELL_SCRIPT_STYLE_VARIABLE = QE_STYLE_TYPE,
SHELL_SCRIPT_STYLE_STRING = QE_STYLE_STRING,
SHELL_SCRIPT_STYLE_OP = QE_STYLE_KEYWORD,
SHELL_SCRIPT_STYLE_KEYWORD = QE_STYLE_KEYWORD,
};
static const char shell_script_keywords[] = {
/* reserved words */
"if|then|elif|else|fi|case|esac|for|while|until|do|done|shift|break|"
"function|return|export|alias|in|select|time|"
/* internal commands */
//"cd|echo|umask|"
};
static int shell_script_get_var(char *buf, int buf_size,
unsigned int *str, int j, int n)
{
int i = 0, c;
while (j < n && qe_isalnum_(c = str[j])) {
if (i < buf_size - 1) {
buf[i++] = c;
}
j++;
}
if (i < buf_size) {
buf[i] = '\0';
}
return j;
}
static int shell_script_has_sep(unsigned int *str, int i, int n) {
return i >= n || qe_findchar(" \t<>|&;()", str[i]);
}
static int shell_script_string(unsigned int *str, int i, int n,
int sep, int escape, int dollar)
{
while (i < n) {
int c = str[i++];
if (c == '\\' && escape && i < n) {
i++;
} else
if (c == '$' && dollar && i < n) {
/* XXX: should highlight variable substitutions */
i++;
} else
if (c == sep) {
break;
}
}
return i;
}
static void shell_script_colorize_line(QEColorizeContext *cp,
unsigned int *str, int n, ModeDef *syn)
{
int i = 0, j, c, start, style, bits = 0;
/* special case sh-bang line */
if (n >= 2 && str[0] == '#' && str[1] == '!') {
SET_COLOR(str, 0, n, SHELL_SCRIPT_STYLE_PREPROCESS);
return;
}
start_cmd:
style = SHELL_SCRIPT_STYLE_COMMAND;
while (i < n && qe_isblank(str[i])) {
i++;
}
while (i < n) {
start = i;
switch (c = str[i++]) {
case '#':
style = SHELL_SCRIPT_STYLE_COMMENT;
i = n;
break;
case '`':
/* XXX: should be a state */
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
goto start_cmd;
case '\'':
i = shell_script_string(str, i, n, c, 0, 0);
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_STRING);
/* XXX: should support multi-line strings? */
continue;
case '"':
i = shell_script_string(str, i, n, c, 1, 1);
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_STRING);
/* XXX: should support multi-line strings? */
continue;
case '\\':
if (i >= n) {
/* Should keep state for next line */
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
continue;
}
/* Do not interpret the next character */
i++;
break;
case '$':
if (i == n || qe_findchar(" \t\"", str[i]))
break;
SET_COLOR1(str, start++, SHELL_SCRIPT_STYLE_OP);
switch (c = str[i++]) {
case '\'':
i = shell_script_string(str, i, n, c, 1, 0);
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_STRING);
continue;
case '(': /* expand command output */
bits = (bits << 2) | 1;
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
goto start_cmd;
case '[': /* expression */
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
for (j = i; i < n; i++) {
if (str[i] == ']')
break;
}
SET_COLOR(str, j, i, SHELL_SCRIPT_STYLE_TEXT);
if (i < n) {
i++;
SET_COLOR(str, i - 1, i, SHELL_SCRIPT_STYLE_OP);
}
continue;
case '{': /* variable substitution with options */
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
/* XXX: should parse variable name or single char */
/* XXX: should support % syntax with regex */
for (j = i; i < n; i++) {
if (str[i] == '}')
break;
}
SET_COLOR(str, j, i, SHELL_SCRIPT_STYLE_VARIABLE);
if (i < n) {
i++;
SET_COLOR(str, i - 1, i, SHELL_SCRIPT_STYLE_OP);
}
continue;
case '$':
case '?':
case '#':
default:
if (qe_isalpha_(c)) {
i = shell_script_get_var(NULL, 0, str, i, n);
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_VARIABLE);
} else {
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
}
continue;
}
continue;
case ' ':
case '\t':
style = SHELL_SCRIPT_STYLE_TEXT;
break;
case '{': /* compound command */
case '}':
/* XXX: should support numeric enumerations */
if (i == n || qe_isblank(str[i])) {
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_OP);
goto start_cmd;
}
style = SHELL_SCRIPT_STYLE_TEXT;
break;
case '>':
case '<':
// XXX: Should support other punctuation syntaxes
if (str[i] == (unsigned int)c) { /* handle >> and << */
i++;
}
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_OP);
// XXX: Should support << syntax
style = SHELL_SCRIPT_STYLE_TEXT;
continue;
case '|':
case '&':
if (str[i] == (unsigned int)c) { /* handle || and && */
i++;
}
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_OP);
goto start_cmd;
case ';':
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
goto start_cmd;
case '(':
bits = (bits << 2) | 2;
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
goto start_cmd;
case ')':
bits = (bits >> 2);
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
goto start_cmd;
case '[':
if (style == SHELL_SCRIPT_STYLE_COMMAND) {
bits = (bits << 2) | 3;
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
style = SHELL_SCRIPT_STYLE_TEXT;
continue;
}
break;
case ']':
if ((bits & 3) == 3) {
bits = (bits >> 2);
SET_COLOR1(str, start, SHELL_SCRIPT_STYLE_OP);
style = SHELL_SCRIPT_STYLE_TEXT;
continue;
}
break;
default:
if (style == SHELL_SCRIPT_STYLE_COMMAND && qe_isalpha_(c)) {
char kbuf[64];
i = shell_script_get_var(kbuf, sizeof kbuf, str, i - 1, n);
if (shell_script_has_sep(str, i, n) && strfind(syn->keywords, kbuf)) {
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_KEYWORD);
if (!strfind("for|case|export|in", kbuf))
goto start_cmd;
else
continue;
}
if (str[i] == '=') {
SET_COLOR(str, start, i, SHELL_SCRIPT_STYLE_VARIABLE);
SET_COLOR1(str, i, SHELL_SCRIPT_STYLE_OP);
i++;
style = SHELL_SCRIPT_STYLE_TEXT;
continue;
}
}
break;
}
SET_COLOR(str, start, i, style);
}
}
static int shell_script_mode_probe(ModeDef *mode, ModeProbeData *p)
{
/* Match on file extension, shbang line and .bashrc .bash_history... */
if (match_extension(p->filename, mode->extensions)
|| match_shell_handler(cs8(p->buf), mode->shell_handlers)
|| (*p->filename == '.'
&& stristart(p->filename + 1, mode->extensions, NULL))) {
return 82;
}
if (stristart(p->filename, ".profile", NULL)) {
/* XXX: Should check the user login shell */
return 80;
}
if (p->buf[0] == '#') {
if (p->buf[1] == '!')
return 60;
if (p->buf[1] == ' ')
return 25;
}
return 1;
}
/* XXX: should have shell specific variations */
static ModeDef sh_mode = {
.name = "Shell",
.alt_name = "sh",
.extensions = "sh",
.shell_handlers = "sh",
.mode_probe = shell_script_mode_probe,
.colorize_func = shell_script_colorize_line,
.keywords = shell_script_keywords,
};
static ModeDef bash_mode = {
.name = "bash",
.extensions = "bash",
.shell_handlers = "bash",
.mode_probe = shell_script_mode_probe,
.colorize_func = shell_script_colorize_line,
.keywords = shell_script_keywords,
};
static ModeDef csh_mode = {
.name = "csh",
.extensions = "csh",
.shell_handlers = "csh",
.mode_probe = shell_script_mode_probe,
.colorize_func = shell_script_colorize_line,
.keywords = shell_script_keywords,
};
static ModeDef ksh_mode = {
.name = "ksh",
.extensions = "ksh",
.shell_handlers = "ksh",
.mode_probe = shell_script_mode_probe,
.colorize_func = shell_script_colorize_line,
.keywords = shell_script_keywords,
};
static ModeDef zsh_mode = {
.name = "zsh",
.extensions = "zsh",
.shell_handlers = "zsh",
.mode_probe = shell_script_mode_probe,
.colorize_func = shell_script_colorize_line,
.keywords = shell_script_keywords,
};
static ModeDef tcsh_mode = {
.name = "tcsh",
.extensions = "tcsh",
.shell_handlers = "tcsh",
.mode_probe = shell_script_mode_probe,
.colorize_func = shell_script_colorize_line,
.keywords = shell_script_keywords,
};
static int shell_script_init(void)
{
qe_register_mode(&sh_mode, MODEF_SYNTAX);
qe_register_mode(&bash_mode, MODEF_SYNTAX);
qe_register_mode(&csh_mode, MODEF_SYNTAX);
qe_register_mode(&ksh_mode, MODEF_SYNTAX);
qe_register_mode(&zsh_mode, MODEF_SYNTAX);
qe_register_mode(&tcsh_mode, MODEF_SYNTAX);
return 0;
}
qe_module_init(shell_script_init);