teqo-backup/scripts/lib/worktree.mjs

/**
 * Pure helpers for `scripts/worktree.mjs` — branch naming derived from the
 * claim-queue issue, shared with the unit tests. Naming: `<code>-<slug>`
 * where `code` is the frontmatter `id` (ex. `C15`) and `slug` is the pt-BR
 * title slugified with the repo's `src/lib/slug.ts` (accents stripped,
 * non-alphanumeric → `-`).
 */

import { slugify } from '../../src/lib/slug.ts'

/**
 * Prefix of every `/plan-issue` planning-worktree branch (`pnpm worktree
 * plan`). Lowercase-led `plans/…`, so it can never collide with a `next`
 * branch — `<Code>-<slug>` is always uppercase-led — neither in branch name
 * nor, structurally, in the port/slot space derived from the branch.
 */
export const PLAN_BRANCH_PREFIX = 'plans/plan-issue'

/** Env var the interactive-terminal `worktree()` function sets to request the opencode launch. */
export const WORKTREE_TERMINAL_ENV = 'TEQO_WORKTREE_TERMINAL'

/**
 * Preset model for the opencode launch — fallback comum aos 3 repos do fluxo
 * (infra-solla, teqo, iara-pwa). Override sem editar código: global via
 * `export OPENCODE_WORKTREE_MODEL=…`, ou por repo na chave
 * `OPENCODE_WORKTREE_MODEL` da `.forgejo/worktree.env`. Ex-OPS78/OPS82: o
 * preset era o flavor `-0731` via Vercel AI Gateway; voltou ao provider
 * direto porque o gateway está sem créditos (erros "positive credit balance"
 * em 24/08/2026) e porque unificar o default reduz a sobrescrita confusa de
 * seleção de modelo do TUI (opencode issues #8349/#13456). OPS93: com
 * `--cheap/--pro/--zen/--go/--alibaba/--glm/--free` o mapa `WORKTREE_MODEL_MAP`
 * escolhe o modelo por invocação; sem flag o preset permanece. OPS95: sem
 * `--variant` na diretiva (o TUI rejeita o flag) — variantes ficam na config
 * global da máquina, via Ctrl+T. OPS101: o fallback passou a usar o nome
 * canônico `deepseek/deepseek-flash` (DeepSeek V4.1 Flash, lançado em
 * 2026-09-10; o alias legado segue servido ao preço Flash, mas pode sumir
 * sem aviso).
 */
export const OPENCODE_PRESET_MODEL =
  process.env.OPENCODE_WORKTREE_MODEL || 'deepseek/deepseek-flash'

/**
 * Per-invocation model map — fixed menu of named flags (OPS93; values
 * corrected in OPS95 after the OPS93 delivery shipped wrong IDs; OPS100 adds
 * `--glm` and `--free`). OPS112 corrects `--go` → DeepSeek V4.1 Flash and
 * `--zen` → Muse Spark 1.3 Free.
 * The directive picks `WORKTREE_MODEL_MAP[flag]` when
 * a single flag is present, otherwise falls back to `OPENCODE_PRESET_MODEL`.
 * No `--variant` is emitted (the TUI yargs rejects the flag — OPS95):
 * variants live on the machine's global config, selectable via Ctrl+T
 * (`opencode run --variant max` remains the headless path).
 */
export const WORKTREE_MODEL_MAP = {
  cheap: 'cheapestinference/deepseek-v4-flash',
  pro: 'deepseek/deepseek-v4-pro',
  zen: 'opencode/muse-spark-1.3-contributor-free',
  go: 'opencode-go/deepseek-v4.1-flash',
  alibaba: 'alibaba-token-plan/deepseek-v4-flash',
  glm: 'opencode-go/glm-5.3-flash',
  free: 'openrouter/openrouter/free',
}

/** Set of flag names that select a model (keys of WORKTREE_MODEL_MAP). */
export const WORKTREE_MODEL_FLAGS = new Set(Object.keys(WORKTREE_MODEL_MAP))

/**
 * Resolve the model for a parsed `flags` bag (from `parseArgs`). At most one
 * model flag may be present — multiple → throw (fail-high, never guess).
 * No flag → `OPENCODE_PRESET_MODEL`.
 */
export const resolveWorktreeModel = (flags = {}) => {
  const active = [...WORKTREE_MODEL_FLAGS].filter((flag) => flags[flag])
  if (active.length > 1) {
    throw new Error(
      `Flags de modelo conflitantes: --${active.join(' --')} (use apenas um de --cheap/--pro/--zen/--go/--alibaba/--glm/--free)`,
    )
  }
  if (active.length === 1) return WORKTREE_MODEL_MAP[active[0]]
  return OPENCODE_PRESET_MODEL
}

/**
 * Prefix of every neutral-worktree branch (`pnpm worktree new`). Lowercase-led
 * `work/…` — structurally disjoint from `next`'s uppercase-led `<Code>-<slug>`
 * and from `plan`'s `plans/plan-issue-…`, in branch name and slot space alike.
 */
export const WORK_BRANCH_PREFIX = 'work'

