From 0356b766bab00fb906a04a0dd891c198fdaac483 Mon Sep 17 00:00:00 2001
From: Pas <74743263+Pasithea0@users.noreply.github.com>
Date: Tue, 22 Jul 2025 09:54:55 -0600
Subject: [PATCH] add a way to disable m3u8 proxying

---
 src/routes/m3u8-proxy.ts | 7 +++++++
 src/routes/ts-proxy.ts   | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/src/routes/m3u8-proxy.ts b/src/routes/m3u8-proxy.ts
index d4e02ee..1373c03 100644
--- a/src/routes/m3u8-proxy.ts
+++ b/src/routes/m3u8-proxy.ts
@@ -363,6 +363,13 @@ export function handleCacheStats(event: any) {
 export default defineEventHandler(async (event) => {
   // Handle CORS preflight requests
   if (isPreflightRequest(event)) return handleCors(event, {});
+
+  if (process.env.DISABLE_M3U8 === 'true') {
+    return sendError(event, createError({
+      statusCode: 404,
+      statusMessage: 'M3U8 proxying is disabled'
+    }));
+  }
   
   if (event.path === '/cache-stats') {
     return handleCacheStats(event);
diff --git a/src/routes/ts-proxy.ts b/src/routes/ts-proxy.ts
index c9d3790..bfc93dd 100644
--- a/src/routes/ts-proxy.ts
+++ b/src/routes/ts-proxy.ts
@@ -7,6 +7,13 @@ const isCacheDisabled = () => process.env.DISABLE_CACHE === 'true';
 export default defineEventHandler(async (event) => {
   // Handle CORS preflight requests
   if (isPreflightRequest(event)) return handleCors(event, {});
+
+  if (process.env.DISABLE_M3U8 === 'true') {
+    return sendError(event, createError({
+      statusCode: 404,
+      statusMessage: 'TS proxying is disabled'
+    }));
+  }
   
   const url = getQuery(event).url as string;
   const headersParam = getQuery(event).headers as string;