Merge pull request #276 from alxtz/master

[tree-sitter] add variable identifier support in assignments expressions

Author: darangi

grammars/tree-sitter-ruby.cson

@@ -161,6 +161,8 @@ scopes:
   '")"': 'punctuation.definition.parameters'
 
   'method > identifier': 'entity.name.function'
+  'assignment > identifier': 'variable'
+
   'singleton_method > identifier:nth-child(3)': 'entity.name.function'
   'setter > identifier': 'entity.name.function'
   'call > identifier:nth-child(2)': 'entity.name.function'

spec/tree-sitter-spec.js

@@ -108,4 +108,26 @@ describe('Tree-sitter Ruby grammar', () => {
       '.source.ruby .keyword.control'
     )
   })
+
+  it('tokenizes variable in assignment expressions', async () => {
+    const editor = await atom.workspace.open('foo.rb')
+    editor.setText(dedent`
+      a = 10
+    `)
+
+    expect(editor.scopeDescriptorForBufferPosition([0, 0]).toString()).toBe(
+      '.source.ruby .variable'
+    )
+  })
+
+  it('does not tokenizes method call in assignment expressions', async () => {
+    const editor = await atom.workspace.open('foo.rb')
+    editor.setText(dedent`
+      foo() = 10
+    `)
+
+    expect(editor.scopeDescriptorForBufferPosition([0, 0]).toString()).not.toBe(
+      '.source.ruby .variable'
+    )
+  })
 })