Theodore Song
65 lines
2.6 KiB
JavaScript
65 lines
2.6 KiB
JavaScript
import { get, put } from "@vercel/blob";
|
|
|
|
const STATE_PATH = process.env.PMA_STATE_PATH || "shared/state.json";
|
|
const AGENTS_KEY = "pma_agents_v2";
|
|
|
|
async function readJsonBlob() {
|
|
const blob = await get(STATE_PATH, { access: "private" });
|
|
if (!blob || blob.statusCode !== 200 || !blob.stream) return null;
|
|
const text = await new Response(blob.stream).text();
|
|
return text ? JSON.parse(text) : null;
|
|
}
|
|
|
|
function agentStateFromItems(items) {
|
|
if (!items || !items[AGENTS_KEY]) return null;
|
|
try {
|
|
return JSON.parse(items[AGENTS_KEY]);
|
|
} catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
export default async function handler(req, res) {
|
|
res.setHeader("Cache-Control", "no-store");
|
|
try {
|
|
if (!process.env.BLOB_READ_WRITE_TOKEN && !process.env.VERCEL_OIDC_TOKEN) {
|
|
return res.status(503).json({ ok: false, error: "Cloud state is not configured" });
|
|
}
|
|
|
|
if (req.method === "GET") {
|
|
return res.status(200).json({ ok: true, state: await readJsonBlob() });
|
|
}
|
|
|
|
if (req.method === "POST") {
|
|
const body = typeof req.body === "string" ? JSON.parse(req.body || "{}") : (req.body || {});
|
|
if (!body || typeof body !== "object" || !body.items || typeof body.items !== "object") {
|
|
return res.status(400).json({ ok: false, error: "Invalid state payload" });
|
|
}
|
|
const current = await readJsonBlob();
|
|
const currentAgents = agentStateFromItems(current && current.items);
|
|
const incomingAgents = agentStateFromItems(body.items);
|
|
if (!body.force && currentAgents && incomingAgents) {
|
|
const sameHour = currentAgents.last_cycle_hour && currentAgents.last_cycle_hour === incomingAgents.last_cycle_hour;
|
|
const differentRun = currentAgents.last_run && incomingAgents.last_run && currentAgents.last_run !== incomingAgents.last_run;
|
|
if (sameHour && differentRun) {
|
|
return res.status(409).json({ ok: false, error: "This cycle hour already has a cloud result", state: current });
|
|
}
|
|
}
|
|
const state = { version: 1, updated_at: new Date().toISOString(), items: body.items };
|
|
await put(STATE_PATH, JSON.stringify(state), {
|
|
access: "private",
|
|
allowOverwrite: true,
|
|
contentType: "application/json",
|
|
cacheControlMaxAge: 60,
|
|
});
|
|
return res.status(200).json({ ok: true, state });
|
|
}
|
|
|
|
res.setHeader("Allow", "GET, POST");
|
|
return res.status(405).json({ ok: false, error: "Method not allowed" });
|
|
} catch (err) {
|
|
const message = err && err.message ? err.message : "State sync failed";
|
|
return res.status(500).json({ ok: false, error: message });
|
|
}
|
|
}
|