Chore: Add shebang to embedded bash documents

server/src/__tests__/embedded-languages.test.ts

@@ -11,6 +11,28 @@ import { generateParser } from '../tree-sitter/parser'
 import { TextDocument } from 'vscode-languageserver-textdocument'
 import { type EmbeddedLanguageType } from '../lib/src/types/embedded-languages'
 import { imports } from '../embedded-languages/python-support'
+import { shebang } from '../embedded-languages/bash-support'
+
+describe('Create basic embedded bash documents', () => {
+  beforeAll(async () => {
+    if (!analyzer.hasParser()) {
+      const parser = await generateParser()
+      analyzer.initialize(parser)
+    }
+    analyzer.resetAnalyzedDocuments()
+  })
+
+  test.each([
+    [
+      'basic',
+      'foo(){\nBAR=""\n}',
+      `${shebang}foo(){\nBAR=""\n}`
+    ]
+  ])('%s', async (description, input, result) => {
+    const embeddedContent = await createEmbeddedContent(input, 'bash')
+    expect(embeddedContent).toEqual(result)
+  })
+})
 
 describe('Create various basic embedded python documents', () => {
   beforeAll(async () => {

server/src/embedded-languages/bash-support.ts

@@ -8,6 +8,8 @@ import * as TreeSitterUtils from '../tree-sitter/utils'
 import { initEmbeddedLanguageDoc, insertTextIntoEmbeddedLanguageDoc } from './utils'
 import { type EmbeddedLanguageDoc } from '../lib/src/types/embedded-languages'
 
+export const shebang = '#!/bin/sh\n'
+
 export const generateBashEmbeddedLanguageDoc = (analyzedDocument: AnalyzedDocument): EmbeddedLanguageDoc => {
   const embeddedLanguageDoc = initEmbeddedLanguageDoc(analyzedDocument.document, 'bash')
   TreeSitterUtils.forEach(analyzedDocument.tree.rootNode, (node) => {
@@ -21,5 +23,6 @@ export const generateBashEmbeddedLanguageDoc = (analyzedDocument: AnalyzedDocume
         return false
     }
   })
+  insertTextIntoEmbeddedLanguageDoc(embeddedLanguageDoc, 0, 0, shebang)
   return embeddedLanguageDoc
 }