import { NextResponse } from 'next/server' import type { NextRequest } from 'next/server' import { isIpAllowed, getClientIp, isValidIp, isValidIPv6 } from '@/lib/ip-utils' export function middleware(request: NextRequest) { // 檢查是否啟用IP白名單 const enableIpWhitelist = process.env.ENABLE_IP_WHITELIST === 'true' if (!enableIpWhitelist) { return NextResponse.next() } // 獲取客戶端IP - 使用與API相同的邏輯 let clientIp = getClientIp(request) // 根據你的環境，優先使用 cf-connecting-ip (支持IPv4和IPv6) const cfConnectingIp = request.headers.get('cf-connecting-ip') if (cfConnectingIp && cfConnectingIp.trim() !== '') { const cleanCfIp = cfConnectingIp.trim() // 检查是否是有效的IPv4或IPv6地址 if (isValidIp(cleanCfIp) || isValidIPv6(cleanCfIp)) { clientIp = cleanCfIp console.log(`[Middleware] 使用 cf-connecting-ip: ${clientIp} ${isValidIPv6(clientIp) ? '(IPv6)' : '(IPv4)'}`) } } // 獲取允許的IP列表 const allowedIps = process.env.ALLOWED_IPS || '' // 調試信息 console.log(`[Middleware] 最終IP檢測: ${clientIp}, 路徑: ${request.nextUrl.pathname}`) console.log(`[Middleware] 白名單狀態: ${enableIpWhitelist}`) console.log(`[Middleware] 允許的IP: ${allowedIps}`) // 檢查IP是否被允許 if (!isIpAllowed(clientIp, allowedIps)) { // 記錄被拒絕的訪問 console.warn(`[Middleware] Access denied for IP: ${clientIp} - Path: ${request.nextUrl.pathname}`) console.warn(`[Middleware] 允許的IP列表: ${allowedIps}`) // 返回403禁止訪問頁面 return new NextResponse( ` 訪問被拒絕

🚫 訪問被拒絕

很抱歉，您的IP地址不在允許的訪問列表中。

您的IP地址：
${clientIp}

如果您認為這是一個錯誤，請聯繫系統管理員。

錯誤代碼：403 Forbidden
時間：${new Date().toLocaleString('zh-TW')}

`, { status: 403, headers: { 'Content-Type': 'text/html; charset=utf-8', }, } ) } // IP檢查通過，繼續處理請求 return NextResponse.next() } // 配置中間件適用的路徑 export const config = { matcher: [ /* * 匹配所有路徑，除了： * - api/ip (IP檢測API，允許訪問) * - api/ip-diagnostic (IP診斷API，允許訪問) * - _next/static (static files) * - _next/image (image optimization files) * - favicon.ico (favicon file) * - icon.png (icon file) */ '/((?!api/ip|api/ip-diagnostic|_next/static|_next/image|favicon.ico|icon.png).*)', ], }