import { parseComfyWorkflow } from '../../src/types/comfyWorkflow'
import { defaultGraph } from '../../src/scripts/defaultGraph'
import fs from 'fs'

const WORKFLOW_DIR = 'tests-ui/workflows'

describe('parseComfyWorkflow', () => {
  it('parses valid workflow', async () => {
    fs.readdirSync(WORKFLOW_DIR).forEach(async (file) => {
      if (file.endsWith('.json')) {
        const data = fs.readFileSync(`${WORKFLOW_DIR}/${file}`, 'utf-8')
        await expect(parseComfyWorkflow(data)).resolves.not.toThrow()
      }
    })
  })

  it('workflow.nodes', async () => {
    const workflow = JSON.parse(JSON.stringify(defaultGraph))
    workflow.nodes = undefined
    await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow()

    workflow.nodes = null
    await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow()

    workflow.nodes = []
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()
  })

  it('workflow.version', async () => {
    const workflow = JSON.parse(JSON.stringify(defaultGraph))
    workflow.version = undefined
    await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow()

    workflow.version = '1.0.1' // Invalid format.
    await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow()

    workflow.version = 1
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()
  })

  it('workflow.extra', async () => {
    const workflow = JSON.parse(JSON.stringify(defaultGraph))
    workflow.extra = undefined
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()

    workflow.extra = null
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()

    workflow.extra = {}
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()

    workflow.extra = { foo: 'bar' } // Should accept extra fields.
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()
  })

  it('workflow.nodes.pos', async () => {
    const workflow = JSON.parse(JSON.stringify(defaultGraph))
    workflow.nodes[0].pos = [1, 2, 3]
    await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow()

    workflow.nodes[0].pos = [1, 2]
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()

    // Should automatically transform the legacy format object to array.
    workflow.nodes[0].pos = { '0': 3, '1': 4 }
    let parsedWorkflow = await parseComfyWorkflow(JSON.stringify(workflow))
    expect(parsedWorkflow.nodes[0].pos).toEqual([3, 4])

    workflow.nodes[0].pos = { 0: 3, 1: 4 }
    parsedWorkflow = await parseComfyWorkflow(JSON.stringify(workflow))
    expect(parsedWorkflow.nodes[0].pos).toEqual([3, 4])
  })

  it('workflow.nodes.widget_values', async () => {
    const workflow = JSON.parse(JSON.stringify(defaultGraph))
    workflow.nodes[0].widgets_values = ['foo', 'bar']
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()

    workflow.nodes[0].widgets_values = 'foo'
    await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow()

    workflow.nodes[0].widgets_values = undefined
    await expect(
      parseComfyWorkflow(JSON.stringify(workflow))
    ).resolves.not.toThrow()

    // The object format of widgets_values is used by VHS nodes to perform
    // dynamic widgets display.
    workflow.nodes[0].widgets_values = { foo: 'bar' }
    const parsedWorkflow = await parseComfyWorkflow(JSON.stringify(workflow))
    expect(parsedWorkflow.nodes[0].widgets_values).toEqual({ foo: 'bar' })
  })
})