5chan/src/lib/media-hosting/__tests__/upload-orchestrator.test.ts

import { beforeEach, describe, expect, it, vi } from 'vitest';
import { orchestrateElectronUpload } from '../upload-orchestrator';
import { uploadToCatbox } from '../../utils/catbox-utils';
import type { ProviderId } from '../types';

vi.mock('../../utils/catbox-utils', () => ({
  uploadToCatbox: vi.fn(),
}));

function createElectronApiMock() {
  return {
    isElectron: true,
    copyToClipboard: vi.fn(async () => ({ success: true })),
    getPlatform: vi.fn(async () => ({ platform: 'darwin' as NodeJS.Platform, arch: 'x64', version: 'v20.0.0' })),
    automateUploadMedia: vi.fn(async (options: { provider: ProviderId }) => ({ url: 'https://i.imgur.com/abc.png', provider: options.provider })),
    getPathForFile: vi.fn((): string | null => '/tmp/image.png'),
  };
}

describe('orchestrateElectronUpload', () => {
  beforeEach(() => {
    vi.clearAllMocks();
    window.electronApi = undefined;
    window.isElectron = false;
  });

  it('uploads via catbox provider directly', async () => {
    vi.mocked(uploadToCatbox).mockResolvedValue('https://files.catbox.moe/a.png');
    const file = new File(['a'], 'a.png', { type: 'image/png' });

    const url = await orchestrateElectronUpload(file, ['catbox']);

    expect(url).toBe('https://files.catbox.moe/a.png');
    expect(uploadToCatbox).toHaveBeenCalledWith(file);
  });

  it('uses electronApi.getPathForFile when File.path is unavailable', async () => {
    const electronApi = createElectronApiMock();
    window.electronApi = electronApi;

    const file = new File(['x'], 'x.png', { type: 'image/png' });
    const url = await orchestrateElectronUpload(file, ['imgur']);

    expect(url).toBe('https://i.imgur.com/abc.png');
    expect(electronApi.getPathForFile).toHaveBeenCalledWith(file);
    expect(electronApi.automateUploadMedia).toHaveBeenCalledWith({
      provider: 'imgur',
      filePath: '/tmp/image.png',
    });
  });

  it('routes ImgBB through Electron automation', async () => {
    const electronApi = createElectronApiMock();
    electronApi.automateUploadMedia = vi.fn(async () => ({ url: 'https://i.ibb.co/example/image.png', provider: 'imgbb' as const }));
    window.electronApi = electronApi;

    const file = new File(['x'], 'x.png', { type: 'image/png' });
    const url = await orchestrateElectronUpload(file, ['imgbb']);

    expect(url).toBe('https://i.ibb.co/example/image.png');
    expect(electronApi.automateUploadMedia).toHaveBeenCalledWith({
      provider: 'imgbb',
      filePath: '/tmp/image.png',
    });
  });

  it('fails with provider attempt details if no file path can be resolved', async () => {
    const electronApi = createElectronApiMock();
    electronApi.getPathForFile = vi.fn((): string | null => null);
    window.electronApi = electronApi;

    const file = new File(['z'], 'z.png', { type: 'image/png' });

    try {
      await orchestrateElectronUpload(file, ['imgur']);
      throw new Error('Expected orchestrateElectronUpload to throw');
    } catch (error) {
      const typedError = error as Error & {
        attempts?: Array<{ provider: string; error?: string; elapsedMs?: number; stage?: string }>;
      };
      expect(typedError.message).toBe('All providers failed');
      expect(typedError.attempts?.[0]?.provider).toBe('imgur');
      expect(typedError.attempts?.[0]?.error).toContain('File path required for Electron automation');
      expect(typedError.attempts?.[0]?.elapsedMs).toBeGreaterThanOrEqual(0);
      expect(typedError.attempts?.[0]?.stage).toBeDefined();
    }
  });

  it('includes stage and matchedSelectors when provider throws block/file-input errors', async () => {
    const electronApi = createElectronApiMock();
    electronApi.automateUploadMedia = vi.fn().mockRejectedValue(new Error('No file input found for imgur. Tried: input[type="file"], #upload'));
    window.electronApi = electronApi;

    const file = new File(['x'], 'x.png', { type: 'image/png' });

    try {
      await orchestrateElectronUpload(file, ['imgur']);
      throw new Error('Expected orchestrateElectronUpload to throw');
    } catch (error) {
      const typedError = error as Error & {
        attempts?: Array<{ provider: string; error?: string; stage?: string; elapsedMs?: number; matchedSelectors?: string[] }>;
      };
      expect(typedError.attempts?.[0]?.provider).toBe('imgur');
      expect(typedError.attempts?.[0]?.stage).toBe('file_input');
      expect(typedError.attempts?.[0]?.matchedSelectors).toEqual(['input[type="file"]', '#upload']);
      expect(typedError.attempts?.[0]?.elapsedMs).toBeGreaterThanOrEqual(0);
    }
  });

  it('parses submit selector failure with matchedSelectors', async () => {
    const electronApi = createElectronApiMock();
    electronApi.automateUploadMedia = vi
      .fn()
      .mockRejectedValue(new Error('No submit button found for imgur. Tried: button[type="submit"], [data-action="upload"], .upload-btn'));
    window.electronApi = electronApi;

    const file = new File(['x'], 'x.png', { type: 'image/png' });

    try {
      await orchestrateElectronUpload(file, ['imgur']);
      throw new Error('Expected orchestrateElectronUpload to throw');
    } catch (error) {
      const typedError = error as Error & {
        attempts?: Array<{ provider: string; stage?: string; matchedSelectors?: string[] }>;
      };
      expect(typedError.attempts?.[0]?.provider).toBe('imgur');
      expect(typedError.attempts?.[0]?.stage).toBe('submit');
      expect(typedError.attempts?.[0]?.matchedSelectors).toEqual(['button[type="submit"]', '[data-action="upload"]', '.upload-btn']);
    }
  });

  it('parses timeout stage when upload or URL extraction times out', async () => {
    const electronApi = createElectronApiMock();
    electronApi.automateUploadMedia = vi.fn().mockRejectedValue(new Error('Upload timeout or no direct URL extracted for imgur (elapsed: 45000ms, timeout: 45000ms)'));
    window.electronApi = electronApi;

    const file = new File(['x'], 'x.png', { type: 'image/png' });

    try {
      await orchestrateElectronUpload(file, ['imgur']);
      throw new Error('Expected orchestrateElectronUpload to throw');
    } catch (error) {
      const typedError = error as Error & {
        attempts?: Array<{ provider: string; stage?: string }>;
      };
      expect(typedError.attempts?.[0]?.provider).toBe('imgur');
      expect(typedError.attempts?.[0]?.stage).toBe('timeout');
    }
  });

  it('parses page_load stage when page fails to load', async () => {
    const electronApi = createElectronApiMock();
    electronApi.automateUploadMedia = vi.fn().mockRejectedValue(new Error('Page load failed: -3 net::ERR_ABORTED'));
    window.electronApi = electronApi;

    const file = new File(['x'], 'x.png', { type: 'image/png' });

    try {
      await orchestrateElectronUpload(file, ['imgur']);
      throw new Error('Expected orchestrateElectronUpload to throw');
    } catch (error) {
      const typedError = error as Error & {
        attempts?: Array<{ provider: string; stage?: string }>;
      };
      expect(typedError.attempts?.[0]?.provider).toBe('imgur');
      expect(typedError.attempts?.[0]?.stage).toBe('page_load');
    }
  });

  it('parses blocked stage when captcha or challenge detected', async () => {
    const electronApi = createElectronApiMock();
    electronApi.automateUploadMedia = vi.fn().mockRejectedValue(new Error('Provider blocked: captcha, login, or challenge detected (imgur), selector: .g-recaptcha'));
    window.electronApi = electronApi;

    const file = new File(['x'], 'x.png', { type: 'image/png' });

    try {
      await orchestrateElectronUpload(file, ['imgur']);
      throw new Error('Expected orchestrateElectronUpload to throw');
    } catch (error) {
      const typedError = error as Error & {
        attempts?: Array<{ provider: string; stage?: string }>;
      };
      expect(typedError.attempts?.[0]?.provider).toBe('imgur');
      expect(typedError.attempts?.[0]?.stage).toBe('blocked');
    }
  });
});