import { Button } from 'antd';
|
import type { ButtonProps } from 'antd';
|
import { useAuthStore } from '@/store/auth';
|
|
interface AuthButtonProps extends ButtonProps {
|
/** 不传表示不做权限限制 */
|
permission?: string;
|
/** 无权限时的表现：隐藏（默认）或禁用 */
|
fallback?: 'hide' | 'disabled';
|
}
|
|
/**
|
* 带按钮级权限校验的 Button。
|
*
|
* <AuthButton permission="system:user:add">新增用户</AuthButton>
|
*/
|
export default function AuthButton({
|
permission,
|
fallback = 'hide',
|
...buttonProps
|
}: AuthButtonProps) {
|
const hasPermission = useAuthStore((state) => state.hasPermission);
|
|
if (permission && !hasPermission(permission)) {
|
return fallback === 'disabled' ? <Button {...buttonProps} disabled /> : null;
|
}
|
|
return <Button {...buttonProps} />;
|
}
|