/**
 * Prefix of every bug-fix worktree branch (`pnpm worktree fix`). Lowercase-led
 * `fix/…` — structurally disjoint from `next`'s uppercase-led `<Code>-<slug>`,
 * from `plan`'s `plans/plan-issue-…` and from `new`'s `work/…`, in branch name
 * and slot space alike.
 */
export const FIX_BRANCH_PREFIX = 'fix'

/** Total branch-name budget — mirrors `branchNameForIssue` (60). */
const NAMESPACE_BRANCH_MAX_LENGTH = 60

/** Strip the leading `<code> — ` (or any dash variant) off a title. */
const stripCodePrefix = (title, code) => {
  const escaped = code.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')
  return title.replace(new RegExp(`^${escaped}\\s*[—–:-]?\\s*`), '')
}

/**
 * `code` = frontmatter `id` (falls back to the leading `ID — ` token of the
 * title); `subject` = the title with that prefix removed.
 */
export const issueCodeAndSubject = (issue) => {
  const title = issue.title ?? ''
  const code = typeof issue.meta?.id === 'string' && issue.meta.id.length > 0 ? issue.meta.id : null
  const subject = code
    ? stripCodePrefix(title, code)
    : title.replace(/^[A-Za-z0-9+]+\s*[—–:-]?\s*/, '')
  return { code, subject }
}

/**
 * `<code>-<slug>` truncated to `maxLength` total characters (always keeping
 * the code). Throws when the issue has no id — fail loudly, never invent.
 */
export const branchNameForIssue = (issue, maxLength = 60) => {
  const { code, subject } = issueCodeAndSubject(issue)
  if (!code) {
    throw new Error(`Issue sem frontmatter id: #${issue.number} ${issue.title}`)
  }
  const slug = slugify(subject) || 'issue'
  const full = `${code}-${slug}`
  if (full.length <= maxLength) return full
  const keep = Math.max(1, maxLength - code.length - 1)
  return `${code}-${slug.slice(0, keep)}`
}

/**
 * Shared branch naming for namespace worktrees NOT tied to the claim queue
 * (`plan`, `new`). Every invocation must land on a DIFFERENT branch so
 * parallel sessions coexist:
 *  - `bag` given → `<prefix><slug>`; if that name is already taken,
 *    `<prefix><slug>-2`, `-3`, …
 *  - no `bag` → the next free sequential `<prefix>1`, `<prefix>2`, …
 * `prefix` carries the namespace AND its separator (`plans/plan-issue-` for
 * plan, `work/` for new), so the sequential and bag forms share one spelling
 * per namespace. `taken` = branch short-names already alive (local refs +
 * origin); when a name is free it is reused only as a name-free slot — the
 * branch is created fresh from `origin/main` each time. Both namespaces are
 * lowercase-led, so neither can collide with `next`'s uppercase-led
 * `<Code>-<slug>` branches.
 */
const namespaceBranchName = ({ prefix, bag = '', taken = new Set(), fallback }) => {
  const hasBag = typeof bag === 'string' && bag.trim().length > 0

  if (!hasBag) {
    for (let n = 1; ; n += 1) {
      const candidate = `${prefix}${n}`
      if (!taken.has(candidate)) return candidate
    }
  }

  const slug = slugify(bag) || fallback
  const base = `${prefix}${slug.slice(0, NAMESPACE_BRANCH_MAX_LENGTH - prefix.length)}`
  if (!taken.has(base)) return base

  for (let n = 2; ; n += 1) {
    const suffix = `-${n}`
    const keep = Math.max(1, NAMESPACE_BRANCH_MAX_LENGTH - prefix.length - suffix.length)
    const candidate = `${prefix}${slug.slice(0, keep)}${suffix}`
    if (!taken.has(candidate)) return candidate
  }
}

/**
 * Launch directive for the opencode TUI, printed by `worktree next`/`plan`
 * right before the `cd <dir>` line when called from the interactive terminal
 * (`WORKTREE_TERMINAL=1`): `launch node scripts/agent-session.mjs start
 * --purpose=<P> --dir=<dir> --model=<preset|map> [--issue=<N>]
 * [--argument="<bag>"]` (OPS110 — o launch deixa de abrir um TUI local dono da
 * sessão e passa a delegar ao `scripts/agent-session.mjs`, que sobe/anexa no
 * servidor compartilhado: o run sobrevive ao fechamento do terminal e pode ser
 * reatado de outro dispositivo). O mapa purpose→skill e a montagem do comando
 * vivem no dono do ciclo de vida (`scripts/lib/agent-session.mjs`,
 * `purposeInvocation`). OPS95 continua: sem `--variant`; modelo vem do mapa
 * `WORKTREE_MODEL_MAP` quando a flag `--cheap/--pro/--zen/--go/--alibaba/--glm/--free`
 * está presente, senão do preset.
 *
 * The shell function (`.agents/shell/worktree.sh`) applies the `cd` first, then
 * tokenizes and executes this line (xargs — quote-aware, never eval) — the dir
 * is always `<root without spaces>/<slugified branch>`, so the line never needs
 * quoting for the path. `next` with an `issueNumber` carries `--issue=<N>`
 * (OPS33: the launch delivers the claimed issue to the agent; the skill reads
 * the rest from GitHub), `fix` (bug description) and `plan` (opening message)
 * with an `argument` carry `--argument="<bag>"` (the text arrives with the
 * skill; quotes/backslashes are stripped because the shell layer tokenizes
 * with xargs, which does not honor backslash escapes). Returns `null` outside
 * the terminal so the `/worktree` opencode command never launches a nested TUI.
 * `sessionScript` é o caminho do CLI de sessão — `worktree.mjs` passa o
 * ABSOLUTO do checkout que emitiu a diretiva (worktrees reabertos/criados
 * antes do merge não têm o arquivo novo; o relativo apontaria para o branch
 * errado). O default relativo existe para os specs.
 * @param {{ dir: string, purpose: string, terminal?: boolean, issueNumber?: number | null, model?: string | null, argument?: string | null, sessionScript?: string }} options
 */
