support for directives

1 files changed, 11 insertions, 0 deletions

diff --git a/lib/scope.js b/lib/scope.js
index 79d6e91d..dfbff4b8 100644
--- a/lib/scope.js
+++ b/lib/scope.js
@@ -82,6 +82,11 @@ AST_Toplevel.DEFMETHOD("figure_out_scope", function(){
             scope = save_scope;
             return true;        // don't descend again in TreeWalker
         }
+        if (node instanceof AST_Directive) {
+            node.scope = scope;
+            push_uniq(scope.directives, node.value);
+            return true;
+        }
         if (node instanceof AST_With) {
             for (var s = scope; s; s = s.parent_scope)
                 s.uses_with = true;
@@ -193,6 +198,7 @@ AST_Toplevel.DEFMETHOD("figure_out_scope", function(){
 });
 
 AST_Scope.DEFMETHOD("init_scope_vars", function(){
+    this.directives = [];     // contains the directives defined in this scope, i.e. "use strict"
     this.variables = {};      // map name to AST_SymbolVar (variables defined in this scope; includes functions)
     this.functions = {};      // map name to AST_SymbolDefun (functions defined in this scope)
     this.uses_with = false;   // will be set to true if this or some nested scope uses the `with` statement
@@ -233,6 +239,11 @@ AST_Scope.DEFMETHOD("find_variable", function(name){
         : (this.parent_scope && this.parent_scope.find_variable(name));
 });
 
+AST_Scope.DEFMETHOD("has_directive", function(value){
+    return this.parent_scope && this.parent_scope.has_directive(value)
+        || (this.directives.indexOf(value) >= 0 ? this : null);
+});
+
 AST_Scope.DEFMETHOD("def_function", function(symbol){
     this.functions[symbol.name] = this.def_variable(symbol);
 });