import React, { useState, useEffect } from "react";
import {
  View,
  ScrollView,
  Pressable,
  TextInput,
  StyleSheet,
  ActivityIndicator,
} from "react-native";
import { useColorScheme } from "nativewind";
import { Text } from "@/components/ui/text";
import { Button } from "@/components/ui/button";
import {
  Trash2,
  Plus,
  Calendar,
  ChevronDown,
  CalendarSearch,
} from "@/lib/icons";
import { ScreenWrapper } from "@/components/ScreenWrapper";
import { FormFlow } from "@/components/FormFlow";
import { CustomerPicker } from "@/components/CustomerPicker";
import { useSirouRouter } from "@sirou/react-native";
import { AppRoutes } from "@/lib/routes";
import { Stack, useLocalSearchParams } from "expo-router";
import { useRouter } from "expo-router";
import { api } from "@/lib/api";
import { toast } from "@/lib/toast-store";
import { PickerModal, SelectOption } from "@/components/PickerModal";
import { CalendarGrid } from "@/components/CalendarGrid";
import { StandardHeader } from "@/components/StandardHeader";

type Item = { id: number; description: string; qty: string; price: string };

const S = StyleSheet.create({
  input: {
    paddingHorizontal: 12,
    paddingVertical: 12,
    fontSize: 12,
    fontWeight: "500",
    borderRadius: 6,
    borderWidth: 1,
    textAlignVertical: "center",
  },
  inputCenter: {
    paddingHorizontal: 12,
    paddingVertical: 10,
    fontSize: 14,
    fontWeight: "500",
    borderRadius: 6,
    borderWidth: 1,
    textAlign: "center",
    textAlignVertical: "center",
  },
});

function useInputColors() {
  const { colorScheme } = useColorScheme();
  const isDark = colorScheme === "dark";
  return {
    bg: isDark ? "rgba(30,30,30,0.8)" : "rgba(241,245,249,0.2)",
    border: isDark ? "rgba(255,255,255,0.08)" : "rgba(203,213,225,0.6)",
    text: isDark ? "#f1f5f9" : "#0f172a",
    placeholder: "rgba(100,116,139,0.45)",
  };
}

function Field({
  label,
  value,
  onChangeText,
  placeholder,
  numeric = false,
  center = false,
  flex,
  multiline = false,
}: {
  label: string;
  value: string;
  onChangeText: (v: string) => void;
  placeholder: string;
  numeric?: boolean;
  center?: boolean;
  flex?: number;
  multiline?: boolean;
}) {
  const c = useInputColors();
  return (
    <View style={flex != null ? { flex } : undefined}>
      <Text className="text-[14px] font-sans-bold mb-1.5 ml-1 text-foreground">
        {label}
      </Text>
      <TextInput
        style={[
          center ? S.inputCenter : S.input,
          { backgroundColor: c.bg, borderColor: c.border, color: c.text },
          multiline
            ? { height: 80, paddingTop: 10, textAlignVertical: "top" }
            : {},
        ]}
        placeholder={placeholder}
        placeholderTextColor={c.placeholder}
        value={value}
        onChangeText={onChangeText}
        keyboardType={numeric ? "numeric" : "default"}
        multiline={multiline}
        autoCorrect={false}
        autoCapitalize="none"
      />
    </View>
  );
}

function PickerField({
  label,
  value,
  onPress,
}: {
  label: string;
  value: string;
  onPress: () => void;
}) {
  const c = useInputColors();
  return (
    <View className="flex-1">
      <Text className="text-[14px] font-sans-bold mb-1.5 ml-1 text-foreground">
        {label}
      </Text>
      <Pressable
        onPress={onPress}
        className="h-11 px-3 border border-border rounded-[6px] flex-row items-center justify-between"
        style={{ backgroundColor: c.bg, borderColor: c.border }}
      >
        <Text className="text-xs font-sans-bold" style={{ color: c.text }}>
          {value}
        </Text>
        <ChevronDown size={14} color={c.text} strokeWidth={3} />
      </Pressable>
    </View>
  );
}

const STEPS = [
  { key: "customer", label: "Customer" },
  { key: "details", label: "Details" },
  { key: "dates", label: "Dates" },
  { key: "items", label: "Items" },
  { key: "totals", label: "Totals" },
  { key: "config", label: "Config" },
];

