From 7697c767214921084e1ba6c1af3fd528e75f19a3 Mon Sep 17 00:00:00 2001
From: Jorrin <jorrinkievit@hotmail.com>
Date: Wed, 10 Jan 2024 21:21:47 +0100
Subject: [PATCH] add headers

---
 src/providers/embeds/upcloud.ts       |  7 +++++++
 src/providers/sources/remotestream.ts | 17 +++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/src/providers/embeds/upcloud.ts b/src/providers/embeds/upcloud.ts
index 7880cc4..0f86cd0 100644
--- a/src/providers/embeds/upcloud.ts
+++ b/src/providers/embeds/upcloud.ts
@@ -4,6 +4,9 @@ import { flags } from '@/entrypoint/utils/targets';
 import { makeEmbed } from '@/providers/base';
 import { Caption, getCaptionTypeFromUrl, labelToLanguageCode } from '@/providers/captions';
 
+const origin = 'https://rabbitstream.net';
+const referer = 'https://rabbitstream.net/';
+
 const { AES, enc } = crypto;
 
 interface StreamRes {
@@ -126,6 +129,10 @@ export const upcloudScraper = makeEmbed({
           playlist: sources.file,
           flags: [flags.CORS_ALLOWED],
           captions,
+          preferredHeaders: {
+            Referer: referer,
+            Origin: origin,
+          },
         },
       ],
     };
diff --git a/src/providers/sources/remotestream.ts b/src/providers/sources/remotestream.ts
index 8a3090b..f0a0ae8 100644
--- a/src/providers/sources/remotestream.ts
+++ b/src/providers/sources/remotestream.ts
@@ -4,6 +4,9 @@ import { NotFoundError } from '@/utils/errors';
 
 const remotestreamBase = atob('aHR0cHM6Ly9mc2IuOG1ldDNkdGpmcmNxY2hjb25xcGtsd3hzeGIyb2N1bWMuc3RyZWFt');
 
+const origin = 'https://remotestre.am';
+const referer = 'https://remotestre.am/';
+
 export const remotestreamScraper = makeSourcerer({
   id: 'remotestream',
   name: 'Remote Stream',
@@ -19,6 +22,9 @@ export const remotestreamScraper = makeSourcerer({
     const streamRes = await ctx.fetcher.full(playlistLink, {
       method: 'HEAD',
       readHeaders: ['content-type'],
+      headers: {
+        Referer: referer,
+      },
     });
     if (!streamRes.headers.get('content-type')?.toLowerCase().includes('application/x-mpegurl'))
       throw new NotFoundError('No watchable item found');
@@ -33,6 +39,10 @@ export const remotestreamScraper = makeSourcerer({
           playlist: playlistLink,
           type: 'hls',
           flags: [flags.CORS_ALLOWED],
+          preferredHeaders: {
+            Referer: referer,
+            Origin: origin,
+          },
         },
       ],
     };
@@ -44,6 +54,9 @@ export const remotestreamScraper = makeSourcerer({
     const streamRes = await ctx.fetcher.full(playlistLink, {
       method: 'HEAD',
       readHeaders: ['content-type'],
+      headers: {
+        Referer: referer,
+      },
     });
     if (!streamRes.headers.get('content-type')?.toLowerCase().includes('application/x-mpegurl'))
       throw new NotFoundError('No watchable item found');
@@ -58,6 +71,10 @@ export const remotestreamScraper = makeSourcerer({
           playlist: playlistLink,
           type: 'hls',
           flags: [flags.CORS_ALLOWED],
+          preferredHeaders: {
+            Referer: referer,
+            Origin: origin,
+          },
         },
       ],
     };