added support for thumbnails

Author: Toasty360

README.md

@@ -24,6 +24,8 @@ This application serves as a proxy for HLS streams, Images and enabling secure a
 
 - `/proxy` - for HLS
 - `/cors` - for Images/Web pages
+- `/image` - for Images/Web pages
+- `/thumbnail` - for thumbnail images
 
 Use the following format to access the proxy:
 

src/cors.js

@@ -1,46 +1,4 @@
-async function handleCorsRequest(request) {
-	const urlParams = new URL(request.url).searchParams;
-	const encodedUrl = urlParams.get('url');
-	const headersBase64 = urlParams.get('headers');
-
-	if (!encodedUrl) {
-		return new Response('Invalid URL format. Must be a valid base64-encoded URL.', {
-			status: 400,
-		});
-	}
-
-	// Decode the base64-encoded URL
-	let targetUrl;
-	try {
-		targetUrl = atob(encodedUrl);
-	} catch (error) {
-		return new Response('Failed to decode URL. Ensure it is base64-encoded.', {
-			status: 400,
-		});
-	}
-
-	let decodedHeaders = {};
-	if (headersBase64) {
-		try {
-			const decodedString = atob(headersBase64);
-			decodedHeaders = JSON.parse(decodedString);
-		} catch (error) {
-			return new Response('Invalid headers format. Must be valid base64-encoded JSON.', {
-				status: 400,
-			});
-		}
-	}
-
-	// Convert the plain JSON headers object to a Headers instance
-	const headers = new Headers();
-	headers.append(
-		'User-Agent',
-		'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/237.84.2.178 Safari/537.36'
-	);
-	for (const [key, value] of Object.entries(decodedHeaders)) {
-		headers.append(key, value);
-	}
-
+async function handleCorsRequest(targetUrl, headers) {
 	try {
 		// Fetch the target URL with the specified headers
 		const response = await fetch(targetUrl, {

src/index.js

@@ -12,6 +12,8 @@
 
 import handleCorsRequest from './cors.js';
 import proxy from './proxy.js';
+import { thumbnailHandler } from './thumbnails.js';
+import { handleRequest } from './utils/handler.js';
 
 export default {
 	async fetch(request) {
@@ -20,15 +22,26 @@ export default {
 		if (url.pathname === '/proxy') {
 			return proxy(request);
 		} else if (url.pathname === '/cors') {
-			return handleCorsRequest(request);
+			const [url, headers] = handleRequest(request);
+			return handleCorsRequest(url, headers);
 		} else if (url.pathname === '/image') {
-			return handleCorsRequest(request);
+			const [url, headers, origin] = handleRequest(request);
+			return handleCorsRequest(url, headers, origin);
+		} else if (url.pathname === '/thumbnail') {
+			const [url, headers, origin] = handleRequest(request);
+			return thumbnailHandler(url, headers, origin);
 		} else if (url.pathname === '/') {
 			return new Response(
 				JSON.stringify({
 					message: 'Welcome to Roxy',
-					Endpoints: ['/proxy', '/cors'],
+					Endpoints: [
+						{ '/proxy': 'For HLS' },
+						{ '/cors': 'For CORS' },
+						{ '/image': 'For Manga Images' },
+						{ '/thumbnail': 'For Thumbnails' },
+					],
 					params: '?url=<Base64-encoded-m3u8-url>&headers=<Base64-encoded-headers>',
+					tip: 'Base64Encoding is optional for /cors, /thumbnail, /image. Encode the url if it gives error. For /proxy, encoding is required.',
 				}),
 				{
 					status: 200,

src/proxy.js

@@ -1,3 +1,5 @@
+import { decodeHeaders } from './utils/handler';
+
 const m3u8ContentTypes = [
 	'application/vnd.apple.mpegurl', // Standard HLS playlist
 	'application/x-mpegurl', // Common alias
@@ -28,22 +30,6 @@ const CACHE_CONTROL_SETTINGS = {
 	ERROR: 'no-store',
 };
 
-function decodeHeaders(base64Headers) {
-	const headers = new Headers();
-	if (!base64Headers) return headers;
-	try {
-		const decodedString = atob(base64Headers);
-		const headersObj = JSON.parse(decodedString);
-
-		Object.entries(headersObj).forEach(([key, value]) => {
-			headers.append(key, value);
-		});
-		return headers;
-	} catch (error) {
-		return null;
-	}
-}
-
 function getCacheSettings(url, content) {
 	if (url.includes('.ts') || url.includes('.m4s')) {
 		return CACHE_CONTROL_SETTINGS.SEGMENT;

src/thumbnails.js

@@ -0,0 +1,25 @@
+export const thumbnailHandler = async (url, headers, origin) => {
+	const resp = await fetch(url, {
+		...headers,
+		redirect: 'follow',
+		'User-Agent':
+			'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/237.84.2.178 Safari/537.36',
+	});
+
+	if (resp.status !== 200) {
+		return new Response(resp.status, resp);
+	}
+	const timestampRegex = /(?<=\d{2}:\d{2}:\d{2}\.\d{3} --> \d{2}:\d{2}:\d{2}\.\d{3}\s)(.*)/gm;
+	const responseBody = await resp.text();
+	const baseUrl = url.substring(0, url.lastIndexOf('/'));
+	const modifiedBody = responseBody.replace(timestampRegex, (match) => {
+		const fullUrl = match.startsWith('http') ? match : match.startsWith('/') ? `${baseUrl}/${match}` : `${baseUrl}/${match}`;
+		return `${origin}/cors?url=${encodeURIComponent(btoa(fullUrl))}`;
+	});
+	return new Response(modifiedBody, {
+		headers: {
+			...resp.headers,
+			'Access-Control-Allow-Origin': '*',
+		},
+	});
+};

src/utils/handler.js

@@ -0,0 +1,40 @@
+export const decodeHeaders = (base64Headers) => {
+	const headers = new Headers();
+	if (!base64Headers) return headers;
+	try {
+		const decodedString = atob(base64Headers);
+		const headersObj = JSON.parse(decodedString);
+
+		Object.entries(headersObj).forEach(([key, value]) => {
+			headers.append(key, value);
+		});
+		headers.append(
+			'User-Agent',
+			'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/237.84.2.178 Safari/537.36'
+		);
+		return headers;
+	} catch (error) {
+		return headers;
+	}
+};
+
+export const handleRequest = (request) => {
+	const url = new URL(request.url);
+	const urlParams = url.searchParams;
+	const encodedUrl = urlParams.get('url');
+	const headersBase64 = urlParams.get('headers');
+	if (!encodedUrl) {
+		return new Response('Url is required!', {
+			status: 400,
+		});
+	}
+	let targetUrl;
+	try {
+		targetUrl = atob(encodedUrl);
+	} catch (error) {
+		targetUrl = encodedUrl;
+	}
+
+	const headers = decodeHeaders(headersBase64);
+	return [targetUrl, headers, url.origin];
+};