Files
new-api-custom/web/src/components/group-badge.tsx
T

134 lines
3.6 KiB
TypeScript

/*
Copyright (C) 2023-2026 QuantumNous
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
For commercial licensing, please contact support@quantumnous.com
*/
import type { ReactNode } from 'react'
import { useTranslation } from 'react-i18next'
import { cn } from '@/lib/utils'
import { StatusBadge, type StatusBadgeProps } from './status-badge'
import { Badge } from './ui/badge'
export function GroupMultiplierBadge(props: {
children?: ReactNode
className?: string
label?: string
ratio?: number | null
}) {
let colorClassName =
'border-muted-foreground/30 bg-muted text-muted-foreground'
if (props.ratio != null && props.ratio > 1) {
colorClassName = 'border-warning/30 bg-warning/10 text-warning'
} else if (props.ratio != null && props.ratio < 1) {
colorClassName = 'border-info/30 bg-info/10 text-info'
}
return (
<Badge
variant='outline'
className={cn(
'relative h-5 min-w-12 rounded-full px-1.5 py-0 text-sm leading-none font-medium shadow-none',
!props.label && 'tabular-nums',
colorClassName,
props.className
)}
>
{props.children}
<span>{props.label ?? `${props.ratio}x`}</span>
</Badge>
)
}
type GroupBadgeProps = Omit<
StatusBadgeProps,
'autoColor' | 'label' | 'variant'
> & {
group?: string | null
label?: string
ratio?: number | null
ratioLabel?: string
containerClassName?: string
}
function getGroupLabel(params: {
labelOverride?: string
groupName?: string
isAutoGroup: boolean
isEmptyGroup: boolean
t: (key: string) => string
}): string {
if (params.labelOverride) return params.labelOverride
if (params.isEmptyGroup) return params.t('User Group')
if (params.isAutoGroup) return params.t('Auto')
return params.groupName ?? ''
}
export function GroupBadge(props: GroupBadgeProps) {
const { t } = useTranslation()
const {
group,
label: labelOverride,
ratio,
ratioLabel,
containerClassName,
copyable = false,
showDot,
className,
...badgeProps
} = props
const groupName = group?.trim()
const isAutoGroup = groupName === 'auto'
const isEmptyGroup = !groupName
const isSpecialGroup = isAutoGroup || isEmptyGroup
const label = getGroupLabel({
labelOverride,
groupName,
isAutoGroup,
isEmptyGroup,
t,
})
const badge = (
<StatusBadge
{...badgeProps}
copyable={copyable}
label={label}
showDot={showDot ?? (isSpecialGroup ? false : undefined)}
variant={isSpecialGroup ? 'neutral' : undefined}
autoColor={isSpecialGroup ? undefined : groupName}
className={cn('min-w-0 shrink overflow-hidden', className)}
/>
)
if (ratio == null && !ratioLabel) {
return badge
}
return (
<span
className={cn(
'inline-flex max-w-full min-w-0 items-center gap-2 text-sm',
containerClassName
)}
>
<span className='max-w-full min-w-0 overflow-hidden'>{badge}</span>
<GroupMultiplierBadge ratio={ratio} label={ratioLabel} />
</span>
)
}