Add fixes to procurement logic and fix rel-time connections

frontend/src/api/hooks/orders.ts

@@ -29,6 +29,9 @@ import {
   GeneratePlanResponse,
   PaginatedProcurementPlans,
   GetProcurementPlansParams,
+  CreatePOsResult,
+  LinkRequirementToPORequest,
+  UpdateDeliveryStatusRequest,
   GetPlanRequirementsParams,
   UpdatePlanStatusParams,
 } from '../types/orders';
@@ -546,3 +549,186 @@ export const useTriggerDailyScheduler = (
   });
 };
 
+// ===== NEW PROCUREMENT FEATURE HOOKS =====
+
+/**
+ * Hook to recalculate a procurement plan
+ */
+export const useRecalculateProcurementPlan = (
+  options?: UseMutationOptions<GeneratePlanResponse, ApiError, { tenantId: string; planId: string }>
+) => {
+  const queryClient = useQueryClient();
+
+  return useMutation<GeneratePlanResponse, ApiError, { tenantId: string; planId: string }>({
+    mutationFn: ({ tenantId, planId }) => OrdersService.recalculateProcurementPlan(tenantId, planId),
+    onSuccess: (data, variables) => {
+      if (data.plan) {
+        // Update the specific plan in cache
+        queryClient.setQueryData(
+          ordersKeys.procurementPlan(variables.tenantId, variables.planId),
+          data.plan
+        );
+      }
+
+      // Invalidate plans list and dashboard
+      queryClient.invalidateQueries({
+        queryKey: ordersKeys.procurement(),
+        predicate: (query) => {
+          return JSON.stringify(query.queryKey).includes(variables.tenantId);
+        },
+      });
+    },
+    ...options,
+  });
+};
+
+/**
+ * Hook to approve a procurement plan
+ */
+export const useApproveProcurementPlan = (
+  options?: UseMutationOptions<ProcurementPlanResponse, ApiError, { tenantId: string; planId: string; approval_notes?: string }>
+) => {
+  const queryClient = useQueryClient();
+
+  return useMutation<ProcurementPlanResponse, ApiError, { tenantId: string; planId: string; approval_notes?: string }>({
+    mutationFn: ({ tenantId, planId, approval_notes }) =>
+      OrdersService.approveProcurementPlan(tenantId, planId, { approval_notes }),
+    onSuccess: (data, variables) => {
+      // Update the specific plan in cache
+      queryClient.setQueryData(
+        ordersKeys.procurementPlan(variables.tenantId, variables.planId),
+        data
+      );
+
+      // Invalidate plans list and dashboard
+      queryClient.invalidateQueries({
+        queryKey: ordersKeys.procurement(),
+        predicate: (query) => {
+          return JSON.stringify(query.queryKey).includes(variables.tenantId);
+        },
+      });
+    },
+    ...options,
+  });
+};
+
+/**
+ * Hook to reject a procurement plan
+ */
+export const useRejectProcurementPlan = (
+  options?: UseMutationOptions<ProcurementPlanResponse, ApiError, { tenantId: string; planId: string; rejection_notes?: string }>
+) => {
+  const queryClient = useQueryClient();
+
+  return useMutation<ProcurementPlanResponse, ApiError, { tenantId: string; planId: string; rejection_notes?: string }>({
+    mutationFn: ({ tenantId, planId, rejection_notes }) =>
+      OrdersService.rejectProcurementPlan(tenantId, planId, { rejection_notes }),
+    onSuccess: (data, variables) => {
+      // Update the specific plan in cache
+      queryClient.setQueryData(
+        ordersKeys.procurementPlan(variables.tenantId, variables.planId),
+        data
+      );
+
+      // Invalidate plans list and dashboard
+      queryClient.invalidateQueries({
+        queryKey: ordersKeys.procurement(),
+        predicate: (query) => {
+          return JSON.stringify(query.queryKey).includes(variables.tenantId);
+        },
+      });
+    },
+    ...options,
+  });
+};
+
+/**
+ * Hook to create purchase orders from procurement plan
+ */
+export const useCreatePurchaseOrdersFromPlan = (
+  options?: UseMutationOptions<CreatePOsResult, ApiError, { tenantId: string; planId: string; autoApprove?: boolean }>
+) => {
+  const queryClient = useQueryClient();
+
+  return useMutation<CreatePOsResult, ApiError, { tenantId: string; planId: string; autoApprove?: boolean }>({
+    mutationFn: ({ tenantId, planId, autoApprove = false }) =>
+      OrdersService.createPurchaseOrdersFromPlan(tenantId, planId, autoApprove),
+    onSuccess: (data, variables) => {
+      // Invalidate procurement plan to refresh requirements status
+      queryClient.invalidateQueries({
+        queryKey: ordersKeys.procurementPlan(variables.tenantId, variables.planId),
+      });
+
+      // Invalidate dashboard
+      queryClient.invalidateQueries({
+        queryKey: ordersKeys.procurementDashboard(variables.tenantId),
+      });
+    },
+    ...options,
+  });
+};
+
+/**
+ * Hook to link a requirement to a purchase order
+ */
+export const useLinkRequirementToPurchaseOrder = (
+  options?: UseMutationOptions<
+    { success: boolean; message: string; requirement_id: string; purchase_order_id: string },
+    ApiError,
+    { tenantId: string; requirementId: string; request: LinkRequirementToPORequest }
+  >
+) => {
+  const queryClient = useQueryClient();
+
+  return useMutation<
+    { success: boolean; message: string; requirement_id: string; purchase_order_id: string },
+    ApiError,
+    { tenantId: string; requirementId: string; request: LinkRequirementToPORequest }
+  >({
+    mutationFn: ({ tenantId, requirementId, request }) =>
+      OrdersService.linkRequirementToPurchaseOrder(tenantId, requirementId, request),
+    onSuccess: (data, variables) => {
+      // Invalidate procurement data to refresh requirements
+      queryClient.invalidateQueries({
+        queryKey: ordersKeys.procurement(),
+        predicate: (query) => {
+          return JSON.stringify(query.queryKey).includes(variables.tenantId);
+        },
+      });
+    },
+    ...options,
+  });
+};
+
+/**
+ * Hook to update delivery status for a requirement
+ */
+export const useUpdateRequirementDeliveryStatus = (
+  options?: UseMutationOptions<
+    { success: boolean; message: string; requirement_id: string; delivery_status: string },
+    ApiError,
+    { tenantId: string; requirementId: string; request: UpdateDeliveryStatusRequest }
+  >
+) => {
+  const queryClient = useQueryClient();
+
+  return useMutation<
+    { success: boolean; message: string; requirement_id: string; delivery_status: string },
+    ApiError,
+    { tenantId: string; requirementId: string; request: UpdateDeliveryStatusRequest }
+  >({
+    mutationFn: ({ tenantId, requirementId, request }) =>
+      OrdersService.updateRequirementDeliveryStatus(tenantId, requirementId, request),
+    onSuccess: (data, variables) => {
+      // Invalidate procurement data to refresh requirements
+      queryClient.invalidateQueries({
+        queryKey: ordersKeys.procurement(),
+        predicate: (query) => {
+          return JSON.stringify(query.queryKey).includes(variables.tenantId);
+        },
+      });
+    },
+    ...options,
+  });
+};
+