File size: 2,903 Bytes
88c4c60 | 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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | import { NextResponse } from "next/server";
import { createProxyPool, getProviderConnections, getProxyPools } from "@/models";
function toBoolean(value) {
if (value === "true") return true;
if (value === "false") return false;
return undefined;
}
const VALID_PROXY_TYPES = ["http", "vercel", "cloudflare", "deno"];
function normalizeProxyPoolInput(body = {}) {
const name = typeof body?.name === "string" ? body.name.trim() : "";
const proxyUrl = typeof body?.proxyUrl === "string" ? body.proxyUrl.trim() : "";
const noProxy = typeof body?.noProxy === "string" ? body.noProxy.trim() : "";
const isActive = body?.isActive === undefined ? true : body.isActive === true;
const strictProxy = body?.strictProxy === true;
const type = VALID_PROXY_TYPES.includes(body?.type) ? body.type : "http";
if (!name) {
return { error: "Name is required" };
}
if (!proxyUrl) {
return { error: "Proxy URL is required" };
}
return { name, proxyUrl, noProxy, isActive, strictProxy, type };
}
function buildUsageMap(connections = []) {
const usageMap = new Map();
for (const connection of connections) {
const proxyPoolId = connection?.providerSpecificData?.proxyPoolId;
if (!proxyPoolId) continue;
usageMap.set(proxyPoolId, (usageMap.get(proxyPoolId) || 0) + 1);
}
return usageMap;
}
// GET /api/proxy-pools - List proxy pools
export async function GET(request) {
try {
const { searchParams } = new URL(request.url);
const isActive = toBoolean(searchParams.get("isActive"));
const includeUsage = searchParams.get("includeUsage") === "true";
const filter = {};
if (isActive !== undefined) {
filter.isActive = isActive;
}
const proxyPools = await getProxyPools(filter);
if (!includeUsage) {
return NextResponse.json({ proxyPools });
}
const connections = await getProviderConnections();
const usageMap = buildUsageMap(connections);
const enrichedProxyPools = proxyPools.map((pool) => ({
...pool,
boundConnectionCount: usageMap.get(pool.id) || 0,
}));
return NextResponse.json({ proxyPools: enrichedProxyPools });
} catch (error) {
console.log("Error fetching proxy pools:", error);
return NextResponse.json({ error: "Failed to fetch proxy pools" }, { status: 500 });
}
}
// POST /api/proxy-pools - Create proxy pool
export async function POST(request) {
try {
const body = await request.json();
const normalized = normalizeProxyPoolInput(body);
if (normalized.error) {
return NextResponse.json({ error: normalized.error }, { status: 400 });
}
const proxyPool = await createProxyPool(normalized);
return NextResponse.json({ proxyPool }, { status: 201 });
} catch (error) {
console.log("Error creating proxy pool:", error);
return NextResponse.json({ error: "Failed to create proxy pool" }, { status: 500 });
}
}
|