"use client";

import { useEffect, useState } from "react";
import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";
import { Button } from "@/components/ui/button";
import { RefreshCw } from "lucide-react";

interface Task {
  id: string;
  headline: string;
  projectName: string;
  status: string;
  dueDate: string | null;
  details?: string | null;
  milestone?: string | null;
  class: string;
}

export function Flow() {
  const [tasks, setTasks] = useState<Task[]>([]);
  const [error, setError] = useState<string | null>(null);
  const [loading, setLoading] = useState(true);
  const [refreshing, setRefreshing] = useState(false);
  const [retryTimeout, setRetryTimeout] = useState<NodeJS.Timeout | null>(null);

  const fetchTasks = async (isRefresh = false) => {
    try {
      if (isRefresh) {
        setRefreshing(true);
      }
      const response = await fetch('/api/leantime/status-labels');
      
      if (response.status === 429) {
        const retryAfter = parseInt(response.headers.get('retry-after') || '60');
        const timeout = setTimeout(() => fetchTasks(), retryAfter * 1000);
        setRetryTimeout(timeout);
        setError(`Rate limit exceeded. Retrying in ${retryAfter} seconds...`);
        return;
      }

      if (!response.ok) {
        throw new Error('Failed to fetch tasks');
      }

      const data = await response.json();
      setTasks(data.tasks || []);
      setError(null);
    } catch (err) {
      console.error('Error fetching tasks:', err);
      setError('Failed to fetch tasks');
    } finally {
      setLoading(false);
      setRefreshing(false);
    }
  };

  useEffect(() => {
    fetchTasks();
    return () => {
      if (retryTimeout) {
        clearTimeout(retryTimeout);
      }
    };
  }, []);

  // Group tasks by project
  const groupedTasks = tasks.reduce((acc, task) => {
    if (!acc[task.projectName]) {
      acc[task.projectName] = [];
    }
    acc[task.projectName].push(task);
    return acc;
  }, {} as Record<string, Task[]>);

  return (
    <Card className="transition-transform duration-500 ease-in-out transform hover:scale-105">
      <CardHeader className="flex flex-row items-center justify-between pb-2">
        <CardTitle className="text-lg font-medium">Flow</CardTitle>
        <Button
          variant="ghost"
          size="icon"
          onClick={() => fetchTasks(true)}
          disabled={refreshing || !!retryTimeout}
          className={refreshing ? 'animate-spin' : ''}
        >
          <RefreshCw className="h-4 w-4" />
        </Button>
      </CardHeader>
      <CardContent>
        {loading ? (
          <div className="flex items-center justify-center py-4">
            <div className="h-4 w-4 animate-spin rounded-full border-2 border-primary border-t-transparent" />
          </div>
        ) : error ? (
          <div className="text-center text-sm text-red-500">{error}</div>
        ) : Object.keys(groupedTasks).length === 0 ? (
          <div className="text-center text-sm text-gray-500">No tasks found</div>
        ) : (
          <div className="space-y-6">
            {Object.entries(groupedTasks).map(([projectName, projectTasks]) => (
              <div key={projectName} className="space-y-2">
                <h3 className="font-medium text-sm">{projectName}</h3>
                <div className="space-y-2">
                  {projectTasks.map((task) => (
                    <div key={task.id} className="flex justify-between items-start text-sm border-b border-gray-100 pb-2">
                      <div className="flex-1">
                        <div className="font-medium">{task.headline}</div>
                      </div>
                      <div className={`ml-4 px-2 py-1 rounded text-xs ${getLabelClass(task.class)}`}>
                        {task.status.toUpperCase()}
                      </div>
                    </div>
                  ))}
                </div>
              </div>
            ))}
          </div>
        )}
      </CardContent>
    </Card>
  );
}

function getLabelClass(className: string): string {
  const classMap: Record<string, string> = {
    'label-default': 'bg-gray-100 text-gray-800',
    'label-success': 'bg-green-100 text-green-800',
    'label-warning': 'bg-yellow-100 text-yellow-800',
    'label-info': 'bg-blue-100 text-blue-800',
    'label-blue': 'bg-blue-100 text-blue-800',
    'label-darker-blue': 'bg-indigo-100 text-indigo-800',
    'label-dark-green': 'bg-emerald-100 text-emerald-800',
  };
  
  return classMap[className] || 'bg-gray-100 text-gray-800';
}