You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.0 KiB
82 lines
2.0 KiB
import { UserType } from '@server/constants/user';
|
|
import type { PermissionCheckOptions } from '@server/lib/permissions';
|
|
import { hasPermission, Permission } from '@server/lib/permissions';
|
|
import type { NotificationAgentKey } from '@server/lib/settings';
|
|
import type { MutatorCallback } from 'swr';
|
|
import useSWR from 'swr';
|
|
|
|
export { Permission, UserType };
|
|
export type { PermissionCheckOptions };
|
|
|
|
export interface User {
|
|
id: number;
|
|
plexUsername?: string;
|
|
username?: string;
|
|
displayName: string;
|
|
email: string;
|
|
avatar: string;
|
|
permissions: number;
|
|
userType: number;
|
|
createdAt: Date;
|
|
updatedAt: Date;
|
|
requestCount: number;
|
|
settings?: UserSettings;
|
|
}
|
|
|
|
type NotificationAgentTypes = Record<NotificationAgentKey, number>;
|
|
|
|
export interface UserSettings {
|
|
discordId?: string;
|
|
region?: string;
|
|
originalLanguage?: string;
|
|
locale?: string;
|
|
notificationTypes: Partial<NotificationAgentTypes>;
|
|
watchlistSyncMovies?: boolean;
|
|
watchlistSyncTv?: boolean;
|
|
}
|
|
|
|
interface UserHookResponse {
|
|
user?: User;
|
|
loading: boolean;
|
|
error: string;
|
|
revalidate: (
|
|
data?: User | Promise<User> | MutatorCallback<User> | undefined,
|
|
shouldRevalidate?: boolean | undefined
|
|
) => Promise<User | undefined>;
|
|
hasPermission: (
|
|
permission: Permission | Permission[],
|
|
options?: PermissionCheckOptions
|
|
) => boolean;
|
|
}
|
|
|
|
export const useUser = ({
|
|
id,
|
|
initialData,
|
|
}: { id?: number; initialData?: User } = {}): UserHookResponse => {
|
|
const {
|
|
data,
|
|
error,
|
|
mutate: revalidate,
|
|
} = useSWR<User>(id ? `/api/v1/user/${id}` : `/api/v1/auth/me`, {
|
|
fallbackData: initialData,
|
|
refreshInterval: 30000,
|
|
errorRetryInterval: 30000,
|
|
shouldRetryOnError: false,
|
|
});
|
|
|
|
const checkPermission = (
|
|
permission: Permission | Permission[],
|
|
options?: PermissionCheckOptions
|
|
): boolean => {
|
|
return hasPermission(permission, data?.permissions ?? 0, options);
|
|
};
|
|
|
|
return {
|
|
user: data,
|
|
loading: !data && !error,
|
|
error,
|
|
hasPermission: checkPermission,
|
|
revalidate,
|
|
};
|
|
};
|