import { describe, expect, it } from 'vitest';
import { prepareArtifactDeployment } from './artifact-runtime';
import {
  buildStreamingArtifactDraft,
  extractStreamingJsonStrings,
} from './streaming-artifact';

describe('extractStreamingJsonStrings', () => {
  it('decodes complete properties and the active unfinished string', () => {
    expect(extractStreamingJsonStrings(
      '{"title":"Demo","file_1_path":"index.html","file_1_content":"<h1>Hello\\nwor',
    )).toEqual({
      values: {
        title: 'Demo',
        file_1_path: 'index.html',
      },
      partial: {
        key: 'file_1_content',
        value: '<h1>Hello\nwor',
      },
    });
  });

  it('holds an unfinished JSON escape until it becomes decodable', () => {
    expect(extractStreamingJsonStrings('{"html":"hello\\u26')).toEqual({
      values: {},
      partial: { key: 'html', value: 'hello' },
    });
  });
});

describe('buildStreamingArtifactDraft', () => {
  it('renders partial HTML before the content property closes', () => {
    const draft = buildStreamingArtifactDraft(
      'artifact_deploy',
      '{"title":"Live","file_1_path":"index.html","file_1_content":"<!doctype html><h1>Hel',
      'draft-live',
    );

    expect(draft?.files).toMatchObject([{
      path: 'index.html',
      content: '<!doctype html><h1>Hel',
      partial: true,
    }]);
    expect(draft?.preview?.source).toContain('<h1>Hel');
  });

  it('shows partial JavaScript in the tree without executing it', () => {
    const draft = buildStreamingArtifactDraft(
      'artifact_deploy',
      '{"file_1_path":"index.html","file_1_content":"<main>Ready</main>","file_2_path":"app.js","file_2_content":"document.querySele',
      'draft-script',
    );

    expect(draft?.files.find((file) => file.path === 'app.js')).toMatchObject({
      content: 'document.querySele',
      partial: true,
    });
    expect(draft?.preview?.files?.some((file) => file.path === 'app.js')).toBe(false);
  });

  it('layers a partial write over the existing artifact without mutating it', () => {
    const original = prepareArtifactDeployment([
      { path: 'index.html', content: '<link rel="stylesheet" href="./style.css"><main>Old</main>' },
      { path: 'style.css', content: 'main { color: red; }' },
    ], 'Original', 'index.html', 'artifact-1');
    const draft = buildStreamingArtifactDraft(
      'artifact_write',
      '{"artifact_id":"artifact-1","file_1_path":"style.css","file_1_content":"main { color: blue',
      'draft-write',
      new Map([[original.id, original]]),
    );

    expect(draft?.files.find((file) => file.path === 'style.css')).toMatchObject({
      content: 'main { color: blue',
      partial: true,
    });
    expect(original.files?.find((file) => file.path === 'style.css')?.content).toBe('main { color: red; }');
    expect(draft?.preview?.files?.find((file) => file.path === 'style.css')?.content).toBe('main { color: blue');
  });
});