a small fix

Author: devkiran

apps/web/ui/partners/add-edit-reward-sheet.tsx

@@ -6,6 +6,7 @@ import { updateRewardAction } from "@/lib/actions/partners/update-reward";
 import { handleMoneyInputChange, handleMoneyKeyDown } from "@/lib/form-utils";
 import { mutatePrefix } from "@/lib/swr/mutate";
 import useProgram from "@/lib/swr/use-program";
+import useRewardPartners from "@/lib/swr/use-reward-partners";
 import useRewards from "@/lib/swr/use-rewards";
 import useWorkspace from "@/lib/swr/use-workspace";
 import { RewardProps } from "@/lib/types";
@@ -105,13 +106,6 @@ function RewardSheetContent({ setIsOpen, event, reward }: RewardSheetProps) {
     (reward) => reward.event === "sale" && reward.partnersCount === 0,
   );
 
-  // const { data: rewardPartners } = useRewardPartners({
-  //   query: {
-  //     rewardId,
-  //   },
-  //   enabled: Boolean(programId && rewardId),
-  // });
-
   useEffect(() => {
     if (reward) {
       setSelectedPartnerType(reward.partnersCount === 0 ? "all" : "specific");
@@ -132,6 +126,20 @@ function RewardSheetContent({ setIsOpen, event, reward }: RewardSheetProps) {
     hasProgramWideSaleReward,
   ]);
 
+  const { data: rewardPartners, loading: isLoadingRewardPartners } =
+    useRewardPartners({
+      query: {
+        rewardId: reward?.id,
+      },
+      enabled: Boolean(reward?.id && program?.id),
+    });
+
+  useEffect(() => {
+    if (rewardPartners && rewardPartners.length > 0) {
+      setValue("partnerIds", rewardPartners);
+    }
+  }, [rewardPartners, setValue]);
+
   const { executeAsync: createReward, isPending: isCreating } = useAction(
     createRewardAction,
     {
@@ -538,12 +546,11 @@ function RewardSheetContent({ setIsOpen, event, reward }: RewardSheetProps) {
 
             {displayPartners && program?.id && (
               <RewardPartnersTable
-                programId={program.id}
-                rewardId={reward?.id}
                 partnerIds={partnerIds || []}
                 setPartners={(value: string[]) => {
                   setValue("partnerIds", value);
                 }}
+                loading={isLoadingRewardPartners}
               />
             )}
           </div>

apps/web/ui/partners/reward-partners-table.tsx

@@ -4,21 +4,19 @@ import usePartners from "@/lib/swr/use-partners";
 import { EnrolledPartnerProps } from "@/lib/types";
 import { Combobox, Table, useTable } from "@dub/ui";
 import { cn, DICEBEAR_AVATAR_URL, pluralize } from "@dub/utils";
-import { useEffect, useMemo, useState } from "react";
+import { useMemo, useState } from "react";
 import { useDebounce } from "use-debounce";
 
 interface RewardPartnersTableProps {
-  programId: string;
-  rewardId?: string;
   partnerIds: string[];
   setPartners: (value: string[]) => void;
+  loading: boolean;
 }
 
 export function RewardPartnersTable({
-  programId,
-  rewardId,
   partnerIds,
   setPartners,
+  loading,
 }: RewardPartnersTableProps) {
   const [search, setSearch] = useState("");
   const [debouncedSearch] = useDebounce(search, 500);
@@ -116,6 +114,7 @@ export function RewardPartnersTable({
     scrollWrapperClassName: "min-h-[40px]",
     resourceName: (p) => `eligible partner${p ? "s" : ""}`,
     getRowId: (row: EnrolledPartnerProps) => row.id,
+    loading,
   });
 
   return (