diff --git a/functions/upload.js b/functions/upload.js index 4b4cb3c..3a79290 100644 --- a/functions/upload.js +++ b/functions/upload.js @@ -1,87 +1,85 @@ -const API_BASE = 'https://temp.imgur.gg/api/upload'; -const COMPLETE_URL = 'https://temp.imgur.gg/api/upload/complete'; -const PING_URL = 'https://temp.imgur.gg/api/ping'; +const API_BASE = "https://temp.imgur.gg/api/upload"; +const COMPLETE_URL = "https://temp.imgur.gg/api/upload/complete"; +const PING_URL = "https://temp.imgur.gg/api/ping"; export async function onRequest(context) { const { request } = context; - if (request.method === 'OPTIONS') { - return new Response(null, { - status: 204, - headers: corsHeaders(), - }); + if (request.method === "OPTIONS") { + return new Response(null, { status: 204, headers: corsHeaders() }); } - if (request.method !== 'POST') { - return jsonError('Method not allowed', 405); + if (request.method !== "POST") { + return jsonError("Method not allowed", 405); } try { - const contentType = request.headers.get('content-type') || ''; + const contentType = request.headers.get("content-type") || ""; let file; let fileName; let fileType; - /* ========================== - HANDLE FILE INPUT - ========================== */ + /* ========================= */ + /* GET FILE */ + /* ========================= */ - if (contentType.includes('application/json')) { + if (contentType.includes("application/json")) { const body = await request.json(); - - if (!body.fileUrl) { - return jsonError('No fileUrl provided', 400); - } + if (!body.fileUrl) return jsonError("No fileUrl provided", 400); const res = await fetch(body.fileUrl); - if (!res.ok) throw new Error('Failed to fetch remote file'); + if (!res.ok) throw new Error("Failed to fetch remote file"); file = await res.arrayBuffer(); - fileName = body.fileName || body.fileUrl.split('/').pop(); - fileType = res.headers.get('content-type') || 'application/octet-stream'; + fileName = body.fileName || body.fileUrl.split("/").pop(); + fileType = res.headers.get("content-type") || "application/octet-stream"; } else { const form = await request.formData(); - const uploaded = form.get('file'); - - if (!uploaded) return jsonError('No file provided', 400); + const uploaded = form.get("file"); + if (!uploaded) return jsonError("No file provided", 400); if (uploaded.size > 500 * 1024 * 1024) { - return jsonError('File exceeds 500MB', 400); + return jsonError("File exceeds 500MB", 400); } file = await uploaded.arrayBuffer(); fileName = uploaded.name; - fileType = uploaded.type || 'application/octet-stream'; + fileType = uploaded.type || "application/octet-stream"; } - /* ========================== - GET UPLOAD METADATA - ========================== */ + /* ========================= */ + /* GET SESSION */ + /* ========================= */ const ping = await fetch(PING_URL, { - method: 'GET', - headers: userAgentHeaders(), + method: "GET", + headers: userAgentHeaders() }); - const cookieHeader = ping.headers.get('set-cookie') || ''; - const sessionCookie = cookieHeader.split(';').find((c) => c.trim().startsWith('_s=')) || ''; + const setCookie = ping.headers.get("set-cookie") || ""; + const sessionCookie = + setCookie.split(";").find(c => c.trim().startsWith("_s=")) || ""; + + /* ========================= */ + /* GET METADATA */ + /* ========================= */ const metadataResp = await fetch(API_BASE, { - method: 'POST', + method: "POST", headers: { - ...userAgentHeaders(), - 'Content-Type': 'application/json', - Cookie: sessionCookie, + "Content-Type": "application/json", + "Cookie": sessionCookie, + ...userAgentHeaders() }, body: JSON.stringify({ files: [ { fileName, fileType, - fileSize: file.byteLength, - }, - ], - }), + fileSize: file.byteLength + } + ] + }) }); const metadataText = await metadataResp.text(); @@ -94,19 +92,19 @@ export async function onRequest(context) { const fileInfo = metadata.files?.[0]; if (!fileInfo || !fileInfo.success) { - throw new Error('Invalid metadata response'); + throw new Error("Invalid metadata response"); } - /* ========================== - HANDLE UPLOAD TYPE - ========================== */ + /* ========================= */ + /* HANDLE UPLOAD */ + /* ========================= */ - let finalData; + let result; if (fileInfo.isMultipart) { - finalData = await handleMultipart(fileInfo, file); + result = await handleMultipart(fileInfo, file, sessionCookie); } else { - finalData = await handleSingle(fileInfo.uploadUrl, file, fileType); + result = await handleSingle(fileInfo.uploadUrl, file, fileType); } const publicUrl = `https://i.imgur.gg/${fileInfo.fileId}-${fileInfo.fileName}`; @@ -115,8 +113,9 @@ export async function onRequest(context) { success: true, url: publicUrl, fileId: fileInfo.fileId, - fileName: fileInfo.fileName, + fileName: fileInfo.fileName }); + } catch (err) { return jsonError(err.message, 500); } @@ -127,16 +126,14 @@ export async function onRequest(context) { /* ===================================================== */ async function handleSingle(uploadUrl, fileBuffer, fileType) { - if (!uploadUrl) { - throw new Error('Missing uploadUrl for single upload'); - } + if (!uploadUrl) throw new Error("Missing uploadUrl"); const res = await fetch(uploadUrl, { - method: 'PUT', + method: "PUT", body: fileBuffer, headers: { - 'Content-Type': fileType, - }, + "Content-Type": fileType + } }); if (!res.ok) { @@ -151,11 +148,11 @@ async function handleSingle(uploadUrl, fileBuffer, fileType) { /* ================= MULTIPART UPLOAD =================== */ /* ===================================================== */ -async function handleMultipart(fileInfo, fileBuffer) { +async function handleMultipart(fileInfo, fileBuffer, sessionCookie) { const { partUrls, partSize, uploadId, fileId } = fileInfo; if (!partUrls || !uploadId) { - throw new Error('Invalid multipart metadata'); + throw new Error("Invalid multipart metadata"); } const parts = []; @@ -166,8 +163,8 @@ async function handleMultipart(fileInfo, fileBuffer) { const chunk = fileBuffer.slice(start, end); const uploadRes = await fetch(partUrls[i].url, { - method: 'PUT', - body: chunk, + method: "PUT", + body: chunk }); if (!uploadRes.ok) { @@ -175,55 +172,59 @@ async function handleMultipart(fileInfo, fileBuffer) { throw new Error(`Multipart part ${i + 1} failed: ${txt}`); } - let etag = uploadRes.headers.get('etag') || ''; - etag = etag.replace(/"/g, ''); // clean quotes + let etag = uploadRes.headers.get("etag") || ""; + etag = etag.replace(/"/g, ""); parts.push({ PartNumber: i + 1, - ETag: `"${etag}"`, + ETag: `"${etag}"` }); } - /* ===== FINALIZE MULTIPART ===== */ + /* ========================= */ + /* FINALIZE UPLOAD */ + /* ========================= */ - const complete = await fetch(COMPLETE_URL, { - method: 'PUT', + const completeResp = await fetch(COMPLETE_URL, { + method: "PUT", headers: { - 'Content-Type': 'application/json', + "Content-Type": "application/json", + "Cookie": sessionCookie, + ...userAgentHeaders() }, body: JSON.stringify({ fileId, uploadId, - parts, - }), + parts + }) }); - const completeText = await complete.text(); + const completeText = await completeResp.text(); - if (!complete.ok) { + if (!completeResp.ok) { throw new Error(`Multipart complete failed: ${completeText}`); } const completeData = JSON.parse(completeText); if (!completeData.success) { - throw new Error('Multipart finalize returned failure'); + throw new Error("Multipart finalize returned failure"); } return completeData; } /* ===================================================== */ -/* ================= UTIL FUNCTIONS ===================== */ +/* ================= UTILITIES ========================= */ /* ===================================================== */ function jsonResponse(obj, status = 200) { return new Response(JSON.stringify(obj), { status, headers: { - 'Content-Type': 'application/json', - ...corsHeaders(), - }, + "Content-Type": "application/json", + ...corsHeaders() + } }); } @@ -233,14 +234,15 @@ function jsonError(message, status) { function corsHeaders() { return { - 'Access-Control-Allow-Origin': '*', - 'Access-Control-Allow-Methods': 'POST, OPTIONS', - 'Access-Control-Allow-Headers': '*', + "Access-Control-Allow-Origin": "*", + "Access-Control-Allow-Methods": "POST, OPTIONS", + "Access-Control-Allow-Headers": "*" }; } function userAgentHeaders() { return { - 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/120.0.0.0', + "User-Agent": + "Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/120.0.0.0" }; }