import React from 'react';
import { BrandIcon } from '@theorchard/suite-icons';

interface AccountIconProps {
    brandName: string;
    testId?: string;
}

const AccountIcon: React.FC<AccountIconProps> = ({ brandName, testId }) => {
    const brand = iconFromBrandName(brandName);
    if (!brand) return null;
    return <BrandIcon brand={brand} size="24" testId={testId} />;
};

export default AccountIcon;

// Copied company icon mapping from frontend-insights
function iconFromBrandName(brand?: string) {
    const name = brand?.toLowerCase();

    if (name?.startsWith('theorchard') || name?.startsWith('hrs')) {
        return 'orchard';
    } else if (name?.startsWith('sme') || name?.startsWith('columbia')) {
        return 'sme';
    } else if (name?.startsWith('awal')) {
        return 'awal';
    } else if (name?.startsWith('ab')) {
        return 'ab';
    } else if (name?.startsWith('msk')) {
        return 'msk';
    } else if (name?.startsWith('ma')) {
        return 'ma';
    } else if (name?.startsWith('drm')) {
        return 'drm';
    } else if (name?.startsWith('knr')) {
        return 'knr';
    } else if (name?.startsWith('altafonte')) {
        return 'altafonte';
    }
    console.warn(`Unsupported brand name: ${brand}`);
}