@fairyhunter13/opentui-core 0.1.112 → 0.1.114

package/src/lib/tree-sitter-styled-text.ts

@@ -0,0 +1,306 @@
+import type { TextChunk } from "../text-buffer.js"
+import { StyledText } from "./styled-text.js"
+import { SyntaxStyle, type StyleDefinition } from "../syntax-style.js"
+import { TreeSitterClient } from "./tree-sitter/client.js"
+import type { SimpleHighlight } from "./tree-sitter/types.js"
+import { createTextAttributes } from "../utils.js"
+import { registerEnvVar, env } from "./env.js"
+
+registerEnvVar({ name: "OTUI_TS_STYLE_WARN", default: false, description: "Enable warnings for missing syntax styles" })
+
+interface ConcealOptions {
+  enabled: boolean
+}
+
+interface Boundary {
+  offset: number
+  type: "start" | "end"
+  highlightIndex: number
+}
+
+function getSpecificity(group: string): number {
+  return group.split(".").length
+}
+
+function shouldSuppressInInjection(group: string, meta: any): boolean {
+  if (meta?.isInjection) {
+    return false
+  }
+
+  // Check if this is a parent block that should be suppressed
+  // TODO: This is language/highlight specific,
+  // not generic enough. Needs a more generic solution.
+  // The styles need to be more like a stack that gets merged
+  // and for a container with injections we just don't push that container style
+  return group === "markup.raw.block"
+}
+
+export function treeSitterToTextChunks(
+  content: string,
+  highlights: SimpleHighlight[],
+  syntaxStyle: SyntaxStyle,
+  options?: ConcealOptions,
+): TextChunk[] {
+  const chunks: TextChunk[] = []
+  const defaultStyle = syntaxStyle.getStyle("default")
+  const concealEnabled = options?.enabled ?? true
+
+  const injectionContainerRanges: Array<{ start: number; end: number }> = []
+  const boundaries: Boundary[] = []
+
+  for (let i = 0; i < highlights.length; i++) {
+    const [start, end, , meta] = highlights[i]
+    if (start === end) continue // Skip zero-length ranges
+    if (meta?.containsInjection) {
+      injectionContainerRanges.push({ start, end })
+    }
+    boundaries.push({ offset: start, type: "start", highlightIndex: i })
+    boundaries.push({ offset: end, type: "end", highlightIndex: i })
+  }
+
+  // Sort boundaries by offset, with ends before starts at same offset
+  // This ensures we close old ranges before opening new ones at the same position
+  boundaries.sort((a, b) => {
+    if (a.offset !== b.offset) return a.offset - b.offset
+    if (a.type === "end" && b.type === "start") return -1
+    if (a.type === "start" && b.type === "end") return 1
+    return 0
+  })
+
+  const activeHighlights = new Set<number>()
+  let currentOffset = 0
+
+  for (let i = 0; i < boundaries.length; i++) {
+    const boundary = boundaries[i]
+
+    if (currentOffset < boundary.offset && activeHighlights.size > 0) {
+      const segmentText = content.slice(currentOffset, boundary.offset)
+
+      const activeGroups: Array<{ group: string; meta: any; index: number }> = []
+      for (const idx of activeHighlights) {
+        const [, , group, meta] = highlights[idx]
+        activeGroups.push({ group, meta, index: idx })
+      }
+
+      // Check if any active highlight has a conceal property
+      // Priority: 1. Check meta.conceal first 2. Check group === "conceal" or starts with "conceal."
+      const concealHighlight = concealEnabled
+        ? activeGroups.find(
+            (h) => h.meta?.conceal !== undefined || h.group === "conceal" || h.group.startsWith("conceal."),
+          )
+        : undefined
+
+      if (concealHighlight) {
+        let replacementText = ""
+
+        if (concealHighlight.meta?.conceal !== undefined) {
+          // If meta.conceal is set, use it (this would come from (#set! conceal "...") if supported)
+          replacementText = concealHighlight.meta.conceal
+        } else if (concealHighlight.group === "conceal.with.space") {
+          // Special group name means replace with space
+          replacementText = " "
+        }
+
+        if (replacementText) {
+          chunks.push({
+            __isChunk: true,
+            text: replacementText,
+            fg: defaultStyle?.fg,
+            bg: defaultStyle?.bg,
+            attributes: defaultStyle
+              ? createTextAttributes({
+                  bold: defaultStyle.bold,
+                  italic: defaultStyle.italic,
+                  underline: defaultStyle.underline,
+                  dim: defaultStyle.dim,
+                })
+              : 0,
+          })
+        }
+      } else {
+        const insideInjectionContainer = injectionContainerRanges.some(
+          (range) => currentOffset >= range.start && currentOffset < range.end,
+        )
+
+        // Filter out highlights that should be suppressed
+        // Suppress highlights when we're inside an injection container
+        const validGroups = activeGroups.filter((h) => {
+          // If we're inside an injection container, suppress all markup.raw.block highlights
+          // This includes both the container itself and any nested markup.raw.block
+          if (insideInjectionContainer && shouldSuppressInInjection(h.group, h.meta)) {
+            return false
+          }
+          return true
+        })
+
+        // Sort groups by specificity (least to most), then by index (earlier to later)
+        // This ensures we merge styles in the correct order: parent styles first, then child overrides
+        const sortedGroups = validGroups.sort((a, b) => {
+          const aSpec = getSpecificity(a.group)
+          const bSpec = getSpecificity(b.group)
+          if (aSpec !== bSpec) return aSpec - bSpec // Lower specificity first
+          return a.index - b.index // Earlier index first
+        })
+
+        // Merge all active styles in order (like CSS cascade)
+        // Later/more specific styles override earlier/less specific ones
+        const mergedStyle: StyleDefinition = {}
+
+        for (const { group } of sortedGroups) {
+          let styleForGroup = syntaxStyle.getStyle(group)
+
+          if (!styleForGroup && group.includes(".")) {
+            // Fallback to base scope
+            const baseName = group.split(".")[0]
+            styleForGroup = syntaxStyle.getStyle(baseName)
+          }
+
+          if (styleForGroup) {
+            // Merge properties - later styles override earlier ones
+            if (styleForGroup.fg !== undefined) mergedStyle.fg = styleForGroup.fg
+            if (styleForGroup.bg !== undefined) mergedStyle.bg = styleForGroup.bg
+            if (styleForGroup.bold !== undefined) mergedStyle.bold = styleForGroup.bold
+            if (styleForGroup.italic !== undefined) mergedStyle.italic = styleForGroup.italic
+            if (styleForGroup.underline !== undefined) mergedStyle.underline = styleForGroup.underline
+            if (styleForGroup.dim !== undefined) mergedStyle.dim = styleForGroup.dim
+          } else {
+            if (group.includes(".")) {
+              const baseName = group.split(".")[0]
+              if (env.OTUI_TS_STYLE_WARN) {
+                console.warn(
+                  `Syntax style not found for group "${group}" or base scope "${baseName}", using default style`,
+                )
+              }
+            } else {
+              if (env.OTUI_TS_STYLE_WARN) {
+                console.warn(`Syntax style not found for group "${group}", using default style`)
+              }
+            }
+          }
+        }
+
+        // Use merged style, falling back to default if nothing was merged
+        const finalStyle = Object.keys(mergedStyle).length > 0 ? mergedStyle : defaultStyle
+
+        chunks.push({
+          __isChunk: true,
+          text: segmentText,
+          fg: finalStyle?.fg,
+          bg: finalStyle?.bg,
+          attributes: finalStyle
+            ? createTextAttributes({
+                bold: finalStyle.bold,
+                italic: finalStyle.italic,
+                underline: finalStyle.underline,
+                dim: finalStyle.dim,
+              })
+            : 0,
+        })
+      }
+    } else if (currentOffset < boundary.offset) {
+      const text = content.slice(currentOffset, boundary.offset)
+      chunks.push({
+        __isChunk: true,
+        text,
+        fg: defaultStyle?.fg,
+        bg: defaultStyle?.bg,
+        attributes: defaultStyle
+          ? createTextAttributes({
+              bold: defaultStyle.bold,
+              italic: defaultStyle.italic,
+              underline: defaultStyle.underline,
+              dim: defaultStyle.dim,
+            })
+          : 0,
+      })
+    }
+
+    if (boundary.type === "start") {
+      activeHighlights.add(boundary.highlightIndex)
+    } else {
+      activeHighlights.delete(boundary.highlightIndex)
+
+      if (concealEnabled) {
+        const [, , group, meta] = highlights[boundary.highlightIndex]
+        if (meta?.concealLines !== undefined) {
+          if (boundary.offset < content.length && content[boundary.offset] === "\n") {
+            currentOffset = boundary.offset + 1
+            continue
+          }
+        }
+
+        // TODO: This is also a query specific workaround, needs improvement
+        if (meta?.conceal !== undefined) {
+          // Skip the next space if we replaced with a space (prevents double spaces like "text] (url)")
+          if (meta.conceal === " ") {
+            if (boundary.offset < content.length && content[boundary.offset] === " ") {
+              currentOffset = boundary.offset + 1
+              continue
+            }
+          }
+          // For heading markers specifically, also skip the trailing space
+          // The group is just "conceal" for heading markers from the markdown query
+          // We need to check if this conceal is NOT from an injection (markdown_inline)
+          else if (meta.conceal === "" && group === "conceal" && !meta.isInjection) {
+            if (boundary.offset < content.length && content[boundary.offset] === " ") {
+              currentOffset = boundary.offset + 1
+              continue
+            }
+          }
+        }
+      }
+    }
+
+    currentOffset = boundary.offset
+  }
+
+  if (currentOffset < content.length) {
+    const text = content.slice(currentOffset)
+    chunks.push({
+      __isChunk: true,
+      text,
+      fg: defaultStyle?.fg,
+      bg: defaultStyle?.bg,
+      attributes: defaultStyle
+        ? createTextAttributes({
+            bold: defaultStyle.bold,
+            italic: defaultStyle.italic,
+            underline: defaultStyle.underline,
+            dim: defaultStyle.dim,
+          })
+        : 0,
+    })
+  }
+
+  return chunks
+}
+
+export interface TreeSitterToStyledTextOptions {
+  conceal?: ConcealOptions
+}
+
+export async function treeSitterToStyledText(
+  content: string,
+  filetype: string,
+  syntaxStyle: SyntaxStyle,
+  client: TreeSitterClient,
+  options?: TreeSitterToStyledTextOptions,
+): Promise<StyledText> {
+  const result = await client.highlightOnce(content, filetype)
+  if (result.highlights && result.highlights.length > 0) {
+    const chunks = treeSitterToTextChunks(content, result.highlights, syntaxStyle, options?.conceal)
+    return new StyledText(chunks)
+  } else {
+    const defaultStyle = syntaxStyle.mergeStyles("default")
+    const chunks: TextChunk[] = [
+      {
+        __isChunk: true,
+        text: content,
+        fg: defaultStyle.fg,
+        bg: defaultStyle.bg,
+        attributes: defaultStyle.attributes,
+      },
+    ]
+    return new StyledText(chunks)
+  }
+}

