app/src/admin/pages/AttendanceCreate.tsx

import { useState } from "react";
import { useQuery } from "@tanstack/react-query";
import { Link as RouterLink } from "react-router-dom";
import { useNavigate } from "react-router-dom";
import Box from "@mui/material/Box";

import { userListOptions } from "../lib/queries/users";
import { useAlert } from "../context/AlertContext";
import { useAuth } from "../context/AuthContext";
import Forbidden from "../components/Forbidden";
import { useApiMutation } from "../lib/queries/mutations";
import { todayLocalStr } from "../utils/formatters";
import {
  Button,
  Card,
  DateField,
  Field,
  LoadingState,
  PageEnter,
  PageHeader,
  Select,
  TextField,
  TimeField,
} from "../ui";

const API_BASE = "/api/admin";

interface CreateForm {
  user_id: string;
  shift_date: string;
  leave_type: string;
  leave_hours: number;
  arrival_date: string;
  arrival_time: string;
  break_start_date: string;
  break_start_time: string;
  break_end_date: string;
  break_end_time: string;
  departure_date: string;
  departure_time: string;
  notes: string;
}

export default function AttendanceCreate() {
  const alert = useAlert();
  const { hasPermission } = useAuth();
  const navigate = useNavigate();
  const { data: usersData, isPending: loading } = useQuery(userListOptions());
  const users = (usersData ?? []).map((u) => ({
    id: u.id,
    name: `${u.first_name} ${u.last_name}`.trim() || u.username,
  }));
  const [submitting, setSubmitting] = useState(false);

  const createMutation = useApiMutation<CreateForm, { message?: string }>({
    url: () => `${API_BASE}/attendance`,
    method: () => "POST",
    invalidate: ["attendance", "users"],
  });

  const [form, setForm] = useState<CreateForm>(() => {
    const today = todayLocalStr();
    return {
      user_id: "",
      shift_date: today,
      leave_type: "work",
      leave_hours: 8,
      arrival_date: today,
      arrival_time: "",
      break_start_date: today,
      break_start_time: "",
      break_end_date: today,
      break_end_time: "",
      departure_date: today,
      departure_time: "",
      notes: "",
    };
  });

  const handleSubmit = async (e: React.FormEvent) => {
    e.preventDefault();

    if (!form.user_id || !form.shift_date) {
      alert.error("Vyplňte zaměstnance a datum směny");
      return;
    }

    setSubmitting(true);

    try {
      const result = await createMutation.mutateAsync(form);
      alert.success(result?.message || "Uloženo");
      navigate(`/attendance/admin?month=${form.shift_date.substring(0, 7)}`);
    } catch (e) {
      alert.error(e instanceof Error ? e.message : "Chyba připojení");
    } finally {
      setSubmitting(false);
    }
  };

  const handleShiftDateChange = (newDate: string) => {
    setForm({
      ...form,
      shift_date: newDate,
      arrival_date: newDate,
      break_start_date: newDate,
      break_end_date: newDate,
      departure_date: newDate,
    });
  };

  const isWorkType = form.leave_type === "work";

  if (!hasPermission("attendance.manage")) return <Forbidden />;

  if (loading) {
    return <LoadingState />;
  }

  return (
    <PageEnter>
      <PageHeader
        title="Přidat záznam docházky"
        actions={
          <Button
            variant="outlined"
            color="inherit"
            component={RouterLink}
            to="/attendance/admin"
          >
            ← Zpět na správu
          </Button>
        }
      />

      <Card sx={{ maxWidth: 640 }}>
        <Box component="form" onSubmit={handleSubmit}>
          {/* Employee + Shift date */}
          <Box
            sx={{
              display: "grid",
              gridTemplateColumns: { xs: "1fr", sm: "1fr 1fr" },
              gap: 2,
            }}
          >
            <Field label="Zaměstnanec" required>
              <Select
                value={form.user_id}
                onChange={(val) => setForm({ ...form, user_id: val })}
                options={[
                  { value: "", label: "Vyberte zaměstnance" },
                  ...users.map((u) => ({
                    value: String(u.id),
                    label: u.name,
                  })),
                ]}
              />
            </Field>
            <Field label="Datum směny" required>
              <DateField
                value={form.shift_date}
                onChange={(val) => handleShiftDateChange(val)}
              />
            </Field>
          </Box>

          {/* Record type */}
          <Field label="Typ záznamu" required>
            <Select
              value={form.leave_type}
              onChange={(val) => setForm({ ...form, leave_type: val })}
              options={[
                { value: "work", label: "Práce" },
                { value: "vacation", label: "Dovolená" },
                { value: "sick", label: "Nemoc" },
                { value: "unpaid", label: "Neplacené volno" },
              ]}
            />
          </Field>

          {/* Leave hours — shown only for non-work types */}
          {!isWorkType && (
            <Field label="Počet hodin" hint="Výchozí 8 hodin pro celý den">
              <TextField
                type="number"
                value={form.leave_hours}
                onChange={(e) =>
                  setForm({
                    ...form,
                    leave_hours: parseFloat(e.target.value),
                  })
                }
                slotProps={{ htmlInput: { min: 0.5, max: 24, step: 0.5 } }}
              />
            </Field>
          )}

          {/* Work-type time fields */}
          {isWorkType && (
            <>
              {/* Arrival */}
              <Box
                sx={{
                  display: "grid",
                  gridTemplateColumns: { xs: "1fr", sm: "1fr 1fr" },
                  gap: 2,
                }}
              >
                <Field label="Příchod - datum">
                  <DateField
                    value={form.arrival_date}
                    onChange={(val) => setForm({ ...form, arrival_date: val })}
                  />
                </Field>
                <Field label="Příchod - čas">
                  <TimeField
                    value={form.arrival_time}
                    onChange={(val) => setForm({ ...form, arrival_time: val })}
                  />
                </Field>
              </Box>

              {/* Break start */}
              <Box
                sx={{
                  display: "grid",
                  gridTemplateColumns: { xs: "1fr", sm: "1fr 1fr" },
                  gap: 2,
                }}
              >
                <Field label="Začátek pauzy - datum">
                  <DateField
                    value={form.break_start_date}
                    onChange={(val) =>
                      setForm({ ...form, break_start_date: val })
                    }
                  />
                </Field>
                <Field label="Začátek pauzy - čas">
                  <TimeField
                    value={form.break_start_time}
                    onChange={(val) =>
                      setForm({ ...form, break_start_time: val })
                    }
                  />
                </Field>
              </Box>

              {/* Break end */}
              <Box
                sx={{
                  display: "grid",
                  gridTemplateColumns: { xs: "1fr", sm: "1fr 1fr" },
                  gap: 2,
                }}
              >
                <Field label="Konec pauzy - datum">
                  <DateField
                    value={form.break_end_date}
                    onChange={(val) =>
                      setForm({ ...form, break_end_date: val })
                    }
                  />
                </Field>
                <Field label="Konec pauzy - čas">
                  <TimeField
                    value={form.break_end_time}
                    onChange={(val) =>
                      setForm({ ...form, break_end_time: val })
                    }
                  />
                </Field>
              </Box>

              {/* Departure */}
              <Box
                sx={{
                  display: "grid",
                  gridTemplateColumns: { xs: "1fr", sm: "1fr 1fr" },
                  gap: 2,
                }}
              >
                <Field label="Odchod - datum">
                  <DateField
                    value={form.departure_date}
                    onChange={(val) =>
                      setForm({ ...form, departure_date: val })
                    }
                  />
                </Field>
                <Field label="Odchod - čas">
                  <TimeField
                    value={form.departure_time}
                    onChange={(val) =>
                      setForm({ ...form, departure_time: val })
                    }
                  />
                </Field>
              </Box>
            </>
          )}

          {/* Notes */}
          <Field label="Poznámka">
            <TextField
              multiline
              minRows={3}
              value={form.notes}
              onChange={(e) => setForm({ ...form, notes: e.target.value })}
            />
          </Field>

          {/* Actions */}
          <Box sx={{ display: "flex", gap: 1.5, justifyContent: "flex-end" }}>
            <Button
              variant="outlined"
              color="inherit"
              component={RouterLink}
              to="/attendance/admin"
            >
              Zrušit
            </Button>
            <Button type="submit" disabled={submitting}>
              {submitting ? "Ukládám..." : "Uložit"}
            </Button>
          </Box>
        </Box>
      </Card>
    </PageEnter>
  );
}