Close atom-community#152 - Add respect shebang

Author: erran

examples/version_info.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python2.7
+import sys
+
+print(sys.version_info)

lib/code-context.coffee

@@ -40,3 +40,21 @@ class CodeContext
     newlineCount = Number(@lineNumber)
     newlines = Array(newlineCount).join("\n")
     "#{newlines}#{code}"
+
+  shebangCommand: ->
+    code = @textSource?.getText()
+    return unless code
+
+    lines = code.split("\n")
+    firstLine = lines[0]
+    return unless firstLine.match(/^#!/)
+    firstLine = firstLine.replace(/^#! /, '#!')
+    paths = firstLine.split(' ')
+    parts = paths[0]?.split('/')
+
+    if parts? and parts.length > 1
+      script = parts[parts.length-1]
+    else
+      script = parts.first.sub('#!', '')
+
+    script = if script == 'env' then paths[paths.length-1] else script

lib/script-view.coffee

@@ -158,7 +158,7 @@ class ScriptView extends View
     try
       if not @runOptions.cmd? or @runOptions.cmd is ''
         # Precondition: lang? and lang of grammarMap
-        commandContext.command = grammarMap[codeContext.lang][codeContext.argType].command
+        commandContext.command = codeContext.shebangCommand() or grammarMap[codeContext.lang][codeContext.argType].command
       else
         commandContext.command = @runOptions.cmd
 

spec/code-context-spec.coffee

@@ -46,3 +46,29 @@ describe 'CodeContext', ->
       code = @codeContext.getCode()
       expect(typeof code).toEqual('string')
       expect(code).toMatch("print 'hello world!'")
+
+  describe 'shebangCommand', ->
+    it 'returns undefined when no code is available', ->
+      expect(@codeContext.shebangCommand()).toBe(undefined)
+
+    it 'returns undefined when no shebang is found', ->
+      @codeContext.textSource = @dummyTextSource
+      expect(@codeContext.shebangCommand()).toBe(undefined)
+
+    it 'returns the command name when a shebang is found', ->
+      @codeContext.textSource = @dummyTextSource
+      @codeContext.textSource.getText = ->
+        "#!/bin/bash\necho 'hello from bash!'"
+      expect(@codeContext.shebangCommand()).toMatch('bash')
+
+      @codeContext.textSource.getText = ->
+        "#!/usr/bin/env ruby\nputs 'hello from ruby!'"
+      expect(@codeContext.shebangCommand()).toMatch('ruby')
+
+      @codeContext.textSource.getText = ->
+        "#!/usr/bin/python2.7\nprint 'hello from python!'"
+      expect(@codeContext.shebangCommand()).toMatch('python2.7')
+
+      @codeContext.textSource.getText = ->
+        "#!/usr/bin/python3\nprint 'hello from python3!'"
+      expect(@codeContext.shebangCommand()).toMatch('python3')