"use client";

import * as React from "react";

import { zodResolver } from "@hookform/resolvers/zod";
import { Building2, CircleDollarSign, ClipboardList, Layers3, StickyNote } from "lucide-react";
import { FormProvider, useForm } from "react-hook-form";

import { withServerValidationErrors } from "@/app/customer/_lib/apply-yii-validation-errors";
import type { StockChildFieldMeta } from "@/app/customer/[tenant]/stock/_lib/stock-child-types";
import {
  EMPTY_STOCK_OPTION_FORM_OPTIONS,
  mapStockOptionFormOptions,
  type StockOptionFormOptions,
} from "@/app/customer/[tenant]/stock/_lib/stock-child-form-options";
import { transactionFieldLabels } from "@/components/form/common/field-labels";
import { formGridClass, formSpanFull } from "@/components/form/common/form-layout";
import { DateField } from "@/components/form/fields/date-field";
import { KnockAmountField } from "@/components/form/fields/knock-amount-field";
import { NumberField } from "@/components/form/fields/number-field";
import { ReferenceLinkField } from "@/components/form/fields/reference-link-field";
import { SelectField } from "@/components/form/fields/select-field";
import { TextAreaField } from "@/components/form/fields/text-area-field";
import { TextField } from "@/components/form/fields/text-field";
import {
  buildStockOptionFormSchema,
  defaultStockOptionFormValues,
  type StockOptionFormValues,
} from "@/components/form/schemas/stock-option-schema";
import { TransactionAmountSummarySidebar } from "@/components/form/transaction-workspace/transaction-form-summary";
import { cn } from "@/lib/utils";

import { TransactionFormSection } from "./form-section";

const DEFAULT_KNOCK_FT_OPTIONS = [
  { value: "1", label: "Knock In" },
  { value: "2", label: "Knock Out" },
];

type StockOptionTransactionFormProps = {
  formId: string;
  initialValues?: StockOptionFormValues;
  formOptionsRaw?: Record<string, Record<string, string>>;
  fieldMeta?: StockChildFieldMeta;
  onSubmit?: (values: StockOptionFormValues) => void | Promise<void>;
};

function toSelectOptions(options: { value: string; label: string }[]) {
  return options.map((option) => ({ value: option.value, label: option.label }));
}

function LockedSection({
  locked,
  children,
}: {
  locked: boolean;
  children: React.ReactNode;
}) {
  return (
    <div className={cn("relative w-full min-w-0")}>
      {locked ? <div className="pointer-events-none absolute inset-0 z-10 rounded-lg bg-background/40" aria-hidden /> : null}
      {children}
    </div>
  );
}

