Save and load scripts

This commit is contained in:
2025-11-03 16:55:46 -08:00
parent 684ae0ecf8
commit 30649488bb
7 changed files with 412 additions and 5 deletions

View File

@@ -0,0 +1,54 @@
import { NextRequest, NextResponse } from "next/server";
import { readFile } from "fs/promises";
import { join } from "path";
import { existsSync } from "fs";
import { SAVE_CONFIG } from "@/lib/saveConfig";
export async function GET(request: NextRequest) {
try {
const { searchParams } = new URL(request.url);
const name = searchParams.get("name");
if (!name) {
return NextResponse.json(
{ error: "Script name is required" },
{ status: 400 }
);
}
// Validate filename (same validation as save)
if (!SAVE_CONFIG.FILENAME_REGEX.test(name)) {
return NextResponse.json(
{ error: "Invalid script name" },
{ status: 400 }
);
}
const saveDir = join(process.cwd(), SAVE_CONFIG.SAVE_DIRECTORY);
const filename = `${name}.rhai`;
const filepath = join(saveDir, filename);
// Check if file exists
if (!existsSync(filepath)) {
return NextResponse.json(
{ error: `Script "${name}" not found` },
{ status: 404 }
);
}
// Read and return file content
const content = await readFile(filepath, "utf8");
return NextResponse.json({
name,
filename,
content,
});
} catch (error) {
console.error("Get script error:", error);
return NextResponse.json(
{ error: "Failed to read script" },
{ status: 500 }
);
}
}

View File

@@ -0,0 +1,31 @@
import { NextResponse } from "next/server";
import { readdir } from "fs/promises";
import { join } from "path";
import { existsSync } from "fs";
import { SAVE_CONFIG } from "@/lib/saveConfig";
export async function GET() {
try {
const saveDir = join(process.cwd(), SAVE_CONFIG.SAVE_DIRECTORY);
// If save directory doesn't exist, return empty array
if (!existsSync(saveDir)) {
return NextResponse.json({ scripts: [] });
}
// Read directory and filter for .rhai files
const files = await readdir(saveDir);
const scripts = files
.filter((file) => file.endsWith(".rhai"))
.map((file) => file.replace(".rhai", ""))
.sort(); // Sort alphabetically
return NextResponse.json({ scripts });
} catch (error) {
console.error("List scripts error:", error);
return NextResponse.json(
{ error: "Failed to list scripts" },
{ status: 500 }
);
}
}

View File

@@ -0,0 +1,86 @@
import { NextRequest, NextResponse } from "next/server";
import { writeFile, mkdir } from "fs/promises";
import { join } from "path";
import { existsSync } from "fs";
import { SAVE_CONFIG } from "@/lib/saveConfig";
export async function POST(request: NextRequest) {
try {
// Check if saving is enabled
if (!SAVE_CONFIG.ENABLE_SAVE) {
return NextResponse.json(
{ error: "Script saving is disabled" },
{ status: 403 }
);
}
const { name, content, secret } = await request.json();
// Validate secret
if (secret !== SAVE_CONFIG.SAVE_SECRET) {
return NextResponse.json(
{ error: "Invalid save secret" },
{ status: 401 }
);
}
// Validate required fields
if (!name || !content) {
return NextResponse.json(
{ error: "Name and content are required" },
{ status: 400 }
);
}
// Validate filename
if (name.length > SAVE_CONFIG.MAX_FILENAME_LENGTH) {
return NextResponse.json(
{
error: `Filename must be ${SAVE_CONFIG.MAX_FILENAME_LENGTH} characters or less`,
},
{ status: 400 }
);
}
if (!SAVE_CONFIG.FILENAME_REGEX.test(name)) {
return NextResponse.json(
{
error: "Filename can only contain alphanumerics, underscores, spaces, and hyphens",
},
{ status: 400 }
);
}
// Ensure save directory exists
const saveDir = join(process.cwd(), SAVE_CONFIG.SAVE_DIRECTORY);
if (!existsSync(saveDir)) {
await mkdir(saveDir, { recursive: true });
}
// Check if file already exists
const filename = `${name}.rhai`;
const filepath = join(saveDir, filename);
if (existsSync(filepath)) {
return NextResponse.json(
{ error: `A script named "${name}" already exists` },
{ status: 409 }
);
}
// Save the file
await writeFile(filepath, content, "utf8");
return NextResponse.json({
success: true,
message: `Script saved as ${filename}`,
filename,
});
} catch (error) {
console.error("Save script error:", error);
return NextResponse.json(
{ error: "Failed to save script" },
{ status: 500 }
);
}
}

