feat: new finality provider table design

src/ui/components/Multistaking/BsnFinalityProviderField/BsnFinalityProviderField.tsx

@@ -30,8 +30,9 @@ export function BsnFinalityProviderField({
     disabled,
   });
 
-  const handleAdd = () => {
-    // TODO: Implement provider selection logic
+  const handleAdd = (providerPk: string) => {
+    // Add selected provider ID to list
+    onChange([...selectedProviderIds, providerPk]);
     onClose();
   };
 
@@ -62,7 +63,12 @@ export function BsnFinalityProviderField({
           />
         )}
       </div>
-      <BsnModal open={open} onAdd={handleAdd} onClose={onClose} />
+      <BsnModal
+        open={open}
+        onAdd={handleAdd}
+        onClose={onClose}
+        selectedProviderIds={selectedProviderIds}
+      />
     </SubSection>
   );
 }

src/ui/components/Multistaking/BsnFinalityProviderField/BsnModal.tsx

@@ -1,30 +1,87 @@
-import { DialogBody, DialogHeader } from "@babylonlabs-io/core-ui";
+import { useEffect, useMemo, useState } from "react";
 
 import { ResponsiveDialog } from "@/ui/components/Modals/ResponsiveDialog";
+import { ChainSelectionModal } from "@/ui/components/Multistaking/ChainSelectionModal/ChainSelectionModal";
+import { FinalityProviderModal } from "@/ui/components/Multistaking/FinalityProviderField/FinalityProviderModal";
+import {
+  FinalityProviderBsnState,
+  useFinalityProviderBsnState,
+} from "@/ui/state/FinalityProviderBsnState";
 
 interface Props {
   open: boolean;
-  defaultFinalityProvider?: string;
-  onAdd: () => void;
+  onAdd: (selectedProviderPk: string) => void;
   onClose: () => void;
+  selectedProviderIds: string[];
 }
 
