oh-my-opencode/src/tools/hashline-edit/normalize-edits.ts

import type { HashlineEdit } from "./types"

type HashlineToolOp = "replace" | "append" | "prepend"

export interface RawHashlineEdit {
  op?: HashlineToolOp
  pos?: string
  end?: string
  lines?: string | string[] | null
}

function normalizeAnchor(value: string | undefined): string | undefined {
  if (typeof value !== "string") return undefined
  const trimmed = value.trim()
  return trimmed === "" ? undefined : trimmed
}

function requireLines(edit: RawHashlineEdit, index: number): string | string[] {
  if (edit.lines === undefined) {
    throw new Error(`Edit ${index}: lines is required for ${edit.op ?? "unknown"}`)
  }
  if (edit.lines === null) {
    return []
  }
  return edit.lines
}

function requireLine(anchor: string | undefined, index: number, op: HashlineToolOp): string {
  if (!anchor) {
    throw new Error(`Edit ${index}: ${op} requires at least one anchor line reference (pos or end)`)
  }
  return anchor
}

function normalizeReplaceEdit(edit: RawHashlineEdit, index: number): HashlineEdit {
  const pos = normalizeAnchor(edit.pos)
  const end = normalizeAnchor(edit.end)
  const anchor = requireLine(pos ?? end, index, "replace")
  const text = requireLines(edit, index)

  if (pos && end) {
    return {
      type: "replace_lines",
      start_line: pos,
      end_line: end,
      text,
    }
  }

  return {
    type: "set_line",
    line: anchor,
    text,
  }
}

function normalizeAppendEdit(edit: RawHashlineEdit, index: number): HashlineEdit {
  const pos = normalizeAnchor(edit.pos)
  const end = normalizeAnchor(edit.end)
  const anchor = pos ?? end
  const text = requireLines(edit, index)

  if (!anchor) {
    return {
      type: "append",
      text,
    }
  }

  return {
    type: "insert_after",
    line: anchor,
    text,
  }
}

function normalizePrependEdit(edit: RawHashlineEdit, index: number): HashlineEdit {
  const pos = normalizeAnchor(edit.pos)
  const end = normalizeAnchor(edit.end)
  const anchor = pos ?? end
  const text = requireLines(edit, index)

  if (!anchor) {
    return {
      type: "prepend",
      text,
    }
  }

  return {
    type: "insert_before",
    line: anchor,
    text,
  }
}

export function normalizeHashlineEdits(rawEdits: RawHashlineEdit[]): HashlineEdit[] {
  return rawEdits.map((rawEdit, index) => {
    const edit = rawEdit ?? {}

    switch (edit.op) {
      case "replace":
        return normalizeReplaceEdit(edit, index)
      case "append":
        return normalizeAppendEdit(edit, index)
      case "prepend":
        return normalizePrependEdit(edit, index)
      default:
        throw new Error(
          `Edit ${index}: unsupported op "${String(edit.op)}". Legacy format was removed; use op/pos/end/lines.`
        )
    }
  })
}