[elpa] master 16cee40 150/271: Pass test for block scopes.

[Jackson Ray Hamilton]

branch: master
commit 16cee407fe10692f34e7dae8478fcc100217e6fb
Author: Jackson Ray Hamilton <jack...@jacksonrayhamilton.com>
Commit: Jackson Ray Hamilton <jack...@jacksonrayhamilton.com>

    Pass test for block scopes.
---
 context-coloring.el |   37 ++++++++++++++++++++++++++++---------
 1 files changed, 28 insertions(+), 9 deletions(-)

diff --git a/context-coloring.el b/context-coloring.el
index 8891c21..99c56a7 100644
--- a/context-coloring.el
+++ b/context-coloring.el
@@ -46,6 +46,23 @@
   "This file's directory.")
 
 
+;;; Customizable options
+
+(defcustom context-coloring-delay 0.25
+  "Delay between a buffer update and colorization.
+
+Increase this if your machine is high-performing. Decrease it if it ain't."
+  :group 'context-coloring)
+
+(defcustom context-coloring-block-scopes nil
+  "If non-nil, add block scopes to the scope hierarchy.
+
+The block-scope-inducing `let' and `const' are introduced in
+ES6. If you are writing ES6 code, then turn this on; otherwise,
+confusion will ensue."
+  :group 'context-coloring)
+
+
 ;;; Local variables
 
 (defvar-local context-coloring-buffer nil
@@ -196,11 +213,19 @@ For example: \"context-coloring-level-1-face\"."
 
 (defsubst context-coloring-js2-scope-level (scope)
   "Gets the level of SCOPE."
-  (let ((level 0))
+  (let ((level 0)
+        enclosing-scope)
     (while (and (not (null scope))
                 (not (null (js2-node-parent scope)))
-                (not (null (setq scope (js2-node-get-enclosing-scope scope)))))
-      (setq level (+ level 1)))
+                (not (null (setq enclosing-scope (js2-node-get-enclosing-scope 
scope)))))
+      (when (or context-coloring-block-scopes
+                (let ((type (js2-scope-type scope)))
+                  (or (= type js2-SCRIPT)
+                      (= type js2-FUNCTION)
+                      (= type js2-CATCH)
+                      (= type js2-WITH))))
+        (setq level (+ level 1)))
+      (setq scope enclosing-scope))
     level))
 
 ;; Obtained from js2-refactor.el/js2r-vars.el
@@ -360,12 +385,6 @@ of the current buffer, then does it."
 
 ;;; Colorization
 
-(defcustom context-coloring-delay 0.25
-  "Delay between a buffer update and colorization.
-
-Increase this if your machine is high-performing. Decrease it if it ain't."
-  :group 'context-coloring)
-
 (defun context-coloring-colorize ()
   "Colors the current buffer by function context."
   (interactive)