Backend added table for shortener visitors

elysia/src/index.ts

@@ -35,20 +35,37 @@ app.post('/link', async ({ body }) => {
 	return { message: 'Success' }
 })
 
-app.get('/:shortenerCode', async ({ params: { shortenerCode }, set }) => {
-	const shortener = await db
-		.selectFrom('shortener')
-		.selectAll()
-		.where('code', '=', shortenerCode)
-		.execute()
+app.get(
+	'/:shortenerCode',
+	async ({ params: { shortenerCode }, set, request }) => {
+		const ip = request.headers.get('x-forwarded-for')
 
-	if (!shortener.length) {
-		set.redirect = '/invalid'
-		return
-	}
+		const geolocation = await (
+			await fetch(`https://api.ipbase.com/v2/info?ip=${ip}`)
+		).json()
 
-	set.redirect = shortener[0].link
-})
+		const shortener = await db
+			.selectFrom('shortener')
+			.selectAll()
+			.where('code', '=', shortenerCode)
+			.orderBy('created_at', 'desc')
+			.execute()
+
+		const visitor_data = {
+			shortener_id: shortener[0].id,
+			country: geolocation.data.location.country.name as string,
+		}
+
+		await db.insertInto('visitor').values(visitor_data).execute()
+
+		if (!shortener.length) {
+			set.redirect = '/invalid'
+			return
+		}
+
+		set.redirect = shortener[0].link
+	}
+)
 
 app.listen(3000)
 

elysia/src/types.ts

@@ -1,15 +1,33 @@
-import { Generated, Insertable, Selectable, Updateable } from 'kysely'
+import {
+	ColumnType,
+	Generated,
+	Insertable,
+	Selectable,
+	Updateable,
+} from 'kysely'
 
 export interface Database {
 	shortener: ShortenerTable
+	visitor: VisitorTable
 }
 
 export interface ShortenerTable {
 	id: Generated<number>
 	link: string
 	code: string
+	created_at: ColumnType<Date, string | undefined, never>
 }
 
 export type Shortener = Selectable<ShortenerTable>
 export type NewShortener = Insertable<ShortenerTable>
 export type ShortenerUpdate = Updateable<ShortenerTable>
+
+export interface VisitorTable {
+	id: Generated<number>
+	shortener_id: number
+	country: string
+	created_at: ColumnType<Date, string | undefined, never>
+}
+
+export type Visitor = Selectable<VisitorTable>
+export type NewVisitor = Insertable<VisitorTable>