package/src/lib/validate-dir-name.ts

@@ -0,0 +1,55 @@
+export function isValidDirectoryName(name: string): boolean {
+  if (!name || typeof name !== "string") {
+    return false
+  }
+
+  if (name.trim().length === 0) {
+    return false
+  }
+
+  const reservedNames = [
+    "CON",
+    "PRN",
+    "AUX",
+    "NUL",
+    "COM1",
+    "COM2",
+    "COM3",
+    "COM4",
+    "COM5",
+    "COM6",
+    "COM7",
+    "COM8",
+    "COM9",
+    "LPT1",
+    "LPT2",
+    "LPT3",
+    "LPT4",
+    "LPT5",
+    "LPT6",
+    "LPT7",
+    "LPT8",
+    "LPT9",
+  ]
+  if (reservedNames.includes(name.toUpperCase())) {
+    return false
+  }
+
+  // Check for invalid characters
+  // Windows: < > : " | ? * \ and control characters (0-31)
+  // Unix: null character and forward slash
+  const invalidChars = /[<>:"|?*/\\\x00-\x1f]/
+  if (invalidChars.test(name)) {
+    return false
+  }
+
+  if (name.endsWith(".") || name.endsWith(" ")) {
+    return false
+  }
+
+  if (name === "." || name === "..") {
+    return false
+  }
+
+  return true
+}