import { render, screen } from '@testing-library/react';
import { MemoryRouter, Route, Routes } from 'react-router-dom';
import { beforeEach, describe, expect, it, vi } from 'vitest';
import { MainLayout } from '@/components/layout/MainLayout';
import { useOpencodeStore } from '@/stores/opencode';
import { useProjectConfigStore } from '@/stores/project-config';
import { createProjectConfig } from '../../shared/project-config';

vi.mock('@/components/layout/Sidebar', () => ({ Sidebar: () => <aside data-testid="sidebar-stub" /> }));
vi.mock('@/components/layout/TitleBar', () => ({
  TitleBar: ({ overlay = false }: { overlay?: boolean }) => (
    <header data-testid="titlebar-stub" data-overlay={overlay ? 'true' : 'false'} />
  ),
}));

function renderLayout(path: string) {
  render(
    <MemoryRouter initialEntries={[path]}>
      <Routes>
        <Route element={<MainLayout />}>
          <Route path="*" element={<div data-testid="route-content" />} />
        </Route>
      </Routes>
    </MemoryRouter>,
  );
}

describe('MainLayout module isolation', () => {
  beforeEach(() => {
    const project = {
      id: 'local-project',
      path: '/tmp/local-project',
      name: 'local-project',
      createdAt: '',
      updatedAt: '',
      lastOpenedAt: '',
    };
    useOpencodeStore.setState({ activeProject: project });
    useProjectConfigStore.setState({
      configsByProjectId: {
        [project.id]: { ...createProjectConfig(), initialized: false },
      },
    });
  });

  it('does not overlay the local project initialization gate on AI painting', () => {
    renderLayout('/image-canvas');

    expect(screen.getByTestId('route-content')).toBeVisible();
    expect(screen.queryByTestId('project-initialization-gate')).not.toBeInTheDocument();
  });

  it('uses the full-height workspace shell for AI painting', () => {
    renderLayout('/image-canvas');

    expect(screen.getByTestId('main-content')).toHaveClass('basis-0', 'h-full', 'overflow-hidden', 'p-0');
    expect(screen.getByTestId('main-content')).not.toHaveClass('p-5', 'sm:p-6');
    expect(screen.getByTestId('titlebar-stub')).toHaveAttribute('data-overlay', 'true');
  });

  it('keeps the local project initialization gate for AI programming routes', () => {
    renderLayout('/opencode-chat');

    expect(screen.getByTestId('project-initialization-gate')).toBeVisible();
    expect(screen.getByTestId('titlebar-stub')).toHaveAttribute('data-overlay', 'false');
  });

  it('keeps the programming workspace padding compensation separate from painting', () => {
    renderLayout('/opencode-chat');

    expect(screen.getByTestId('main-content')).toHaveClass('basis-0', 'overflow-hidden', 'p-0', 'sm:p-6');
  });
});
