worklenz/worklenz-frontend/src/components/enhanced-kanban/EnhancedKanbanTaskCard.tsx

import React, { useCallback, useMemo, useState } from 'react';
import { useSortable, defaultAnimateLayoutChanges } from '@dnd-kit/sortable';
import { CSS } from '@dnd-kit/utilities';
import { IProjectTask } from '@/types/project/projectTasksViewModel.types';
import { useAppSelector } from '@/hooks/useAppSelector';
import './EnhancedKanbanTaskCard.css';
import Flex from 'antd/es/flex';
import Tag from 'antd/es/tag';
import Tooltip from 'antd/es/tooltip';
import Progress from 'antd/es/progress';
import Button from 'antd/es/button';
import { useAppDispatch } from '@/hooks/useAppDispatch';
import { setShowTaskDrawer, setSelectedTaskId } from '@/features/task-drawer/task-drawer.slice';
import PrioritySection from '../board/taskCard/priority-section/priority-section';
import Typography from 'antd/es/typography';
import CustomAvatarGroup from '../board/custom-avatar-group';
import CustomDueDatePicker from '../board/custom-due-date-picker';
import { themeWiseColor } from '@/utils/themeWiseColor';
import { ForkOutlined } from '@ant-design/icons';
import { Dayjs } from 'dayjs';
import dayjs from 'dayjs';
import { CaretDownFilled, CaretRightFilled } from '@ant-design/icons';
import { fetchBoardSubTasks } from '@/features/enhanced-kanban/enhanced-kanban.slice';
import { Divider } from 'antd';
import { List } from 'antd';
import { Skeleton } from 'antd';
import { PlusOutlined } from '@ant-design/icons';
import BoardSubTaskCard from '@/pages/projects/projectView/board/board-section/board-sub-task-card/board-sub-task-card';
import BoardCreateSubtaskCard from '@/pages/projects/projectView/board/board-section/board-sub-task-card/board-create-sub-task-card';
import { useTranslation } from 'react-i18next';

interface EnhancedKanbanTaskCardProps {
  task: IProjectTask;
  sectionId: string;
  isActive?: boolean;
  isDragOverlay?: boolean;
  isDropTarget?: boolean;
}
// Priority and status colors - moved outside component to avoid recreation
const PRIORITY_COLORS = {
  critical: '#ff4d4f',
  high: '#ff7a45',
  medium: '#faad14',
  low: '#52c41a',
} as const;

