]> code.delx.au - gnu-emacs-elpa/blob - context-coloring.el
Add elisp benchmarks.
[gnu-emacs-elpa] / context-coloring.el
1 ;;; context-coloring.el --- Highlight by scope -*- lexical-binding: t; -*-
2
3 ;; Copyright (C) 2014-2015 Free Software Foundation, Inc.
4
5 ;; Author: Jackson Ray Hamilton <jackson@jacksonrayhamilton.com>
6 ;; Version: 6.3.0
7 ;; Keywords: convenience faces tools
8 ;; Package-Requires: ((emacs "24") (js2-mode "20150126"))
9 ;; URL: https://github.com/jacksonrayhamilton/context-coloring
10
11 ;; This file is part of GNU Emacs.
12
13 ;; This program is free software; you can redistribute it and/or modify
14 ;; it under the terms of the GNU General Public License as published by
15 ;; the Free Software Foundation, either version 3 of the License, or
16 ;; (at your option) any later version.
17
18 ;; This program is distributed in the hope that it will be useful,
19 ;; but WITHOUT ANY WARRANTY; without even the implied warranty of
20 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 ;; GNU General Public License for more details.
22
23 ;; You should have received a copy of the GNU General Public License
24 ;; along with this program. If not, see <http://www.gnu.org/licenses/>.
25
26 ;;; Commentary:
27
28 ;; Highlights code by scope. Top-level scopes are one color, second-level
29 ;; scopes are another color, and so on. Variables retain the color of the scope
30 ;; in which they are defined. A variable defined in an outer scope referenced
31 ;; by an inner scope is colored the same as the outer scope.
32
33 ;; By default, comments and strings are still highlighted syntactically.
34
35 ;;; Code:
36
37 (require 'js2-mode)
38
39
40 ;;; Utilities
41
42 (defun context-coloring-join (strings delimiter)
43 "Join a list of STRINGS with the string DELIMITER."
44 (mapconcat 'identity strings delimiter))
45
46 (defsubst context-coloring-trim-right (string)
47 "Remove leading whitespace from STRING."
48 (if (string-match "[ \t\n\r]+\\'" string)
49 (replace-match "" t t string)
50 string))
51
52 (defsubst context-coloring-trim-left (string)
53 "Remove trailing whitespace from STRING."
54 (if (string-match "\\`[ \t\n\r]+" string)
55 (replace-match "" t t string)
56 string))
57
58 (defsubst context-coloring-trim (string)
59 "Remove leading and trailing whitespace from STRING."
60 (context-coloring-trim-left (context-coloring-trim-right string)))
61
62
63 ;;; Faces
64
65 (defun context-coloring-defface (level tty light dark)
66 "Define a face for LEVEL with colors for TTY, LIGHT and DARK
67 backgrounds."
68 (let ((face (intern (format "context-coloring-level-%s-face" level)))
69 (doc (format "Context coloring face, level %s." level)))
70 (custom-declare-face
71 face
72 `((((type tty)) (:foreground ,tty))
73 (((background light)) (:foreground ,light))
74 (((background dark)) (:foreground ,dark)))
75 doc
76 :group 'context-coloring)))
77
78 (defun context-coloring-defface-neutral (level)
79 "Define a face for LEVEL with the default neutral colors."
80 (context-coloring-defface level nil "#3f3f3f" "#cdcdcd"))
81
82 (context-coloring-defface 0 nil "#000000" "#ffffff")
83 (context-coloring-defface 1 "yellow" "#008b8b" "#00ffff")
84 (context-coloring-defface 2 "green" "#0000ff" "#87cefa")
85 (context-coloring-defface 3 "cyan" "#483d8b" "#b0c4de")
86 (context-coloring-defface 4 "blue" "#a020f0" "#eedd82")
87 (context-coloring-defface 5 "magenta" "#a0522d" "#98fb98")
88 (context-coloring-defface 6 "red" "#228b22" "#7fffd4")
89 (context-coloring-defface-neutral 7)
90
91 (defvar context-coloring-maximum-face nil
92 "Index of the highest face available for coloring.")
93
94 (defvar context-coloring-original-maximum-face nil
95 "Fallback value for `context-coloring-maximum-face' when all
96 themes have been disabled.")
97
98 (setq context-coloring-maximum-face 7)
99
100 (setq context-coloring-original-maximum-face
101 context-coloring-maximum-face)
102
103 ;; Theme authors can have up to 26 levels: 1 (0th) for globals, 24 (1st-24th)
104 ;; for nested levels, and 1 (25th) for infinity.
105 (dotimes (number 18)
106 (context-coloring-defface-neutral (+ number context-coloring-maximum-face 1)))
107
108
109 ;;; Face functions
110
111 (defsubst context-coloring-level-face (level)
112 "Return the symbol for a face with LEVEL."
113 ;; `concat' is faster than `format' here.
114 (intern-soft
115 (concat "context-coloring-level-" (number-to-string level) "-face")))
116
117 (defsubst context-coloring-bounded-level-face (level)
118 "Return the symbol for a face with LEVEL, bounded by
119 `context-coloring-maximum-face'."
120 (context-coloring-level-face (min level context-coloring-maximum-face)))
121
122
123 ;;; Colorization utilities
124
125 (defsubst context-coloring-colorize-region (start end level)
126 "Color characters from the 1-indexed START point (inclusive) to
127 the END point (exclusive) with the face corresponding to LEVEL."
128 (add-text-properties
129 start
130 end
131 `(face ,(context-coloring-bounded-level-face level))))
132
133 (make-obsolete-variable
134 'context-coloring-comments-and-strings
135 "use `context-coloring-syntactic-comments' and
136 `context-coloring-syntactic-strings' instead."
137 "6.1.0")
138
139 (defcustom context-coloring-syntactic-comments t
140 "If non-nil, also color comments using `font-lock'."
141 :group 'context-coloring)
142
143 (defcustom context-coloring-syntactic-strings t
144 "If non-nil, also color strings using `font-lock'."
145 :group 'context-coloring)
146
147 (defun context-coloring-font-lock-syntactic-comment-function (state)
148 "Tell `font-lock' to color a comment but not a string."
149 (if (nth 3 state) nil font-lock-comment-face))
150
151 (defun context-coloring-font-lock-syntactic-string-function (state)
152 "Tell `font-lock' to color a string but not a comment."
153 (if (nth 3 state) font-lock-string-face nil))
154
155 (defsubst context-coloring-maybe-colorize-comments-and-strings (&optional min max)
156 "Color the current buffer's comments or strings if
157 `context-coloring-syntactic-comments' or
158 `context-coloring-syntactic-strings' are non-nil."
159 (when (or context-coloring-syntactic-comments
160 context-coloring-syntactic-strings)
161 (let ((min (or min (point-min)))
162 (max (or max (point-max)))
163 (font-lock-syntactic-face-function
164 (cond
165 ((and context-coloring-syntactic-comments
166 (not context-coloring-syntactic-strings))
167 'context-coloring-font-lock-syntactic-comment-function)
168 ((and context-coloring-syntactic-strings
169 (not context-coloring-syntactic-comments))
170 'context-coloring-font-lock-syntactic-string-function)
171 (t
172 font-lock-syntactic-face-function))))
173 (save-excursion
174 (font-lock-fontify-syntactically-region min max)
175 ;; TODO: Make configurable at the dispatch level.
176 (when (eq major-mode 'emacs-lisp-mode)
177 (font-lock-fontify-keywords-region min max))))))
178
179
180 ;;; js2-mode colorization
181
182 (defvar-local context-coloring-js2-scope-level-hash-table nil
183 "Associate `js2-scope' structures and with their scope
184 levels.")
185
186 (defcustom context-coloring-js-block-scopes nil
187 "If non-nil, also color block scopes in the scope hierarchy in JavaScript.
188
189 The block-scoped `let' and `const' are introduced in ES6. Enable
190 this for ES6 code; disable it elsewhere.
191
192 Supported modes: `js2-mode'"
193 :group 'context-coloring)
194
195 (defsubst context-coloring-js2-scope-level (scope)
196 "Return the level of SCOPE."
197 (cond ((gethash scope context-coloring-js2-scope-level-hash-table))
198 (t
199 (let ((level 0)
200 (current-scope scope)
201 enclosing-scope)
202 (while (and current-scope
203 (js2-node-parent current-scope)
204 (setq enclosing-scope
205 (js2-node-get-enclosing-scope current-scope)))
206 (when (or context-coloring-js-block-scopes
207 (let ((type (js2-scope-type current-scope)))
208 (or (= type js2-SCRIPT)
209 (= type js2-FUNCTION)
210 (= type js2-CATCH))))
211 (setq level (+ level 1)))
212 (setq current-scope enclosing-scope))
213 (puthash scope level context-coloring-js2-scope-level-hash-table)))))
214
215 (defsubst context-coloring-js2-local-name-node-p (node)
216 "Determine if NODE is a `js2-name-node' representing a local
217 variable."
218 (and (js2-name-node-p node)
219 (let ((parent (js2-node-parent node)))
220 (not (or (and (js2-object-prop-node-p parent)
221 (eq node (js2-object-prop-node-left parent)))
222 (and (js2-prop-get-node-p parent)
223 ;; For nested property lookup, the node on the left is a
224 ;; `js2-prop-get-node', so this always works.
225 (eq node (js2-prop-get-node-right parent))))))))
226
227 (defvar-local context-coloring-point-max nil
228 "Cached value of `point-max'.")
229
230 (defsubst context-coloring-js2-colorize-node (node level)
231 "Color NODE with the color for LEVEL."
232 (let ((start (js2-node-abs-pos node)))
233 (context-coloring-colorize-region
234 start
235 (min
236 ;; End
237 (+ start (js2-node-len node))
238 ;; Somes nodes (like the ast when there is an unterminated multiline
239 ;; comment) will stretch to the value of `point-max'.
240 context-coloring-point-max)
241 level)))
242
243 (defun context-coloring-js2-colorize ()
244 "Color the current buffer using the abstract syntax tree
245 generated by `js2-mode'."
246 ;; Reset the hash table; the old one could be obsolete.
247 (setq context-coloring-js2-scope-level-hash-table (make-hash-table :test 'eq))
248 (setq context-coloring-point-max (point-max))
249 (with-silent-modifications
250 (js2-visit-ast
251 js2-mode-ast
252 (lambda (node end-p)
253 (when (null end-p)
254 (cond
255 ((js2-scope-p node)
256 (context-coloring-js2-colorize-node
257 node
258 (context-coloring-js2-scope-level node)))
259 ((context-coloring-js2-local-name-node-p node)
260 (let* ((enclosing-scope (js2-node-get-enclosing-scope node))
261 (defining-scope (js2-get-defining-scope
262 enclosing-scope
263 (js2-name-node-name node))))
264 ;; The tree seems to be walked lexically, so an entire scope will
265 ;; be colored, including its name nodes, before they are reached.
266 ;; Coloring the nodes defined in that scope would be redundant, so
267 ;; don't do it.
268 (when (not (eq defining-scope enclosing-scope))
269 (context-coloring-js2-colorize-node
270 node
271 (context-coloring-js2-scope-level defining-scope))))))
272 ;; The `t' indicates to search children.
273 t)))
274 (context-coloring-maybe-colorize-comments-and-strings)))
275
276
277 ;;; Emacs Lisp colorization
278
279 (defsubst context-coloring-forward-sws ()
280 "Move forward through whitespace and comments."
281 (while (forward-comment 1)))
282
283 (defsubst context-coloring-get-syntax-code ()
284 (syntax-class (syntax-after (point))))
285
286 (defsubst context-coloring-exact-regexp (word)
287 "Create a regexp that matches exactly WORD."
288 (concat "\\`" (regexp-quote word) "\\'"))
289
290 (defsubst context-coloring-exact-or-regexp (words)
291 "Create a regexp that matches any exact word in WORDS."
292 (context-coloring-join
293 (mapcar 'context-coloring-exact-regexp words) "\\|"))
294
295 (defconst context-coloring-elisp-defun-regexp
296 (context-coloring-exact-or-regexp
297 '("defun" "defun*" "defsubst" "defmacro"
298 "cl-defun" "cl-defsubst" "cl-defmacro")))
299
300 (defconst context-coloring-elisp-lambda-regexp
301 (context-coloring-exact-regexp "lambda"))
302
303 (defconst context-coloring-elisp-let-regexp
304 (context-coloring-exact-regexp "let"))
305
306 (defconst context-coloring-elisp-let*-regexp
307 (context-coloring-exact-regexp "let*"))
308
309 (defconst context-coloring-elisp-arglist-arg-regexp
310 "\\`[^&:]")
311
312 (defconst context-coloring-ignored-word-regexp
313 (context-coloring-join (list "\\`[-+]?[0-9]"
314 "\\`[&:].+"
315 (context-coloring-exact-or-regexp
316 '("t" "nil" "." "?")))
317 "\\|"))
318
319 (defconst context-coloring-WORD-CODE 2)
320 (defconst context-coloring-SYMBOL-CODE 3)
321 (defconst context-coloring-OPEN-PARENTHESIS-CODE 4)
322 (defconst context-coloring-CLOSE-PARENTHESIS-CODE 5)
323 (defconst context-coloring-EXPRESSION-PREFIX-CODE 6)
324 (defconst context-coloring-STRING-QUOTE-CODE 7)
325 (defconst context-coloring-ESCAPE-CODE 9)
326 (defconst context-coloring-COMMENT-START-CODE 11)
327 (defconst context-coloring-COMMENT-END-CODE 12)
328
329 (defconst context-coloring-OCTOTHORPE-CHAR (string-to-char "#"))
330 (defconst context-coloring-APOSTROPHE-CHAR (string-to-char "'"))
331 (defconst context-coloring-OPEN-PARENTHESIS-CHAR (string-to-char "("))
332 (defconst context-coloring-COMMA-CHAR (string-to-char ","))
333 (defconst context-coloring-AT-CHAR (string-to-char "@"))
334 (defconst context-coloring-BACKTICK-CHAR (string-to-char "`"))
335
336 (defvar context-coloring-elisp-scope-stack '())
337
338 (defsubst context-coloring-elisp-make-scope (level)
339 (list
340 :level level
341 :variables '()))
342
343 (defsubst context-coloring-elisp-scope-get-level (scope)
344 (plist-get scope :level))
345
346 (defsubst context-coloring-elisp-scope-add-variable (scope variable)
347 (plist-put scope :variables (cons variable (plist-get scope :variables))))
348
349 (defsubst context-coloring-elisp-scope-has-variable (scope variable)
350 (member variable (plist-get scope :variables)))
351
352 (defsubst context-coloring-elisp-get-variable-level (variable)
353 (let* ((scope-stack context-coloring-elisp-scope-stack)
354 scope
355 level)
356 (while (and scope-stack (not level))
357 (setq scope (car scope-stack))
358 (cond
359 ((context-coloring-elisp-scope-has-variable scope variable)
360 (setq level (context-coloring-elisp-scope-get-level scope)))
361 (t
362 (setq scope-stack (cdr scope-stack)))))
363 ;; Assume a global variable.
364 (or level 0)))
365
366 (defsubst context-coloring-elisp-current-scope-level ()
367 (cond
368 ((car context-coloring-elisp-scope-stack)
369 (context-coloring-elisp-scope-get-level (car context-coloring-elisp-scope-stack)))
370 (t
371 0)))
372
373 (defsubst context-coloring-elisp-push-scope ()
374 (push (context-coloring-elisp-make-scope
375 (1+ (context-coloring-elisp-current-scope-level)))
376 context-coloring-elisp-scope-stack))
377
378 (defsubst context-coloring-elisp-pop-scope ()
379 (pop context-coloring-elisp-scope-stack))
380
381 (defsubst context-coloring-elisp-add-variable (variable)
382 (context-coloring-elisp-scope-add-variable
383 (car context-coloring-elisp-scope-stack)
384 variable))
385
386 (defun context-coloring-elisp-parse-arg (callback)
387 (let (arg-pos
388 arg-end
389 arg-string)
390 (setq arg-pos (point))
391 (forward-sexp)
392 (setq arg-end (point))
393 (setq arg-string (buffer-substring-no-properties
394 arg-pos
395 arg-end))
396 (when (string-match-p
397 context-coloring-elisp-arglist-arg-regexp
398 arg-string)
399 (funcall callback arg-string))))
400
401 (defun context-coloring-elisp-parse-let-varlist (type)
402 (let ((varlist '())
403 syntax-code)
404 ;; Enter.
405 (forward-char)
406 (while (/= (setq syntax-code (context-coloring-get-syntax-code))
407 context-coloring-CLOSE-PARENTHESIS-CODE)
408 (cond
409 ((= syntax-code context-coloring-OPEN-PARENTHESIS-CODE)
410 (forward-char)
411 (context-coloring-forward-sws)
412 (setq syntax-code (context-coloring-get-syntax-code))
413 (when (or (= syntax-code context-coloring-WORD-CODE)
414 (= syntax-code context-coloring-SYMBOL-CODE))
415 (context-coloring-elisp-parse-arg
416 (lambda (var)
417 (push var varlist)))
418 (context-coloring-forward-sws)
419 (setq syntax-code (context-coloring-get-syntax-code))
420 (when (/= syntax-code context-coloring-CLOSE-PARENTHESIS-CODE)
421 (context-coloring-elisp-colorize-sexp)))
422 (context-coloring-forward-sws)
423 ;; Skip past the closing parenthesis.
424 (forward-char))
425 ((or (= syntax-code context-coloring-WORD-CODE)
426 (= syntax-code context-coloring-SYMBOL-CODE))
427 (context-coloring-elisp-parse-arg
428 (lambda (var)
429 (push var varlist)))))
430 (when (eq type 'let*)
431 (context-coloring-elisp-add-variable (pop varlist)))
432 (context-coloring-forward-sws))
433 (when (eq type 'let)
434 (while varlist
435 (context-coloring-elisp-add-variable (pop varlist))))
436 ;; Exit.
437 (forward-char)))
438
439 (defun context-coloring-elisp-parse-arglist ()
440 (let (syntax-code)
441 ;; Enter.
442 (forward-char)
443 (while (/= (setq syntax-code (context-coloring-get-syntax-code))
444 context-coloring-CLOSE-PARENTHESIS-CODE)
445 (cond
446 ((or (= syntax-code context-coloring-WORD-CODE)
447 (= syntax-code context-coloring-SYMBOL-CODE))
448 (context-coloring-elisp-parse-arg
449 (lambda (arg)
450 (context-coloring-elisp-add-variable arg))))
451 (t
452 (forward-sexp)))
453 (context-coloring-forward-sws))
454 ;; Exit.
455 (forward-char)))
456
457 (defun context-coloring-elisp-colorize-defun (&optional anonymous-p
458 let-type)
459 (let ((start (point))
460 end
461 stop
462 syntax-code
463 defun-name-pos
464 defun-name-end)
465 (context-coloring-elisp-push-scope)
466 ;; Color the whole sexp.
467 (forward-sexp)
468 (setq end (point))
469 (context-coloring-colorize-region
470 start
471 end
472 (context-coloring-elisp-current-scope-level))
473 (goto-char start)
474 ;; Skip past the "defun".
475 (skip-syntax-forward "^w_")
476 (forward-sexp)
477 (context-coloring-forward-sws)
478 (setq stop nil)
479 (unless anonymous-p
480 ;; Check for the defun's name.
481 (setq syntax-code (context-coloring-get-syntax-code))
482 (cond
483 ((or (= syntax-code context-coloring-WORD-CODE)
484 (= syntax-code context-coloring-SYMBOL-CODE))
485 ;; Color the defun's name with the top-level color.
486 (setq defun-name-pos (point))
487 (forward-sexp)
488 (setq defun-name-end (point))
489 (context-coloring-colorize-region defun-name-pos defun-name-end 0)
490 (context-coloring-forward-sws))
491 (t
492 (setq stop t))))
493 (cond
494 (stop
495 ;; Skip it.
496 (goto-char start)
497 (forward-sexp))
498 (t
499 (setq syntax-code (context-coloring-get-syntax-code))
500 (cond
501 ((= syntax-code context-coloring-OPEN-PARENTHESIS-CODE)
502 (cond
503 (let-type
504 (context-coloring-elisp-parse-let-varlist let-type))
505 (t
506 (context-coloring-elisp-parse-arglist)))
507 ;; Colorize the rest of the function.
508 (context-coloring-elisp-colorize-region (point) (1- end))
509 ;; Exit the defun.
510 (forward-char))
511 (t
512 ;; Skip it.
513 (goto-char start)
514 (forward-sexp)))))
515 (context-coloring-elisp-pop-scope)
516 (context-coloring-elisp-colorize-comments-and-strings-in-region start end)))
517
518 (defun context-coloring-elisp-colorize-lambda ()
519 (context-coloring-elisp-colorize-defun t))
520
521 (defun context-coloring-elisp-colorize-let ()
522 (context-coloring-elisp-colorize-defun t 'let))
523
524 (defun context-coloring-elisp-colorize-let* ()
525 (context-coloring-elisp-colorize-defun t 'let*))
526
527 (defun context-coloring-elisp-colorize-parenthesized-sexp ()
528 (let ((start (point))
529 end
530 syntax-code
531 child-0-pos
532 child-0-end
533 child-0-string)
534 (forward-sexp)
535 (setq end (point))
536 (goto-char start)
537 (forward-char)
538 (context-coloring-forward-sws)
539 (setq syntax-code (context-coloring-get-syntax-code))
540 ;; Figure out if the sexp is a special form.
541 (cond
542 ((when (or (= syntax-code context-coloring-WORD-CODE)
543 (= syntax-code context-coloring-SYMBOL-CODE))
544 (setq child-0-pos (point))
545 (forward-sexp)
546 (setq child-0-end (point))
547 (setq child-0-string (buffer-substring-no-properties
548 child-0-pos
549 child-0-end))
550 (cond
551 ((string-match-p context-coloring-elisp-defun-regexp child-0-string)
552 (goto-char start)
553 (context-coloring-elisp-colorize-defun)
554 t)
555 ((string-match-p context-coloring-elisp-lambda-regexp child-0-string)
556 (goto-char start)
557 (context-coloring-elisp-colorize-lambda)
558 t)
559 ((string-match-p context-coloring-elisp-let-regexp child-0-string)
560 (goto-char start)
561 (context-coloring-elisp-colorize-let)
562 t)
563 ((string-match-p context-coloring-elisp-let*-regexp child-0-string)
564 (goto-char start)
565 (context-coloring-elisp-colorize-let*)
566 t)
567 (t
568 nil))))
569 ;; Not a special form; just colorize the remaining region.
570 (t
571 (context-coloring-colorize-region
572 start
573 end
574 (context-coloring-elisp-current-scope-level))
575 (context-coloring-elisp-colorize-region (point) (1- end))
576 (forward-char)))))
577
578 (defun context-coloring-elisp-colorize-symbol ()
579 (let (symbol-pos
580 symbol-end
581 symbol-string)
582 (setq symbol-pos (point))
583 (forward-sexp)
584 (setq symbol-end (point))
585 (setq symbol-string (buffer-substring-no-properties
586 symbol-pos
587 symbol-end))
588 (cond
589 ((string-match-p context-coloring-ignored-word-regexp symbol-string))
590 (t
591 (context-coloring-colorize-region
592 symbol-pos
593 symbol-end
594 (context-coloring-elisp-get-variable-level
595 symbol-string))))))
596
597 (defun context-coloring-elisp-colorize-expression-prefix ()
598 (let ((char (char-after))
599 (start (point))
600 (end (progn (forward-sexp)
601 (point))))
602 (cond
603 ((or (= char context-coloring-APOSTROPHE-CHAR)
604 (= char context-coloring-OCTOTHORPE-CHAR))
605 (context-coloring-elisp-colorize-comments-and-strings-in-region start end))
606 ((= char context-coloring-BACKTICK-CHAR)
607 (goto-char start)
608 (while (> end (progn (forward-char)
609 (point)))
610 (setq char (char-after))
611 (when (= char context-coloring-COMMA-CHAR)
612 (forward-char)
613 (when (= (char-after) context-coloring-AT-CHAR)
614 ;; If we don't do this "@" could be interpreted as a symbol.
615 (forward-char))
616 (context-coloring-forward-sws)
617 (context-coloring-elisp-colorize-sexp)))
618 (context-coloring-elisp-colorize-comments-and-strings-in-region start end)))))
619
620 (defvar context-coloring-parse-interruptable-p t
621 "Set this to nil to force parse to continue until finished.")
622
623 (defconst context-coloring-elisp-sexps-per-pause 1000
624 "Pause after this many iterations to check for user input.
625 If user input is pending, stop the parse. This makes for a
626 smoother user experience for large files.
627
628 As of this writing, emacs lisp colorization seems to run at about
629 60,000 iterations per second. A default value of 1000 should
630 provide visually \"instant\" updates at 60 frames per second.")
631
632 (defvar context-coloring-elisp-sexp-count 0)
633
634 (defun context-coloring-elisp-increment-sexp-count ()
635 (setq context-coloring-elisp-sexp-count
636 (1+ context-coloring-elisp-sexp-count))
637 (when (and (zerop (% context-coloring-elisp-sexp-count
638 context-coloring-elisp-sexps-per-pause))
639 context-coloring-parse-interruptable-p
640 (input-pending-p))
641 (throw 'interrupted t)))
642
643 (defun context-coloring-elisp-colorize-sexp ()
644 (let (syntax-code)
645 (context-coloring-elisp-increment-sexp-count)
646 (setq syntax-code (context-coloring-get-syntax-code))
647 (cond
648 ((= syntax-code context-coloring-OPEN-PARENTHESIS-CODE)
649 (context-coloring-elisp-colorize-parenthesized-sexp))
650 ((or (= syntax-code context-coloring-WORD-CODE)
651 (= syntax-code context-coloring-SYMBOL-CODE))
652 (context-coloring-elisp-colorize-symbol))
653 ((= syntax-code context-coloring-EXPRESSION-PREFIX-CODE)
654 (context-coloring-elisp-colorize-expression-prefix))
655 ((= syntax-code context-coloring-STRING-QUOTE-CODE)
656 (context-coloring-elisp-colorize-string))
657 (t
658 (forward-char)))))
659
660 (defun context-coloring-elisp-colorize-comment ()
661 (let ((start (point)))
662 (context-coloring-elisp-increment-sexp-count)
663 (context-coloring-forward-sws)
664 (context-coloring-maybe-colorize-comments-and-strings
665 start
666 (point))))
667
668 (defun context-coloring-elisp-colorize-string ()
669 (let ((start (point)))
670 (context-coloring-elisp-increment-sexp-count)
671 (forward-sexp)
672 (context-coloring-maybe-colorize-comments-and-strings
673 start
674 (point))))
675
676 (defun context-coloring-elisp-colorize-comments-and-strings-in-region (start end)
677 (let (syntax-code)
678 (goto-char start)
679 (while (> end (progn (skip-syntax-forward "^<\"\\" end)
680 (point)))
681 (setq syntax-code (context-coloring-get-syntax-code))
682 (cond
683 ((= syntax-code context-coloring-STRING-QUOTE-CODE)
684 (context-coloring-elisp-colorize-string))
685 ((= syntax-code context-coloring-COMMENT-START-CODE)
686 (context-coloring-elisp-colorize-comment))
687 ((= syntax-code context-coloring-ESCAPE-CODE)
688 (forward-char 2))
689 (t
690 (forward-char))))))
691
692 (defun context-coloring-elisp-colorize-region (start end)
693 (let (syntax-code)
694 (goto-char start)
695 (while (> end (progn (skip-syntax-forward "^()w_'<\"\\" end)
696 (point)))
697 (setq syntax-code (context-coloring-get-syntax-code))
698 (cond
699 ((or (= syntax-code context-coloring-OPEN-PARENTHESIS-CODE)
700 (= syntax-code context-coloring-WORD-CODE)
701 (= syntax-code context-coloring-SYMBOL-CODE)
702 (= syntax-code context-coloring-EXPRESSION-PREFIX-CODE)
703 (= syntax-code context-coloring-STRING-QUOTE-CODE))
704 (context-coloring-elisp-colorize-sexp))
705 ((= syntax-code context-coloring-COMMENT-START-CODE)
706 (context-coloring-elisp-colorize-comment))
707 ((= syntax-code context-coloring-ESCAPE-CODE)
708 (forward-char 2))
709 (t
710 (forward-char))))))
711
712 (defun context-coloring-elisp-colorize (start end)
713 (setq context-coloring-elisp-sexp-count 0)
714 (setq context-coloring-elisp-scope-stack '())
715 (let ((inhibit-point-motion-hooks t)
716 (case-fold-search nil)
717 ;; This is a recursive-descent parser, so give it a big stack.
718 (max-lisp-eval-depth (max max-lisp-eval-depth 3000))
719 (max-specpdl-size (max max-specpdl-size 3000)))
720 (context-coloring-elisp-colorize-region start end)))
721
722 (defun context-coloring-elisp-colorize-changed-region (start end)
723 (with-silent-modifications
724 (save-excursion
725 (let ((start (progn (goto-char start)
726 (beginning-of-defun)
727 (point)))
728 (end (progn (goto-char end)
729 (end-of-defun)
730 (point))))
731 (context-coloring-elisp-colorize start end)))))
732
733 (defun context-coloring-elisp-colorize-buffer ()
734 (interactive)
735 (with-silent-modifications
736 (save-excursion
737 (context-coloring-elisp-colorize (point-min) (point-max)))))
738
739 (defalias 'ccecb 'context-coloring-elisp-colorize-buffer)
740
741
742 ;;; Shell command scopification / colorization
743
744 (defun context-coloring-apply-tokens (tokens)
745 "Process a vector of TOKENS to apply context-based coloring to
746 the current buffer. Tokens are 3 integers: start, end, level.
747 The vector is flat, with a new token occurring after every 3rd
748 element."
749 (with-silent-modifications
750 (let ((i 0)
751 (len (length tokens)))
752 (while (< i len)
753 (context-coloring-colorize-region
754 (elt tokens i)
755 (elt tokens (+ i 1))
756 (elt tokens (+ i 2)))
757 (setq i (+ i 3))))
758 (context-coloring-maybe-colorize-comments-and-strings)))
759
760 (defun context-coloring-parse-array (array)
761 "Parse ARRAY as a flat JSON array of numbers."
762 (let ((braceless (substring (context-coloring-trim array) 1 -1)))
763 (cond
764 ((> (length braceless) 0)
765 (vconcat
766 (mapcar 'string-to-number (split-string braceless ","))))
767 (t
768 (vector)))))
769
770 (defvar-local context-coloring-scopifier-process nil
771 "The single scopifier process that can be running.")
772
773 (defun context-coloring-kill-scopifier ()
774 "Kill the currently-running scopifier process."
775 (when (not (null context-coloring-scopifier-process))
776 (delete-process context-coloring-scopifier-process)
777 (setq context-coloring-scopifier-process nil)))
778
779 (defun context-coloring-scopify-shell-command (command callback)
780 "Invoke a scopifier via COMMAND, read its response
781 asynchronously and invoke CALLBACK with its output."
782
783 ;; Prior running tokenization is implicitly obsolete if this function is
784 ;; called.
785 (context-coloring-kill-scopifier)
786
787 ;; Start the process.
788 (setq context-coloring-scopifier-process
789 (start-process-shell-command "scopifier" nil command))
790
791 (let ((output ""))
792
793 ;; The process may produce output in multiple chunks. This filter
794 ;; accumulates the chunks into a message.
795 (set-process-filter
796 context-coloring-scopifier-process
797 (lambda (_process chunk)
798 (setq output (concat output chunk))))
799
800 ;; When the process's message is complete, this sentinel parses it as JSON
801 ;; and applies the tokens to the buffer.
802 (set-process-sentinel
803 context-coloring-scopifier-process
804 (lambda (_process event)
805 (when (equal "finished\n" event)
806 (funcall callback output))))))
807
808 (defun context-coloring-send-buffer-to-scopifier ()
809 "Give the scopifier process its input so it can begin
810 scopifying."
811 (process-send-region
812 context-coloring-scopifier-process
813 (point-min) (point-max))
814 (process-send-eof
815 context-coloring-scopifier-process))
816
817 (defun context-coloring-scopify-and-colorize (command &optional callback)
818 "Invoke a scopifier via COMMAND with the current buffer's contents,
819 read the scopifier's response asynchronously and apply a parsed
820 list of tokens to `context-coloring-apply-tokens'.
821
822 Invoke CALLBACK when complete."
823 (let ((buffer (current-buffer)))
824 (context-coloring-scopify-shell-command
825 command
826 (lambda (output)
827 (let ((tokens (context-coloring-parse-array output)))
828 (with-current-buffer buffer
829 (context-coloring-apply-tokens tokens))
830 (setq context-coloring-scopifier-process nil)
831 (when callback (funcall callback))))))
832 (context-coloring-send-buffer-to-scopifier))
833
834
835 ;;; Dispatch
836
837 (defvar context-coloring-dispatch-hash-table (make-hash-table :test 'eq)
838 "Map dispatch strategy names to their corresponding property
839 lists, which contain details about the strategies.")
840
841 (defvar context-coloring-mode-hash-table (make-hash-table :test 'eq)
842 "Map major mode names to dispatch property lists.")
843
844 (defun context-coloring-get-dispatch-for-mode (mode)
845 "Return the dispatch for MODE (or a derivative mode)."
846 (let ((parent mode)
847 dispatch)
848 (while (and parent
849 (not (setq dispatch (gethash parent context-coloring-mode-hash-table)))
850 (setq parent (get parent 'derived-mode-parent))))
851 dispatch))
852
853 (defun context-coloring-define-dispatch (symbol &rest properties)
854 "Define a new dispatch named SYMBOL with PROPERTIES.
855
856 A \"dispatch\" is a property list describing a strategy for
857 coloring a buffer. There are three possible strategies: Parse
858 and color in a single function (`:colorizer'), parse in a
859 function that returns scope data (`:scopifier'), or parse with a
860 shell command that returns scope data (`:command'). In the
861 latter two cases, the scope data will be used to automatically
862 color the buffer.
863
864 PROPERTIES must include `:modes' and one of `:colorizer',
865 `:scopifier' or `:command'.
866
867 `:modes' - List of major modes this dispatch is valid for.
868
869 `:colorizer' - Symbol referring to a function that parses and
870 colors the buffer.
871
872 `:scopifier' - Symbol referring to a function that parses the
873 buffer a returns a flat vector of start, end and level data.
874
875 `:executable' - Optional name of an executable required by
876 `:command'.
877
878 `:command' - Shell command to execute with the current buffer
879 sent via stdin, and with a flat JSON array of start, end and
880 level data returned via stdout.
881
882 `:version' - Minimum required version that should be printed when
883 executing `:command' with a \"--version\" flag. The version
884 should be numeric, e.g. \"2\", \"19700101\", \"1.2.3\",
885 \"v1.2.3\" etc.
886
887 `:setup' - Arbitrary code to set up this dispatch when
888 `context-coloring-mode' is enabled.
889
890 `:teardown' - Arbitrary code to tear down this dispatch when
891 `context-coloring-mode' is disabled."
892 (let ((modes (plist-get properties :modes))
893 (colorizer (plist-get properties :colorizer))
894 (scopifier (plist-get properties :scopifier))
895 (command (plist-get properties :command)))
896 (when (null modes)
897 (error "No mode defined for dispatch"))
898 (when (not (or colorizer
899 scopifier
900 command))
901 (error "No colorizer, scopifier or command defined for dispatch"))
902 (puthash symbol properties context-coloring-dispatch-hash-table)
903 (dolist (mode modes)
904 (puthash mode properties context-coloring-mode-hash-table))))
905
906
907 ;;; Colorization
908
909 (defvar context-coloring-colorize-hook nil
910 "Hooks to run after coloring a buffer.")
911
912 (defun context-coloring-colorize (&optional callback)
913 "Color the current buffer by function context.
914
915 Invoke CALLBACK when complete; see `context-coloring-dispatch'."
916 (interactive)
917 (context-coloring-dispatch
918 (lambda ()
919 (when callback (funcall callback))
920 (run-hooks 'context-coloring-colorize-hook))))
921
922 (defvar-local context-coloring-changed nil
923 "Indication that the buffer has changed recently, which implies
924 that it should be colored again by
925 `context-coloring-colorize-idle-timer' if that timer is being
926 used.")
927
928 (defun context-coloring-change-function (_start _end _length)
929 "Register a change so that a buffer can be colorized soon."
930 ;; Tokenization is obsolete if there was a change.
931 (context-coloring-kill-scopifier)
932 (setq context-coloring-changed t))
933
934 (defun context-coloring-maybe-colorize (buffer)
935 "Colorize the current buffer if it has changed."
936 (when (and (eq buffer (current-buffer))
937 context-coloring-changed)
938 (setq context-coloring-changed nil)
939 (context-coloring-colorize)))
940
941
942 ;;; Versioning
943
944 (defun context-coloring-parse-version (string)
945 "Extract segments of a version STRING into a list. \"v1.0.0\"
946 produces (1 0 0), \"19700101\" produces (19700101), etc."
947 (let (version)
948 (while (string-match "[0-9]+" string)
949 (setq version (append version
950 (list (string-to-number (match-string 0 string)))))
951 (setq string (substring string (match-end 0))))
952 version))
953
954 (defun context-coloring-check-version (expected actual)
955 "Check that version EXPECTED is less than or equal to ACTUAL."
956 (let ((expected (context-coloring-parse-version expected))
957 (actual (context-coloring-parse-version actual))
958 (continue t)
959 (acceptable t))
960 (while (and continue expected)
961 (let ((an-expected (car expected))
962 (an-actual (car actual)))
963 (cond
964 ((> an-actual an-expected)
965 (setq acceptable t)
966 (setq continue nil))
967 ((< an-actual an-expected)
968 (setq acceptable nil)
969 (setq continue nil))))
970 (setq expected (cdr expected))
971 (setq actual (cdr actual)))
972 acceptable))
973
974 (defvar context-coloring-check-scopifier-version-hook nil
975 "Hooks to run after checking the scopifier version.")
976
977 (defun context-coloring-check-scopifier-version (&optional callback)
978 "Asynchronously invoke CALLBACK with a predicate indicating
979 whether the current scopifier version satisfies the minimum
980 version number required for the current major mode."
981 (let ((dispatch (context-coloring-get-dispatch-for-mode major-mode)))
982 (when dispatch
983 (let ((version (plist-get dispatch :version))
984 (command (plist-get dispatch :command)))
985 (context-coloring-scopify-shell-command
986 (context-coloring-join (list command "--version") " ")
987 (lambda (output)
988 (if (context-coloring-check-version version output)
989 (progn
990 (when callback (funcall callback t)))
991 (when callback (funcall callback nil)))
992 (run-hooks 'context-coloring-check-scopifier-version-hook)))))))
993
994
995 ;;; Themes
996
997 (defvar context-coloring-theme-hash-table (make-hash-table :test 'eq)
998 "Map theme names to theme properties.")
999
1000 (defun context-coloring-theme-p (theme)
1001 "Return t if THEME is defined, nil otherwise."
1002 (and (gethash theme context-coloring-theme-hash-table)))
1003
1004 (defconst context-coloring-level-face-regexp
1005 "context-coloring-level-\\([[:digit:]]+\\)-face"
1006 "Extract a level from a face.")
1007
1008 (defvar context-coloring-originally-set-theme-hash-table
1009 (make-hash-table :test 'eq)
1010 "Cache custom themes who originally set their own
1011 `context-coloring-level-N-face' faces.")
1012
1013 (defun context-coloring-theme-originally-set-p (theme)
1014 "Return t if there is a `context-coloring-level-N-face'
1015 originally set for THEME, nil otherwise."
1016 (let (originally-set)
1017 (cond
1018 ;; `setq' might return a non-nil value for the sake of this `cond'.
1019 ((setq
1020 originally-set
1021 (gethash
1022 theme
1023 context-coloring-originally-set-theme-hash-table))
1024 (eq originally-set 'yes))
1025 (t
1026 (let* ((settings (get theme 'theme-settings))
1027 (tail settings)
1028 found)
1029 (while (and tail (not found))
1030 (and (eq (nth 0 (car tail)) 'theme-face)
1031 (string-match
1032 context-coloring-level-face-regexp
1033 (symbol-name (nth 1 (car tail))))
1034 (setq found t))
1035 (setq tail (cdr tail)))
1036 found)))))
1037
1038 (defun context-coloring-cache-originally-set (theme originally-set)
1039 "Remember if THEME had colors originally set for it. If
1040 ORIGINALLY-SET is non-nil, it did, otherwise it didn't."
1041 ;; Caching whether a theme was originally set is kind of dirty, but we have to
1042 ;; do it to remember the past state of the theme. There are probably some
1043 ;; edge cases where caching will be an issue, but they are probably rare.
1044 (puthash
1045 theme
1046 (if originally-set 'yes 'no)
1047 context-coloring-originally-set-theme-hash-table))
1048
1049 (defun context-coloring-warn-theme-originally-set (theme)
1050 "Warn the user that the colors for THEME are already originally
1051 set."
1052 (warn "Context coloring colors for theme `%s' are already defined" theme))
1053
1054 (defun context-coloring-theme-highest-level (theme)
1055 "Return the highest level N of a face like
1056 `context-coloring-level-N-face' set for THEME, or `-1' if there
1057 is none."
1058 (let* ((settings (get theme 'theme-settings))
1059 (tail settings)
1060 face-string
1061 number
1062 (found -1))
1063 (while tail
1064 (and (eq (nth 0 (car tail)) 'theme-face)
1065 (setq face-string (symbol-name (nth 1 (car tail))))
1066 (string-match
1067 context-coloring-level-face-regexp
1068 face-string)
1069 (setq number (string-to-number
1070 (substring face-string
1071 (match-beginning 1)
1072 (match-end 1))))
1073 (> number found)
1074 (setq found number))
1075 (setq tail (cdr tail)))
1076 found))
1077
1078 (defun context-coloring-apply-theme (theme)
1079 "Apply THEME's properties to its respective custom theme,
1080 which must already exist and which *should* already be enabled."
1081 (let* ((properties (gethash theme context-coloring-theme-hash-table))
1082 (colors (plist-get properties :colors))
1083 (level -1))
1084 ;; Only clobber when we have to.
1085 (when (custom-theme-enabled-p theme)
1086 (setq context-coloring-maximum-face (- (length colors) 1)))
1087 (apply
1088 'custom-theme-set-faces
1089 theme
1090 (mapcar
1091 (lambda (color)
1092 (setq level (+ level 1))
1093 `(,(context-coloring-level-face level) ((t (:foreground ,color)))))
1094 colors))))
1095
1096 (defun context-coloring-define-theme (theme &rest properties)
1097 "Define a context theme named THEME for coloring scope levels.
1098
1099 PROPERTIES is a property list specifiying the following details:
1100
1101 `:aliases': List of symbols of other custom themes that these
1102 colors are applicable to.
1103
1104 `:colors': List of colors that this context theme uses.
1105
1106 `:override': If non-nil, this context theme is intentionally
1107 overriding colors set by a custom theme. Don't set this non-nil
1108 unless there is a custom theme you want to use which sets
1109 `context-coloring-level-N-face' faces that you want to replace.
1110
1111 `:recede': If non-nil, this context theme should not apply its
1112 colors if a custom theme already sets
1113 `context-coloring-level-N-face' faces. This option is
1114 optimistic; set this non-nil if you would rather confer the duty
1115 of picking colors to a custom theme author (if / when he ever
1116 gets around to it).
1117
1118 By default, context themes will always override custom themes,
1119 even if those custom themes set `context-coloring-level-N-face'
1120 faces. If a context theme does override a custom theme, a
1121 warning will be raised, at which point you may want to enable the
1122 `:override' option, or just delete your context theme and opt to
1123 use your custom theme's author's colors instead.
1124
1125 Context themes only work for the custom theme with the highest
1126 precedence, i.e. the car of `custom-enabled-themes'."
1127 (let ((aliases (plist-get properties :aliases))
1128 (override (plist-get properties :override))
1129 (recede (plist-get properties :recede)))
1130 (dolist (name (append `(,theme) aliases))
1131 (puthash name properties context-coloring-theme-hash-table)
1132 (when (custom-theme-p name)
1133 (let ((originally-set (context-coloring-theme-originally-set-p name)))
1134 (context-coloring-cache-originally-set name originally-set)
1135 ;; In the particular case when you innocently define colors that a
1136 ;; custom theme originally set, warn. Arguably this only has to be
1137 ;; done at enable time, but it is probably more useful to do it at
1138 ;; definition time for prompter feedback.
1139 (when (and originally-set
1140 (not recede)
1141 (not override))
1142 (context-coloring-warn-theme-originally-set name))
1143 ;; Set (or overwrite) colors.
1144 (when (not (and originally-set
1145 recede))
1146 (context-coloring-apply-theme name)))))))
1147
1148 (defun context-coloring-enable-theme (theme)
1149 "Apply THEME if its colors are not already set, else just set
1150 `context-coloring-maximum-face' to the correct value for THEME."
1151 (let* ((properties (gethash theme context-coloring-theme-hash-table))
1152 (recede (plist-get properties :recede))
1153 (override (plist-get properties :override)))
1154 (cond
1155 (recede
1156 (let ((highest-level (context-coloring-theme-highest-level theme)))
1157 (cond
1158 ;; This can be true whether originally set by a custom theme or by a
1159 ;; context theme.
1160 ((> highest-level -1)
1161 (setq context-coloring-maximum-face highest-level))
1162 ;; It is possible that the corresponding custom theme did not exist at
1163 ;; the time of defining this context theme, and in that case the above
1164 ;; condition proves the custom theme did not originally set any faces,
1165 ;; so we have license to apply the context theme for the first time
1166 ;; here.
1167 (t
1168 (context-coloring-apply-theme theme)))))
1169 (t
1170 (let ((originally-set (context-coloring-theme-originally-set-p theme)))
1171 ;; Cache now in case the context theme was defined after the custom
1172 ;; theme.
1173 (context-coloring-cache-originally-set theme originally-set)
1174 (when (and originally-set
1175 (not override))
1176 (context-coloring-warn-theme-originally-set theme))
1177 (context-coloring-apply-theme theme))))))
1178
1179 (defadvice enable-theme (after context-coloring-enable-theme (theme) activate)
1180 "Enable colors for context themes just-in-time."
1181 (when (and (not (eq theme 'user)) ; Called internally by `enable-theme'.
1182 (custom-theme-p theme) ; Guard against non-existent themes.
1183 (context-coloring-theme-p theme))
1184 (when (= (length custom-enabled-themes) 1)
1185 ;; Cache because we can't reliably figure it out in reverse.
1186 (setq context-coloring-original-maximum-face
1187 context-coloring-maximum-face))
1188 (context-coloring-enable-theme theme)))
1189
1190 (defadvice disable-theme (after context-coloring-disable-theme (theme) activate)
1191 "Update `context-coloring-maximum-face'."
1192 (when (custom-theme-p theme) ; Guard against non-existent themes.
1193 (let ((enabled-theme (car custom-enabled-themes)))
1194 (if (context-coloring-theme-p enabled-theme)
1195 (progn
1196 (context-coloring-enable-theme enabled-theme))
1197 ;; Assume we are back to no theme; act as if nothing ever happened.
1198 ;; This is still prone to intervention, but rather extraordinarily.
1199 (setq context-coloring-maximum-face
1200 context-coloring-original-maximum-face)))))
1201
1202 (context-coloring-define-theme
1203 'ample
1204 :recede t
1205 :colors '("#bdbdb3"
1206 "#baba36"
1207 "#6aaf50"
1208 "#5180b3"
1209 "#ab75c3"
1210 "#cd7542"
1211 "#df9522"
1212 "#454545"))
1213
1214 (context-coloring-define-theme
1215 'anti-zenburn
1216 :recede t
1217 :colors '("#232333"
1218 "#6c1f1c"
1219 "#401440"
1220 "#0f2050"
1221 "#205070"
1222 "#336c6c"
1223 "#23733c"
1224 "#6b400c"
1225 "#603a60"
1226 "#2f4070"
1227 "#235c5c"))
1228
1229 (context-coloring-define-theme
1230 'grandshell
1231 :recede t
1232 :colors '("#bebebe"
1233 "#5af2ee"
1234 "#b2baf6"
1235 "#f09fff"
1236 "#efc334"
1237 "#f6df92"
1238 "#acfb5a"
1239 "#888888"))
1240
1241 (context-coloring-define-theme
1242 'leuven
1243 :recede t
1244 :colors '("#333333"
1245 "#0000ff"
1246 "#6434a3"
1247 "#ba36a5"
1248 "#d0372d"
1249 "#036a07"
1250 "#006699"
1251 "#006fe0"
1252 "#808080"))
1253
1254 (context-coloring-define-theme
1255 'monokai
1256 :recede t
1257 :colors '("#f8f8f2"
1258 "#66d9ef"
1259 "#a1efe4"
1260 "#a6e22e"
1261 "#e6db74"
1262 "#fd971f"
1263 "#f92672"
1264 "#fd5ff0"
1265 "#ae81ff"))
1266
1267 (context-coloring-define-theme
1268 'solarized
1269 :recede t
1270 :aliases '(solarized-light
1271 solarized-dark
1272 sanityinc-solarized-light
1273 sanityinc-solarized-dark)
1274 :colors '("#839496"
1275 "#268bd2"
1276 "#2aa198"
1277 "#859900"
1278 "#b58900"
1279 "#cb4b16"
1280 "#dc322f"
1281 "#d33682"
1282 "#6c71c4"
1283 "#69b7f0"
1284 "#69cabf"
1285 "#b4c342"
1286 "#deb542"
1287 "#f2804f"
1288 "#ff6e64"
1289 "#f771ac"
1290 "#9ea0e5"))
1291
1292 (context-coloring-define-theme
1293 'spacegray
1294 :recede t
1295 :colors '("#ffffff"
1296 "#89aaeb"
1297 "#c189eb"
1298 "#bf616a"
1299 "#dca432"
1300 "#ebcb8b"
1301 "#b4eb89"
1302 "#89ebca"))
1303
1304 (context-coloring-define-theme
1305 'tango
1306 :recede t
1307 :colors '("#2e3436"
1308 "#346604"
1309 "#204a87"
1310 "#5c3566"
1311 "#a40000"
1312 "#b35000"
1313 "#c4a000"
1314 "#8ae234"
1315 "#8cc4ff"
1316 "#ad7fa8"
1317 "#ef2929"
1318 "#fcaf3e"
1319 "#fce94f"))
1320
1321 (context-coloring-define-theme
1322 'zenburn
1323 :recede t
1324 :colors '("#dcdccc"
1325 "#93e0e3"
1326 "#bfebbf"
1327 "#f0dfaf"
1328 "#dfaf8f"
1329 "#cc9393"
1330 "#dc8cc3"
1331 "#94bff3"
1332 "#9fc59f"
1333 "#d0bf8f"
1334 "#dca3a3"))
1335
1336
1337 ;;; Change detection
1338
1339 (defvar-local context-coloring-colorize-idle-timer nil
1340 "The currently-running idle timer.")
1341
1342 (defcustom context-coloring-delay 0.25
1343 "Delay between a buffer update and colorization.
1344
1345 Increase this if your machine is high-performing. Decrease it if
1346 it ain't.
1347
1348 Supported modes: `js-mode', `js3-mode', `emacs-lisp-mode'"
1349 :group 'context-coloring)
1350
1351 (defun context-coloring-setup-idle-change-detection ()
1352 "Setup idle change detection."
1353 (add-hook
1354 'after-change-functions 'context-coloring-change-function nil t)
1355 (add-hook
1356 'kill-buffer-hook 'context-coloring-teardown-idle-change-detection nil t)
1357 (setq context-coloring-colorize-idle-timer
1358 (run-with-idle-timer
1359 context-coloring-delay
1360 t
1361 'context-coloring-maybe-colorize
1362 (current-buffer))))
1363
1364 (defun context-coloring-teardown-idle-change-detection ()
1365 "Teardown idle change detection."
1366 (context-coloring-kill-scopifier)
1367 (when context-coloring-colorize-idle-timer
1368 (cancel-timer context-coloring-colorize-idle-timer))
1369 (remove-hook
1370 'kill-buffer-hook 'context-coloring-teardown-idle-change-detection t)
1371 (remove-hook
1372 'after-change-functions 'context-coloring-change-function t))
1373
1374
1375 ;;; Built-in dispatches
1376
1377 (context-coloring-define-dispatch
1378 'javascript-node
1379 :modes '(js-mode js3-mode)
1380 :executable "scopifier"
1381 :command "scopifier"
1382 :version "v1.1.1")
1383
1384 (context-coloring-define-dispatch
1385 'javascript-js2
1386 :modes '(js2-mode)
1387 :colorizer 'context-coloring-js2-colorize
1388 :setup
1389 (lambda ()
1390 (add-hook 'js2-post-parse-callbacks 'context-coloring-colorize nil t))
1391 :teardown
1392 (lambda ()
1393 (remove-hook 'js2-post-parse-callbacks 'context-coloring-colorize t)))
1394
1395 (context-coloring-define-dispatch
1396 'emacs-lisp
1397 :modes '(emacs-lisp-mode)
1398 :colorizer 'context-coloring-elisp-colorize-buffer
1399 :setup 'context-coloring-setup-idle-change-detection
1400 :teardown 'context-coloring-teardown-idle-change-detection)
1401
1402 (defun context-coloring-dispatch (&optional callback)
1403 "Determine the optimal track for scopification / coloring of
1404 the current buffer, then execute it.
1405
1406 Invoke CALLBACK when complete. It is invoked synchronously for
1407 elisp tracks, and asynchronously for shell command tracks."
1408 (let* ((dispatch (context-coloring-get-dispatch-for-mode major-mode))
1409 (colorizer (plist-get dispatch :colorizer))
1410 (scopifier (plist-get dispatch :scopifier))
1411 (command (plist-get dispatch :command))
1412 interrupted-p)
1413 (cond
1414 ((or colorizer scopifier)
1415 (setq interrupted-p
1416 (catch 'interrupted
1417 (cond
1418 (colorizer
1419 (funcall colorizer))
1420 (scopifier
1421 (context-coloring-apply-tokens (funcall scopifier))))))
1422 (cond
1423 (interrupted-p
1424 (setq context-coloring-changed t))
1425 (t
1426 (when callback (funcall callback)))))
1427 (command
1428 (context-coloring-scopify-and-colorize command callback)))))
1429
1430
1431 ;;; Minor mode
1432
1433 ;;;###autoload
1434 (define-minor-mode context-coloring-mode
1435 "Context-based code coloring, inspired by Douglas Crockford."
1436 nil " Context" nil
1437 (if (not context-coloring-mode)
1438 (progn
1439 (let ((dispatch (context-coloring-get-dispatch-for-mode major-mode)))
1440 (when dispatch
1441 (let ((command (plist-get dispatch :command))
1442 (teardown (plist-get dispatch :teardown)))
1443 (when command
1444 (context-coloring-teardown-idle-change-detection))
1445 (when teardown
1446 (funcall teardown)))))
1447 (font-lock-mode)
1448 (jit-lock-mode t))
1449
1450 ;; Font lock is incompatible with this mode; the converse is also true.
1451 (font-lock-mode 0)
1452 (jit-lock-mode nil)
1453
1454 ;; ...but we do use font-lock functions here.
1455 (font-lock-set-defaults)
1456
1457 ;; Safely change the valye of this function as necessary.
1458 (make-local-variable 'font-lock-syntactic-face-function)
1459
1460 (let ((dispatch (context-coloring-get-dispatch-for-mode major-mode)))
1461 (if dispatch
1462 (progn
1463 (let ((command (plist-get dispatch :command))
1464 (version (plist-get dispatch :version))
1465 (executable (plist-get dispatch :executable))
1466 (setup (plist-get dispatch :setup))
1467 (colorize-initially-p t))
1468 (when command
1469 ;; Shell commands recolor on change, idly.
1470 (cond
1471 ((and executable
1472 (null (executable-find executable)))
1473 (message "Executable \"%s\" not found" executable)
1474 (setq colorize-initially-p nil))
1475 (version
1476 (context-coloring-check-scopifier-version
1477 (lambda (sufficient-p)
1478 (if sufficient-p
1479 (progn
1480 (context-coloring-setup-idle-change-detection)
1481 (context-coloring-colorize))
1482 (message "Update to the minimum version of \"%s\" (%s)"
1483 executable version))))
1484 (setq colorize-initially-p nil))
1485 (t
1486 (context-coloring-setup-idle-change-detection))))
1487 (when setup
1488 (funcall setup))
1489 ;; Colorize once initially.
1490 (when colorize-initially-p
1491 (let ((context-coloring-parse-interruptable-p nil))
1492 (context-coloring-colorize)))))
1493 (when (null dispatch)
1494 (message "Context coloring is not available for this major mode"))))))
1495
1496 (provide 'context-coloring)
1497
1498 ;;; context-coloring.el ends here