mirror of https://github.com/TZGyn/shortener
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.5 KiB
TypeScript
62 lines
1.5 KiB
TypeScript
import { Elysia, t } from 'elysia'
|
|
import { nanoid } from 'nanoid'
|
|
import { db } from './database'
|
|
import { createLinkSchema } from './zodSchema'
|
|
import { cors } from '@elysiajs/cors'
|
|
import { jsonArrayFrom } from 'kysely/helpers/postgres'
|
|
import { login, signup } from './auth'
|
|
|
|
const fallback_url = Bun.env.FALLBACK_URL ?? 'https://shortener.tzgyn.com'
|
|
|
|
const app = new Elysia().use(cors())
|
|
|
|
app.get('/', () => 'Hello Elysia')
|
|
app.get('/invalid', () => 'Invalid Shortener')
|
|
|
|
app.get(
|
|
'/:shortenerCode',
|
|
async ({ params: { shortenerCode }, set, request }) => {
|
|
const ip = request.headers.get('x-forwarded-for')
|
|
|
|
const geolocation = await (
|
|
await fetch(`https://api.ipbase.com/v2/info?ip=${ip}`)
|
|
).json()
|
|
|
|
try {
|
|
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,
|
|
country_code: geolocation.data.location.country
|
|
.alpha2 as string,
|
|
}
|
|
|
|
await db.insertInto('visitor').values(visitor_data).execute()
|
|
|
|
if (!shortener.length) {
|
|
set.redirect = '/invalid'
|
|
return
|
|
}
|
|
|
|
set.redirect = shortener[0].link
|
|
} catch {
|
|
set.redirect = fallback_url
|
|
}
|
|
}
|
|
)
|
|
|
|
app.listen(Bun.env.PORT ?? 3000)
|
|
|
|
console.log(
|
|
`🦊 Elysia is running at ${app.server?.hostname}:${app.server?.port}`,
|
|
`${Bun.env.PGDATABASE}`
|
|
)
|
|
|
|
export type App = typeof app
|