export default function EditProformaScreen() {
  const nav = useSirouRouter<AppRoutes>();
  const router = useRouter();
  const { id } = useLocalSearchParams();
  const isEdit = !!id;

  const [loading, setLoading] = useState(isEdit);
  const [submitting, setSubmitting] = useState(false);
  const [step, setStep] = useState(0);

  const [proformaNumber, setProformaNumber] = useState("");
  const [customerId, setCustomerId] = useState("");
  const [customerName, setCustomerName] = useState("");
  const [customerEmail, setCustomerEmail] = useState("");
  const [customerPhone, setCustomerPhone] = useState("");
  const [selectedCustomers, setSelectedCustomers] = useState<any[]>([]);
  const [currency, setCurrency] = useState("USD");
  const [description, setDescription] = useState("");
  const [notes, setNotes] = useState("");
  const [taxAmount, setTaxAmount] = useState("");
  const [discountAmount, setDiscountAmount] = useState("");

  const [issueDate, setIssueDate] = useState(new Date());
  const [dueDate, setDueDate] = useState(new Date());

  const [items, setItems] = useState<Item[]>([
    { id: 1, description: "", qty: "", price: "" },
  ]);

  const [currencyModal, setCurrencyModal] = useState(false);
  const [issueModal, setIssueModal] = useState(false);
  const [dueModal, setDueModal] = useState(false);

  useEffect(() => {
    if (isEdit) {
      fetchProforma();
    }
  }, [id]);

  const fetchProforma = async () => {
    try {
      setLoading(true);
      const data = await api.proforma.getById({ params: { id: id as string } });
      setProformaNumber(data.proformaNumber || "");
      setCustomerName(data.customerName || "");
      setCustomerEmail(data.customerEmail || "");
      setCustomerPhone(data.customerPhone || "");
      setCurrency(data.currency || "USD");
      setDescription(data.description || "");
      setNotes(data.notes || "");
      setTaxAmount(
        String(
          typeof data.taxAmount === "object"
            ? data.taxAmount?.value
            : data.taxAmount || "",
        ),
      );
      setDiscountAmount(
        String(
          typeof data.discountAmount === "object"
            ? data.discountAmount?.value
            : data.discountAmount || "",
        ),
      );
      setIssueDate(new Date(data.issueDate));
      setDueDate(new Date(data.dueDate));
      setItems(
        data.items?.map((item: any, idx: number) => ({
          id: idx + 1,
          description: item.description || "",
          qty: String(item.quantity || ""),
          price: String(item.unitPrice?.value || item.unitPrice || ""),
        })) || [{ id: 1, description: "", qty: "", price: "" }],
      );
    } catch (error) {
      toast.error("Error", "Failed to load proforma");
    } finally {
      setLoading(false);
    }
  };

  const addItem = () => {
    const newId = Math.max(...items.map((i) => i.id)) + 1;
    setItems([...items, { id: newId, description: "", qty: "", price: "" }]);
  };

  const removeItem = (id: number) => {
    if (items.length > 1) {
      setItems(items.filter((i) => i.id !== id));
    }
  };

  const updateItem = (id: number, field: keyof Item, value: string) => {
    setItems(items.map((i) => (i.id === id ? { ...i, [field]: value } : i)));
  };

  const calculateSubtotal = () => {
    return items.reduce((acc, item) => {
      const qty = parseFloat(item.qty) || 0;
      const price = parseFloat(item.price) || 0;
      return acc + qty * price;
    }, 0);
  };

  const calculateTotal = () => {
    const subtotal = calculateSubtotal();
    const tax = parseFloat(taxAmount) || 0;
    const discount = parseFloat(discountAmount) || 0;
    return subtotal + tax - discount;
  };

  const handleNext = () => {
    if (step === 0 && !customerName.trim()) {
      toast.error("Validation", "Customer name is required");
      return;
    }
    if (step === 1 && !proformaNumber.trim()) {
      toast.error("Validation", "Proforma number is required");
      return;
    }
    setStep(step + 1);
  };

  const handleSubmit = async () => {
    if (!proformaNumber || !customerName) {
      toast.error("Error", "Please fill required fields");
      return;
    }

    setSubmitting(true);
    try {
      const payload = {
        proformaNumber,
        customerName,
        customerEmail,
        customerPhone,
        amount: calculateTotal(),
        currency,
        issueDate: issueDate.toISOString(),
        dueDate: dueDate.toISOString(),
        description,
        notes,
        taxAmount: parseFloat(taxAmount) || 0,
        discountAmount: parseFloat(discountAmount) || 0,
        items: items.map((item) => ({
          description: item.description,
          quantity: parseFloat(item.qty) || 0,
          unitPrice: parseFloat(item.price) || 0,
          total: (parseFloat(item.qty) || 0) * (parseFloat(item.price) || 0),
        })),
      };

      if (isEdit) {
        await api.proforma.update({
          params: { id: id as string },
          body: payload,
        });
        toast.success("Success", "Proforma updated successfully");
      } else {
        await api.proforma.create({ body: payload });
        toast.success("Success", "Proforma created successfully");
      }

      nav.back();
    } catch (error: any) {
      toast.error("Error", error.message || "Failed to save proforma");
    } finally {
      setSubmitting(false);
    }
  };

  if (loading) {
    return (
      <ScreenWrapper className="bg-background">
        <Stack.Screen options={{ headerShown: false }} />
        <StandardHeader
          title={isEdit ? "Edit Proforma" : "Create Proforma"}
          showBack
        />
        <View className="flex-1 justify-center items-center">
          <ActivityIndicator color="#ea580c" size="large" />
        </View>
      </ScreenWrapper>
    );
  }

  const currencies = ["USD", "EUR", "GBP", "CAD"];

  return (
    <ScreenWrapper className="bg-background">
      <Stack.Screen options={{ headerShown: false }} />
      <FormFlow
        steps={STEPS}
        currentStep={step}
        onNext={handleNext}
        onBack={() => setStep(step - 1)}
        onComplete={handleSubmit}
        loading={submitting}
        completeLabel={isEdit ? "Update Proforma" : "Create Proforma"}
      >
        {step === 0 && (
          <View className="gap-5">
            <Text className="text-[18px] font-sans-bold text-foreground tracking-tight">
              Customer Details
            </Text>
            <View className="gap-4">
              <View>
                <Text className="text-[14px] font-sans-bold mb-1.5 ml-1 text-foreground">
                  Customer Name
                </Text>
                <CustomerPicker
                  selectedIds={customerId ? [customerId] : []}
                  selectedCustomers={selectedCustomers}
                  onSelect={(ids, customers) => {
                    setCustomerId(ids[0] || "");
                    setSelectedCustomers(customers);
                    if (customers[0]) {
                      setCustomerName(customers[0].name);
                      setCustomerEmail(customers[0].email);
                      setCustomerPhone(customers[0].phone?.replace("+251", "") || "");
                    } else {
                      setCustomerName("");
                      setCustomerEmail("");
                      setCustomerPhone("");
                    }
                  }}
                  placeholder="Select or search for a customer"
                />
              </View>
              <Field
                label="Customer Email"
                value={customerEmail}
                onChangeText={setCustomerEmail}
                placeholder="billing@acme.com"
              />
              <Field
                label="Customer Phone"
                value={customerPhone}
                onChangeText={setCustomerPhone}
                placeholder="+1234567890"
              />
            </View>
          </View>
        )}

        {step === 1 && (
          <View className="gap-5">
            <Text className="text-[18px] font-sans-bold text-foreground tracking-tight">
              Proforma Details
            </Text>
            <View className="bg-card rounded-[6px] gap-4">
              <Field
                label="Proforma Number"
                value={proformaNumber}
                onChangeText={setProformaNumber}
                placeholder="e.g. PF-2024-001"
              />
              <Field
                label="Description"
                value={description}
                onChangeText={setDescription}
                placeholder="Brief description"
              />
              <Field
                label="Notes"
                value={notes}
                onChangeText={setNotes}
                placeholder="Additional notes"
                multiline
              />
            </View>
          </View>
        )}

        {step === 2 && (
          <View className="gap-5">
            <Text className="text-[18px] font-sans-bold text-foreground tracking-tight">
              Dates
            </Text>
            <View className="bg-card rounded-[6px] gap-4">
              <PickerField
                label="Issue Date"
                value={issueDate.toLocaleDateString()}
                onPress={() => setIssueModal(true)}
              />
              <PickerField
                label="Due Date"
                value={dueDate.toLocaleDateString()}
                onPress={() => setDueModal(true)}
              />
            </View>
          </View>
        )}

        {step === 3 && (
          <View className="gap-5">
            <View className="flex-row items-center justify-between">
              <Text className="text-[18px] font-sans-bold text-foreground tracking-tight">
                Items
              </Text>
              <Pressable
                onPress={addItem}
                className="flex-row items-center gap-1 px-3 py-1.5 rounded-[6px] bg-primary/10 border border-primary/20"
              >
                <Plus color="#ea580c" size={10} strokeWidth={2.5} />
                <Text className="text-primary text-[8px] font-sans-bold">
                  Add
                </Text>
              </Pressable>
            </View>
            <View className="gap-3">
              {items.map((item) => (
                <View
                  key={item.id}
                  className="bg-card rounded-[6px] p-4 border border-border"
                >
                  <View className="flex-row justify-between items-center mb-3">
                    <Text className="text-[14px] font-sans-bold text-foreground">
                      Item {item.id}
                    </Text>
                    <Pressable onPress={() => removeItem(item.id)} hitSlop={8}>
                      <Trash2 color="#ef4444" size={16} />
                    </Pressable>
                  </View>
                  <Field
                    label="Description"
                    placeholder="e.g. UI Design"
                    value={item.description}
                    onChangeText={(v) => updateItem(item.id, "description", v)}
                  />
                  <View className="flex-row gap-3 mt-4">
                    <Field
                      label="Qty"
                      placeholder="1"
                      numeric
                      center
                      value={item.qty}
                      onChangeText={(v) => updateItem(item.id, "qty", v)}
                      flex={1}
                    />
                    <Field
                      label="Price"
                      placeholder="0.00"
                      numeric
                      value={item.price}
                      onChangeText={(v) => updateItem(item.id, "price", v)}
                      flex={3}
                    />
                  </View>
                </View>
              ))}
            </View>
          </View>
        )}

        {step === 4 && (
          <View className="gap-5">
            <Text className="text-[18px] font-sans-bold text-foreground tracking-tight">
              Totals
            </Text>
            <View className="bg-card rounded-[6px] p-4 border border-border gap-4">
              <View className="flex-row justify-between">
                <Text className="text-foreground font-sans-medium">Subtotal</Text>
                <Text className="text-foreground font-sans-bold">
                  {currency} {calculateSubtotal().toFixed(2)}
                </Text>
              </View>
              <Field
                label="Tax Amount"
                value={taxAmount}
                onChangeText={setTaxAmount}
                placeholder="0.00"
                numeric
              />
              <Field
                label="Discount Amount"
                value={discountAmount}
                onChangeText={setDiscountAmount}
                placeholder="0.00"
                numeric
              />
              <View className="border-t border-border pt-4 flex-row justify-between">
                <Text className="text-foreground font-sans-bold text-lg">
                  Total
                </Text>
                <Text className="text-foreground font-sans-bold text-lg">
                  {currency} {calculateTotal().toFixed(2)}
                </Text>
              </View>
            </View>
          </View>
        )}

        {step === 5 && (
          <View className="gap-5">
            <Text className="text-[18px] font-sans-bold text-foreground tracking-tight">
              Configuration
            </Text>
            <View className="bg-card rounded-[6px] gap-4">
              <PickerField
                label="Currency"
                value={currency}
                onPress={() => setCurrencyModal(true)}
              />
            </View>
          </View>
        )}
      </FormFlow>

      <PickerModal
        visible={currencyModal}
        title="Select Currency"
        onClose={() => setCurrencyModal(false)}
      >
        {currencies.map((curr) => (
          <SelectOption
            key={curr}
            label={curr}
            value={curr}
            selected={curr === currency}
            onSelect={(v) => {
              setCurrency(v);
              setCurrencyModal(false);
            }}
          />
        ))}
      </PickerModal>

      <PickerModal
        visible={issueModal}
        title="Select Issue Date"
        onClose={() => setIssueModal(false)}
      >
        <CalendarGrid
          onSelect={(dateStr: string) => {
            setIssueDate(new Date(dateStr));
            setIssueModal(false);
          }}
          selectedDate={issueDate.toISOString().substring(0, 10)}
        />
      </PickerModal>

      <PickerModal
        visible={dueModal}
        title="Select Due Date"
        onClose={() => setDueModal(false)}
      >
        <CalendarGrid
          onSelect={(dateStr: string) => {
            setDueDate(new Date(dateStr));
            setDueModal(false);
          }}
          selectedDate={dueDate.toISOString().substring(0, 10)}
        />
      </PickerModal>
    </ScreenWrapper>
  );
}
