Add Link Aliases feature

closes #495
dubinc · Jun 25, 2024 · 39954b9 · 39954b9
1 parent 0e0a62e
commit 39954b9
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 2 deletions.
diff --git a/apps/web/lib/middleware/link.ts b/apps/web/lib/middleware/link.ts
@@ -103,6 +103,7 @@ export default async function LinkMiddleware(
     geo,
     expiredUrl,
     doIndex,
+    parentLinkId,
   } = link;
 
   // by default, we only index default dub domain links (e.g. dub.sh)
@@ -190,6 +191,7 @@ export default async function LinkMiddleware(
       recordClick({
         req,
         linkId,
+        parentLinkId,
         clickId,
         ...(url && { url: getFinalUrl(url, { req }) }),
       }),

diff --git a/apps/web/lib/tinybird/record-click.ts b/apps/web/lib/tinybird/record-click.ts
@@ -18,11 +18,13 @@ import { ratelimit } from "../upstash";
 export async function recordClick({
   req,
   linkId,
+  parentLinkId,
   clickId,
   url,
 }: {
   req: NextRequest;
   linkId: string;
+  parentLinkId?: string;
   clickId?: string;
   url?: string;
 }) {
@@ -56,8 +58,13 @@ export async function recordClick({
           timestamp: new Date(Date.now()).toISOString(),
           identity_hash,
           click_id: clickId || nanoid(16),
-          link_id: linkId,
-          alias_link_id: "",
+          /* 
+            if the link is a child alias of another link, 
+              log the click under the parent link ID
+            else log the click under the link ID
+          */
+          link_id: parentLinkId || linkId,
+          alias_link_id: parentLinkId ? linkId : "",
           url: url || "",
           ip:
             // only record IP if it's a valid IP and not from EU

diff --git a/apps/web/lib/types.ts b/apps/web/lib/types.ts
@@ -37,6 +37,7 @@ export interface RedisLinkProps {
   geo?: object;
   doIndex?: boolean;
   projectId?: string;
+  parentLinkId: string;
 }
 
 export interface EdgeLinkProps {

diff --git a/apps/web/prisma/schema.prisma b/apps/web/prisma/schema.prisma
@@ -268,6 +268,10 @@ model Link {
   // Comments on the particular shortlink
   comments String? @db.LongText
 
+  parentLink Link? @relation("LinkAliases", fields: [parentLinkId], references: [id], onDelete: NoAction, onUpdate: NoAction)
+  parentLinkId String?
+  childLinks Link[] @relation("LinkAliases")
+
   @@unique([domain, key])
   @@unique([projectId, externalId])
   @@index(projectId)
@@ -280,6 +284,7 @@ model Link {
   @@index(lastClicked)
   @@index(userId)
   @@fulltext([key, url])
+  @@index([parentLinkId])
 }
 
 model Tag {