File size: 1,788 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 | "use server";
import { NextResponse } from "next/server";
import { getMitmAlias, setMitmAliasAll } from "@/models";
import { getMitmStatus } from "@/mitm/manager";
import { writeAliasForTool } from "@/lib/mitmAliasCache";
// GET - Get MITM aliases for a tool
export async function GET(request) {
try {
const { searchParams } = new URL(request.url);
const toolName = searchParams.get("tool");
const aliases = await getMitmAlias(toolName || undefined);
return NextResponse.json({ aliases });
} catch (error) {
console.log("Error fetching MITM aliases:", error.message);
return NextResponse.json({ error: "Failed to fetch aliases" }, { status: 500 });
}
}
// PUT - Save MITM aliases for a specific tool
export async function PUT(request) {
try {
const { tool, mappings } = await request.json();
if (!tool || !mappings || typeof mappings !== "object") {
return NextResponse.json({ error: "tool and mappings required" }, { status: 400 });
}
// Check if DNS is enabled for this tool
const status = await getMitmStatus();
if (!status.dnsStatus || !status.dnsStatus[tool]) {
return NextResponse.json(
{ error: `DNS must be enabled for ${tool} before editing model mappings` },
{ status: 403 }
);
}
const filtered = {};
for (const [alias, model] of Object.entries(mappings)) {
if (model && model.trim()) {
filtered[alias] = model.trim();
}
}
await setMitmAliasAll(tool, filtered);
writeAliasForTool(tool, filtered);
return NextResponse.json({ success: true, aliases: filtered });
} catch (error) {
console.log("Error saving MITM aliases:", error.message);
return NextResponse.json({ error: "Failed to save aliases" }, { status: 500 });
}
}
|