vndangkhoa/kv-music
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

session cookies for chunked uploads

Browse source
This commit is contained in:
Eduard Prigoana 2026-02-21 19:23:18 +02:00 committed by GitHub
parent bb427facf4
commit 19b5cc5b1d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 82 additions and 80 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

162
functions/upload.js
View file

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