-export function BsnModal({
-  open,
-  // defaultFinalityProvider,
-  // onAdd,
-  onClose,
-}: Props) {
+enum BsnModalPage {
+  CHAIN = "CHAIN",
+  FP = "FP",
+}
+
+export function BsnModal({ open, onAdd, onClose, selectedProviderIds }: Props) {
+  const [selectedChainId, setSelectedChainId] = useState<string | null>(null);
+  const [page, setPage] = useState<BsnModalPage>(BsnModalPage.CHAIN);
+
+  const { getRegisteredFinalityProvider } = useFinalityProviderBsnState();
+
+  const hasBabylonProvider = useMemo(
+    () =>
+      selectedProviderIds.some((pk) => {
+        const fp = getRegisteredFinalityProvider(pk);
+        return !fp?.bsnId;
+      }),
+    [selectedProviderIds, getRegisteredFinalityProvider],
+  );
+
+  const disabledChainIds = useMemo(() => {
+    const set = new Set<string>();
+    selectedProviderIds.forEach((pk) => {
+      const fp = getRegisteredFinalityProvider(pk);
+      set.add(fp?.bsnId || "");
+    });
+    return Array.from(set);
+  }, [selectedProviderIds, getRegisteredFinalityProvider]);
+
+  useEffect(() => {
+    if (!open) {
+      return;
+    }
+    setPage(BsnModalPage.CHAIN);
+    setSelectedChainId(null);
+  }, [open]);
+
+  const handleChainNext = (chainId: string) => {
+    setSelectedChainId(chainId);
+    setPage(BsnModalPage.FP);
+  };
+
+  const handleProviderAdd = (providerPk: string) => {
+    onAdd(providerPk);
+  };
+
   return (
     <ResponsiveDialog open={open} onClose={onClose} className="w-[52rem]">
-      <DialogHeader
-        title="Select BSN and Finality Provider"
-        onClose={onClose}
-        className="text-accent-primary"
-      />
-      <DialogBody className="flex flex-col mb-4 mt-4 text-accent-primary">
-        {/* TODO: Implement BSN selection functionality */}
-      </DialogBody>
+      {page === BsnModalPage.CHAIN && (
+        <ChainSelectionModal
+          onNext={handleChainNext}
+          onClose={onClose}
+          disableNonBabylon={!hasBabylonProvider}
+          disabledChainIds={disabledChainIds}
+        />
+      )}
+      {page === BsnModalPage.FP && selectedChainId !== null && (
+        <FinalityProviderBsnState bsnId={selectedChainId}>
+          <FinalityProviderModal
+            open={true}
+            defaultFinalityProvider=""
+            onClose={onClose}
+            onAdd={handleProviderAdd}
+            onBack={() => setPage(BsnModalPage.CHAIN)}
+          />
+        </FinalityProviderBsnState>
+      )}
     </ResponsiveDialog>
   );
 }

src/ui/components/Multistaking/ChainSelectionModal/ChainSelectionModal.tsx

@@ -5,10 +5,12 @@ import {
   DialogHeader,
   Text,
 } from "@babylonlabs-io/core-ui";
+import { useQuery } from "@tanstack/react-query";
 import { PropsWithChildren, useState } from "react";
 import { MdOutlineInfo } from "react-icons/md";
 import { twMerge } from "tailwind-merge";
 
+import { getBSNs } from "@/ui/api/getBsn";
 import { chainLogos } from "@/ui/constants";
 
 const SubSection = ({
@@ -80,12 +82,21 @@ const ChainButton = ({
 export const ChainSelectionModal = ({
   onNext,
   onClose,
+  disableNonBabylon = false,
+  disabledChainIds = [],
 }: {
   onNext: (selectedChain: string) => void;
   onClose: () => void;
+  disableNonBabylon?: boolean;
+  disabledChainIds?: string[];
 }) => {
   const [selected, setSelected] = useState<string | null>(null);
 
+  const { data: bsns, isLoading } = useQuery({
+    queryKey: ["API_BSN_LIST"],
+    queryFn: getBSNs,
+  });
+
   return (
     <>
       <DialogHeader
@@ -101,47 +112,48 @@ export const ChainSelectionModal = ({
           rewards.
         </div>
         <div className="overflow-x-auto flex flex-col gap-2 mt-10">
-          <ChainButton
-            logo={chainLogos.babylon}
-            title="Babylon"
-            selected={selected === "babylon"}
-            onClick={() => setSelected("babylon")}
-          />
-          <ChainButton
-            logo={chainLogos.cosmos}
-            title="Cosmos"
-            disabled
-            onClick={() => setSelected("cosmos")}
-          />
-          <ChainButton
-            logo={chainLogos.ethereum}
-            title="Ethereum"
-            disabled
-            onClick={() => setSelected("ethereum")}
-          />
-          <ChainButton
-            logo={chainLogos.sui}
-            title="Sui"
-            disabled
-            onClick={() => setSelected("sui")}
-          />
+          {isLoading && <div>Loading...</div>}
+          {bsns?.map((bsn) => {
+            const logo =
+              bsn.id === ""
+                ? chainLogos.babylon
+                : (chainLogos as Record<string, string>)[bsn.id] ||
+                  chainLogos.placeholder;
+
+            const isDisabled =
+              (disableNonBabylon && bsn.id !== "") ||
+              disabledChainIds.includes(bsn.id);
+
+            return (
+              <ChainButton
+                key={bsn.id}
+                logo={logo}
+                title={bsn.name}
+                selected={selected === bsn.id}
+                disabled={isDisabled}
+                onClick={() => !isDisabled && setSelected(bsn.id)}
+              />
+            );
+          })}
         </div>
-        <SubSection className="text-base text-[#387085] gap-3 flex-row mt-4">
-          <div>
-            <MdOutlineInfo size={22} />
-          </div>
-          <div>
-            Babylon must be the first BSN you add before selecting others. Once
-            added, you can choose additional BSNs to multistake.
-          </div>
-        </SubSection>
+        {disableNonBabylon && (
+          <SubSection className="text-base text-[#387085] gap-3 flex-row mt-4">
+            <div>
+              <MdOutlineInfo size={22} />
+            </div>
+            <div>
+              Babylon must be the first BSN you add before selecting others.
+              Once added, you can choose additional BSNs to multistake.
+            </div>
+          </SubSection>
+        )}
       </DialogBody>
 
       <DialogFooter className="flex justify-end">
         <Button
           variant="contained"
-          onClick={() => selected && onNext(selected)}
-          disabled={!selected}
+          onClick={() => selected !== null && onNext(selected)}
+          disabled={selected === null}
         >
           Next
         </Button>

src/ui/components/Multistaking/FinalityProviderField/FinalityProviders.tsx

@@ -1,6 +1,6 @@
-import { FinalityProviderFilter } from "@/ui/components/Staking/FinalityProviders/FinalityProviderFilter";
-import { FinalityProviderSearch } from "@/ui/components/Staking/FinalityProviders/FinalityProviderSearch";
-import { FinalityProviderTable } from "@/ui/components/Staking/FinalityProviders/FinalityProviderTable";
+import { FinalityProviderFilter } from "@/ui/components/Multistaking/FinalityProviders/FinalityProviderFilter";
+import { FinalityProviderSearch } from "@/ui/components/Multistaking/FinalityProviders/FinalityProviderSearch";
+import { FinalityProviderTable } from "@/ui/components/Multistaking/FinalityProviders/FinalityProviderTable";
 
 interface Props {
   selectedFP: string;

src/ui/components/Multistaking/FinalityProviders/FinalityProviderFilter.tsx

@@ -0,0 +1,23 @@
+import { Select } from "@babylonlabs-io/core-ui";
+
+import { useFinalityProviderBsnState } from "@/ui/state/FinalityProviderBsnState";
+
+const options = [
+  { value: "active", label: "Active" },
+  { value: "inactive", label: "Inactive" },
+];
+
+export const FinalityProviderFilter = () => {
+  const { filter, handleFilter } = useFinalityProviderBsnState();
+
+  return (
+    <Select
+      options={options}
+      onSelect={(value) => handleFilter("status", value.toString())}
+      placeholder="Select Status"
+      value={filter.search ? "" : filter.status}
+      disabled={Boolean(filter.search)}
+      renderSelectedOption={(option) => `Showing ${option.label}`}
+    />
+  );
+};

src/ui/components/Multistaking/FinalityProviders/FinalityProviderSearch.tsx

@@ -0,0 +1,43 @@
+import { Input } from "@babylonlabs-io/core-ui";
+import { useCallback } from "react";
+import { MdCancel } from "react-icons/md";
+import { RiSearchLine } from "react-icons/ri";
+
+import { useFinalityProviderBsnState } from "@/ui/state/FinalityProviderBsnState";
+
+export const FinalityProviderSearch = () => {
+  const { filter, handleFilter } = useFinalityProviderBsnState();
+
+  const onSearchChange = useCallback(
+    (e: React.ChangeEvent<HTMLInputElement>) => {
+      handleFilter("search", e.target.value);
+    },
+    [handleFilter],
+  );
+
+  const onClearSearch = useCallback(() => {
+    handleFilter("search", "");
+  }, [handleFilter]);
+
+  const searchSuffix = filter.search ? (
+    <button
+      onClick={onClearSearch}
+      className="opacity-60 hover:opacity-100 transition-opacity"
+    >
+      <MdCancel size={18} className="text-secondary-strokeDark" />
+    </button>
+  ) : (
+    <span className="text-secondary-strokeDark">
+      <RiSearchLine size={20} />
+    </span>
+  );
+
+  return (
+    <Input
+      placeholder="Search by Name or Public Key"
+      suffix={searchSuffix}
+      value={filter.search}
+      onChange={onSearchChange}
+    />
+  );
+};