/* Based on https://github.com/antirez/kilo * * ----------------------------------------------------------------------- * * Copyright (C) 2016 Salvatore Sanfilippo * * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are * met: * * * Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #define KILO_VERSION "0.0.1" #ifdef __linux__ #define _POSIX_C_SOURCE 200809L #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #include "lua.h" #include "teliva.h" /* Syntax highlight types */ #define HL_NORMAL 0 #define HL_NONPRINT 1 #define HL_COMMENT 2 /* Single line comment. */ #define HL_MLCOMMENT 3 /* Multi-line comment. */ #define HL_KEYWORD1 4 #define HL_KEYWORD2 5 #define HL_STRING 6 #define HL_NUMBER 7 #define HL_MATCH 8 /* Search match. */ struct editorSyntax { char **keywords; char singleline_comment_start[2]; char *multiline_comment_start; char *multiline_comment_end; }; /* This structure represents a single line of the file we are editing. */ typedef struct erow { int idx; /* Row index in the file, zero-based. */ int size; /* Size of the row, excluding the null term. */ int rsize; /* Size of the rendered row. */ char *chars; /* Row content. */ char *render; /* Row content "rendered" for screen (for TABs). */ unsigned char *hl; /* Syntax highlight type for each character in render.*/ int hl_oc; /* Row had open comment at end in last syntax highlight check. */ } erow; typedef struct hlcolor { int r,g,b; } hlcolor; #define LINE_NUMBER_SPACE 4 struct editorConfig { int cx,cy; /* Cursor x and y position in characters */ int cols; /* viewport width */ int rowoff; /* Offset of row displayed. */ int coloff; /* Offset of column displayed. */ int numrows; /* Number of rows */ erow *row; /* Rows */ int dirty; /* File modified but not saved. */ char *filename; /* Currently open filename */ struct editorSyntax *syntax; /* Current syntax highlight, or NULL. */ }; static struct editorConfig E; /* =========================== Syntax highlights DB ========================= * * In order to add a new syntax, define two arrays with a list of file name * matches and keywords. The file name matches are used in order to match * a given syntax with a given file name: if a match pattern starts with a * dot, it is matched as the last past of the filename, for example ".c". * Otherwise the pattern is just searched inside the filenme, like "Makefile"). * *

; To demonstrate generic functions, we'll construct a factorial function with
; separate base and recursive clauses. Compare factorial.mu.

; factorial n = n*factorial(n-1)
(function factorial [
  (default-space:space-address <- new space:literal 30:literal)
  (n:integer <- input 0:literal)
  (x:integer <- subtract n:integer 1:literal)
  (subresult:integer <- factorial x:integer)
  (result:integer <- multiply subresult:integer n:integer)
  (reply result:integer)
])

; factorial 0 = 1
(function factorial [
  (default-space:space-address <- new space:literal 30:literal)
  (n:integer <- input 0:literal)
  { begin
    (zero?:boolean <- equal n:integer 0:literal)
    (break-unless zero?:boolean)
    (reply 1:literal)
  }
])

(function main [
  (1:integer <- factorial 5:literal)
  ($print (("result: " literal)))
  (print-integer nil:literal/terminal 1:integer)
  ($print (("\n" literal)))
])

