openharness

import type { PaneView, QuestionRow, QuestionView } from '../../lib/askQuestion.js'
import { earlierDialogEnd } from '../../lib/dialogEnd.js'

function stripAnsi(value: string): string {
  return value.replace(/\u001b\[[0-9;:]*[A-Za-z]/g, '')
}

/**
 * Parse Grok 1.0.0's live questionnaire.
 *
 * Measured shape:
 *   |  Which color should I report?
 *   |  1 (o) Red   Report the color Red
 *   |  2 (o) Blue  Report the color Blue
 *   |  z (o) Type your answer here
 *   |  up/down navigate ... Enter:submit
 *
 * Descriptions share the row with labels, separated by 2+ spaces. The `z` row opens free text and must
 * not be offered as an option. Digits select and submit directly (verified by pressing `2`).
 *
 * Grok's PERMISSION prompt reuses this exact row syntax and only reworded the footer, so it needs a
 * second footer alternative rather than a second parser (`__fixtures__/permission-grok.txt`):
 *
 *   \u2503  Fetch Bitcoin price from CoinGecko
 *   \u2503  curl -s https://api.coingecko.com/api/v3/simple/price?ids=bitcoin
 *   \u2503
 *   \u2503  1 (\u25cf) Yes, and don't ask again for anything (always-approve mode)
 *   \u2503  2 (\u25cb) Yes, proceed
 *   \u2503  3 (\u25cb) No, reject (type to add feedback)
 *
 *   1/3:select  \u2502  Tab:next option  \u2502  Ctrl+o:always-approve  \u2502  Ctrl+c:cancel  \u2502  Esc:scrollback
 *
 * `1/3:select` is the range of digits the dialog accepts, and pressing one both selects AND submits here
 * too \u2014 verified by pressing `2` on a live prompt and watching the command run. The question line the
 * parser lands on is the COMMAND rather than grok's own one-line summary above it, which is the more
 * useful of the two on a device screen. Only that footer (`Ctrl+o:always-approve`) or the approval's own
 * first row (`always-approve mode`) marks the dialog a permission prompt, and only on the LIVE dialog: an
 * answered approval still in scrollback made the questionnaire under it read as one.
 */
export function parseGrokQuestionPane(capture: string): PaneView {
  const lines = stripAnsi(capture).replace(/\u00a0/g, ' ').split('\n')
  const footer = lines.findLastIndex((line) => /enter\s*:\s*submit|\d+\/\d+\s*:\s*select/i.test(line))
  if (footer < 0) return null

  // Nothing above an earlier dialog's end is this dialog's: not a row, not the question.
  const floor = earlierDialogEnd(lines, footer, 28)
  const rows: QuestionRow[] = []
  let start = -1
  for (let i = footer - 1; i > floor && footer - i <= 20; i--) {
    const match = /^\s*[┃|]?\s*([0-9a-z])\s+\([^)]*\)\s+(.+?)\s*$/i.exec(lines[i])
    if (!match) continue
    const label = match[2].split(/\s{2,}/)[0].trim()
    rows.unshift({ number: match[1], label, checked: /\([●x*]\)/i.test(lines[i]) })
    start = i
  }
  if (!rows.length || start < 0) return null

  let question = ''
  let top = start
  for (let i = start - 1; i > floor && start - i <= 8; i--) {
    const line = lines[i].replace(/^\s*[┃|]\s?/, '').trim()
    if (!line) continue
    question = line
    top = i
    break
  }
  if (!question) return null

  const typeRow = rows.find((row) => /^type your answer here$/i.test(row.label)) ?? null
  const view: QuestionView = {
    kind: 'question',
    question,
    rows: rows.filter((row) => row !== typeRow),
    // The measured 1.0.0 dialog is single-select; do not infer checkbox semantics from its `(○)` marker.
    multi: false,
    typeRow,
  }
  // Grok prints this footer only for tool approval, never its questionnaire.
  return /always-approve|Ctrl\+o:/i.test(lines.slice(top).join('\n')) ? { ...view, permission: true } : view
}