export const opencodeLaunchDirective = ({
  dir,
  purpose,
  terminal = false,
  issueNumber = null,
  model = null,
  argument = null,
  sessionScript = 'scripts/agent-session.mjs',
}) => {
  if (!terminal) return null
  const selectedModel = model ?? OPENCODE_PRESET_MODEL
  const args = [
    'node',
    sessionScript,
    'start',
    `--purpose=${purpose}`,
    `--dir=${dir}`,
    `--model=${selectedModel}`,
  ]
  // The issue suffix belongs to `next` alone — `plan`/`new`/`fix` never carry
  // a claimed issue (fail-safe: a stray issueNumber must not break them).
  if (purpose === 'next' && issueNumber) args.push(`--issue=${issueNumber}`)
  // The bag suffix belongs to `fix` (bug description) and `plan` (opening
  // message) — the text arrives with the skill. JSON.stringify quotes the
  // value (it carries spaces); xargs strips the quotes at execution time.
  if (purpose === 'fix' || purpose === 'plan') {
    const sanitized = typeof argument === 'string' ? argument.replace(/["\\]/g, '').trim() : ''
    if (sanitized) args.push(`--argument=${JSON.stringify(sanitized)}`)
  }
  return `launch ${args.join(' ')}`
}

/**
 * Headless launch for the OPS106 auto-unblock wrapper: the same `/bug-fix`
 * contract as `fix`, no TUI and no `--prompt` (the headless CLI has
 * `--command` as a first-class option — the report rides as the message/args).
 * The directive is written to a FILE, never stdout: the provisioning path
 * inherits stdio from git/pnpm/seed children, so stdout cannot be a clean JSON
 * channel. Emitted by `scripts/worktree.mjs fix --headless --directive <path>`.
 */
export const OPENCODE_HEADLESS_COMMAND = 'bug-fix'

/**
 * opencode `run` argv for the auto-unblock agent (model + report, no shell).
 * @param {{ model?: string, report?: string, command?: string }} [options]
 */
export const opencodeHeadlessArgs = ({
  model,
  report,
  command = OPENCODE_HEADLESS_COMMAND,
} = {}) => {
  if (typeof model !== 'string' || model.length === 0) {
    throw new Error('opencodeHeadlessArgs: model ausente')
  }
  if (typeof report !== 'string' || report.trim().length === 0) {
    throw new Error('opencodeHeadlessArgs: report vazio')
  }
  return ['opencode', 'run', '--model', model, '--auto', '--command', command, report]
}

/**
 * Machine-readable `--headless` directive persisted for the wrapper.
 * @param {{ dir?: string, branch?: string, model?: string, report?: string }} [options]
 */
export const headlessDirective = ({ dir, branch, model, report } = {}) => {
  if (typeof dir !== 'string' || dir.length === 0) {
    throw new Error('headlessDirective: dir ausente')
  }
  return { dir, branch: branch ?? '', model, argv: opencodeHeadlessArgs({ model, report }) }
}

/**
 * Branch for a `/plan-issue` planning worktree — see `namespaceBranchName`
 * (namespace `plans/plan-issue-…`, fallback label `plano`).
 */
export const planBranchName = ({ bag = '', taken = new Set() }) =>
  namespaceBranchName({ prefix: `${PLAN_BRANCH_PREFIX}-`, bag, taken, fallback: 'plano' })

/**
 * Branch for a neutral worktree (`pnpm worktree new`) — see
 * `namespaceBranchName` (namespace `work/…`, fallback label `work`).
 */
export const workBranchName = ({ bag = '', taken = new Set() }) =>
  namespaceBranchName({ prefix: `${WORK_BRANCH_PREFIX}/`, bag, taken, fallback: 'work' })

/**
 * Branch for a bug-fix worktree (`pnpm worktree fix`) — see
 * `namespaceBranchName` (namespace `fix/…`, fallback label `fix`).
 */
export const fixBranchName = ({ bag = '', taken = new Set() }) =>
  namespaceBranchName({ prefix: `${FIX_BRANCH_PREFIX}/`, bag, taken, fallback: 'fix' })