import { describe, expect, it } from 'vitest'

import { aggregateSessions, renderJson, renderTable } from '../src/sessions-report.js'
import type { ClassifiedTurn, ProjectSummary, SessionSummary } from '../src/types.js'

function makeProject(): ProjectSummary {
  const turn: ClassifiedTurn = {
    userMessage: 'build it',
    timestamp: '2026-07-10T10:00:00.000Z',
    sessionId: 'session-1',
    category: 'feature',
    retries: 0,
    hasEdits: true,
    assistantCalls: [{
      provider: 'claude',
      model: 'claude-sonnet-4-5',
      usage: {
        inputTokens: 100,
        outputTokens: 20,
        cacheCreationInputTokens: 30,
        cacheReadInputTokens: 400,
        cachedInputTokens: 0,
        reasoningTokens: 0,
        webSearchRequests: 0,
      },
      costUSD: 0.12,
      tools: [],
      mcpTools: [],
      skills: [],
      subagentTypes: [],
      hasAgentSpawn: false,
      hasPlanMode: false,
      speed: 'standard',
      timestamp: '2026-07-10T10:01:00.000Z',
      bashCommands: [],
      deduplicationKey: 'call-1',
    }],
  }
  const session: SessionSummary = {
    sessionId: 'session-1',
    project: 'codeburn',
    firstTimestamp: '2026-07-10T10:00:00.000Z',
    lastTimestamp: '2026-07-10T10:05:00.000Z',
    totalCostUSD: 0.12,
    totalSavingsUSD: 0.03,
    totalInputTokens: 100,
    totalOutputTokens: 20,
    totalReasoningTokens: 0,
    totalCacheReadTokens: 400,
    totalCacheWriteTokens: 30,
    apiCalls: 1,
    turns: [turn],
    modelBreakdown: {
      'claude-sonnet-4-5': { calls: 1, costUSD: 0.12, tokens: turn.assistantCalls[0]!.usage, savingsUSD: 0.03 },
    },
    toolBreakdown: {},
    mcpBreakdown: {},
    bashBreakdown: {},
    categoryBreakdown: {} as SessionSummary['categoryBreakdown'],
    skillBreakdown: {},
    subagentBreakdown: {},
  }
  return {
    project: 'codeburn',
    projectPath: '/tmp/codeburn',
    sessions: [session],
    totalCostUSD: 0.12,
    totalSavingsUSD: 0.03,
    totalApiCalls: 1,
    totalProxiedCostUSD: 0,
  }
}

describe('sessions JSON emitter', () => {
  it('flattens SessionSummary fields into the exact JSON row shape', () => {
    const rows = aggregateSessions([makeProject()])
    const parsed = JSON.parse(renderJson(rows))

    expect(parsed).toEqual([{
      sessionId: 'session-1',
      project: 'codeburn',
      provider: 'claude',
      models: ['claude-sonnet-4-5'],
      cost: 0.12,
      savingsUSD: 0.03,
      calls: 1,
      turns: 1,
      inputTokens: 100,
      outputTokens: 20,
      cacheReadTokens: 400,
      cacheWriteTokens: 30,
      startedAt: '2026-07-10T10:00:00.000Z',
      endedAt: '2026-07-10T10:05:00.000Z',
      durationMs: 300_000,
    }])
  })

  it('renders a simple table', () => {
    const output = renderTable(aggregateSessions([makeProject()]))
    expect(output).toContain('SESSION')
    expect(output).toContain('session-1')
    expect(output).toContain('claude-sonnet-4-5')
  })
})