Viren070/AIOStreams
1
0
Fork 0
mirror of https://github.com/Viren070/AIOStreams.git synced 2025-12-01 23:14:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: add DISABLE_TORRENTIO and DISABLE_TORRENTIO_MESSAGE env vars

Browse Source
This commit is contained in:
Viren070
2025-01-20 22:44:28 +00:00
parent fb787236a1
commit 064e54c7ff
4 changed files with 85 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/addon/src/config.ts
+55 -9
View File
@@ -92,6 +92,13 @@ export function validateConfig(config: Config): {
}
for (const addon of config.addons) {
if (Settings.DISABLE_TORRENTIO && addon.id === 'torrentio') {
return createResponse(
false,
'torrentioDisabled',
Settings.DISABLE_TORRENTIO_MESSAGE
);
}
// if torbox addon is enabled, torbox service must be enabled and torbox api key must be set
if (addon.id === 'torbox') {
const torboxService = config.services.find(
@@ -134,19 +141,58 @@ export function validateConfig(config: Config): {
if (
option.id.toLowerCase().includes('url') &&
addon.options[option.id] &&
addon.options[option.id]?.match(
/^E-[0-9a-fA-F]{32}-[0-9a-fA-F]+$/
) === null
addon.options[option.id]
) {
try {
new URL(addon.options[option.id] as string);
} catch (_) {
if (
Settings.DISABLE_TORRENTIO &&
addon.options[option.id]?.match(/torrentio.strem.fun/) !== null
) {
// if torrentio is disabled, don't allow the user to set URLs with torrentio.strem.fun
return createResponse(
false,
'invalidUrl',
`Invalid URL for ${option.label}`
'torrentioDisabled',
Settings.DISABLE_TORRENTIO_MESSAGE
);
} else if (
Settings.DISABLE_TORRENTIO &&
addon.options[option.id]?.match(/stremthru.elfhosted.com/) !== null
) {
// if torrentio is disabled, we need to inspect the stremthru URL to see if it's using torrentio
try {
const url = new URL(addon.options[option.id] as string);
// get the component before manifest.json
const pathComponents = url.pathname.split('/');
if (pathComponents.includes('manifest.json')) {
const index = pathComponents.indexOf('manifest.json');
const componentBeforeManifest = pathComponents[index - 1];
// base64 decode the component before manifest.json
const decodedComponent = atob(componentBeforeManifest);
const stremthruData = JSON.parse(decodedComponent);
if (stremthruData?.manifest_url?.match(/torrentio.strem.fun/)) {
return createResponse(
false,
'torrentioDisabled',
Settings.DISABLE_TORRENTIO_MESSAGE
);
}
}
} catch (_) {
// ignore
}
} else if (
addon.options[option.id]?.match(
/^E-[0-9a-fA-F]{32}-[0-9a-fA-F]+$/
) === null
) {
try {
new URL(addon.options[option.id] as string);
} catch (_) {
return createResponse(
false,
'invalidUrl',
` Invalid URL for ${option.label}`
);
}
}
}
packages/addon/src/server.ts
+10
View File
@@ -210,6 +210,16 @@ app.post('/encrypt-user-data', (req, res) => {
res.status(500).json({ success: false, message: 'Secret key not set' });
return;
}
const { valid, errorCode, errorMessage } = validateConfig(JSON.parse(data));
if (!valid) {
console.error(
`|ERR| server > /encrypt-user-data: Invalid config: ${errorCode} - ${errorMessage}`
);
res
.status(200)
.json({ success: false, message: errorMessage, error: errorMessage });
return;
}
const encryptedData = compressAndEncrypt(data);
console.log(`|DBG| server > /encrypt-user-data: Encrypted data`);
res.status(200).json({ success: true, data: encryptedData });
packages/frontend/src/app/configure/page.tsx
+14 -5
View File
@@ -277,15 +277,24 @@ export default function Configure() {
if (!response) {
throw new Error('encrypt-user-data failed: no response within timeout');
}
if (!response.ok) {
throw new Error(
`encrypt-user-data failed with status ${response.status}`
`encrypt-user-data failed with status ${response.status} and statusText ${response.statusText}`
);
}
const data = await response.json();
if (!data.success)
if (!data.success) {
if (data.error) {
return {
success: false,
manifest: null,
message: data.error,
};
}
throw new Error(`Encryption service failed, ${data.message}`);
}
const encryptedConfig = data.data;
return {
@@ -342,7 +351,7 @@ export default function Configure() {
if (!manifestUrl.success || !manifestUrl.manifest) {
setDisableButtons(false);
toast.update(id, {
render: 'Failed to generate manifest URL',
render: manifestUrl.message || 'Failed to generate manifest URL',
type: 'error',
autoClose: 5000,
isLoading: false,
@@ -387,7 +396,7 @@ export default function Configure() {
const manifestUrl = await getManifestUrl();
if (!manifestUrl.success || !manifestUrl.manifest) {
toast.update(id, {
render: 'Failed to generate manifest URL',
render: manifestUrl.message || 'Failed to generate manifest URL',
type: 'error',
autoClose: 5000,
isLoading: false,
@@ -436,7 +445,7 @@ export default function Configure() {
const manifestUrl = await getManifestUrl();
if (!manifestUrl.success || !manifestUrl.manifest) {
toast.update(id, {
render: 'Failed to generate manifest URL',
render: manifestUrl.message || 'Failed to generate manifest URL',
type: 'error',
autoClose: 5000,
isLoading: false,
packages/utils/src/settings.ts
+6
View File
@@ -95,4 +95,10 @@ export class Settings {
public static readonly LOG_SENSITIVE_INFO = process.env.LOG_SENSITIVE_INFO
? process.env.LOG_SENSITIVE_INFO === 'true'
: false;
public static readonly DISABLE_TORRENTIO = process.env.DISABLE_TORRENTIO
? process.env.DISABLE_TORRENTIO === 'true'
: false;
public static readonly DISABLE_TORRENTIO_MESSAGE =
process.env.DISABLE_TORRENTIO_MESSAGE ||
'The Torrentio addon has been disabled, please remove it to use this addon.';
}