forked from CaliCastle/cali.so
-
Notifications
You must be signed in to change notification settings - Fork 0
/
middleware.ts
28 lines (22 loc) · 695 Bytes
/
middleware.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import kv from '@vercel/kv'
import { type NextRequest, NextResponse } from 'next/server'
import { kvKeys } from '~/config/kv'
import { env } from '~/env.mjs'
import countries from '~/lib/countries.json'
export const config = {
matcher: '/',
}
export async function middleware(req: NextRequest) {
const { geo } = req
if (geo && env.VERCEL_ENV === 'production') {
const country = geo.country
const city = geo.city
const countryInfo = countries.find((x) => x.cca2 === country)
if (!countryInfo) {
return NextResponse.next()
}
const flag = countryInfo.flag
await kv.set(kvKeys.currentVisitor, { country, city, flag })
}
return NextResponse.next()
}