diff --git a/functions/[slug].js b/functions/[slug].js index 23d5311..08bc470 100644 --- a/functions/[slug].js +++ b/functions/[slug].js @@ -1,7 +1,18 @@ -export async function onRequestGet({ request, params, env, next }) { +export async function onRequestGet({ request, params, env, next, waitUntil }) { if (['abuse', 'admin', 'api', 'dash', 'acceptable-use'].includes(params.slug)) return next(); try { const dest = await env.KV_EV.get(params.slug); + if (dest) { + waitUntil((async () => { + let ref = 'direct'; + try { ref = new URL(request.headers.get('Referer')).hostname } catch {} + const stmt = ` + INSERT INTO analytics (slug, referrer, count) VALUES (?, ?, 1) + ON CONFLICT(slug, referrer) DO UPDATE SET count = count + 1; + `; + await env.D1_EV.prepare(stmt).bind(params.slug, ref).run().catch(console.error); + })()); + } const url = dest ? `https://${dest}` : new URL('/', request.url).href; return Response.redirect(url, dest ? 301 : 302); } catch (e) {