From dcb7cfec276d93eeb139a15c140c8f59e99ff9cd Mon Sep 17 00:00:00 2001
From: 0d0 <0d0acre@esiliati.org>
Date: Sun, 23 Feb 2025 04:55:09 +0100
Subject: [PATCH] Add loader and new formats

---
 src/lib/common/supportedFormats.json |   6 ++
 src/lib/components/Loader.svelte     |   6 +-
 src/lib/server/helpers.ts            |  14 ++++
 src/lib/server/ytdlp.ts              |  23 ++++--
 src/routes/+page.svelte              | 112 +++++++++++++++++++++------
 src/routes/download/+server.ts       |  27 ++++---
 6 files changed, 148 insertions(+), 40 deletions(-)
 create mode 100644 src/lib/common/supportedFormats.json
 create mode 100644 src/lib/server/helpers.ts

diff --git a/src/lib/common/supportedFormats.json b/src/lib/common/supportedFormats.json
new file mode 100644
index 0000000..80187b0
--- /dev/null
+++ b/src/lib/common/supportedFormats.json
@@ -0,0 +1,6 @@
+{
+    "mp3": "audio/mpeg",
+    "mp4": "video/mp4",
+    "opus": "audio/ogg",
+    "wav": "audio/wav"
+}
\ No newline at end of file
diff --git a/src/lib/components/Loader.svelte b/src/lib/components/Loader.svelte
index 62b43fc..df33172 100644
--- a/src/lib/components/Loader.svelte
+++ b/src/lib/components/Loader.svelte
@@ -1,6 +1,8 @@
 <div
-	class="bg-opacity-50 absolute inset-0 z-10 flex items-center justify-center bg-white"
+	class=" backdrop-blur-xs bg-black/20 absolute inset-0 z-10 flex items-center justify-center "
 	id="spinner"
 >
-	<div class="h-20 w-20 animate-spin rounded-full border-t-2 border-b-2 border-gray-900"></div>
+	<div class="w-[150px]">
+		<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 200 200"><linearGradient id="a3"><stop offset="0" stop-color="#FF156D" stop-opacity="0"></stop><stop offset="1" stop-color="#FF156D"></stop></linearGradient><circle fill="none" stroke="url(#a3)" stroke-width="15" stroke-linecap="round" stroke-dasharray="0 44 0 44 0 44 0 44 0 360" cx="100" cy="100" r="70" transform-origin="center"><animateTransform type="rotate" attributeName="transform" calcMode="discrete" dur="2" values="360;324;288;252;216;180;144;108;72;36" repeatCount="indefinite"></animateTransform></circle></svg>
+	</div>
 </div>
diff --git a/src/lib/server/helpers.ts b/src/lib/server/helpers.ts
new file mode 100644
index 0000000..af5111d
--- /dev/null
+++ b/src/lib/server/helpers.ts
@@ -0,0 +1,14 @@
+import formats from '$lib/common/supportedFormats.json';
+
+const formatMime = new Map(Object.entries(formats))
+
+export const mimeTypeMap = formatMime;
+export const contentTypeFromFormat = (format: string): string => {
+    const toReturn: string | undefined = formatMime.get(format)
+
+    if (!toReturn) {
+        throw new Error("Unsupported format")
+    }
+
+    return toReturn;
+}
\ No newline at end of file
diff --git a/src/lib/server/ytdlp.ts b/src/lib/server/ytdlp.ts
index f37ddef..9afded2 100644
--- a/src/lib/server/ytdlp.ts
+++ b/src/lib/server/ytdlp.ts
@@ -1,6 +1,9 @@
 import { create } from 'youtube-dl-exec';
 import { env } from '$env/dynamic/private';
 import { spawn } from 'node:child_process';
+import supportedFormats from '$lib/common/supportedFormats.json';
+import { mimeTypeMap } from '$lib/server/helpers';
+
 const YTDLP_PATH: string = env.YTDLP_PATH as string;
 
 export const ytdl = create(YTDLP_PATH);
