From c5d3db7993b5ca39b48d055a5b42a57113757839 Mon Sep 17 00:00:00 2001
From: Shav Kinderlehrer <shav@trinket.icu>
Date: Mon, 16 Oct 2023 14:08:23 -0400
Subject: Shorten hash + add '+' operator

---
 src/get.ts  | 11 ++++++++++-
 src/post.ts |  2 +-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/get.ts b/src/get.ts
index fb853a3..adba0fc 100644
--- a/src/get.ts
+++ b/src/get.ts
@@ -25,7 +25,14 @@ export async function get_id(
   context: { params: Record<"id", string>, headers: Record<string, string | null> },
   sql: Sql) {
 
-  const id = context.params.id;
+  let id = context.params.id;
+
+  let get_query = false;
+  if (id.endsWith("+")) {
+    id = id.slice(0, id.length - 1);
+    get_query = true;
+  }
+
   let db_res = await sql`
   SELECT * FROM urls
   WHERE id=${id}
@@ -33,6 +40,8 @@ export async function get_id(
 
   if (!db_res.length) {
     return new Response(`url for id '${id}' not found.`, { status: 404 });
+  } else if (get_query) {
+    return new Response(`trkt.in/${id} -> ${db_res[0]['url']}`);
   } else {
     await sql`
     INSERT INTO tracking (id,clicks)
diff --git a/src/post.ts b/src/post.ts
index 538ff00..94643d4 100644
--- a/src/post.ts
+++ b/src/post.ts
@@ -20,7 +20,7 @@ export async function post_c(body: any, sql: Sql) {
     custom_id = false;
     const hasher = new Bun.CryptoHasher("md5");
     hasher.update(form.url.href);
-    form.id = hasher.digest("hex").slice(0, 11);
+    form.id = hasher.digest("hex").slice(0, 4);
   }
 
   const valid_re = /^[A-z0-9_-]+$/;
-- 
cgit v1.2.3