File size: 7,757 Bytes
6111b2b | 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 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 | "use server";
import { NextResponse } from "next/server";
import fs from "fs/promises";
import path from "path";
import { requireCliToolsAuth } from "@/lib/api/requireCliToolsAuth";
import {
ensureCliConfigWriteAllowed,
getCliPrimaryConfigPath,
getCliRuntimeStatus,
} from "@/shared/services/cliRuntime";
import { createBackup } from "@/shared/services/backupService";
import { saveCliToolLastConfigured, deleteCliToolLastConfigured } from "@/lib/db/cliToolState";
import { cliModelConfigSchema } from "@/shared/validation/schemas";
import { isValidationFailure, validateBody } from "@/shared/validation/helpers";
import { resolveApiKey } from "@/shared/services/apiKeyResolver";
const getDroidSettingsPath = () => getCliPrimaryConfigPath("droid");
const getDroidDir = () => path.dirname(getDroidSettingsPath());
// Read current settings.json
const readSettings = async () => {
try {
const settingsPath = getDroidSettingsPath();
const content = await fs.readFile(settingsPath, "utf-8");
return JSON.parse(content);
} catch (error: any) {
if (error.code === "ENOENT") return null;
throw error;
}
};
// Check if settings has OmniRoute customModels
const hasOmniRouteConfig = (settings: any) => {
if (!settings || !settings.customModels) return false;
return settings.customModels.some((m) => m.id === "custom:OmniRoute-0");
};
// GET - Check droid CLI and read current settings
export async function GET(request: Request) {
const authError = await requireCliToolsAuth(request);
if (authError) return authError;
try {
const runtime = await getCliRuntimeStatus("droid");
if (!runtime.installed || !runtime.runnable) {
return NextResponse.json({
installed: runtime.installed,
runnable: runtime.runnable,
command: runtime.command,
commandPath: runtime.commandPath,
runtimeMode: runtime.runtimeMode,
reason: runtime.reason,
settings: null,
message:
runtime.installed && !runtime.runnable
? "Factory Droid CLI is installed but not runnable"
: "Factory Droid CLI is not installed",
});
}
const settings = await readSettings();
return NextResponse.json({
installed: runtime.installed,
runnable: runtime.runnable,
command: runtime.command,
commandPath: runtime.commandPath,
runtimeMode: runtime.runtimeMode,
reason: runtime.reason,
settings,
hasOmniRoute: hasOmniRouteConfig(settings),
settingsPath: getDroidSettingsPath(),
});
} catch (error) {
console.log("Error checking droid settings:", error);
return NextResponse.json({ error: "Failed to check droid settings" }, { status: 500 });
}
}
// POST - Update OmniRoute customModels (merge with existing settings)
export async function POST(request: Request) {
const authError = await requireCliToolsAuth(request);
if (authError) return authError;
let rawBody;
try {
rawBody = await request.json();
} catch {
return NextResponse.json(
{
error: {
message: "Invalid request",
details: [{ field: "body", message: "Invalid JSON body" }],
},
},
{ status: 400 }
);
}
try {
const writeGuard = ensureCliConfigWriteAllowed();
if (writeGuard) {
return NextResponse.json({ error: writeGuard }, { status: 403 });
}
// (#549) Extract keyId BEFORE validation — Zod strips unknown fields!
const keyId = typeof rawBody?.keyId === "string" ? rawBody.keyId.trim() : null;
const validation = validateBody(cliModelConfigSchema, rawBody);
if (isValidationFailure(validation)) {
return NextResponse.json({ error: validation.error }, { status: 400 });
}
const { baseUrl, model } = validation.data;
const apiKey = await resolveApiKey(keyId, validation.data.apiKey);
const droidDir = getDroidDir();
const settingsPath = getDroidSettingsPath();
// Ensure directory exists
await fs.mkdir(droidDir, { recursive: true });
// Backup current settings before modifying
await createBackup("droid", settingsPath);
// Read existing settings or create new
let settings: Record<string, any> = {};
try {
const existingSettings = await fs.readFile(settingsPath, "utf-8");
settings = JSON.parse(existingSettings);
} catch {
/* No existing settings */
}
// Ensure customModels array exists
if (!settings.customModels) {
settings.customModels = [];
}
// Remove existing OmniRoute config if any
settings.customModels = settings.customModels.filter((m) => m.id !== "custom:OmniRoute-0");
// Normalize baseUrl to ensure /v1 suffix
const normalizedBaseUrl = baseUrl.endsWith("/v1") ? baseUrl : `${baseUrl}/v1`;
// Add new OmniRoute config
const customModel = {
model: model,
id: "custom:OmniRoute-0",
index: 0,
baseUrl: normalizedBaseUrl,
apiKey: apiKey || "your_api_key",
displayName: model,
maxOutputTokens: 131072,
noImageSupport: false,
provider: "openai",
};
settings.customModels.unshift(customModel);
// Write settings
await fs.writeFile(settingsPath, JSON.stringify(settings, null, 2));
// Persist last-configured timestamp
try {
saveCliToolLastConfigured("droid");
} catch {
/* non-critical */
}
return NextResponse.json({
success: true,
message: "Factory Droid settings applied successfully!",
settingsPath,
});
} catch (error) {
console.log("Error updating droid settings:", error);
return NextResponse.json({ error: "Failed to update droid settings" }, { status: 500 });
}
}
// DELETE - Remove OmniRoute customModels only (keep other settings)
export async function DELETE(request: Request) {
const authError = await requireCliToolsAuth(request);
if (authError) return authError;
try {
const writeGuard = ensureCliConfigWriteAllowed();
if (writeGuard) {
return NextResponse.json({ error: writeGuard }, { status: 403 });
}
const settingsPath = getDroidSettingsPath();
// Backup current settings before resetting
await createBackup("droid", settingsPath);
// Read existing settings
let settings: Record<string, any> = {};
try {
const existingSettings = await fs.readFile(settingsPath, "utf-8");
settings = JSON.parse(existingSettings);
} catch (error: any) {
if (error.code === "ENOENT") {
return NextResponse.json({
success: true,
message: "No settings file to reset",
});
}
throw error;
}
// Remove OmniRoute customModels
if (settings.customModels) {
settings.customModels = settings.customModels.filter((m) => m.id !== "custom:OmniRoute-0");
// Remove customModels array if empty
if (settings.customModels.length === 0) {
delete settings.customModels;
}
}
// Write updated settings
await fs.writeFile(settingsPath, JSON.stringify(settings, null, 2));
// Clear last-configured timestamp
try {
deleteCliToolLastConfigured("droid");
} catch {
/* non-critical */
}
return NextResponse.json({
success: true,
message: "OmniRoute settings removed successfully",
});
} catch (error) {
console.log("Error resetting droid settings:", error);
return NextResponse.json({ error: "Failed to reset droid settings" }, { status: 500 });
}
}
|