import { useState } from 'react';
import type { Meta, StoryObj } from '@storybook/react';
import { ProvidersStep } from './ProvidersStep';
import type {
  ProviderData,
  ProviderFilterValues,
  ProviderSortBy,
  ListViewMode,
} from './ProvidersStep';
import { EMPTY_FILTER_VALUES } from './ProvidersStep';
import { Navigation } from '../../organisms/Navigation';
import Box from '@mui/material/Box';

// ─── Helpers ─────────────────────────────────────────────────────────────────

const FALogo = () => (
  <Box sx={{ display: 'flex', alignItems: 'center' }}>
    <Box
      component="img"
      src="/brandlogo/logo-full.svg"
      alt="Funeral Arranger"
      sx={{ height: 28, display: { xs: 'none', md: 'block' } }}
    />
    <Box
      component="img"
      src="/brandlogo/logo-short.svg"
      alt="Funeral Arranger"
      sx={{ height: 28, display: { xs: 'block', md: 'none' } }}
    />
  </Box>
);

const nav = (
  <Navigation
    logo={<FALogo />}
    items={[
      { label: 'FAQ', href: '/faq' },
      { label: 'Contact Us', href: '/contact' },
      { label: 'Log in', href: '/login' },
    ]}
  />
);

const mockProviders: ProviderData[] = [
  {
    id: 'parsons',
    name: 'H.Parsons Funeral Directors',
    location: 'Wentworth, NSW',
    verified: true,
    imageUrl: 'https://placehold.co/600x200/E8E0D6/8B6F47?text=H.Parsons',
    logoUrl: 'https://placehold.co/64x64/FEF9F5/BA834E?text=HP',
    rating: 4.6,
    reviewCount: 7,
    startingPrice: 900,
    distanceKm: 2.3,
    description:
      'H.Parsons delivers premium funeral services with exceptional care and support, guiding families through every step with empathy and expertise.',
  },
  {
    id: 'rankins',
    name: 'Rankins Funeral Services',
    location: 'Wollongong, NSW',
    verified: true,
    imageUrl: 'https://placehold.co/600x200/D7E1E2/4C5B6B?text=Rankins',
    logoUrl: 'https://placehold.co/64x64/F2F5F6/4C5B6B?text=R',
    rating: 4.8,
    reviewCount: 23,
    startingPrice: 1200,
    distanceKm: 5.1,
  },
  {
    id: 'wollongong-city',
    name: 'Wollongong City Funerals',
    location: 'Wollongong, NSW',
    verified: false,
    rating: 4.2,
    reviewCount: 15,
    startingPrice: 750,
    distanceKm: 6.8,
  },
  {
    id: 'easy-funerals',
    name: 'Easy Funerals',
    location: 'Sydney, NSW',
    verified: true,
    imageUrl: 'https://placehold.co/600x200/F0F7F0/3B7A3B?text=Easy+Funerals',
    logoUrl: 'https://placehold.co/64x64/F0F7F0/3B7A3B?text=EF',
    rating: 4.5,
    reviewCount: 42,
    startingPrice: 850,
    distanceKm: 12.4,
  },
  {
    id: 'botanical',
    name: 'Botanical Funerals',
    location: 'Newtown, NSW',
    verified: false,
    rating: 4.9,
    reviewCount: 8,
    startingPrice: 2100,
    distanceKm: 15.0,
  },
];

// ─── Meta ────────────────────────────────────────────────────────────────────

const meta: Meta<typeof ProvidersStep> = {
  title: 'Pages/ProvidersStep',
  component: ProvidersStep,
  tags: ['autodocs'],
  parameters: {
    layout: 'fullscreen',
  },
};

export default meta;
type Story = StoryObj<typeof ProvidersStep>;

// ─── Interactive (default) ──────────────────────────────────────────────────

