middleware.ts

import { geolocation, next } from '@vercel/edge';

// List of blocked countries (ISO 3166-1 alpha-2 country codes)
// North Korea (KP), Iran (IR), Syria (SY), Cuba (CU), Crimea (UA-43), Luhansk (UA-09), Donetsk (UA-14)
const BLOCKED_COUNTRIES = [
    'KP',
    'IR',
    'SY',
    'CU',
    'UA-43',
    'UA-09',
    'UA-14',
    'UNKNOWN',
];

export const config = {
    matcher: ['/((?!api|_next/static|_next/image|favicon.ico).*)'],
};

export default function middleware(request: Request) {
    const url = new URL(request.url);
    const { country } = geolocation(request);
    const countryCode = country || 'UNKNOWN';

    // Skip if the app is running locally
    if (url.hostname == 'localhost' || url.hostname == '127.0.0.1') {
        console.log('Running locally, bypassing country check');
        return next();
    }

    if (
        BLOCKED_COUNTRIES.includes(countryCode)
    ) {
        console.log(`Country ${countryCode} is blocked`)
        return new Response('AI agent app not available in your country', { status: 403 });
    }

    return next();
}