diff --git a/functions/api/links/update.js b/functions/api/links/update.js index 2475fb7..ecef833 100644 --- a/functions/api/links/update.js +++ b/functions/api/links/update.js @@ -4,7 +4,7 @@ export async function onRequestPost({ request, env }) { const vR = await fetch("https://challenges.cloudflare.com/turnstile/v0/siteverify", { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify({ secret: env.TURNSTILE_KEY, response: token }) }); if (!(await vR.json()).success) return new Response("CAPTCHA verification failed.", { status: 403 }); - const { slug, destination_url, username, pass_hash } = body; + const { slug, destination_url, analytics_enabled, username, pass_hash } = body; if (!slug || !destination_url || !username || !pass_hash) return new Response("Missing fields", { status: 400 }); const user = await env.D1_EV.prepare("SELECT pass_hash, custom_slugs FROM users WHERE username = ?").bind(username).first(); @@ -17,7 +17,10 @@ export async function onRequestPost({ request, env }) { let url = destination_url.startsWith("http") ? destination_url : `https://${destination_url}`; try { new URL(url) } catch { return new Response("Invalid destination URL", { status: 400 }) } - await env.KV_EV.put(slug, url.replace(/^https?:\/\//, "")); + const dest_no_proto = url.replace(/^https?:\/\//, ""); + const kvValue = analytics_enabled ? `✺${dest_no_proto}` : dest_no_proto; + + await env.KV_EV.put(slug, kvValue); return Response.json({ success: true }); } catch (e) { return new Response(e.message, { status: 500 });