import { useState } from "react";
import { Button, Group, Modal, Stack, Text, TextInput, Textarea } from "@mantine/core";
import toast from "react-hot-toast";

import { trainSchedulingService } from "@/services/trainScheduling.service";

export function RescheduleTrainDialog({
  scheduleId,
  currentBookingIds,
  opened,
  onClose,
  onComplete,
}: {
  scheduleId: string;
  currentBookingIds: string[];
  opened: boolean;
  onClose: () => void;
  onComplete?: () => void;
}) {
  const [newDepartureDate, setNewDepartureDate] = useState("");
  const [reason, setReason] = useState("");
  const [loading, setLoading] = useState(false);

  const handleSubmit = async () => {
    if (!newDepartureDate) {
      toast.error("Select a new departure date");
      return;
    }
    setLoading(true);
    try {
      await trainSchedulingService.maintenanceReschedule(scheduleId, {
        incomingBookingIds: currentBookingIds,
        newDepartureDate: new Date(newDepartureDate).toISOString(),
        reason,
      });
      toast.success("Train rescheduled for maintenance");
      onComplete?.();
      onClose();
    } catch {
      toast.error("Reschedule failed");
    } finally {
      setLoading(false);
    }
  };

  return (
    <Modal opened={opened} onClose={onClose} title="Reschedule train (maintenance)" radius="lg">
      <Stack gap="md">
        <Text size="sm" c="dimmed">
          Updates departure and rebalances bookings on this train. Displaced bookings return to
          the operations queue when capacity is insufficient.
        </Text>
        <TextInput
          label="New departure"
          type="datetime-local"
          value={newDepartureDate}
          onChange={(e) => setNewDepartureDate(e.target.value)}
        />
        <Textarea
          label="Reason"
          placeholder="e.g. Locomotive maintenance"
          value={reason}
          onChange={(e) => setReason(e.target.value)}
        />
        <Group justify="flex-end">
          <Button variant="default" onClick={onClose}>
            Cancel
          </Button>
          <Button loading={loading} onClick={handleSubmit}>
            Reschedule
          </Button>
        </Group>
      </Stack>
    </Modal>
  );
}
