'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} />;
}