/** Full filter panel — tradition, funeral type, verified, online, price range */
export const Default: Story = {
  render: () => {
    const [query, setQuery] = useState('');
    const [filters, setFilters] = useState<ProviderFilterValues>(EMPTY_FILTER_VALUES);
    const [sort, setSort] = useState<ProviderSortBy>('recommended');
    const [view, setView] = useState<ListViewMode>('list');

    const filtered = mockProviders.filter((p) =>
      p.location.toLowerCase().includes(query.toLowerCase()),
    );

    return (
      <ProvidersStep
        providers={filtered}
        onSelectProvider={(id) => alert(`Navigate to provider: ${id}`)}
        searchQuery={query}
        onSearchChange={setQuery}
        filterValues={filters}
        onFilterChange={setFilters}
        sortBy={sort}
        onSortChange={setSort}
        viewMode={view}
        onViewModeChange={setView}
        onBack={() => alert('Back')}
        navigation={nav}
      />
    );
  },
};

// ─── With active filters ────────────────────────────────────────────────────

/** Filters pre-applied — location, tradition, type, verified, price cap */
export const WithActiveFilters: Story = {
  render: () => {
    const [query, setQuery] = useState('Wollongong, NSW');
    const [filters, setFilters] = useState<ProviderFilterValues>({
      tradition: 'Catholic',
      funeralTypes: ['service_and_cremation'],
      verifiedOnly: true,
      onlineArrangements: false,
      priceRange: [0, 2000],
    });
    const [sort, setSort] = useState<ProviderSortBy>('nearest');
    const [view, setView] = useState<ListViewMode>('list');

    const filtered = mockProviders.filter((p) =>
      p.location.toLowerCase().includes(query.toLowerCase()),
    );

    return (
      <ProvidersStep
        providers={filtered}
        onSelectProvider={(id) => alert(`Navigate to provider: ${id}`)}
        searchQuery={query}
        onSearchChange={setQuery}
        filterValues={filters}
        onFilterChange={setFilters}
        sortBy={sort}
        onSortChange={setSort}
        viewMode={view}
        onViewModeChange={setView}
        onBack={() => alert('Back')}
        navigation={nav}
      />
    );
  },
};

// ─── Pre-planning variant ───────────────────────────────────────────────────

/** Pre-planning flow — softer copy */
export const PrePlanning: Story = {
  render: () => {
    const [query, setQuery] = useState('');
    const [filters, setFilters] = useState<ProviderFilterValues>(EMPTY_FILTER_VALUES);

    return (
      <ProvidersStep
        providers={mockProviders}
        onSelectProvider={(id) => alert(`Navigate to provider: ${id}`)}
        searchQuery={query}
        onSearchChange={setQuery}
        filterValues={filters}
        onFilterChange={setFilters}
        onBack={() => alert('Back')}
        navigation={nav}
        isPrePlanning
      />
    );
  },
};

// ─── Empty results ──────────────────────────────────────────────────────────

/** Search yielded no results */
export const EmptyResults: Story = {
  render: () => {
    const [query, setQuery] = useState('xyz');
    const [filters, setFilters] = useState<ProviderFilterValues>(EMPTY_FILTER_VALUES);

    return (
      <ProvidersStep
        providers={[]}
        onSelectProvider={() => {}}
        searchQuery={query}
        onSearchChange={setQuery}
        filterValues={filters}
        onFilterChange={setFilters}
        onBack={() => alert('Back')}
        navigation={nav}
      />
    );
  },
};

// ─── Single provider ────────────────────────────────────────────────────────

/** Only one provider available */
export const SingleProvider: Story = {
  render: () => {
    const [query, setQuery] = useState('');
    const [filters, setFilters] = useState<ProviderFilterValues>(EMPTY_FILTER_VALUES);

    return (
      <ProvidersStep
        providers={[mockProviders[0]]}
        onSelectProvider={(id) => alert(`Navigate to provider: ${id}`)}
        searchQuery={query}
        onSearchChange={setQuery}
        filterValues={filters}
        onFilterChange={setFilters}
        onBack={() => alert('Back')}
        navigation={nav}
      />
    );
  },
};