export function StockOptionTransactionForm({
  formId,
  initialValues,
  formOptionsRaw,
  fieldMeta,
  onSubmit,
}: StockOptionTransactionFormProps) {
  const [formOptions, setFormOptions] = React.useState<StockOptionFormOptions>(
    EMPTY_STOCK_OPTION_FORM_OPTIONS,
  );

  const lockParent = fieldMeta?.lockParentSection ?? false;
  const requireSecurityFields = !lockParent;

  const resolver = React.useMemo(
    () => zodResolver(buildStockOptionFormSchema(requireSecurityFields)),
    [requireSecurityFields],
  );

  const form = useForm<StockOptionFormValues>({
    resolver,
    defaultValues: { ...defaultStockOptionFormValues, ...initialValues },
    mode: "onBlur",
  });

  React.useEffect(() => {
    form.reset({ ...defaultStockOptionFormValues, ...initialValues });
  }, [form, initialValues]);

  const quantity = form.watch("quantity");
  const price = form.watch("price");
  React.useEffect(() => {
    const amount = Number(quantity || 0) * Number(price || 0);
    form.setValue("amount", Number.isFinite(amount) ? amount : 0, { shouldDirty: true });
  }, [form, price, quantity]);

  React.useEffect(() => {
    if (formOptionsRaw) {
      setFormOptions(mapStockOptionFormOptions(formOptionsRaw));
    }
  }, [formOptionsRaw]);

  const knockFtOptions =
    formOptions.knockFtTypes.length > 0 ? formOptions.knockFtTypes : DEFAULT_KNOCK_FT_OPTIONS;

  return (
    <FormProvider {...form}>
      <form
        id={formId}
        className="w-full min-w-0"
        noValidate
        onSubmit={form.handleSubmit(withServerValidationErrors(form.setError, onSubmit))}
      >
        <div className="grid w-full min-w-0 grid-cols-1 gap-6 xl:grid-cols-[minmax(0,1fr)_18rem] xl:gap-8">
          <div className="flex min-w-0 flex-col gap-5 lg:gap-6">
            <LockedSection locked={lockParent}>
              <TransactionFormSection title="Security" icon={<Layers3 className="size-4" />} contentClassName={formGridClass}>
                {lockParent ? <TextField control={form.control} name="uid" label="Ref. ID" readOnly /> : null}
                <TextField control={form.control} name="ticker" label={transactionFieldLabels.ticker} />
                <TextField control={form.control} name="isin" label={transactionFieldLabels.isin} className={formSpanFull} />
              </TransactionFormSection>
            </LockedSection>

            <TransactionFormSection title="Option details" icon={<ClipboardList className="size-4" />} contentClassName={formGridClass}>
              <DateField control={form.control} name="t_date" label="Placement Date" />
              <NumberField control={form.control} name="premium" label="Premium Paid / Received" />
              <DateField control={form.control} name="m_date" label="Expiration Date" />
              <SelectField control={form.control} name="t_o_o" label="Type of Option" options={toSelectOptions(formOptions.optionTypes)} searchable />
              <SelectField control={form.control} name="transaction" label="Type of Transaction" options={toSelectOptions(formOptions.transactionTypes)} searchable />
              <SelectField control={form.control} name="o_s" label="Option Style" options={toSelectOptions(formOptions.optionStyles)} searchable />
              <TextField control={form.control} name="r_no" label={transactionFieldLabels.r_no} />
              <NumberField control={form.control} name="quantity" label="Quantity under Option" step="0.0001" />
              <TextAreaField control={form.control} name="b_notes" label={transactionFieldLabels.b_notes} className={formSpanFull} />
            </TransactionFormSection>

            <TransactionFormSection title="Pricing" icon={<CircleDollarSign className="size-4" />} contentClassName={formGridClass}>
              <NumberField control={form.control} name="price" label="Strike Price" step="0.0001" />
              <NumberField control={form.control} name="t_price" label="Spot Price" step="0.0001" />
              <DateField control={form.control} name="p_date" label="Option Start Date" />
              <NumberField control={form.control} name="amount" label="Max Exposure" readOnly />
              <NumberField control={form.control} name="commission" label={transactionFieldLabels.commission} />
              <TextField control={form.control} name="b_1" label={transactionFieldLabels.b_1} />
              <TextField control={form.control} name="b_2" label={transactionFieldLabels.b_2} />
              <SelectField control={form.control} name="p_currency" label="Currency (If Different)" options={toSelectOptions(formOptions.currencies)} searchable />
            </TransactionFormSection>

            <TransactionFormSection title="Bank & fees" icon={<Building2 className="size-4" />} contentClassName={formGridClass}>
              <SelectField control={form.control} name="bank" label={transactionFieldLabels.bank} options={toSelectOptions(formOptions.banks)} searchable />
              <div className="hidden">
                <TextField control={form.control} name="a_class" label={transactionFieldLabels.a_class} />
                <TextField control={form.control} name="a_type" label={transactionFieldLabels.a_type} />
              </div>
            </TransactionFormSection>

            <TransactionFormSection title="Notes" icon={<StickyNote className="size-4" />} contentClassName={formGridClass}>
              <TextAreaField control={form.control} name="remarks" label={transactionFieldLabels.remarks} className={formSpanFull} />
              <TextAreaField control={form.control} name="purpose" label={transactionFieldLabels.purpose} className={formSpanFull} />
              <TextField control={form.control} name="r_m" label={transactionFieldLabels.r_m} />
              <ReferenceLinkField control={form.control} name="r_link" />
              <KnockAmountField
                control={form.control}
                knockFtName="knock_ft"
                knockName="knock"
                knockTName="knock_t"
                knockFtOptions={knockFtOptions}
              />
            </TransactionFormSection>
          </div>

          <TransactionAmountSummarySidebar
            formOptions={{
              banks: formOptions.banks,
              transactionTypes: formOptions.transactionTypes,
              executionTypes: [],
              currencies: formOptions.currencies,
              assetTypes: [],
              optionTypes: formOptions.optionTypes,
              optionStyles: formOptions.optionStyles,
            }}
            subheadlineField="isin"
            effectiveDateField="m_date"
            effectiveDateLabel="Expiration date"
            showExecutionBadge={false}
            secondaryBadgeField="t_o_o"
            secondaryBadgeOptions={formOptions.optionTypes}
            grossLabel="Max Exposure"
            netLabel="Net Amount"
          />
        </div>
      </form>
    </FormProvider>
  );
}