import { mkdtemp, rm, writeFile } from 'fs/promises'
import { join } from 'path'
import { tmpdir } from 'os'
import { afterEach, beforeEach, describe, expect, it } from 'vitest'
import { convertContentBlocks, convertContentBlocksForAgent } from '../../packages/server/src/services/hermes/run-chat/content-blocks'

let tempDir = ''

describe('run chat content blocks', () => {
  beforeEach(async () => {
    tempDir = await mkdtemp(join(tmpdir(), 'hermes-content-blocks-'))
  })

  afterEach(async () => {
    if (tempDir) await rm(tempDir, { recursive: true, force: true })
  })

  it('keeps API image conversion as base64 input_image only', async () => {
    const imagePath = join(tempDir, 'image.png')
    await writeFile(imagePath, Buffer.from([1, 2, 3]))

    const parts = await convertContentBlocks([
      { type: 'text', text: 'animate this' },
      { type: 'image', name: 'image.png', path: imagePath, media_type: 'image/png' },
    ])

    expect(parts).toHaveLength(2)
    expect(parts[0]).toEqual({ type: 'input_text', text: 'animate this' })
    expect(parts[1].type).toBe('input_image')
    expect(parts[1].image_url).toMatch(/^data:image\/png;base64,/)
    expect(JSON.stringify(parts)).not.toContain('Local image path for tools')
  })

  it('adds local file path text for bridge agents while preserving the image data', async () => {
    const imagePath = join(tempDir, 'image.png')
    await writeFile(imagePath, Buffer.from([1, 2, 3]))

    const parts = await convertContentBlocksForAgent([
      { type: 'text', text: 'animate this' },
      { type: 'image', name: 'image.png', path: imagePath, media_type: 'image/png' },
    ])

    expect(parts).toHaveLength(3)
    expect(parts[0]).toEqual({ type: 'text', text: 'animate this' })
    expect(parts[1]).toEqual({
      type: 'text',
      text: `[Attached image: image.png]\nLocal image path for tools: ${imagePath}`,
    })
    expect(parts[2].type).toBe('image_url')
    expect(parts[2].image_url?.url).toMatch(/^data:image\/png;base64,/)
  })
})