@@ -21,19 +24,27 @@ export async function getYouTubeMetadata(link: string) {
  * Streams the YouTube video/audio using youtube-dl-exec
  */
 export function streamYouTube(link: string, format: string): ReadableStream<Uint8Array> {
+	const mimeType: string | undefined = mimeTypeMap.get(format)
+	
+	if (!mimeType) {
+		throw new Error("Unsupported format");
+	}
+
 	return new ReadableStream({
 		start(controller) {
 			const args = [
 				'-o',
 				'-',
-				format === 'mp3' ? '--embed-metadata' : '',
-				'--format',
-				format === 'mp3' ? 'bestaudio' : 'best',
-				'--audio-format',
-				format === 'mp3' ? 'mp3' : '',
-				'--no-playlist'
 			].filter(Boolean);
 
+			if(mimeType?.includes('audio')) {
+				args.push(...['--extract-audio', '--embed-metadata', '--embed-thumbnail', '--audio-format', format])
+			} else if (mimeType.includes('video')) {
+				args.push(...['--embed-metadata', '--embed-thumbnail', '--format', format])
+			}
+
+			console.info(`yt-dlp ${args.join(' ')} ${link}`)
+
 			const process = spawn('yt-dlp', [...args, link], { stdio: ['ignore', 'pipe', 'ignore'] });
 
 			process.stdout.on('data', (chunk) => controller.enqueue(chunk));
diff --git a/src/routes/+page.svelte b/src/routes/+page.svelte
index 098ef91..0e2f7e8 100644
--- a/src/routes/+page.svelte
+++ b/src/routes/+page.svelte
@@ -1,35 +1,80 @@
 <script lang="ts">
-	import {PUBLIC_VERSION} from '$env/static/public';
+	import { PUBLIC_VERSION } from '$env/static/public';
+	import supportedFormats from '$lib/common/supportedFormats.json';
+	import Loader from '$lib/components/Loader.svelte';
+	import { onMount } from 'svelte';
+
 	let source = $state('youtube');
 	let link = $state('');
 	let format = $state('mp3');
 	let showModal = $state(false);
 	let href = $state('');
-	// let formats = ['ogg', 'mp3', 'mp4']
-	const formats = [{ value: 'mp3', label: 'MP3' }];
+	let disabled = $state(true);
+	let metadata = $state(true);
+	let downloading = $state(false);
+
+	const formats = Object.keys(supportedFormats).map((f) => {
+		return { value: f, label: f.toUpperCase() };
+	});
 
 	const toggleModal = () => {
 		showModal = !showModal;
 	};
 
-	const handleSubmit = async (e: SubmitEvent) => {
-		e.preventDefault();
+	onMount(() => {
+		document.cookie = 'downloading=0'
+	});
 
+	const onClick = () => {
+		link = '';
+		downloading = true;
+		document.cookie = 'downloading=1'
+
+		const id = setInterval(() => {
+			if (document.cookie.includes('downloading=0')) {
+				console.log("FINITO!")
+				downloading = false && clearInterval(id);
+			}
+		}, 1000);
+	};
+
+	const createAnchor = () => {
 		console.log({
 			source,
 			link,
-			format
+			format,
+			metadata
 		});
 
+		if (!(source && link && format)) {
+			disabled = true;
+			return;
+		}
+
+		try {
+			new URL(link);
+			disabled = false;
+		} catch (err) {
+			/*
+			if (err.code === 'ERR_INVALID_URL') {
+
+			}
+			*/
+			disabled = true;
+			return;
+		}
+
 		const searchParams = new URLSearchParams();
 		searchParams.append('source', source);
 		searchParams.append('link', link);
 		searchParams.append('format', format);
+		searchParams.append('metadata', metadata);
 
 		href = `/download?${searchParams.toString()}`;
 	};
 
 	$effect(() => {
+		createAnchor();
 		// Auto selected the radio button based on url regex
 		if (link.includes('spotify')) {
 			source = 'spotify';
@@ -43,6 +88,9 @@
 	id="wrapper"
 	class="relative mx-auto rounded-lg bg-black p-6 text-green-500 shadow-lg sm:max-w-sm sm:border-4 sm:border-green-500 md:mt-10 md:max-w-md lg:max-w-lg 2xl:max-w-2xl"
 >
+	<div id="loader" class={[{ downloading }]}>
+		<Loader />
+	</div>
 	<!-- Info Icon -->
 	<button
 		onclick={toggleModal}
@@ -57,7 +105,7 @@
 	</button>
 
 	<h1 id="title" class="mb-6 text-center text-xl">🐙 Scaricatore 🐙</h1>
-	<form class="space-y-6" onsubmit={handleSubmit}>
+	<form class="space-y-6">
 		<!-- Source Selection -->
 		<fieldset class="space-y-4">
 			<legend class="text-green-400">Choose Source:</legend>
@@ -122,17 +170,28 @@
 			</select>
 		</div>
 
-		<!-- Submit Button -->
-		<button
-			type="submit"
-			class="w-full rounded-lg border-4 border-pink-700 bg-pink-500 px-4 py-3 text-black transition hover:bg-pink-600 active:border-yellow-500"
-		>
-			Create download link
-		</button>
+		<!-- Format Selection -->
+		<div>
+			<label for="metadata" class="mb-2 block text-green-400"> Metadata </label>
+			<input
+				type="checkbox"
+				id="metadata"
+				name="metadata"
+				bind:checked={metadata}
+				class="rounded-lg border-4 border-green-500 bg-green-200 px-4 py-3 text-black focus:border-pink-500 focus:outline-none"
+			/>
+		</div>
 
-		{#if href}
-			<a class="download-link" {href}> Download </a>
-		{/if}
+		<a
+			id="btn-download"
+			{href}
+			onclick={onClick}
+			class="{disabled
+				? 'disabled'
+				: ''} block w-full rounded-lg border-4 border-pink-700 bg-pink-500 px-4 py-3 text-center text-4xl text-black transition hover:bg-pink-600 active:border-yellow-500"
+		>
+			DOWNLOAD
+		</a>
 	</form>
 </div>
 
@@ -167,12 +226,11 @@
 		font-size: 12px;
 	}
 
-	.download-link {
-		margin: 0 auto;
-		padding: 5px;
+	#loader {
+		display: none;
+	}
+	#loader.downloading {
 		display: block;
-		text-decoration: underline;
-		text-align: center;
 	}
 	.not-available {
 		text-decoration-line: line-through;
@@ -203,4 +261,14 @@
 	#title {
 		font-size: 22px;
 	}
+
+	#btn-download {
+		font-size: 24px;
+		transition: opacity 0.3s ease;
+	}
+	a.disabled {
+		pointer-events: none;
+		cursor: default;
+		opacity: 0.4;
+	}
 </style>
diff --git a/src/routes/download/+server.ts b/src/routes/download/+server.ts
index e740c5a..305f0b4 100644
--- a/src/routes/download/+server.ts
+++ b/src/routes/download/+server.ts
@@ -1,12 +1,14 @@
 import { error } from '@sveltejs/kit';
 import type { RequestHandler } from './$types';
-import { getYouTubeMetadata, streamYouTube } from '$lib/server/ytdlp';
+import { getYouTubeMetadata, streamYouTube, ytdl } from '$lib/server/ytdlp';
+import { contentTypeFromFormat, mimeTypeMap } from '$lib/server/helpers';
 
 export const GET: RequestHandler = async ({ url }) => {
 	// Get query params
 	const link = url.searchParams.get('link');
 	const format = url.searchParams.get('format'); // mp3, mp4
 	const source = url.searchParams.get('source'); // youtube or spotify
+	const metadata = url.searchParams.get('metadata');
 
 	// Validate input
 	if (!link || !format || !source) {
@@ -18,22 +20,27 @@ export const GET: RequestHandler = async ({ url }) => {
 	}
 
 	try {
-		// Fetch metadata for filename
-		const metadata = await getYouTubeMetadata(link);
-		const { title, uploader } = metadata;
-		const safeTitle = `${uploader} - ${title}`;
-		const filename = `${safeTitle}.${format}`;
+		let filename = `you-clicked-no-metadata-so-i-cant-put-a-correct-name.${format}`;
+
+		if (metadata) {
+			// Fetch metadata for filename
+			const metadata = await getYouTubeMetadata(link);
+			const { title, uploader } = metadata;
+			const safeTitle = `${uploader} - ${title}`;
+			filename = `${safeTitle}.${format}`;
+		}
 
-		console.log(filename);
 		// Stream video/audio
 		return new Response(streamYouTube(link, format), {
 			headers: {
-				'Content-Type': format === 'mp3' ? 'audio/mpeg' : 'video/mp4',
-				'Content-Disposition': `attachment; filename="${filename}"`
+				'Content-Type': contentTypeFromFormat(format),
+				'Content-Disposition': `attachment; filename="${filename}"`,
+				'Set-Cookie': 'downloading=0'
 			}
 		});
 	} catch (err) {
-		console.error('Error fetching metadata:', err);
+		console.error(err)
+		console.error('Error fetching metadata:');
 		throw error(500, 'Failed to fetch video metadata');
 	}
 };