@fairyhunter13/opentui-core 0.1.91 → 0.1.94

package/src/lib/tree-sitter/parser.worker.ts

@@ -1,1001 +0,0 @@
-// Use type-only imports for type safety; values loaded dynamically to prevent WASM OOB crash
-import type { Parser, Query, Tree, Language, Edit, QueryCapture, Range } from "web-tree-sitter"
-import { mkdir } from "fs/promises"
-import * as path from "path"
-import type {
-  HighlightRange,
-  HighlightResponse,
-  SimpleHighlight,
-  FiletypeParserOptions,
-  PerformanceStats,
-  InjectionMapping,
-} from "./types.js"
-import { DownloadUtils } from "./download-utils.js"
-import { isMainThread } from "worker_threads"
-import { isBunfsPath, normalizeBunfsPath } from "../bunfs.js"
-
-// Runtime references loaded dynamically in initialize() to prevent WASM heap
-// initialization before Parser.init() is called (causes OOB memory access crash)
-let TreeSitter: { Parser: typeof Parser; Query: typeof Query; Language: typeof Language }
-
-const self = globalThis
-
-type ParserState = {
-  parser: Parser
-  tree: Tree
-  queries: {
-    highlights: Query
-    injections?: Query
-  }
-  filetype: string
-  content: string
-  injectionMapping?: InjectionMapping
-}
-
-interface FiletypeParser {
-  filetype: string
-  queries: {
-    highlights: Query
-    injections?: Query
-  }
-  language: Language
-  injectionMapping?: InjectionMapping
-}
-
-interface ReusableParserState {
-  parser: Parser
-  filetypeParser: FiletypeParser
-  queries: {
-    highlights: Query
-    injections?: Query
-  }
-}
-
-class ParserWorker {
-  private bufferParsers: Map<number, ParserState> = new Map()
-  private filetypeParserOptions: Map<string, FiletypeParserOptions> = new Map()
-  private filetypeParsers: Map<string, FiletypeParser> = new Map()
-  private filetypeParserPromises: Map<string, Promise<FiletypeParser | undefined>> = new Map()
-  private reusableParsers: Map<string, ReusableParserState> = new Map()
-  private reusableParserPromises: Map<string, Promise<ReusableParserState | undefined>> = new Map()
-  private initializePromise: Promise<void> | undefined
-  public performance: PerformanceStats
-  private dataPath: string | undefined
-  private tsDataPath: string | undefined
-  private initialized: boolean = false
-
-  constructor() {
-    this.performance = {
-      averageParseTime: 0,
-      parseTimes: [],
-      averageQueryTime: 0,
-      queryTimes: [],
-    }
-  }
-
-  private async fetchQueries(sources: string[], filetype: string): Promise<string> {
-    if (!this.tsDataPath) {
-      return ""
-    }
-    return DownloadUtils.fetchHighlightQueries(sources, this.tsDataPath, filetype)
-  }
-
-  async initialize({ dataPath }: { dataPath: string }) {
-    if (this.initializePromise) {
-      return this.initializePromise
-    }
-    this.initializePromise = new Promise(async (resolve, reject) => {
-      this.dataPath = dataPath
-      this.tsDataPath = path.join(dataPath, "tree-sitter")
-
-      try {
-        await mkdir(path.join(this.tsDataPath, "languages"), { recursive: true })
-        await mkdir(path.join(this.tsDataPath, "queries"), { recursive: true })
-
-        // Dynamic import: load web-tree-sitter at runtime to prevent WASM OOB crash
-        const mod = await import("web-tree-sitter")
-        const P = (mod.default ?? mod) as unknown as typeof Parser
-        TreeSitter = {
-          Parser: P,
-          Query: (P as unknown as Record<string, unknown>).Query as typeof Query,
-          Language: (P as unknown as Record<string, unknown>).Language as typeof Language,
-        }
-
-        let { default: treeWasm } = await import("web-tree-sitter/tree-sitter.wasm" as string, {
-          with: { type: "wasm" },
-        })
-
-        if (isBunfsPath(treeWasm)) {
-          treeWasm = normalizeBunfsPath(path.parse(treeWasm).base)
-        }
-
-        await TreeSitter.Parser.init({
-          locateFile() {
-            return treeWasm
-          },
-        })
-
-        this.initialized = true
-        resolve()
-      } catch (error) {
-        reject(error)
-      }
-    })
-    return this.initializePromise
-  }
-
-  public addFiletypeParser(filetypeParser: FiletypeParserOptions) {
-    this.filetypeParserOptions.set(filetypeParser.filetype, filetypeParser)
-  }
-
-  private async createQueries(
-    filetypeParser: FiletypeParserOptions,
-    language: Language,
-  ): Promise<
-    | {
-        highlights: Query
-        injections?: Query
-      }
-    | undefined
-  > {
-    try {
-      const highlightQueryContent = await this.fetchQueries(filetypeParser.queries.highlights, filetypeParser.filetype)
-      if (!highlightQueryContent) {
-        console.error("Failed to fetch highlight queries for:", filetypeParser.filetype)
-        return undefined
-      }
-
-      const highlightsQuery = new TreeSitter.Query(language, highlightQueryContent)
-      const result: { highlights: Query; injections?: Query } = {
-        highlights: highlightsQuery,
-      }
-
-      if (filetypeParser.queries.injections && filetypeParser.queries.injections.length > 0) {
-        const injectionQueryContent = await this.fetchQueries(
-          filetypeParser.queries.injections,
-          filetypeParser.filetype,
-        )
-        if (injectionQueryContent) {
-          result.injections = new TreeSitter.Query(language, injectionQueryContent)
-        }
-      }
-
-      return result
-    } catch (error) {
-      console.error("Error creating queries for", filetypeParser.filetype, filetypeParser.queries)
-      console.error(error)
-      return undefined
-    }
-  }
-
-  private async loadLanguage(languageSource: string): Promise<Language | undefined> {
-    if (!this.initialized || !this.tsDataPath) {
-      return undefined
-    }
-
-    const result = await DownloadUtils.downloadOrLoad(languageSource, this.tsDataPath, "languages", ".wasm", false)
-
-    if (result.error) {
-      console.error(`Error loading language ${languageSource}:`, result.error)
-      return undefined
-    }
-
-    if (!result.filePath) {
-      return undefined
-    }
-
-    // Normalize path for Windows compatibility - tree-sitter expects forward slashes
-    const normalizedPath = result.filePath.replaceAll("\\", "/")
-
-    try {
-      const language = await TreeSitter.Language.load(normalizedPath)
-      return language
-    } catch (error) {
-      console.error(`Error loading language from ${normalizedPath}:`, error)
-      return undefined
-    }
-  }
-
-  private async resolveFiletypeParser(filetype: string): Promise<FiletypeParser | undefined> {
-    if (this.filetypeParsers.has(filetype)) {
-      return this.filetypeParsers.get(filetype)
-    }
-
-    if (this.filetypeParserPromises.has(filetype)) {
-      return this.filetypeParserPromises.get(filetype)
-    }
-
-    const loadingPromise = this.loadFiletypeParser(filetype)
-    this.filetypeParserPromises.set(filetype, loadingPromise)
-
-    try {
-      const result = await loadingPromise
-      if (result) {
-        this.filetypeParsers.set(filetype, result)
-      }
-      return result
-    } finally {
-      this.filetypeParserPromises.delete(filetype)
-    }
-  }
-
-  private async loadFiletypeParser(filetype: string): Promise<FiletypeParser | undefined> {
-    const filetypeParserOptions = this.filetypeParserOptions.get(filetype)
-    if (!filetypeParserOptions) {
-      return undefined
-    }
-    const language = await this.loadLanguage(filetypeParserOptions.wasm)
-    if (!language) {
-      return undefined
-    }
-    const queries = await this.createQueries(filetypeParserOptions, language)
-    if (!queries) {
-      console.error("Failed to create queries for:", filetype)
-      return undefined
-    }
-    const filetypeParser: FiletypeParser = {
-      ...filetypeParserOptions,
-      queries,
-      language,
-    }
-    return filetypeParser
-  }
-
-  public async preloadParser(filetype: string) {
-    return this.resolveFiletypeParser(filetype)
-  }
-
-  private async getReusableParser(filetype: string): Promise<ReusableParserState | undefined> {
-    if (this.reusableParsers.has(filetype)) {
-      return this.reusableParsers.get(filetype)
-    }
-
-    if (this.reusableParserPromises.has(filetype)) {
-      return this.reusableParserPromises.get(filetype)
-    }
-
-    const creationPromise = this.createReusableParser(filetype)
-    this.reusableParserPromises.set(filetype, creationPromise)
-
-    try {
-      const result = await creationPromise
-      if (result) {
-        this.reusableParsers.set(filetype, result)
-      }
-      return result
-    } finally {
-      this.reusableParserPromises.delete(filetype)
-    }
-  }
-
-  private async createReusableParser(filetype: string): Promise<ReusableParserState | undefined> {
-    const filetypeParser = await this.resolveFiletypeParser(filetype)
-    if (!filetypeParser) {
-      return undefined
-    }
-
-    const parser = new TreeSitter.Parser()
-    parser.setLanguage(filetypeParser.language)
-
-    const reusableState: ReusableParserState = {
-      parser,
-      filetypeParser,
-      queries: filetypeParser.queries,
-    }
-
-    return reusableState
-  }
-
-  async handleInitializeParser(
-    bufferId: number,
-    version: number,
-    content: string,
-    filetype: string,
-    messageId: string,
-  ) {
-    const filetypeParser = await this.resolveFiletypeParser(filetype)
-
-    if (!filetypeParser) {
-      self.postMessage({
-        type: "PARSER_INIT_RESPONSE",
-        bufferId,
-        messageId,
-        hasParser: false,
-        warning: `No parser available for filetype ${filetype}`,
-      })
-      return
-    }
-
-    const parser = new TreeSitter.Parser()
-    parser.setLanguage(filetypeParser.language)
-    const tree = parser.parse(content)
-    if (!tree) {
-      self.postMessage({
-        type: "PARSER_INIT_RESPONSE",
-        bufferId,
-        messageId,
-        hasParser: false,
-        error: "Failed to parse buffer",
-      })
-      return
-    }
-
-    const parserState: ParserState = {
-      parser,
-      tree,
-      queries: filetypeParser.queries,
-      filetype,
-      content,
-      injectionMapping: filetypeParser.injectionMapping,
-    }
-    this.bufferParsers.set(bufferId, parserState)
-
-    self.postMessage({
-      type: "PARSER_INIT_RESPONSE",
-      bufferId,
-      messageId,
-      hasParser: true,
-    })
-    const highlights = await this.initialQuery(parserState)
-    self.postMessage({
-      type: "HIGHLIGHT_RESPONSE",
-      bufferId,
-      version,
-      ...highlights,
-    })
-  }
-
-  private async initialQuery(parserState: ParserState) {
-    const query = parserState.queries.highlights
-    const matches: QueryCapture[] = query.captures(parserState.tree.rootNode)
-    let injectionRanges = new Map<string, Array<{ start: number; end: number }>>()
-
-    if (parserState.queries.injections) {
-      const injectionResult = await this.processInjections(parserState)
-      matches.push(...injectionResult.captures)
-      injectionRanges = injectionResult.injectionRanges
-    }
-
-    return this.getHighlights(parserState, matches, injectionRanges)
-  }
-
-  private getNodeText(node: any, content: string): string {
-    return content.substring(node.startIndex, node.endIndex)
-  }
-
-  private async processInjections(
-    parserState: ParserState,
-  ): Promise<{ captures: QueryCapture[]; injectionRanges: Map<string, Array<{ start: number; end: number }>> }> {
-    const injectionMatches: QueryCapture[] = []
-    const injectionRanges = new Map<string, Array<{ start: number; end: number }>>()
-
-    if (!parserState.queries.injections) {
-      return { captures: injectionMatches, injectionRanges }
-    }
-
-    const content = parserState.content
-    const injectionCaptures = parserState.queries.injections.captures(parserState.tree.rootNode)
-    const languageGroups = new Map<string, Array<{ node: any; name: string }>>()
-
-    // Use the injection mapping stored in the parser state
-    const injectionMapping = parserState.injectionMapping
-
-    for (const capture of injectionCaptures) {
-      const captureName = capture.name
-
-      if (captureName === "injection.content" || captureName.includes("injection")) {
-        const nodeType = capture.node.type
-        let targetLanguage: string | undefined
-
-        // First, check if there's a direct node type mapping
-        if (injectionMapping?.nodeTypes && injectionMapping.nodeTypes[nodeType]) {
-          targetLanguage = injectionMapping.nodeTypes[nodeType]
-        } else if (nodeType === "code_fence_content") {
-          // For code fence content, try to extract language from info_string
-          const parent = capture.node.parent
-          if (parent) {
-            const infoString = parent.children.find((child: any) => child.type === "info_string")
-            if (infoString) {
-              const languageNode = infoString.children.find((child: any) => child.type === "language")
-              if (languageNode) {
-                const languageName = this.getNodeText(languageNode, content)
-
-                if (injectionMapping?.infoStringMap && injectionMapping.infoStringMap[languageName]) {
-                  targetLanguage = injectionMapping.infoStringMap[languageName]
-                } else {
-                  targetLanguage = languageName
-                }
-              }
-            }
-          }
-        }
-
-        if (targetLanguage) {
-          if (!languageGroups.has(targetLanguage)) {
-            languageGroups.set(targetLanguage, [])
-          }
-          languageGroups.get(targetLanguage)!.push({ node: capture.node, name: capture.name })
-        }
-      }
-    }
-
-    // Process each language group
-    for (const [language, captures] of languageGroups.entries()) {
-      const injectedParser = await this.getReusableParser(language)
-
-      if (!injectedParser) {
-        console.warn(`No parser found for injection language: ${language}`)
-        continue
-      }
-
-      // Track injection ranges for this language
-      if (!injectionRanges.has(language)) {
-        injectionRanges.set(language, [])
-      }
-
-      const parser = injectedParser.parser
-      for (const { node: injectionNode } of captures) {
-        try {
-          // Record the injection range
-          injectionRanges.get(language)!.push({
-            start: injectionNode.startIndex,
-            end: injectionNode.endIndex,
-          })
-
-          const injectionContent = this.getNodeText(injectionNode, content)
-          const tree = parser.parse(injectionContent)
-
-          if (tree) {
-            const matches = injectedParser.queries.highlights.captures(tree.rootNode)
-
-            // Create new QueryCapture objects with offset positions
-            for (const match of matches) {
-              // Calculate offset positions by creating a new capture with adjusted node properties
-              // Store the injected query reference so we can look up properties correctly
-              const offsetCapture: QueryCapture & { _injectedQuery?: Query } = {
-                name: match.name,
-                patternIndex: match.patternIndex,
-                _injectedQuery: injectedParser.queries.highlights, // Store the correct query reference
-                node: {
-                  ...match.node,
-                  startPosition: {
-                    row: match.node.startPosition.row + injectionNode.startPosition.row,
-                    column:
-                      match.node.startPosition.row === 0
-                        ? match.node.startPosition.column + injectionNode.startPosition.column
-                        : match.node.startPosition.column,
-                  },
-                  endPosition: {
-                    row: match.node.endPosition.row + injectionNode.startPosition.row,
-                    column:
-                      match.node.endPosition.row === 0
-                        ? match.node.endPosition.column + injectionNode.startPosition.column
-                        : match.node.endPosition.column,
-                  },
-                  startIndex: match.node.startIndex + injectionNode.startIndex,
-                  endIndex: match.node.endIndex + injectionNode.startIndex,
-                } as any, // Cast to any since we're creating a pseudo-node
-              }
-
-              injectionMatches.push(offsetCapture)
-            }
-
-            tree.delete()
-          }
-        } catch (error) {
-          console.error(`Error processing injection for language ${language}:`, error)
-        }
-      }
-
-      // NOTE: Do NOT call parser.delete() here - this is a reusable parser!
-    }
-
-    return { captures: injectionMatches, injectionRanges }
-  }
-
-  private editToRange(edit: Edit): Range {
-    return {
-      startPosition: {
-        column: edit.startPosition.column,
-        row: edit.startPosition.row,
-      },
-      endPosition: {
-        column: edit.newEndPosition.column,
-        row: edit.newEndPosition.row,
-      },
-      startIndex: edit.startIndex,
-      endIndex: edit.newEndIndex,
-    }
-  }
-
-  async handleEdits(
-    bufferId: number,
-    content: string,
-    edits: Edit[],
-  ): Promise<{ highlights?: HighlightResponse[]; warning?: string; error?: string }> {
-    const parserState = this.bufferParsers.get(bufferId)
-    if (!parserState) {
-      return { warning: "No parser state found for buffer" }
-    }
-
-    parserState.content = content
-
-    for (const edit of edits) {
-      parserState.tree.edit(edit)
-    }
-
-    const startParse = performance.now()
-
-    const newTree = parserState.parser.parse(content, parserState.tree)
-
-    const endParse = performance.now()
-    const parseTime = endParse - startParse
-    this.performance.parseTimes.push(parseTime)
-    if (this.performance.parseTimes.length > 10) {
-      this.performance.parseTimes.shift()
-    }
-    this.performance.averageParseTime =
-      this.performance.parseTimes.reduce((acc, time) => acc + time, 0) / this.performance.parseTimes.length
-
-    if (!newTree) {
-      return { error: "Failed to parse buffer" }
-    }
-
-    const changedRanges = parserState.tree.getChangedRanges(newTree)
-    parserState.tree = newTree
-
-    const startQuery = performance.now()
-    const matches: QueryCapture[] = []
-
-    if (changedRanges.length === 0) {
-      edits.forEach((edit) => {
-        const range = this.editToRange(edit)
-        changedRanges.push(range)
-      })
-    }
-
-    for (const range of changedRanges) {
-      let node = parserState.tree.rootNode.descendantForPosition(range.startPosition, range.endPosition)
-
-      if (!node) {
-        continue
-      }
-
-      // If we got the root node, query with range to limit scope
-      if (node.equals(parserState.tree.rootNode)) {
-        // WHY ARE RANGES NOT WORKING!?
-        // The changed ranges are not returning anything in some cases
-        // Even this shit somehow returns many lines before the actual range,
-        // and even though expanded by 1000 bytes it does not capture much beyond the actual range.
-        // So freaking weird.
-        const rangeCaptures = parserState.queries.highlights.captures(
-          node,
-          // WTF!?
-          {
-            startIndex: range.startIndex - 100,
-            endIndex: range.endIndex + 1000,
-          },
-        )
-        matches.push(...rangeCaptures)
-        continue
-      }
-
-      while (node && !this.nodeContainsRange(node, range)) {
-        node = node.parent
-      }
-
-      if (!node) {
-        node = parserState.tree.rootNode
-      }
-
-      const nodeCaptures = parserState.queries.highlights.captures(node)
-      matches.push(...nodeCaptures)
-    }
-
-    let injectionRanges = new Map<string, Array<{ start: number; end: number }>>()
-    if (parserState.queries.injections) {
-      const injectionResult = await this.processInjections(parserState)
-      // Only add injection matches that are in the changed ranges
-      // This is a simplification - ideally we'd only process injections in changed ranges
-      matches.push(...injectionResult.captures)
-      injectionRanges = injectionResult.injectionRanges
-    }
-
-    const endQuery = performance.now()
-    const queryTime = endQuery - startQuery
-    this.performance.queryTimes.push(queryTime)
-    if (this.performance.queryTimes.length > 10) {
-      this.performance.queryTimes.shift()
-    }
-    this.performance.averageQueryTime =
-      this.performance.queryTimes.reduce((acc, time) => acc + time, 0) / this.performance.queryTimes.length
-
-    return this.getHighlights(parserState, matches, injectionRanges)
-  }
-
-  private nodeContainsRange(node: any, range: any): boolean {
-    return (
-      node.startPosition.row <= range.startPosition.row &&
-      node.endPosition.row >= range.endPosition.row &&
-      (node.startPosition.row < range.startPosition.row || node.startPosition.column <= range.startPosition.column) &&
-      (node.endPosition.row > range.endPosition.row || node.endPosition.column >= range.endPosition.column)
-    )
-  }
-
-  private getHighlights(
-    parserState: ParserState,
-    matches: QueryCapture[],
-    injectionRanges?: Map<string, Array<{ start: number; end: number }>>,
-  ): { highlights: HighlightResponse[] } {
-    const lineHighlights: Map<number, Map<number, HighlightRange>> = new Map()
-    const droppedHighlights: Map<number, Map<number, HighlightRange>> = new Map()
-
-    for (const match of matches) {
-      const node = match.node
-      const startLine = node.startPosition.row
-      const endLine = node.endPosition.row
-
-      const highlight = {
-        startCol: node.startPosition.column,
-        endCol: node.endPosition.column,
-        group: match.name,
-      }
-
-      if (!lineHighlights.has(startLine)) {
-        lineHighlights.set(startLine, new Map())
-        droppedHighlights.set(startLine, new Map())
-      }
-      if (lineHighlights.get(startLine)?.has(node.id)) {
-        droppedHighlights.get(startLine)?.set(node.id, lineHighlights.get(startLine)?.get(node.id)!)
-      }
-      lineHighlights.get(startLine)?.set(node.id, highlight)
-
-      if (startLine !== endLine) {
-        for (let line = startLine + 1; line <= endLine; line++) {
-          if (!lineHighlights.has(line)) {
-            lineHighlights.set(line, new Map())
-          }
-          const hl: HighlightRange = {
-            startCol: 0,
-            endCol: node.endPosition.column,
-            group: match.name,
-          }
-          lineHighlights.get(line)?.set(node.id, hl)
-        }
-      }
-    }
-
-    return {
-      highlights: Array.from(lineHighlights.entries()).map(([line, lineHighlights]) => ({
-        line,
-        highlights: Array.from(lineHighlights.values()),
-        droppedHighlights: droppedHighlights.get(line) ? Array.from(droppedHighlights.get(line)!.values()) : [],
-      })),
-    }
-  }
-
-  private getSimpleHighlights(
-    matches: QueryCapture[],
-    injectionRanges: Map<string, Array<{ start: number; end: number }>>,
-  ): SimpleHighlight[] {
-    const highlights: SimpleHighlight[] = []
-
-    const flatInjectionRanges: Array<{ start: number; end: number; lang: string }> = []
-    for (const [lang, ranges] of injectionRanges.entries()) {
-      for (const range of ranges) {
-        flatInjectionRanges.push({ ...range, lang })
-      }
-    }
-
-    for (const match of matches) {
-      const node = match.node
-
-      let isInjection = false
-      let injectionLang: string | undefined
-      let containsInjection = false
-      for (const injRange of flatInjectionRanges) {
-        if (node.startIndex >= injRange.start && node.endIndex <= injRange.end) {
-          isInjection = true
-          injectionLang = injRange.lang
-          break
-        } else if (node.startIndex <= injRange.start && node.endIndex >= injRange.end) {
-          containsInjection = true
-          break
-        }
-      }
-
-      const matchQuery = (match as any)._injectedQuery
-      const patternProperties = matchQuery?.setProperties?.[match.patternIndex]
-
-      const concealValue = patternProperties?.conceal ?? match.setProperties?.conceal
-      const concealLines = patternProperties?.conceal_lines ?? match.setProperties?.conceal_lines
-
-      const meta: any = {}
-      if (isInjection && injectionLang) {
-        meta.isInjection = true
-        meta.injectionLang = injectionLang
-      }
-      if (containsInjection) {
-        meta.containsInjection = true
-      }
-      if (concealValue !== undefined) {
-        meta.conceal = concealValue
-      }
-      if (concealLines !== undefined) {
-        meta.concealLines = concealLines
-      }
-
-      if (Object.keys(meta).length > 0) {
-        highlights.push([node.startIndex, node.endIndex, match.name, meta])
-      } else {
-        highlights.push([node.startIndex, node.endIndex, match.name])
-      }
-    }
-
-    highlights.sort((a, b) => a[0] - b[0])
-
-    return highlights
-  }
-
-  async handleResetBuffer(
-    bufferId: number,
-    version: number,
-    content: string,
-  ): Promise<{ highlights?: HighlightResponse[]; warning?: string; error?: string }> {
-    const parserState = this.bufferParsers.get(bufferId)
-    if (!parserState) {
-      return { warning: "No parser state found for buffer" }
-    }
-
-    parserState.content = content
-
-    const newTree = parserState.parser.parse(content)
-
-    if (!newTree) {
-      return { error: "Failed to parse buffer during reset" }
-    }
-
-    parserState.tree = newTree
-    const matches = parserState.queries.highlights.captures(parserState.tree.rootNode)
-
-    let injectionRanges = new Map<string, Array<{ start: number; end: number }>>()
-    if (parserState.queries.injections) {
-      const injectionResult = await this.processInjections(parserState)
-      matches.push(...injectionResult.captures)
-      injectionRanges = injectionResult.injectionRanges
-    }
-
-    return this.getHighlights(parserState, matches, injectionRanges)
-  }
-
-  disposeBuffer(bufferId: number): void {
-    const parserState = this.bufferParsers.get(bufferId)
-    if (!parserState) {
-      return
-    }
-
-    parserState.tree.delete()
-    parserState.parser.delete()
-
-    this.bufferParsers.delete(bufferId)
-  }
-
-  async handleOneShotHighlight(content: string, filetype: string, messageId: string): Promise<void> {
-    const reusableState = await this.getReusableParser(filetype)
-
-    if (!reusableState) {
-      self.postMessage({
-        type: "ONESHOT_HIGHLIGHT_RESPONSE",
-        messageId,
-        hasParser: false,
-        warning: `No parser available for filetype ${filetype}`,
-      })
-      return
-    }
-
-    // Markdown Parser BUG: For markdown, ensure content ends with newline so closing delimiters are parsed correctly
-    // The tree-sitter markdown parser only creates closing delimiter nodes when followed by newline
-    const parseContent = filetype === "markdown" && content.endsWith("```") ? content + "\n" : content
-
-    const tree = reusableState.parser.parse(parseContent)
-
-    if (!tree) {
-      self.postMessage({
-        type: "ONESHOT_HIGHLIGHT_RESPONSE",
-        messageId,
-        hasParser: false,
-        error: "Failed to parse content",
-      })
-      return
-    }
-
-    try {
-      const matches = reusableState.filetypeParser.queries.highlights.captures(tree.rootNode)
-
-      let injectionRanges = new Map<string, Array<{ start: number; end: number }>>()
-      if (reusableState.filetypeParser.queries.injections) {
-        const parserState: ParserState = {
-          parser: reusableState.parser,
-          tree,
-          queries: reusableState.filetypeParser.queries,
-          filetype,
-          content,
-          injectionMapping: reusableState.filetypeParser.injectionMapping,
-        }
-        const injectionResult = await this.processInjections(parserState)
-
-        matches.push(...injectionResult.captures)
-        injectionRanges = injectionResult.injectionRanges
-      }
-
-      const highlights = this.getSimpleHighlights(matches, injectionRanges)
-
-      self.postMessage({
-        type: "ONESHOT_HIGHLIGHT_RESPONSE",
-        messageId,
-        hasParser: true,
-        highlights,
-      })
-    } finally {
-      tree.delete()
-    }
-  }
-
-  async updateDataPath(dataPath: string): Promise<void> {
-    this.dataPath = dataPath
-    this.tsDataPath = path.join(dataPath, "tree-sitter")
-
-    try {
-      await mkdir(path.join(this.tsDataPath, "languages"), { recursive: true })
-      await mkdir(path.join(this.tsDataPath, "queries"), { recursive: true })
-    } catch (error) {
-      throw new Error(`Failed to update data path: ${error}`)
-    }
-  }
-
-  async clearCache(): Promise<void> {
-    if (!this.dataPath || !this.tsDataPath) {
-      throw new Error("No data path configured")
-    }
-
-    const { rm } = await import("fs/promises")
-
-    try {
-      const treeSitterPath = path.join(this.dataPath, "tree-sitter")
-
-      await rm(treeSitterPath, { recursive: true, force: true })
-
-      await mkdir(path.join(treeSitterPath, "languages"), { recursive: true })
-      await mkdir(path.join(treeSitterPath, "queries"), { recursive: true })
-
-      this.filetypeParsers.clear()
-      this.filetypeParserPromises.clear()
-      this.reusableParsers.clear()
-      this.reusableParserPromises.clear()
-    } catch (error) {
-      throw new Error(`Failed to clear cache: ${error}`)
-    }
-  }
-}
-if (!isMainThread) {
-  const worker = new ParserWorker()
-
-  function logMessage(type: "log" | "error" | "warn", ...args: any[]) {
-    self.postMessage({
-      type: "WORKER_LOG",
-      logType: type,
-      data: args,
-    })
-  }
-  console.log = (...args) => logMessage("log", ...args)
-  console.error = (...args) => logMessage("error", ...args)
-  console.warn = (...args) => logMessage("warn", ...args)
-
-  // @ts-ignore - we'll fix this in the future for sure
-  self.onmessage = async (e: MessageEvent) => {
-    const { type, bufferId, version, content, filetype, edits, filetypeParser, messageId, dataPath } = e.data
-
-    try {
-      switch (type) {
-        case "INIT":
-          try {
-            await worker.initialize({ dataPath })
-            self.postMessage({ type: "INIT_RESPONSE" })
-          } catch (error) {
-            self.postMessage({
-              type: "INIT_RESPONSE",
-              error: error instanceof Error ? error.stack || error.message : String(error),
-            })
-          }
-          break
-
-        case "ADD_FILETYPE_PARSER":
-          worker.addFiletypeParser(filetypeParser)
-          break
-
-        case "PRELOAD_PARSER":
-          const maybeParser = await worker.preloadParser(filetype)
-          self.postMessage({ type: "PRELOAD_PARSER_RESPONSE", messageId, hasParser: !!maybeParser })
-          break
-
-        case "INITIALIZE_PARSER":
-          await worker.handleInitializeParser(bufferId, version, content, filetype, messageId)
-          break
-
-        case "HANDLE_EDITS":
-          const response = await worker.handleEdits(bufferId, content, edits)
-          if (response.highlights && response.highlights.length > 0) {
-            self.postMessage({ type: "HIGHLIGHT_RESPONSE", bufferId, version, ...response })
-          } else if (response.warning) {
-            self.postMessage({ type: "WARNING", bufferId, warning: response.warning })
-          } else if (response.error) {
-            self.postMessage({ type: "ERROR", bufferId, error: response.error })
-          }
-          break
-
-        case "GET_PERFORMANCE":
-          self.postMessage({ type: "PERFORMANCE_RESPONSE", performance: worker.performance, messageId })
-          break
-
-        case "RESET_BUFFER":
-          const resetResponse = await worker.handleResetBuffer(bufferId, version, content)
-          if (resetResponse.highlights && resetResponse.highlights.length > 0) {
-            self.postMessage({ type: "HIGHLIGHT_RESPONSE", bufferId, version, ...resetResponse })
-          } else if (resetResponse.warning) {
-            self.postMessage({ type: "WARNING", bufferId, warning: resetResponse.warning })
-          } else if (resetResponse.error) {
-            self.postMessage({ type: "ERROR", bufferId, error: resetResponse.error })
-          }
-          break
-
-        case "DISPOSE_BUFFER":
-          worker.disposeBuffer(bufferId)
-          self.postMessage({ type: "BUFFER_DISPOSED", bufferId })
-          break
-
-        case "ONESHOT_HIGHLIGHT":
-          await worker.handleOneShotHighlight(content, filetype, messageId)
-          break
-
-        case "UPDATE_DATA_PATH":
-          try {
-            await worker.updateDataPath(dataPath)
-            self.postMessage({ type: "UPDATE_DATA_PATH_RESPONSE", messageId })
-          } catch (error) {
-            self.postMessage({
-              type: "UPDATE_DATA_PATH_RESPONSE",
-              messageId,
-              error: error instanceof Error ? error.message : String(error),
-            })
-          }
-          break
-
-        case "CLEAR_CACHE":
-          try {
-            await worker.clearCache()
-            self.postMessage({ type: "CLEAR_CACHE_RESPONSE", messageId })
-          } catch (error) {
-            self.postMessage({
-              type: "CLEAR_CACHE_RESPONSE",
-              messageId,
-              error: error instanceof Error ? error.message : String(error),
-            })
-          }
-          break
-
-        default:
-          self.postMessage({
-            type: "ERROR",
-            bufferId,
-            error: `Unknown message type: ${type}`,
-          })
-      }
-    } catch (error) {
-      self.postMessage({
-        type: "ERROR",
-        bufferId,
-        error: error instanceof Error ? error.stack || error.message : String(error),
-      })
-    }
-  }
-}