雨霖铃
88 lines
2.2 KiB
TypeScript
88 lines
2.2 KiB
TypeScript
import { useEffect, useState } from 'react';
|
|
import { MessageAction, sendMessageToContent } from '@/utils/messages';
|
|
import type { RestorerStatus } from './constants';
|
|
|
|
const UNSUPPORTED_PROTOCOLS = new Set([
|
|
'chrome:',
|
|
'chrome-extension:',
|
|
'about:',
|
|
'edge:',
|
|
'brave:',
|
|
]);
|
|
|
|
function isUnsupportedPage(url: string | undefined): boolean {
|
|
if (!url) return true;
|
|
try {
|
|
const protocol = new URL(url).protocol;
|
|
return UNSUPPORTED_PROTOCOLS.has(protocol);
|
|
} catch {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
function deriveStatus(isUnsupported: boolean, isUnlocked: boolean): RestorerStatus {
|
|
if (isUnsupported) return 'unsupported';
|
|
return isUnlocked ? 'unlocked' : 'locked';
|
|
}
|
|
|
|
export interface UseRightClickRestorerReturn {
|
|
domain: string;
|
|
isLoading: boolean;
|
|
status: RestorerStatus;
|
|
unlock: () => Promise<void>;
|
|
}
|
|
|
|
export function useRightClickRestorer(): UseRightClickRestorerReturn {
|
|
const [domain, setDomain] = useState<string>('');
|
|
const [isLoading, setIsLoading] = useState<boolean>(true);
|
|
const [isUnlocked, setIsUnlocked] = useState<boolean>(false);
|
|
const [isUnsupported, setIsUnsupported] = useState<boolean>(false);
|
|
|
|
useEffect(() => {
|
|
const load = async () => {
|
|
try {
|
|
const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
|
|
const url = tab?.url;
|
|
|
|
setDomain(url ? new URL(url).hostname : '');
|
|
|
|
if (isUnsupportedPage(url)) {
|
|
setIsUnsupported(true);
|
|
return;
|
|
}
|
|
|
|
const response = await sendMessageToContent(MessageAction.QUERY_RIGHT_CLICK_STATUS);
|
|
if (response?.success) {
|
|
setIsUnlocked(response.restored);
|
|
}
|
|
} catch (err) {
|
|
console.error('[RightClickRestorer] Failed to load state:', err);
|
|
} finally {
|
|
setIsLoading(false);
|
|
}
|
|
};
|
|
|
|
void load();
|
|
}, []);
|
|
|
|
async function unlock() {
|
|
if (isUnsupported) return;
|
|
|
|
try {
|
|
const response = await sendMessageToContent(MessageAction.RESTORE_RIGHT_CLICK);
|
|
if (response?.success) {
|
|
setIsUnlocked(response.restored);
|
|
}
|
|
} catch (err) {
|
|
console.error('[RightClickRestorer] Failed to unlock:', err);
|
|
}
|
|
}
|
|
|
|
return {
|
|
domain,
|
|
isLoading,
|
|
status: deriveStatus(isUnsupported, isUnlocked),
|
|
unlock,
|
|
};
|
|
}
|