import React, { useState } from 'react';
import { useQuery } from '@tanstack/react-query';
import { useNavigate } from 'react-router-dom';
import { Header } from '@/components/layout/Header';
import { Input } from '@/components/ui/input';
import { Button } from '@/components/ui/button';
import { Badge } from '@/components/ui/badge';
import { Dialog, DialogContent } from '@/components/ui/dialog';
import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from '@/components/ui/select';
import { Search, Filter, Building, Plus } from 'lucide-react';
import { supabase } from '@/integrations/supabase/client';
import OperatorCard from '@/components/operators/OperatorCard';
import { OperatorForm } from '@/components/admin/OperatorForm';
import { Company } from '@/types/database';
import { useAuth } from '@/hooks/useAuth';
import { useUserRole } from '@/hooks/useUserRole';
import { toast } from '@/hooks/use-toast';
import { submitCompanyCreation } from '@/lib/companyHelpers';

const Operators = () => {
  const navigate = useNavigate();
  const { user } = useAuth();
  const { isModerator } = useUserRole();
  const [searchTerm, setSearchTerm] = useState('');
  const [sortBy, setSortBy] = useState('name');
  const [filterBy, setFilterBy] = useState('all');
  const [isCreateModalOpen, setIsCreateModalOpen] = useState(false);

  const { data: operators, isLoading } = useQuery({
    queryKey: ['operators'],
    queryFn: async () => {
      // Get companies that are park operators with park counts
      const { data, error } = await supabase
        .from('companies')
        .select(`
          *,
          parks:parks!operator_id(count)
        `)
        .in('id', 
          await supabase
            .from('parks')
            .select('operator_id')
            .not('operator_id', 'is', null)
            .then(({ data }) => data?.map(park => park.operator_id) || [])
        )
        .order('name');

      if (error) throw error;
      
      // Transform the data to include park_count
      const transformedData = data?.map(company => ({
        ...company,
        park_count: company.parks?.[0]?.count || 0
      })) || [];
      
      return transformedData as (Company & { park_count: number })[];
    },
  });

  const handleCreateSubmit = async (data: any) => {
    try {
      if (!user) {
        navigate('/auth');
        return;
      }

      await submitCompanyCreation(
        data,
        'operator',
        user.id
      );

      toast({
        title: "Operator Submitted",
        description: "Your submission has been sent for review."
      });

      setIsCreateModalOpen(false);
    } catch (error: any) {
      toast({
        title: "Error",
        description: error.message || "Failed to submit operator.",
        variant: "destructive"
      });
    }
  };

  const filteredAndSortedOperators = React.useMemo(() => {
    if (!operators) return [];

    let filtered = operators.filter(operator => {
      const matchesSearch = operator.name.toLowerCase().includes(searchTerm.toLowerCase()) ||
                          (operator.description && operator.description.toLowerCase().includes(searchTerm.toLowerCase()));
      
      if (filterBy === 'all') return matchesSearch;
      if (filterBy === 'with-rating') return matchesSearch && operator.average_rating > 0;
      if (filterBy === 'established') return matchesSearch && operator.founded_year;
      
      return matchesSearch;
    });

    // Sort
    filtered.sort((a, b) => {
      switch (sortBy) {
        case 'name':
          return a.name.localeCompare(b.name);
        case 'rating':
          return (b.average_rating || 0) - (a.average_rating || 0);
        case 'founded':
          return (b.founded_year || 0) - (a.founded_year || 0);
        case 'reviews':
          return (b.review_count || 0) - (a.review_count || 0);
        default:
          return 0;
      }
    });

    return filtered;
  }, [operators, searchTerm, sortBy, filterBy]);

  return (
    <div className="min-h-screen bg-background">
      <Header />
      
      <main className="container mx-auto px-4 py-8">
        {/* Page Header */}
        <div className="mb-8">
          <div className="flex items-center gap-3 mb-4">
            <Building className="w-8 h-8 text-primary" />
            <h1 className="text-4xl font-bold">Park Operators</h1>
          </div>
          <p className="text-lg text-muted-foreground mb-4">
            Discover companies that operate theme parks
          </p>
          
          <div className="flex items-center justify-between">
            <div className="flex flex-wrap items-center gap-1.5 sm:gap-2">
              <Badge variant="secondary" className="flex items-center justify-center text-sm sm:text-base px-2 py-0.5 sm:px-3 sm:py-1 whitespace-nowrap">
                {filteredAndSortedOperators?.length || 0} operators
              </Badge>
              {searchTerm && (
                <Badge variant="outline" className="flex items-center justify-center text-xs sm:text-sm px-2 py-0.5 whitespace-nowrap">
                  Searching: "{searchTerm}"
                </Badge>
              )}
            </div>

            <div className="flex items-center gap-2">
              <Button 
                onClick={() => {
                  if (!user) {
                    navigate('/auth');
                  } else {
                    setIsCreateModalOpen(true);
                  }
                }}
                className="gap-2"
              >
                <Plus className="w-4 h-4" />
                Add Operator
              </Button>
            </div>
          </div>
        </div>

        {/* Search and Filters */}
        <div className="space-y-3 mb-6">
          <div className="relative">
            <Search className="absolute left-3 top-1/2 transform -translate-y-1/2 text-muted-foreground h-4 w-4" />
            <Input
              placeholder="Search park operators..."
              value={searchTerm}
              onChange={(e) => setSearchTerm(e.target.value)}
              className="pl-10 h-11"
            />
          </div>
          
          <div className="flex gap-2">
            <Select value={sortBy} onValueChange={setSortBy}>
              <SelectTrigger className="flex-1 h-10">
                <SelectValue placeholder="Sort by" />
              </SelectTrigger>
              <SelectContent>
                <SelectItem value="name">Name</SelectItem>
                <SelectItem value="rating">Rating</SelectItem>
                <SelectItem value="founded">Founded Year</SelectItem>
                <SelectItem value="reviews">Review Count</SelectItem>
              </SelectContent>
            </Select>

            <Select value={filterBy} onValueChange={setFilterBy}>
              <SelectTrigger className="flex-1 h-10">
                <Filter className="h-4 w-4 mr-2" />
                <SelectValue placeholder="Filter" />
              </SelectTrigger>
              <SelectContent>
                <SelectItem value="all">All</SelectItem>
                <SelectItem value="with-rating">Rated</SelectItem>
                <SelectItem value="established">Est.</SelectItem>
              </SelectContent>
            </Select>
          </div>
        </div>


        {/* Loading State */}
        {isLoading && (
          <div className="text-center py-12">
            <div className="animate-spin rounded-full h-8 w-8 border-b-2 border-primary mx-auto"></div>
            <p className="text-muted-foreground mt-4">Loading park operators...</p>
          </div>
        )}

        {/* Operators Grid */}
        {!isLoading && filteredAndSortedOperators && (
          <div className="grid grid-cols-1 sm:grid-cols-2 lg:grid-cols-3 xl:grid-cols-4 gap-4 md:gap-6">
            {filteredAndSortedOperators.map((operator) => (
              <OperatorCard key={operator.id} company={operator} />
            ))}
          </div>
        )}

        {/* Empty State */}
        {!isLoading && filteredAndSortedOperators?.length === 0 && (
          <div className="text-center py-12">
            <Building className="h-12 w-12 text-muted-foreground mx-auto mb-4" />
            <h3 className="text-lg font-semibold mb-2">No park operators found</h3>
            <p className="text-muted-foreground">
              {searchTerm 
                ? "Try adjusting your search terms or filters"
                : "No park operators are currently available"
              }
            </p>
          </div>
        )}

        {/* Create Modal */}
        <Dialog open={isCreateModalOpen} onOpenChange={setIsCreateModalOpen}>
          <DialogContent className="max-w-4xl max-h-[90vh] overflow-y-auto">
            <OperatorForm
              onSubmit={handleCreateSubmit}
              onCancel={() => setIsCreateModalOpen(false)}
            />
          </DialogContent>
        </Dialog>
      </main>
    </div>
  );
};

export default Operators;
