mirror of
https://github.com/bytedream/stream-bypass.git
synced 2026-08-04 21:10:40 +02:00
fix per domain disable not working if host is in iframe
This commit is contained in:
@@ -1,11 +1,25 @@
|
||||
import { BackgroundMessageData, BackgroundMessageType } from '@/lib/communication';
|
||||
import { getHost } from '@/lib/host';
|
||||
import { HostSettings, UrlReferer } from '@/lib/settings';
|
||||
|
||||
export default defineBackground(() => {
|
||||
browser.runtime.onMessage.addListener(async (message) => {
|
||||
if (message.action == 'ff2mpv') {
|
||||
await browser.runtime.sendNativeMessage('ff2mpv', { url: message.url });
|
||||
browser.runtime.onMessage.addListener(async (message, sender) => {
|
||||
const messageType = message.type as BackgroundMessageType;
|
||||
const messageData = message.data as BackgroundMessageData<typeof messageType>;
|
||||
|
||||
let response;
|
||||
switch (messageType) {
|
||||
case BackgroundMessageType.Ff2mpv:
|
||||
await browser.runtime.sendNativeMessage('ff2mpv', { url: messageData!.url });
|
||||
break;
|
||||
case BackgroundMessageType.RequestTabUrl:
|
||||
response = sender.tab?.url ?? null;
|
||||
break;
|
||||
default:
|
||||
return;
|
||||
}
|
||||
|
||||
return response;
|
||||
});
|
||||
|
||||
// the following listener is only available in mv2
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
import { BackgroundMessageType, sendBackgroundMessage } from '@/lib/communication';
|
||||
import { getHost, HostMatchType, hosts, type HostMatch } from '@/lib/host';
|
||||
import { FF2MPVSettings } from '@/lib/settings';
|
||||
import { FF2MPVSettings, PerDomainSettings } from '@/lib/settings';
|
||||
|
||||
export default defineContentScript({
|
||||
matches: [
|
||||
@@ -16,6 +17,18 @@ async function main() {
|
||||
const host = await getHost(window.location.host);
|
||||
if (!host) return;
|
||||
|
||||
// if current windows is an iframe, check if the parent has any disabled hosts
|
||||
if (window !== window.top) {
|
||||
const tabUrl = await sendBackgroundMessage(BackgroundMessageType.RequestTabUrl, undefined);
|
||||
if (tabUrl) {
|
||||
const domain = new URL(tabUrl).host;
|
||||
|
||||
const perDomainSetting = await PerDomainSettings.get(domain);
|
||||
if (perDomainSetting.allDisabled) return;
|
||||
else if (perDomainSetting.disabledHostIds.indexOf(host.id) !== -1) return;
|
||||
}
|
||||
}
|
||||
|
||||
let re = null;
|
||||
for (const regex of host.regex) {
|
||||
if ((re = document.body.innerHTML.match(regex)) !== null) {
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
import Hls from 'hls.js';
|
||||
import { listenMessages, MessageType, sendMessage } from '@/lib/communication';
|
||||
import { listenTabMessages, sendTabBroadcast, TabMessageType } from '@/lib/communication';
|
||||
import { HostMatchType, hosts } from '@/lib/host';
|
||||
import { UrlReferer } from '@/lib/settings';
|
||||
|
||||
@@ -56,7 +56,7 @@ export async function play(videoElem: HTMLVideoElement) {
|
||||
|
||||
function initCommunication(id: string, url: string, domain: string) {
|
||||
const notifyActiveMatch = () =>
|
||||
sendMessage(MessageType.NotifyActiveMatch, {
|
||||
sendTabBroadcast(TabMessageType.NotifyActiveMatch, {
|
||||
id: id,
|
||||
url: url,
|
||||
domain: domain
|
||||
@@ -67,8 +67,8 @@ function initCommunication(id: string, url: string, domain: string) {
|
||||
|
||||
// if an extension popup is opened, the listener will recognize it's active match request and send the match/player
|
||||
// data
|
||||
const cancel = listenMessages((type) => {
|
||||
if (type !== MessageType.RequestActiveMatch) return;
|
||||
const cancel = listenTabMessages((type) => {
|
||||
if (type !== TabMessageType.RequestActiveMatch) return;
|
||||
notifyActiveMatch();
|
||||
});
|
||||
window.onbeforeunload = cancel;
|
||||
|
||||
@@ -8,7 +8,7 @@
|
||||
import CurrentSite from '@/entrypoints/popup/pages/main/CurrentSite.svelte';
|
||||
import Header from '@/entrypoints/popup/pages/main/Header.svelte';
|
||||
import Match from '@/entrypoints/popup/pages/main/Match.svelte';
|
||||
import { listenMessages, MessageType, sendMessageToActiveTab } from '@/lib/communication';
|
||||
import { listenTabMessages, sendTabMessageToActiveTab, TabMessageType } from '@/lib/communication';
|
||||
import { hosts, type Host } from '@/lib/host';
|
||||
import { HostSettings } from '@/lib/settings';
|
||||
|
||||
@@ -27,31 +27,22 @@
|
||||
HostSettings.getAllHostsDisabled().then((val) => (allHostsDisabled = val));
|
||||
|
||||
/* lifecycle */
|
||||
const cancel = listenMessages((type, data) => {
|
||||
if (type !== MessageType.NotifyActiveMatch) return;
|
||||
const cancel = listenTabMessages((type, data) => {
|
||||
if (type !== TabMessageType.NotifyActiveMatch) return;
|
||||
const match = {
|
||||
host: hosts.find((host) => host.id === data.id)!,
|
||||
url: data.url,
|
||||
domain: data.domain
|
||||
};
|
||||
currentMatch = match;
|
||||
currentDomain = match.domain;
|
||||
});
|
||||
sendMessageToActiveTab(MessageType.RequestActiveMatch, undefined);
|
||||
sendTabMessageToActiveTab(TabMessageType.RequestActiveMatch, undefined);
|
||||
|
||||
browser.tabs
|
||||
.query({ active: true, currentWindow: true })
|
||||
.then((tabs) => {
|
||||
if (currentDomain) return;
|
||||
const url = tabs[0]?.url;
|
||||
if (!url) return;
|
||||
try {
|
||||
currentDomain = new URL(url).hostname;
|
||||
} catch {
|
||||
/* ignore */
|
||||
}
|
||||
})
|
||||
.catch(() => {});
|
||||
browser.tabs.query({ active: true, currentWindow: true }).then((tabs) => {
|
||||
const url = tabs[0]?.url;
|
||||
if (!url) return;
|
||||
currentDomain = new URL(url).hostname;
|
||||
});
|
||||
|
||||
onDestroy(cancel);
|
||||
</script>
|
||||
|
||||
@@ -3,7 +3,7 @@
|
||||
import Divider from '@/entrypoints/popup/components/Divider.svelte';
|
||||
import Header from '@/entrypoints/popup/pages/site-config/Header.svelte';
|
||||
import PerSiteHostsList from '@/entrypoints/popup/pages/site-config/PerSiteHostsList.svelte';
|
||||
import { listenMessages, MessageType, sendMessageToActiveTab } from '@/lib/communication';
|
||||
import { listenTabMessages, sendTabMessageToActiveTab, TabMessageType } from '@/lib/communication';
|
||||
import { hosts, type Host } from '@/lib/host';
|
||||
|
||||
/* types */
|
||||
@@ -18,8 +18,8 @@
|
||||
let currentDomain = $state<string | null>(null);
|
||||
|
||||
/* lifecycle */
|
||||
const cancel = listenMessages((type, data) => {
|
||||
if (type !== MessageType.NotifyActiveMatch) return;
|
||||
const cancel = listenTabMessages((type, data) => {
|
||||
if (type !== TabMessageType.NotifyActiveMatch) return;
|
||||
const match = {
|
||||
host: hosts.find((host) => host.id === data.id)!,
|
||||
url: data.url,
|
||||
@@ -28,7 +28,7 @@
|
||||
currentMatch = match;
|
||||
currentDomain = match.domain;
|
||||
});
|
||||
sendMessageToActiveTab(MessageType.RequestActiveMatch, undefined);
|
||||
sendTabMessageToActiveTab(TabMessageType.RequestActiveMatch, undefined);
|
||||
|
||||
browser.tabs
|
||||
.query({ active: true, currentWindow: true })
|
||||
|
||||
+34
-10
@@ -1,36 +1,60 @@
|
||||
export enum MessageType {
|
||||
// --- tab communication --- //
|
||||
export enum TabMessageType {
|
||||
RequestActiveMatch,
|
||||
NotifyActiveMatch
|
||||
}
|
||||
|
||||
export type MessageData<T extends MessageType> = {
|
||||
[MessageType.RequestActiveMatch]: undefined;
|
||||
[MessageType.NotifyActiveMatch]: {
|
||||
export type TabMessageData<T extends TabMessageType> = {
|
||||
[TabMessageType.RequestActiveMatch]: undefined;
|
||||
[TabMessageType.NotifyActiveMatch]: {
|
||||
id: string;
|
||||
url: string;
|
||||
domain: string;
|
||||
};
|
||||
}[T];
|
||||
|
||||
export async function sendMessage<T extends MessageType>(message: T, data: MessageData<T>) {
|
||||
export async function sendTabBroadcast<T extends TabMessageType>(message: T, data: TabMessageData<T>) {
|
||||
await browser.runtime.sendMessage({ type: message, data: data });
|
||||
}
|
||||
|
||||
export async function sendMessageToActiveTab<T extends MessageType>(message: T, data: MessageData<T>) {
|
||||
export async function sendTabMessageToActiveTab<T extends TabMessageType>(message: T, data: TabMessageData<T>) {
|
||||
const tabs = await browser.tabs.query({ active: true, currentWindow: true });
|
||||
|
||||
await browser.tabs.sendMessage(tabs[0].id!, { type: message, data: data }).catch(() => {});
|
||||
}
|
||||
|
||||
export function listenMessages(listener: (type: MessageType, data: any) => void): () => void {
|
||||
const callback = (callbackData: { type: MessageType; data: any }) => {
|
||||
export function listenTabMessages(listener: (type: TabMessageType, data: any) => void): () => void {
|
||||
const callback = (callbackData: { type: TabMessageType; data: any }) => {
|
||||
const { type, data } = callbackData;
|
||||
listener(type, data);
|
||||
};
|
||||
|
||||
browser.runtime.onMessage.addListener(callback);
|
||||
return () => browser.runtime.onMessage.removeListener(callback);
|
||||
}
|
||||
|
||||
return () => {
|
||||
browser.runtime.onMessage.removeListener(callback);
|
||||
// --- background script communication --- //
|
||||
export enum BackgroundMessageType {
|
||||
Ff2mpv,
|
||||
RequestTabUrl
|
||||
}
|
||||
|
||||
export type BackgroundMessageData<T extends BackgroundMessageType> = {
|
||||
[BackgroundMessageType.Ff2mpv]: {
|
||||
url: string;
|
||||
};
|
||||
[BackgroundMessageType.RequestTabUrl]: undefined;
|
||||
}[T];
|
||||
|
||||
export type BackgroundMessageReply<T extends BackgroundMessageType> = {
|
||||
[BackgroundMessageType.Ff2mpv]: undefined;
|
||||
[BackgroundMessageType.RequestTabUrl]: string | null;
|
||||
}[T];
|
||||
|
||||
export async function sendBackgroundMessage<T extends BackgroundMessageType>(
|
||||
message: T,
|
||||
data: BackgroundMessageData<T>
|
||||
): Promise<BackgroundMessageReply<T>> {
|
||||
const response = await browser.runtime.sendMessage({ type: message, data: data });
|
||||
return response;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user