Fix the shebang parser

Author: skovhus

server/src/util/__tests__/shebang.test.ts

@@ -9,23 +9,18 @@ describe('hasBashShebang', () => {
     expect(hasBashShebang(`#!/usr/bin/env python2.7\n# set -x`)).toBe(false)
   })
 
-  it('returns true for "#!/bin/sh -"', () => {
-    expect(hasBashShebang('#!/bin/sh -')).toBe(true)
-    expect(hasBashShebang('#!/bin/sh - ')).toBe(true)
+  it('returns false for "#!/usr/bin/pwsh"', () => {
+    expect(hasBashShebang('#!/usr/bin/pwsh')).toBe(false)
   })
 
-  it('returns true for "#!/usr/bin/env bash"', () => {
-    expect(hasBashShebang('#!/usr/bin/env bash')).toBe(true)
-    expect(hasBashShebang('#!/usr/bin/env bash ')).toBe(true)
-  })
-
-  it('returns true for "#!/bin/sh"', () => {
-    expect(hasBashShebang('#!/bin/sh')).toBe(true)
-    expect(hasBashShebang('#!/bin/sh ')).toBe(true)
-  })
-
-  it('returns true for "#!/bin/bash"', () => {
-    expect(hasBashShebang('#!/bin/bash')).toBe(true)
-    expect(hasBashShebang('#!/bin/bash ')).toBe(true)
+  test.each([
+    ['#!/bin/sh -'],
+    ['#!/usr/bin/env bash'],
+    ['#!/bin/sh'],
+    ['#!/bin/bash'],
+    ['#!/bin/bash -u'],
+  ])('returns true for %p', (command) => {
+    expect(hasBashShebang(command)).toBe(true)
+    expect(hasBashShebang(`${command} `)).toBe(true)
   })
 })

server/src/util/shebang.ts

@@ -6,11 +6,19 @@ export function getShebang(fileContent: string): string | null {
     return null
   }
 
-  return match[1].replace('-', '').trim()
+  return match[1]
 }
 
+/**
+ * Check if the given shebang is a bash shebang.
+ */
 export function isBashShebang(shebang: string): boolean {
-  return shebang.endsWith('bash') || shebang.endsWith('sh')
+  return (
+    shebang.startsWith('/bin/bash') ||
+    shebang.startsWith('/bin/sh') ||
+    shebang.startsWith('/usr/bin/env bash') ||
+    shebang.startsWith('/usr/bin/env sh')
+  )
 }
 
 export function hasBashShebang(fileContent: string): boolean {