const EnhancedKanbanTaskCard: React.FC<EnhancedKanbanTaskCardProps> = React.memo(({
  task,
  sectionId,
  isActive = false,
  isDragOverlay = false,
  isDropTarget = false
}) => {
  const dispatch = useAppDispatch();
  const { t } = useTranslation('kanban-board');
  const themeMode = useAppSelector(state => state.themeReducer.mode);
  const [showNewSubtaskCard, setShowNewSubtaskCard] = useState(false);
  const [dueDate, setDueDate] = useState<Dayjs | null>(
    task?.end_date ? dayjs(task?.end_date) : null
  );
  const [isSubTaskShow, setIsSubTaskShow] = useState(false);
  const projectId = useAppSelector(state => state.projectReducer.projectId);
  const {
    attributes,
    listeners,
    setNodeRef,
    transform,
    transition,
    isDragging,
  } = useSortable({
    id: task.id!,
    data: {
      type: 'task',
      task,
    },
    disabled: isDragOverlay,
    animateLayoutChanges: defaultAnimateLayoutChanges,
  });

  const style = {
    transform: CSS.Transform.toString(transform),
    transition,
    opacity: isDragging ? 0.5 : 1,
    backgroundColor: themeMode === 'dark' ? '#292929' : '#fafafa',
  };

  const handleCardClick = useCallback((e: React.MouseEvent, id: string) => {
    // Prevent the event from propagating to parent elements
    e.stopPropagation();

    // Don't handle click if we're dragging
    if (isDragging) return;
    dispatch(setSelectedTaskId(id));
    dispatch(setShowTaskDrawer(true));
  }, [dispatch, isDragging]);

  const renderLabels = useMemo(() => {
    if (!task?.labels?.length) return null;

    return (
      <>
        {task.labels.slice(0, 2).map((label: any) => (
          <Tag key={label.id} style={{ marginRight: '2px' }} color={label?.color_code}>
            <span style={{ color: themeMode === 'dark' ? '#383838' : '', fontSize: 10 }}>
              {label.name}
            </span>
          </Tag>
        ))}
        {task.labels.length > 2 && <Tag>+ {task.labels.length - 2}</Tag>}
      </>
    );
  }, [task.labels, themeMode]);



  const handleSubTaskExpand = useCallback(() => {
    console.log('handleSubTaskExpand', task, projectId);
    if (task && task.id && projectId) {
      if (task.show_sub_tasks) {
        // If subtasks are already loaded, just toggle visibility
        setIsSubTaskShow(prev => !prev);
      } else {
        // If subtasks need to be fetched, show the section first with loading state
        setIsSubTaskShow(true);
        dispatch(fetchBoardSubTasks({ taskId: task.id, projectId }));
      }
    }
  }, [task, projectId, dispatch]);

  const handleSubtaskButtonClick = useCallback((e: React.MouseEvent) => {
    e.stopPropagation();
    handleSubTaskExpand();
  }, [handleSubTaskExpand]);

  const handleAddSubtaskClick = useCallback((e: React.MouseEvent) => {
    e.stopPropagation();
    setShowNewSubtaskCard(true);
  }, []);

  return (
    <div
      ref={setNodeRef}
      style={style}
      className={`enhanced-kanban-task-card ${isActive ? 'active' : ''} ${isDragging ? 'dragging' : ''} ${isDragOverlay ? 'drag-overlay' : ''} ${isDropTarget ? 'drop-target' : ''}`}
      {...attributes}
      {...listeners}
    >
      <div className="task-content" onClick={e => handleCardClick(e, task.id || '')}>
        <Flex align="center" justify="space-between" className="mb-2">
          <Flex>
            {renderLabels}
          </Flex>

          <Tooltip title={` ${task?.completed_count} / ${task?.total_tasks_count}`}>
            <Progress type="circle" percent={task?.complete_ratio} size={24} strokeWidth={(task.complete_ratio || 0) >= 100 ? 9 : 7} />
          </Tooltip>
        </Flex>
        <Flex gap={4} align="center">
          {/* Action Icons */}
          <div
            className="w-2 h-2 rounded-full"
            style={{ backgroundColor: task.priority_color || '#d9d9d9' }}
          />
          <Typography.Text
            style={{ fontWeight: 500 }}
            ellipsis={{ tooltip: task.name }}
          >
            {task.name}
          </Typography.Text>
        </Flex>
        <Flex
          align="center"
          justify="space-between"
          style={{
            marginBlock: 8,
          }}
        >
          {task && <CustomAvatarGroup task={task} sectionId={sectionId} />}

          <Flex gap={4} align="center">
            <CustomDueDatePicker task={task} onDateChange={setDueDate} />

            {/* Subtask Section */}
            <Button
              onClick={handleSubtaskButtonClick}
              size="small"
              style={{
                padding: 0,
              }}
              type="text"
            >
              <Tag
                bordered={false}
                style={{
                  display: 'flex',
                  alignItems: 'center',
                  margin: 0,
                  backgroundColor: themeWiseColor('white', '#1e1e1e', themeMode),
                }}
              >
                <ForkOutlined rotate={90} />
                <span>{task.sub_tasks_count}</span>
                {isSubTaskShow ? <CaretDownFilled /> : <CaretRightFilled />}
              </Tag>
            </Button>
          </Flex>
        </Flex>
        <Flex vertical gap={8}>
          {isSubTaskShow && (
            <Flex vertical>
              <Divider style={{ marginBlock: 0 }} />
              <List>
                {task.sub_tasks_loading && (
                  <List.Item>
                    <Skeleton active paragraph={{ rows: 2 }} title={false} style={{ marginTop: 8 }} />
                  </List.Item>
                )}

                {!task.sub_tasks_loading && task?.sub_tasks &&
                  task?.sub_tasks.map((subtask: any) => (
                    <BoardSubTaskCard key={subtask.id} subtask={subtask} sectionId={sectionId} />
                  ))}

                {showNewSubtaskCard && (
                  <BoardCreateSubtaskCard
                    sectionId={sectionId}
                    parentTaskId={task.id || ''}
                    setShowNewSubtaskCard={setShowNewSubtaskCard}
                  />
                )}
              </List>
              <Button
                type="text"
                style={{
                  width: 'fit-content',
                  borderRadius: 6,
                  boxShadow: 'none',
                }}
                icon={<PlusOutlined />}
                onClick={handleAddSubtaskClick}
              >
                {t('addSubtask', 'Add Subtask')}
              </Button>
            </Flex>
          )}
        </Flex>
      </div>
    </div>
  );
});

export default EnhancedKanbanTaskCard;