365 lines
10 KiB
TypeScript
Executable File
365 lines
10 KiB
TypeScript
Executable File
'use client';
|
|
|
|
import * as React from 'react';
|
|
import { useState, useRef } from 'react';
|
|
import { Button } from '@nice/ui/components/button';
|
|
import { Popover, PopoverContent, PopoverTrigger } from '@nice/ui/components/popover';
|
|
import {
|
|
Command,
|
|
CommandEmpty,
|
|
CommandGroup,
|
|
CommandInput,
|
|
CommandItem,
|
|
CommandList,
|
|
} from '@nice/ui/components/command';
|
|
import { IconChevronDown, IconCheck, IconShield, IconX } from '@tabler/icons-react';
|
|
import { cn } from '@nice/ui/lib/utils';
|
|
import { useTRPC } from '@fenghuo/client';
|
|
import { useQuery } from '@tanstack/react-query';
|
|
import { Badge } from '@nice/ui/components/badge';
|
|
|
|
// 角色接口定义
|
|
export interface Role {
|
|
id: string;
|
|
name: string;
|
|
slug: string;
|
|
description?: string | null;
|
|
permissions: string[];
|
|
isSystem: boolean;
|
|
isActive: boolean;
|
|
createdAt: Date;
|
|
updatedAt: Date;
|
|
}
|
|
|
|
// 角色选择器属性
|
|
interface RoleSelectorProps {
|
|
value?: string | string[]; // 支持单选和多选
|
|
onValueChange?: (value: string | string[]) => void;
|
|
placeholder?: string;
|
|
className?: string;
|
|
disabled?: boolean;
|
|
allowClear?: boolean;
|
|
multiple?: boolean; // 是否支持多选
|
|
showDescription?: boolean; // 是否显示角色描述
|
|
showBadge?: boolean; // 是否显示系统角色标识
|
|
includeInactive?: boolean; // 是否包含非活跃角色
|
|
modal?: boolean; // 是否为模态模式，用于在 Dialog 中解决滚轮问题
|
|
}
|
|
|
|
// 角色项组件
|
|
interface RoleItemProps {
|
|
role: Role;
|
|
isSelected: boolean;
|
|
showDescription?: boolean;
|
|
showBadge?: boolean;
|
|
onSelect: () => void;
|
|
}
|
|
|
|
function RoleItem({ role, isSelected, showDescription = true, showBadge = true, onSelect }: RoleItemProps) {
|
|
return (
|
|
<CommandItem
|
|
value={`${role.id}-${role.name}`}
|
|
onSelect={onSelect}
|
|
className={cn(
|
|
'flex items-center justify-between cursor-pointer p-3',
|
|
'hover:bg-primary/10 hover:text-primary hover:font-medium',
|
|
isSelected && 'bg-accent text-primary font-medium',
|
|
)}
|
|
>
|
|
<div className="flex items-center gap-3 flex-1 min-w-0">
|
|
{/* 角色图标 */}
|
|
<IconShield className="h-4 w-4 text-muted-foreground shrink-0" />
|
|
|
|
{/* 角色信息 */}
|
|
<div className="flex-1 min-w-0">
|
|
<div className="flex items-center gap-2">
|
|
<span className="font-medium truncate">{role.name}</span>
|
|
{showBadge && role.isSystem && (
|
|
<Badge variant="secondary" className="text-xs">
|
|
系统
|
|
</Badge>
|
|
)}
|
|
{showBadge && !role.isActive && (
|
|
<Badge variant="outline" className="text-xs">
|
|
已禁用
|
|
</Badge>
|
|
)}
|
|
</div>
|
|
{showDescription && role.description && (
|
|
<p className="text-sm text-muted-foreground truncate mt-0.5">{role.description}</p>
|
|
)}
|
|
</div>
|
|
</div>
|
|
|
|
{/* 选中状态指示 */}
|
|
{isSelected && <IconCheck className="h-4 w-4 text-primary shrink-0" />}
|
|
</CommandItem>
|
|
);
|
|
}
|
|
|
|
// 主组件
|
|
export function RoleSelect({
|
|
value,
|
|
onValueChange,
|
|
placeholder = '选择角色',
|
|
className,
|
|
disabled = false,
|
|
allowClear = true,
|
|
multiple = false,
|
|
showDescription = true,
|
|
showBadge = true,
|
|
includeInactive = false,
|
|
modal = false,
|
|
}: RoleSelectorProps) {
|
|
const [open, setOpen] = useState(false);
|
|
const [searchValue, setSearchValue] = useState('');
|
|
const containerRef = useRef<HTMLDivElement>(null);
|
|
|
|
const trpc = useTRPC();
|
|
|
|
// 使用 tRPC 获取角色列表
|
|
const {
|
|
data: roleData,
|
|
isLoading,
|
|
error,
|
|
} = useQuery({
|
|
...trpc.role.findManyWithPagination.queryOptions({
|
|
page: 1,
|
|
pageSize: 100, // 获取足够多的角色
|
|
where: includeInactive ? undefined : { isActive: true }, // 根据配置过滤活跃角色
|
|
}),
|
|
});
|
|
|
|
const roles = roleData?.items || [];
|
|
|
|
// 过滤角色（根据搜索关键词）
|
|
const filteredRoles = React.useMemo(() => {
|
|
if (!searchValue.trim()) {
|
|
return roles;
|
|
}
|
|
|
|
const searchTerm = searchValue.toLowerCase();
|
|
return roles.filter(
|
|
(role) =>
|
|
role.name.toLowerCase().includes(searchTerm) ||
|
|
role.description?.toLowerCase().includes(searchTerm) ||
|
|
role.slug.toLowerCase().includes(searchTerm),
|
|
);
|
|
}, [roles, searchValue]);
|
|
|
|
// 获取选中的角色
|
|
const selectedRoles = React.useMemo(() => {
|
|
if (!value) return [];
|
|
|
|
const selectedIds = Array.isArray(value) ? value : [value];
|
|
return roles.filter((role) => selectedIds.includes(role.id));
|
|
}, [roles, value]);
|
|
|
|
// 处理选择逻辑
|
|
const handleSelect = (roleId: string) => {
|
|
if (!onValueChange) return;
|
|
|
|
if (multiple) {
|
|
const currentValues = Array.isArray(value) ? value : value ? [value] : [];
|
|
|
|
if (currentValues.includes(roleId)) {
|
|
// 取消选择
|
|
const newValues = currentValues.filter((id) => id !== roleId);
|
|
onValueChange(newValues);
|
|
} else {
|
|
// 添加选择
|
|
onValueChange([...currentValues, roleId]);
|
|
}
|
|
} else {
|
|
// 单选模式
|
|
onValueChange(roleId);
|
|
setOpen(false);
|
|
}
|
|
};
|
|
|
|
// 清除选择
|
|
const handleClear = (e: React.MouseEvent) => {
|
|
e.preventDefault();
|
|
e.stopPropagation();
|
|
onValueChange?.(multiple ? [] : '');
|
|
};
|
|
|
|
// 处理单个角色移除（仅多选模式）
|
|
const handleRemoveRole = (roleId: string, e: React.MouseEvent) => {
|
|
e.preventDefault();
|
|
e.stopPropagation();
|
|
if (multiple) {
|
|
const currentValues = Array.isArray(value) ? value : value ? [value] : [];
|
|
const newValues = currentValues.filter((id) => id !== roleId);
|
|
onValueChange?.(newValues);
|
|
}
|
|
};
|
|
|
|
// 渲染触发器内容
|
|
const renderTriggerContent = () => {
|
|
if (selectedRoles.length === 0) {
|
|
return placeholder;
|
|
}
|
|
|
|
if (multiple) {
|
|
if (selectedRoles.length === 1) {
|
|
return selectedRoles[0]!.name;
|
|
} else {
|
|
return (
|
|
<div className="flex flex-wrap gap-1 justify-start">
|
|
{selectedRoles.slice(0, 2).map((role) => (
|
|
<Badge key={role.id} variant="secondary" className="text-xs px-1.5 py-0.5 h-5 max-w-[120px]">
|
|
<span className="truncate">{role.name}</span>
|
|
<span
|
|
onClick={(e) => handleRemoveRole(role.id, e)}
|
|
className="ml-1 hover:bg-muted-foreground/20 rounded-sm p-0.5 cursor-pointer inline-flex items-center justify-center"
|
|
role="button"
|
|
tabIndex={0}
|
|
aria-label={`移除角色 ${role.name}`}
|
|
onKeyDown={(e) => {
|
|
if (e.key === 'Enter' || e.key === ' ') {
|
|
e.preventDefault();
|
|
handleRemoveRole(role.id, e as any);
|
|
}
|
|
}}
|
|
>
|
|
<IconX className="h-2.5 w-2.5" />
|
|
</span>
|
|
</Badge>
|
|
))}
|
|
{selectedRoles.length > 2 && (
|
|
<Badge variant="secondary" className="text-xs px-1.5 py-0.5 h-5">
|
|
+{selectedRoles.length - 2}
|
|
</Badge>
|
|
)}
|
|
</div>
|
|
);
|
|
}
|
|
}
|
|
|
|
return selectedRoles[0]!.name;
|
|
};
|
|
|
|
// 检查是否选中
|
|
const isSelected = (roleId: string) => {
|
|
if (!value) return false;
|
|
return Array.isArray(value) ? value.includes(roleId) : value === roleId;
|
|
};
|
|
|
|
if (error) {
|
|
console.error('加载角色列表失败:', error);
|
|
}
|
|
|
|
return (
|
|
<div ref={containerRef}>
|
|
<Popover open={open} onOpenChange={setOpen} modal={modal}>
|
|
<PopoverTrigger asChild>
|
|
<Button
|
|
variant="outline"
|
|
role="combobox"
|
|
aria-expanded={open}
|
|
className={cn(
|
|
'w-full justify-start text-left font-normal',
|
|
!selectedRoles.length && 'text-muted-foreground',
|
|
className,
|
|
)}
|
|
disabled={disabled}
|
|
>
|
|
<div className="flex items-center gap-2 flex-1 min-w-0">
|
|
<IconShield className="h-4 w-4 text-muted-foreground shrink-0" />
|
|
<div className="flex-1 min-w-0 text-left">{renderTriggerContent()}</div>
|
|
</div>
|
|
|
|
<div className="flex items-center gap-1 shrink-0 ml-auto">
|
|
{allowClear && selectedRoles.length > 0 && (
|
|
<span
|
|
onClick={handleClear}
|
|
className="flex items-center justify-center w-4 h-4 rounded-sm hover:bg-muted/50 text-muted-foreground hover:text-foreground transition-colors cursor-pointer"
|
|
role="button"
|
|
tabIndex={0}
|
|
aria-label="清除选择"
|
|
onKeyDown={(e) => {
|
|
if (e.key === 'Enter' || e.key === ' ') {
|
|
e.preventDefault();
|
|
handleClear(e as any);
|
|
}
|
|
}}
|
|
>
|
|
<IconX className="h-3 w-3" />
|
|
</span>
|
|
)}
|
|
<IconChevronDown className="h-4 w-4 shrink-0 opacity-50" />
|
|
</div>
|
|
</Button>
|
|
</PopoverTrigger>
|
|
|
|
<PopoverContent className="w-full p-0" style={{ width: 'var(--radix-popover-trigger-width)' }} align="start">
|
|
<Command shouldFilter={false}>
|
|
<CommandInput
|
|
placeholder="搜索角色..."
|
|
value={searchValue}
|
|
onValueChange={setSearchValue}
|
|
className="h-9"
|
|
/>
|
|
<CommandList className="max-h-[250px] overflow-y-auto">
|
|
{isLoading ? (
|
|
<CommandEmpty>加载中...</CommandEmpty>
|
|
) : filteredRoles.length === 0 ? (
|
|
<CommandEmpty>未找到角色</CommandEmpty>
|
|
) : (
|
|
<CommandGroup>
|
|
{/* 多选模式下显示已选择数量 */}
|
|
{multiple && selectedRoles.length > 0 && (
|
|
<div className="px-2 py-1.5 text-xs text-muted-foreground border-b">
|
|
已选择 {selectedRoles.length} 个角色
|
|
</div>
|
|
)}
|
|
{filteredRoles.map((role) => (
|
|
<RoleItem
|
|
key={role.id}
|
|
role={role}
|
|
isSelected={isSelected(role.id)}
|
|
showDescription={showDescription}
|
|
showBadge={showBadge}
|
|
onSelect={() => handleSelect(role.id)}
|
|
/>
|
|
))}
|
|
</CommandGroup>
|
|
)}
|
|
</CommandList>
|
|
</Command>
|
|
</PopoverContent>
|
|
</Popover>
|
|
</div>
|
|
);
|
|
}
|
|
|
|
// 导出便捷的单选和多选组件
|
|
export function SingleRoleSelector(
|
|
props: Omit<RoleSelectorProps, 'multiple' | 'onValueChange'> & {
|
|
onValueChange?: (value: string) => void;
|
|
},
|
|
) {
|
|
const handleValueChange = (value: string | string[]) => {
|
|
if (props.onValueChange && typeof value === 'string') {
|
|
props.onValueChange(value);
|
|
}
|
|
};
|
|
|
|
return <RoleSelect {...props} multiple={false} onValueChange={handleValueChange} />;
|
|
}
|
|
|
|
export function MultipleRoleSelector(
|
|
props: Omit<RoleSelectorProps, 'multiple' | 'onValueChange'> & {
|
|
onValueChange?: (value: string[]) => void;
|
|
},
|
|
) {
|
|
const handleValueChange = (value: string | string[]) => {
|
|
if (props.onValueChange && Array.isArray(value)) {
|
|
props.onValueChange(value);
|
|
}
|
|
};
|
|
|
|
return <RoleSelect {...props} multiple={true} onValueChange={handleValueChange} />;
|
|
}
|