Changes

Co-authored-by: renee-png <262607627+renee-png@users.noreply.github.com>

supabase/migrations/20260419012447_24de63a7-2b0f-4524-abd1-4da095af9cb7.sql

@@ -0,0 +1,42 @@
+-- Add case_id to trust_ledger_entries to track per-case trust balances
+ALTER TABLE public.trust_ledger_entries
+  ADD COLUMN IF NOT EXISTS case_id uuid REFERENCES public.cases(id) ON DELETE SET NULL;
+
+CREATE INDEX IF NOT EXISTS trust_ledger_case_idx
+  ON public.trust_ledger_entries (case_id, entry_date DESC);
+
+-- Backfill case_id from linked invoices where possible
+UPDATE public.trust_ledger_entries t
+   SET case_id = i.case_id
+  FROM public.invoices i
+ WHERE t.case_id IS NULL
+   AND i.case_id IS NOT NULL
+   AND (t.source_invoice_id = i.id OR t.applied_invoice_id = i.id);
+
+-- Update retainer-payment trigger to copy case_id from the invoice
+CREATE OR REPLACE FUNCTION public.tg_trust_deposit_from_payment()
+ RETURNS trigger
+ LANGUAGE plpgsql
+ SECURITY DEFINER
+ SET search_path TO 'public'
+AS $function$
+DECLARE
+  v_client_id uuid;
+  v_case_id uuid;
+  v_is_retainer boolean;
+BEGIN
+  SELECT i.client_id, i.case_id, i.is_retainer
+    INTO v_client_id, v_case_id, v_is_retainer
+  FROM public.invoices i WHERE i.id = NEW.invoice_id;
+
+  IF v_is_retainer IS TRUE AND v_client_id IS NOT NULL AND NEW.amount > 0 THEN
+    INSERT INTO public.trust_ledger_entries
+      (client_id, case_id, entry_date, entry_type, amount, note,
+       source_invoice_id, source_payment_id, created_by)
+    VALUES
+      (v_client_id, v_case_id, NEW.paid_on, 'deposit', NEW.amount,
+       'Retainer payment', NEW.invoice_id, NEW.id, NEW.created_by);
+  END IF;
+  RETURN NEW;
+END;
+$function$;