Merge pull request #34 from petems/fix_variables_starting_with_underscore

Allows variables starting with underscores

Author: Wliu

grammars/puppet.cson

@@ -410,7 +410,7 @@
         'captures':
           '1':
             'name': 'punctuation.definition.variable.puppet'
-        'match': '(\\$)([a-zA-Zx7f-xff\\$]|::)([a-zA-Z0-9_x7f-xff\\$]|::)*\\b'
+        'match': '(\\$)((::)?[a-z]\\w*)*((::)?[a-z_]\\w*)\\b'
         'name': 'variable.other.readwrite.global.puppet'
       }
       {

spec/puppet-spec.coffee

@@ -55,3 +55,20 @@ describe "Puppet grammar", ->
     it "tokenizes require => variable as a parameter", ->
       {tokens} = grammar.tokenizeLine("require => Class['foo']")
       expect(tokens[0]).toEqual value: 'require ', scopes: ['source.puppet', 'constant.other.key.puppet']
+
+    it "tokenizes regular variables", ->
+      {tokens} = grammar.tokenizeLine('$foo')
+      expect(tokens[0]).toEqual value: '$', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet', 'punctuation.definition.variable.puppet']
+      expect(tokens[1]).toEqual value: 'foo', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet']
+
+      {tokens} = grammar.tokenizeLine('$_foo')
+      expect(tokens[0]).toEqual value: '$', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet', 'punctuation.definition.variable.puppet']
+      expect(tokens[1]).toEqual value: '_foo', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet']
+
+      {tokens} = grammar.tokenizeLine('$_foo_')
+      expect(tokens[0]).toEqual value: '$', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet', 'punctuation.definition.variable.puppet']
+      expect(tokens[1]).toEqual value: '_foo_', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet']
+
+      {tokens} = grammar.tokenizeLine('$::foo')
+      expect(tokens[0]).toEqual value: '$', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet', 'punctuation.definition.variable.puppet']
+      expect(tokens[1]).toEqual value: '::foo', scopes: ['source.puppet', 'variable.other.readwrite.global.puppet']