case of column 0, we need to move the current line * on the right of the previous one. */ filecol = E.row[filerow-1].size; editorRowAppendString(&E.row[filerow-1],row->chars,row->size); editorDelRow(filerow); row = NULL; if (E.cy == 0) E.rowoff--; else E.cy--; E.cx = filecol; if (E.cx >= E.cols) { int shift = (E.cols-E.cx)+1; E.cx -= shift; E.coloff += shift; } } else { editorRowDelChar(row,filecol-1); if (E.cx == 0 && E.coloff) E.coloff--; else E.cx--; } if (row) editorUpdateRow(row); E.dirty++; } static void editorUncommentCursorRow() { erow *row = &E.row[E.rowoff+E.cy]; E.coloff = 0; E.cx = 4; editorUpdateRow(row); editorDelChar(); editorDelChar(); editorDelChar(); editorDelChar(); } static void editorCommentCursorRow() { erow *row = &E.row[E.rowoff+E.cy]; editorRowInsertChar(row, 0, ' '); editorRowInsertChar(row, 0, '?'); editorRowInsertChar(row, 0, '-'); editorRowInsertChar(row, 0, '-'); E.coloff = 0; E.cx = 0; } /* Load the specified program in the editor memory and returns 0 on success * or 1 on error. */ int editorOpen(char *filename) { FILE *fp; E.dirty = 0; free(E.filename); size_t fnlen = strlen(filename)+1; E.filename = malloc(fnlen); memcpy(E.filename,filename,fnlen); fp = fopen(filename,"r"); if (!fp) { if (errno != ENOENT) { perror("Opening file"); exit(1); } return 1; } char *line = NULL; size_t linecap = 0; ssize_t linelen; while((linelen = getline(&line,&linecap,fp)) != -1) { if (linelen && (line[linelen-1] == '\n' || line[linelen-1] == '\r')) line[--linelen] = '\0'; editorInsertRow(E.numrows,line,linelen); } free(line); fclose(fp); E.dirty = 0; return 0; } /* Save the current file on disk. Return 0 on success, 1 on error. */ static int editorSaveToDisk(void) { int len; char *buf = editorRowsToString(&len); int fd = open(E.filename,O_RDWR|O_CREAT,0644); if (fd == -1) goto writeerr; /* Use truncate + a single write(2) call in order to make saving * a bit safer, under the limits of what we can do in a small editor. */ if (ftruncate(fd,len) == -1) goto writeerr; if (write(fd,buf,len) != len) goto writeerr; close(fd); free(buf); E.dirty = 0; return 0; writeerr: free(buf); if (fd != -1) close(fd); /* TODO: better error handling. */ /* I haven't gotten to this yet since we have version control. */ endwin(); printf("Can't save! I/O error: %s",strerror(errno)); exit(1); return 1; } /* ============================= Terminal update ============================ */ extern char *Previous_error; extern void draw_menu_item(const char* key, const char* name); static void editorMenu(void) { attrset(A_REVERSE); for (int x = 0; x < COLS; ++x) mvaddch(LINES-1, x, ' '); attrset(A_NORMAL); extern int menu_column; menu_column = 2; draw_menu_item("^e", "run"); if (Previous_error != NULL) { attron(A_BOLD); draw_menu_item("^c", "abort"); attroff(A_BOLD); } draw_menu_item("^g", "go"); draw_menu_item("^b", "big picture"); draw_menu_item("^f", "find"); draw_menu_item("^h", "backspace"); draw_menu_item("^l", "end of line"); /* draw_menu_item("^/|^-|^_", "(un)comment line"); */ attroff(A_REVERSE); mvaddstr(LINES-1, menu_column, " ^/"); attron(COLOR_PAIR(COLOR_PAIR_MENU_ALTERNATE)); addstr("|"); attroff(COLOR_PAIR(COLOR_PAIR_MENU_ALTERNATE)); addstr("^-"); attron(COLOR_PAIR(COLOR_PAIR_MENU_ALTERNATE)); addstr("|"); attroff(COLOR_PAIR(COLOR_PAIR_MENU_ALTERNATE)); addstr("^_ "); menu_column += 10; attron(A_REVERSE); mvaddstr(LINES-1, menu_column, " (un)comment line "); menu_column += 18; attrset(A_NORMAL); } extern void draw_string_on_menu (const char* s); static void editorFindMenu(void) { attrset(A_REVERSE); for (int x = 0; x < COLS; ++x) mvaddch(LINES-1, x, ' '); attrset(A_NORMAL); extern int menu_column; menu_column = 2; draw_menu_item("^x", "cancel"); draw_menu_item("Enter", "submit"); draw_menu_item("^h", "backspace"); draw_menu_item("^u", "clear"); /* draw_menu_item("←|↑", "previous"); */ attroff(A_REVERSE); mvaddstr(LINES-1, menu_column, " ←|↑ "); menu_column += 5; /* strlen isn't sufficient */ attron(A_REVERSE); draw_string_on_menu("previous"); /* draw_menu_item("↓|→", "next"); */ attroff(A_REVERSE); mvaddstr(LINES-1, menu_column, " ↓|→ "); menu_column += 5; /* strlen isn't sufficient */ attron(A_REVERSE); draw_string_on_menu("next"); attrset(A_NORMAL); } static void editorGoMenu(void) { attrset(A_REVERSE); for (int x = 0; x < COLS; ++x) mvaddch(LINES-1, x, ' '); attrset(A_NORMAL); extern int menu_column; menu_column = 2; draw_menu_item("^x", "cancel"); draw_menu_item("Enter", "submit"); draw_menu_item("^h", "backspace"); draw_menu_item("^u", "clear"); attrset(A_NORMAL); } extern int render_previous_error(); static void editorRefreshScreen(void (*menu_func)(void)) { int y; erow *r; int current_color = -1; curs_set(0); clear(); attrset(A_NORMAL); /* Draw all line numbers first so they don't mess up curses state later * when rendering lines. */ for (y = 0; y < LINES-1; y++) { int filerow = E.rowoff+y; if (filerow >= E.numrows) { continue; } mvaddstr(y, 0, ""); attron(COLOR_PAIR(COLOR_PAIR_FADE)); printw("%3d ", filerow+1); // LINE_NUMBER_SPACE-1 attroff(COLOR_PAIR(COLOR_PAIR_FADE)); } for (y = 0; y < LINES-1; y++) { int filerow = E.rowoff+y; if (filerow >= E.numrows) { continue; } r = &E.row[filerow]; int len = r->rsize - E.coloff; mvaddstr(y, LINE_NUMBER_SPACE, ""); if (len > 0) { if (len > E.cols) len = E.cols; char *c = r->render+E.coloff; unsigned char *hl = r->hl+E.coloff; int j; for (j = 0; j < len; j++) { if (hl[j] == HL_NONPRINT) { char sym; attron(A_REVERSE); if (c[j] <= 26) sym = '@'+c[j]; else sym = '?'; addch(sym); attroff(A_REVERSE); } else if (hl[j] == HL_NORMAL) { if (current_color != -1) { attrset(A_NORMAL); current_color = -1; } addch(c[j]); } else { int color = editorSyntaxToColorPair(hl[j]); if (color != current_color) { attrset(COLOR_PAIR(color)); current_color = color; } addch(c[j]); } } } } render_previous_error(); (*menu_func)(); /* Put cursor at its current position. Note that the horizontal position * at which the cursor is displayed may be different compared to 'E.cx' * because of TABs. */ int j; int cx = 0; int filerow = E.rowoff+E.cy; erow *row = (filerow >= E.numrows) ? NULL : &E.row[filerow]; if (row) { for (j = E.coloff; j < (E.cx+E.coloff); j++) { if (j < row->size && row->chars[j] == TAB) cx += 7-((cx)%8); cx++; } } mvaddstr(E.cy, cx+LINE_NUMBER_SPACE, ""); curs_set(1); } /* =============================== Find mode ================================ */ #define KILO_QUERY_LEN 256 static void editorFind() { char query[KILO_QUERY_LEN+1] = {0}; int qlen = 0; int last_match = -1; /* Last line where a match was found. -1 for none. */ int find_next = 0; /* if 1 search next, if -1 search prev. */ int saved_hl_line = -1; /* No saved HL */ char *saved_hl = NULL; #define FIND_RESTORE_HL do { \ if (saved_hl) { \ memcpy(E.row[saved_hl_line].hl,saved_hl, E.row[saved_hl_line].rsize); \ free(saved_hl); \ saved_hl = NULL; \ } \ } while (0) /* Save the cursor position in order to restore it later. */ int saved_cx = E.cx, saved_cy = E.cy; int saved_coloff = E.coloff, saved_rowoff = E.rowoff; while(1) { editorRefreshScreen(editorFindMenu); mvprintw(LINES-2, 0, "Find: %s", query); int c = getch(); if (c == KEY_BACKSPACE || c == DELETE || c == CTRL_H) { if (qlen != 0) query[--qlen] = '\0'; last_match = -1; } else if (c == CTRL_X || c == ENTER) { if (c == CTRL_X) { E.cx = saved_cx; E.cy = saved_cy; E.coloff = saved_coloff; E.rowoff = saved_rowoff; } FIND_RESTORE_HL; return; } else if (c == CTRL_U) { qlen = 0; query[qlen] = '\0'; } else if (c == KEY_RIGHT || c == KEY_DOWN) { find_next = 1; } else if (c == KEY_LEFT || c == KEY_UP) { find_next = -1; } else if (isprint(c)) { if (qlen < KILO_QUERY_LEN) { query[qlen++] = c; query[qlen] = '\0'; last_match = -1; } } /* Search occurrence. */ if (last_match == -1) find_next = 1; if (find_next) { char *match = NULL; int match_offset = 0; int i, current = last_match; for (i = 0; i < E.numrows; i++) { current += find_next; if (current == -1) current = E.numrows-1; else if (current == E.numrows) current = 0; match = strstr(E.row[current].render,query); if (match) { match_offset = match-E.row[current].render; break; } } find_next = 0; /* Highlight */ FIND_RESTORE_HL; if (match) { erow *row = &E.row[current]; last_match = current; if (row->hl) { saved_hl_line = current; saved_hl = malloc(row->rsize); memcpy(saved_hl,row->hl,row->rsize); memset(row->hl+match_offset,HL_MATCH,qlen); } E.cy = 0; E.cx = match_offset; E.rowoff = current; E.coloff = 0; /* Scroll horizontally as needed. */ if (E.cx > E.cols) { int diff = E.cx - E.cols; E.cx -= diff; E.coloff += diff; } } } } } /* ========================= Editor events handling ======================== */ static int editorAtStartOfLine() { return E.coloff == 0 && E.cx == 0; } /* Handle cursor position change because arrow keys were pressed. */ static void editorMoveCursor(int key) { int filerow = E.rowoff+E.cy; int filecol = E.coloff+E.cx; int rowlen; erow *row = (filerow >= E.numrows) ? NULL : &E.row[filerow]; switch(key) { case KEY_LEFT: if (E.cx == 0) { if (E.coloff) { E.coloff--; } else { if (filerow > 0) { E.cy--; E.cx = E.row[filerow-1].size; if (E.cx > E.cols-1) { E.coloff = E.cx-E.cols+1; E.cx = E.cols-1; } } } } else { E.cx -= 1; } break; case KEY_RIGHT: if (row && filecol < row->size) { if (E.cx == E.cols-1) { E.coloff++; } else { E.cx += 1; } } else if (row && filecol == row->size) { E.cx = 0; E.coloff = 0; if (E.cy == LINES-1-1) { E.rowoff++; } else { E.cy += 1; } } break; case KEY_UP: if (E.cy == 0) { if (E.rowoff) E.rowoff--; } else { E.cy -= 1; } break; case KEY_DOWN: if (filerow < E.numrows) { if (E.cy == LINES-1-1) { E.rowoff++; } else { E.cy += 1; } } break; } /* Fix cx if the current line has not enough chars. */ filerow = E.rowoff+E.cy; filecol = E.coloff+E.cx; row = (filerow >= E.numrows) ? NULL : &E.row[filerow]; rowlen = row ? row->size : 0; if (filecol > rowlen) { E.cx -= filecol-rowlen; if (E.cx < 0) { E.coloff += E.cx; E.cx = 0; } } } int identifier_char(char c) { /* keep sync'd with llex */ return isalnum(c) || c == '_'; } void word_at_cursor(char* out, int capacity) { erow* row = &E.row[E.rowoff + E.cy]; int cidx = E.coloff + E.cx; int len = 0; memset(out, 0, capacity); if (row == NULL) return; /* scan back to first identifier char */ while (cidx > 0) { --cidx; if (!identifier_char(row->chars[cidx])) { ++cidx; break; } } /* now scan forward */ for (len = 0; cidx+len < row->size; ++len) { if (!identifier_char(row->chars[cidx+len])) break; } if (len < capacity) strncpy(out, &row->chars[cidx], len); } /* Jump to some definition. */ extern void save_to_current_definition_and_editor_buffer(lua_State *L, char *name); extern void load_editor_buffer_to_current_definition_in_image(lua_State *L); extern char Current_definition[]; #define CURRENT_DEFINITION_LEN 256 static void editorGo(lua_State* L) { char query[CURRENT_DEFINITION_LEN+1] = {0}; int qlen = 0; if (strlen(Current_definition) > 0) { /* We're currently editing a definition. Save it. */ editorSaveToDisk(); load_editor_buffer_to_current_definition_in_image(L); } word_at_cursor(query, CURRENT_DEFINITION_LEN); qlen = strlen(query); while(1) { editorRefreshScreen(editorGoMenu); mvprintw(LINES-2, 0, "Go to: %s", query); int c = getch(); if (c == KEY_BACKSPACE || c == DELETE || c == CTRL_H) { if (qlen != 0) query[--qlen] = '\0'; } else if (c == CTRL_X || c == ENTER) { if (c == ENTER) { save_to_current_definition_and_editor_buffer(L, query); clearEditor(); editorOpen("teliva_editor_buffer"); } return; } else if (c == CTRL_U) { qlen = 0; query[qlen] = '\0'; } else if (isprint(c)) { if (qlen < CURRENT_DEFINITION_LEN) { query[qlen++] = c; query[qlen] = '\0'; } } } } /* Process events arriving from the standard input, which is, the user * is typing stuff on the terminal. */ static int Quit = 0; static int Back_to_big_picture = 0; extern void save_editor_state(int rowoff, int coloff, int cy, int cx); static void editorProcessKeypress(lua_State* L) { int c = getch(); //? mvprintw(LINES-3, 60, "key: %d\n", c); //? getch(); switch(c) { case ENTER: { editorInsertNewline(); /* auto-indent */ erow* prevrow = &E.row[E.rowoff + E.cy - 1]; for (int x = 0; x < prevrow->size && prevrow->chars[x] == ' '; ++x) editorInsertChar(' '); } break; case CTRL_C: if (Previous_error != NULL) exit(1); break; case CTRL_E: /* Save and quit. */ editorSaveToDisk(); save_editor_state(E.rowoff, E.coloff, E.cy, E.cx); Quit = 1; break; case CTRL_G: /* Go to a different definition. */ editorGo(L); break; case CTRL_B: /* Go to big-picture view. */ editorSaveToDisk(); Quit = 1; Back_to_big_picture = 1; break; case CTRL_F: editorFind(); break; case KEY_BACKSPACE: case DELETE: case CTRL_H: editorDelChar(); break; case KEY_NPAGE: case KEY_PPAGE: if (c == KEY_PPAGE && E.cy != 0) E.cy = 0; else if (c == KEY_NPAGE && E.cy != LINES-1-1) E.cy = LINES-1-1; { int times = LINES-1; while(times--) editorMoveCursor(c == KEY_PPAGE ? KEY_UP : KEY_DOWN); } break; case CTRL_A: while (!editorAtStartOfLine()) editorMoveCursor(KEY_LEFT); break; case CTRL_L: while (1) { editorMoveCursor(KEY_RIGHT); if (editorAtStartOfLine()) { editorMoveCursor(KEY_LEFT); break; } } break; case CTRL_U: while (!editorAtStartOfLine()) editorDelChar(); break; case CTRL_K: while (1) { editorMoveCursor(KEY_RIGHT); if (editorAtStartOfLine()) { editorMoveCursor(KEY_LEFT); break; } editorDelChar(); } break; case CTRL_SLASH: /* same as CTRL_UNDERSCORE */ if (starts_with(E.row[E.rowoff+E.cy].chars, "--? ")) editorUncommentCursorRow(); else editorCommentCursorRow(); break; case KEY_UP: case KEY_DOWN: case KEY_LEFT: case KEY_RIGHT: editorMoveCursor(c); break; case TAB: /* insert 2 spaces */ editorInsertChar(' '); editorInsertChar(' '); break; default: if (c >= ' ') editorInsertChar(c); break; } } static void initEditor(void) { E.cx = 0; E.cy = 0; E.rowoff = 0; E.coloff = 0; E.numrows = 0; E.row = NULL; E.dirty = 0; E.filename = NULL; E.syntax = &HLDB[0]; } /* return true if user chose to back into the big picture view */ int edit(lua_State* L, char* filename) { Quit = 0; Back_to_big_picture = 0; initEditor(); editorOpen(filename); while(!Quit) { E.cols = COLS-LINE_NUMBER_SPACE; /* update on resize */ editorRefreshScreen(editorMenu); editorProcessKeypress(L); } return Back_to_big_picture; } /* return true if user chose to back into the big picture view */ int edit_from(lua_State* L, char* filename, int rowoff, int coloff, int cy, int cx) { Quit = 0; Back_to_big_picture = 0; initEditor(); E.rowoff = rowoff; E.coloff = coloff; E.cy = cy; E.cx = cx; editorOpen(filename); while(!Quit) { E.cols = COLS-LINE_NUMBER_SPACE; /* update on resize */ editorRefreshScreen(editorMenu); editorProcessKeypress(L); } return Back_to_big_picture; } int resumeEdit(lua_State* L) { Quit = 0; Back_to_big_picture = 0; while(!Quit) { E.cols = COLS-LINE_NUMBER_SPACE; /* update on resize */ editorRefreshScreen(editorMenu); editorProcessKeypress(L); } return Back_to_big_picture; } /* vim:tabstop=4:shiftwidth=0:expandtab:softtabstop=-1 */