View File

@@ -73,6 +73,13 @@ export default function Playground() {
const [bulkRounds, setBulkRounds] = useState(1000);
const [selectedAgent, setSelectedAgent] = useState("Random");
const [isHelpOpen, setIsHelpOpen] = useState(false);
const [scriptName, setScriptName] = useState("");
const [saveSecret, setSaveSecret] = useState("");
const [isSaving, setIsSaving] = useState(false);
const [availableAgents, setAvailableAgents] = useState<string[]>([]);
const [savedScripts, setSavedScripts] = useState<Record<string, string>>(
{}
);
const editorRef = useRef<any>(null);
const resultRef = useRef<HTMLTextAreaElement>(null);
@@ -81,6 +88,65 @@ export default function Playground() {
const runDisabled = isScriptRunning || !isEditorReady;
const stopDisabled = !isScriptRunning;
// Fetch saved scripts and update available agents
const loadSavedScripts = useCallback(async () => {
try {
const response = await fetch("/api/list-scripts");
const data = await response.json();
if (response.ok) {
const scripts = data.scripts || [];
const scriptContents: Record<string, string> = {};
// Fetch content for each saved script
await Promise.all(
scripts.map(async (scriptName: string) => {
try {
const scriptResponse = await fetch(
`/api/get-script?name=${encodeURIComponent(
scriptName
)}`
);
const scriptData = await scriptResponse.json();
if (scriptResponse.ok) {
scriptContents[scriptName] = scriptData.content;
}
} catch (error) {
console.error(
`Failed to load script ${scriptName}:`,
error
);
}
})
);
setSavedScripts(scriptContents);
// Combine hardcoded agents with saved scripts, ensuring Self and Random are first
const combinedAgents = [
"Self",
"Random",
...Object.keys(AGENTS).filter(
(key) => key !== "Self" && key !== "Random"
),
...scripts,
];
setAvailableAgents(combinedAgents);
}
} catch (error) {
console.error("Failed to load saved scripts:", error);
// Fallback to hardcoded agents only
setAvailableAgents(Object.keys(AGENTS));
}
}, []);
// Load saved scripts on component mount
useEffect(() => {
loadSavedScripts();
}, [loadSavedScripts]);
const runHuman = useCallback(async () => {
if (resultRef.current) {
resultRef.current.value = "";
@@ -141,11 +207,15 @@ export default function Playground() {
terminalRef.current?.clear();
terminalRef.current?.focus();
const agentCode = AGENTS[selectedAgent as keyof typeof AGENTS];
// Get script content from either hardcoded agents or saved scripts
const hardcodedAgent = AGENTS[selectedAgent as keyof typeof AGENTS];
const savedScript = savedScripts[selectedAgent];
const agentScript = hardcodedAgent || savedScript;
const blueScript =
agentCode || (editorRef.current?.getValue() ?? "");
agentScript || (editorRef.current?.getValue() ?? "");
const redScript = editorRef.current?.getValue() ?? "";
const opponentName = agentCode ? selectedAgent : "script";
const opponentName = agentScript ? selectedAgent : "script";
await startScriptBulk(
redScript,
@@ -182,12 +252,87 @@ export default function Playground() {
}
setIsScriptRunning(false);
}, [runDisabled, bulkRounds, selectedAgent]);
}, [runDisabled, bulkRounds, selectedAgent, savedScripts]);
const stopScriptHandler = useCallback(() => {
stopScript();
}, []);
const saveScript = useCallback(async () => {
if (!scriptName.trim()) {
alert("Please enter a script name");
return;
}
if (!saveSecret.trim()) {
alert("Please enter the save secret");
return;
}
if (!editorRef.current) {
alert("No script content to save");
return;
}
setIsSaving(true);
try {
const response = await fetch("/api/save-script", {
method: "POST",
headers: {
"Content-Type": "application/json",
},
body: JSON.stringify({
name: scriptName.trim(),
content: editorRef.current.getValue(),
secret: saveSecret.trim(),
}),
});
const result = await response.json();
if (response.ok) {
alert(result.message);
setScriptName("");
// Reload saved scripts to include the new one
loadSavedScripts();
} else {
alert(result.error || "Failed to save script");
}
} catch (error) {
console.error("Save error:", error);
alert("Network error: Failed to save script");
}
setIsSaving(false);
}, [scriptName, saveSecret, loadSavedScripts]);
const copyScriptToEditor = useCallback(() => {
if (!selectedAgent || selectedAgent === "Self") {
alert("Please select a script to copy to the editor");
return;
}
// Get the script content
const hardcodedAgent = AGENTS[selectedAgent as keyof typeof AGENTS];
const savedScript = savedScripts[selectedAgent];
const scriptContent = hardcodedAgent || savedScript;
if (!scriptContent) {
alert("No script content available for the selected agent");
return;
}
// Warn user about losing current content
const confirmed = confirm(
`This will replace your current script with "${selectedAgent}". Your current work will be lost. Continue?`
);
if (confirmed && editorRef.current) {
editorRef.current.setValue(scriptContent);
}
}, [selectedAgent, savedScripts]);
return (
<div className={styles.playgroundRoot}>
<header className={styles.header}>
@@ -250,12 +395,69 @@ export default function Playground() {
<div className={styles.configField}>
<label>Bulk opponent</label>
<AgentSelector
agents={Object.keys(AGENTS)}
agents={availableAgents}
selectedAgent={selectedAgent}
onSelect={setSelectedAgent}
placeholder="Select an agent..."
/>
</div>
<div className={styles.configField}>
<Button
variant="info"
onClick={copyScriptToEditor}
disabled={
!selectedAgent ||
selectedAgent === "Self"
}
>
Copy to Editor
</Button>
</div>
<div className={styles.saveSection}>
<div className={styles.configField}>
<label>Script name</label>
<input
type="text"
className={styles.saveInput}
placeholder="Enter script name..."
value={scriptName}
onChange={(e) =>
setScriptName(
e.target.value
)
}
maxLength={32}
/>
</div>
<div className={styles.configField}>
<label>Save secret</label>
<input
type="password"
className={styles.saveInput}
placeholder="Enter save secret..."
value={saveSecret}
onChange={(e) =>
setSaveSecret(
e.target.value
)
}
/>
</div>
<div className={styles.configField}>
<Button
variant="primary"
onClick={saveScript}
loading={isSaving}
disabled={isSaving}
>
Save Script
</Button>
</div>
</div>
</div>
}
/>

View File

@@ -0,0 +1,8 @@
export const SAVE_CONFIG = {
ENABLE_SAVE: true,
SAVE_SECRET: "save",
SAVE_DIRECTORY: "./data/scripts",
MAX_FILENAME_LENGTH: 32,
FILENAME_REGEX: /^[a-zA-Z0-9_\s-]+$/,
} as const;

View File

@@ -117,6 +117,31 @@
margin-bottom: 6px;
}
.saveInput {
width: 100%;
padding: 6px 8px;
background: #3c3c3c;
color: #e0e0e0;
border: 1px solid #555;
border-radius: 3px;
font-size: 13px;
outline: none;
}
.saveInput:focus {
border-color: #007acc;
box-shadow: 0 0 3px rgba(0, 122, 204, 0.3);
}
.saveInput::placeholder {
color: #999;
}
.saveSection {
margin-top: 24px;
padding-top: 16px;
border-top: 1px solid #444;
}
.helpPanel {
padding: 16px;