Arkie-Library-Frontend/src/wallet/WalletBrandIcon.tsx

import type { WalletKind } from "./injected";

const logos: Partial<Record<WalletKind, string>> = {
  tokenPocket: "/assets/ark-library/wallets/tokenpocket.svg",
  imToken: "/assets/ark-library/wallets/imtoken.svg",
};

export function WalletBrandIcon({
  kind,
  size = 28,
}: {
  kind: WalletKind;
  size?: number;
}) {
  const logo = logos[kind];
  if (!logo) return null;

  return (
    <img
      src={logo}
      alt=""
      aria-hidden="true"
      width={size}
      height={size}
      className="inline-flex shrink-0 rounded-lg"
    />
  );
}