import { expect, story } from "../../stories/playwright/story"

story("requests live output only when the shell is expanded", async ({ mount }) => {
  const root = await mount("current-session-terminal-work--live-user-command", { args: { expanded: false } })
  const shell = root.locator('[data-component="session-shell-message"]')
  const trigger = shell.locator('[data-slot="collapsible-trigger"]')
  await expect(trigger).toHaveAttribute("aria-expanded", "false")
  await expect(root.getByLabel("Output requests")).toHaveText("0")
  await root.getByRole("button", { name: "Update output", exact: true }).click()
  await expect(root.getByLabel("Output requests")).toHaveText("0")
  await trigger.click()
  await expect(shell.locator('[data-slot="bash-result"]')).toHaveText("ready\nnext line\n")
  await expect(root.getByLabel("Output requests")).toHaveText(/^[1-9]\d*$/)
  await trigger.click()
  await expect(shell.locator('[data-component="bash-output"]')).toHaveCount(0)
  await root.getByRole("button", { name: "Update output", exact: true }).click()
  await trigger.click()
  await expect(shell.locator('[data-slot="bash-result"]')).toHaveText("ready\nnext line\nnext line\n")
})

story("streams user shell output and retains the saved completion", async ({ mount }) => {
  const root = await mount("current-session-terminal-work--live-user-command")
  const shell = root.locator('[data-component="session-shell-message"]')
  const output = shell.locator('[data-slot="bash-result"]')
  await expect(shell.locator('[data-slot="bash-command"]')).toHaveText("printf ready")
  await expect(output).toHaveText("ready\n")
  await root.getByRole("button", { name: "Update output", exact: true }).click()
  await expect(output).toHaveText("ready\nnext line\n")
  await root.getByRole("button", { name: "Complete command", exact: true }).click()
  await expect(output).toHaveText("ready\nnext line\nfinished\n")
  await expect(shell.locator('[data-kind="tool-error-card"]')).toHaveCount(0)
})

for (const [outcome, error] of [
  ["nonzero", "Command exited with code 1"],
  ["timeout", "Command timed out"],
  ["killed", "Command cancelled"],
]) {
  story(`shows a direct shell ${outcome} outcome even without output`, async ({ mount }) => {
    const root = await mount("current-session-terminal-work--live-user-command", { args: { outcome, output: false } })
    const shell = root.locator('[data-component="session-shell-message"]')
    await root.getByRole("button", { name: "Complete command", exact: true }).click()
    await expect(shell.locator('[data-kind="tool-error-card"]')).toContainText(error)
    await expect(shell.locator('[data-slot="bash-command"]')).toHaveText("printf ready")
    await expect(shell.locator('[data-slot="bash-result"]')).toHaveCount(0)
  })
}

for (const [id, output] of [
  ["running-a-user-command", "Starting Storybook manager...\nBuilding preview..."],
  ["user-command-completed", " M packages/session-ui/src/timeline/session-timeline.tsx"],
]) {
  story(`keeps captured output without a live reader: ${id}`, async ({ mount }) => {
    const root = await mount(`current-session-terminal-work--${id}`)
    await expect(root.locator('[data-component="session-shell-message"] [data-slot="bash-result"]')).toHaveText(output)
  })
}

story("copies the shell command and its output separately", async ({ mount, page }) => {
  const root = await mount("current-session-terminal-work--expanded-shell")
  const shell = root.locator('[data-component="bash-output"]')
  const command = (await shell.locator('[data-slot="bash-command"]').textContent())!
  const output = (await shell.locator('[data-slot="bash-result"]').textContent())!
  const clipboard = async () => (await page.evaluate(() => navigator.clipboard.readText())).replaceAll("\r\n", "\n")
  await page.context().grantPermissions(["clipboard-read", "clipboard-write"])
  await shell.hover()
  await shell.locator('[data-slot="bash-result"]').getByRole("button", { name: "Copy output", exact: true }).click()
  await expect(shell.locator('[data-slot="bash-result"]').getByRole("button", { name: "Copied" })).toBeVisible()
  expect(await clipboard()).toBe(output)
  await shell.locator('[data-slot="bash-command"]').getByRole("button", { name: "Copy", exact: true }).click()
  expect(await clipboard()).toBe(command)
  expect(command).toContain("opencode2 api get /openapi.json")
  expect(output).toContain('"name":"OpenCode"')
})