osmWorkflow/scripts/road-workbench.js

#!/usr/bin/env node
"use strict";

const fs = require("fs");
const http = require("http");
const path = require("path");
const { execFileSync } = require("child_process");
const { readAreaConfig } = require("./lib/area-config");
const { loadOverrides, validateOverrides, writeJsonAtomic } = require("./lib/native-road");
const { generate, validateDocument, runtime } = require("./lib/native-traffic-signals");
const { compileArea, parseArgs } = require("./compile-native-roads");
const { convertGeoJson } = require("./lib/gaode-junction-reference");

const repoRoot = path.resolve(__dirname, "..");

function main() {
  const args = parseArgs(process.argv.slice(2));
  const configPath = path.resolve(args.config || path.join(repoRoot, "config", "areas", "nantaizi-lake-innovation-valley.json"));
  if (args.noCompile !== "true") compileArea(configPath);
  const area = readAreaConfig(configPath, { repoRoot });
  const junctionReference = args.junctionReference ? readJunctionReference(path.resolve(args.junctionReference)) : null;
  // `--debug` surfaces advisory compiler findings that have no geometry layer of
  // their own — currently the complex-junction candidates. Off by default so the
  // normal editing view stays uncluttered.
  const debug = args.debug === "true";
  const port = Number(args.port || 8787);
  if (!Number.isInteger(port) || port < 1024 || port > 65535) throw new Error("--port must be an integer in [1024, 65535].");
  const server = http.createServer((request, response) => handle(request, response, area, configPath, junctionReference, debug));
  server.on("error", (error) => {
    console.error(`Road Workbench failed to listen: ${error.message}`);
    process.exitCode = 1;
  });
  server.listen(port, "127.0.0.1", () => console.log(`Road Workbench: http://127.0.0.1:${port}/${debug ? "  (debug: 复杂路口候选已开启)" : ""}`));
}

function handle(request, response, area, configPath, junctionReference, debug = false) {
  const url = new URL(request.url, "http://127.0.0.1");
  if (request.method === "GET" && url.pathname === "/") return sendFile(response, path.join(repoRoot, "scripts", "workbench", "index.html"), "text/html; charset=utf-8");
  if (request.method === "GET" && url.pathname === "/app.js") return sendFile(response, path.join(repoRoot, "scripts", "workbench", "app.js"), "text/javascript; charset=utf-8");
  if (request.method === "GET" && url.pathname === "/app.css") return sendFile(response, path.join(repoRoot, "scripts", "workbench", "app.css"), "text/css; charset=utf-8");
  if (request.method === "GET" && url.pathname.startsWith("/vendor/")) return sendVendorFile(response, url.pathname);
  if (request.method === "GET" && url.pathname === "/api/state") return sendJson(response, 200, state(area, junctionReference, debug));
  if (request.method === "POST" && url.pathname === "/api/traffic-signals") return readBody(request).then((body) => {
    const document = validateDocument(body, fs.readFileSync(area.input, "utf8"));
    writeJsonAtomic(area.outputs.nativeTrafficSignals, document);
    sendJson(response, 200, { ok: true, trafficSignals: document, runtime: runtime(document) });
  }).catch((error) => sendJson(response, 400, { ok: false, error: error.message }));
  if (request.method === "POST" && url.pathname === "/api/traffic-signals/generate") return Promise.resolve().then(() => {
    const compiled = readCompiled(area);
    const generated = generate(fs.readFileSync(area.input, "utf8"), readLayer(path.join(area.outputs.nativeRoadDir, "layers", "vehicle_stop_lines.geojson")), readLayer(path.join(area.outputs.nativeRoadDir, "layers", "intersection_surface.geojson")));
    const current = validateDocument(readJson(area.outputs.nativeTrafficSignals), fs.readFileSync(area.input, "utf8"));
    const present = new Set(current.assemblies.features.map((feature) => feature.properties.signal_uid));
    current.assemblies.features.push(...generated.assemblies.features.filter((feature) => !present.has(feature.properties.signal_uid)));
    writeJsonAtomic(area.outputs.nativeTrafficSignals, current);
    sendJson(response, 200, { ok: true, trafficSignals: current, runtime: runtime(current), generated: generated.assemblies.features.length, compiled: Boolean(compiled) });
  }).catch((error) => sendJson(response, 400, { ok: false, error: error.message }));
  if (request.method === "POST" && url.pathname === "/api/overrides") return readBody(request).then((body) => {
    const compiled = readCompiled(area);
    const overrides = validateOverrides(body, { roads: compiled.model.roads, endpoints: compiled.model.endpoints });
    writeJsonAtomic(area.outputs.nativeRoadOverrides, overrides);
    sendJson(response, 200, { ok: true, overrides });
  }).catch((error) => sendJson(response, 400, { ok: false, error: error.message }));
  if (request.method === "POST" && url.pathname === "/api/junction-clusters") return readBody(request).then((body) => {
    if (!debug) throw new Error("该接口仅在 --debug 模式下可用。");
    const added = addJunctionCluster(configPath, body, readCompiled(area));
    compileFresh(configPath);
    const refreshed = readAreaConfig(configPath, { repoRoot });
    sendJson(response, 200, { ok: true, added, ...state(refreshed, junctionReference, debug) });
  }).catch((error) => sendJson(response, 400, { ok: false, error: error.message }));
  if (request.method === "POST" && url.pathname === "/api/compile") return Promise.resolve().then(() => {
    compileFresh(configPath);
    sendJson(response, 200, state(area, junctionReference));
  }).catch((error) => sendJson(response, 500, { ok: false, error: error.message }));
  sendJson(response, 404, { error: "Not found" });
}

function state(area, junctionReference = null, debug = false) {
  const nativeDir = area.outputs.nativeRoadDir;
  const osm2streetsRoadSurface = path.join(area.outputs.geojsonDir, "road_surface.geojson");
  const trafficSignals = fs.existsSync(area.outputs.nativeTrafficSignals)
    ? validateDocument(readJson(area.outputs.nativeTrafficSignals), fs.readFileSync(area.input, "utf8"))
    : { schema: "native-traffic-signals/v1", provenance: "empty", assemblies: { type: "FeatureCollection", features: [] } };
  const trafficRuntime = runtime(trafficSignals);
  const compiled = readCompiled(area);
  return { areaId: area.id, debug: debug ? { junctionCandidates: junctionCandidates(compiled) } : null, compiled, overrides: loadOverrides(area.outputs.nativeRoadOverrides), trafficSignals, trafficRuntime, comparison: readJson(path.join(nativeDir, "comparison.json")), junctionReference, layers: { nativeRoadSurface: readLayer(path.join(nativeDir, "layers", "road_surface.geojson")), edgeLines: readLayer(path.join(nativeDir, "layers", "edge_lines.geojson")), nativeSidewalkSurface: readLayer(path.join(nativeDir, "layers", "sidewalk_surface.geojson")), nativeIntersectionSurface: readLayer(path.join(nativeDir, "layers", "intersection_surface.geojson")), laneCenterlines: readLayer(path.join(nativeDir, "layers", "lane_centerlines.geojson")), laneSeparators: readLayer(path.join(nativeDir, "layers", "lane_separators.geojson")), centerLines: readLayer(path.join(nativeDir, "layers", "center_lines.geojson")), directionArrows: readLayer(path.join(nativeDir, "layers", "direction_arrows.geojson")), turnArrows: readLayer(path.join(nativeDir, "layers", "turn_arrows.geojson")), crosswalks: readLayer(path.join(nativeDir, "layers", "crosswalks.geojson")), vehicleStopLines: readLayer(path.join(nativeDir, "layers", "vehicle_stop_lines.geojson")), connectors: readLayer(path.join(nativeDir, "layers", "connectors.geojson")), osm2streetsRoadSurface: fs.existsSync(osm2streetsRoadSurface) ? readLayer(osm2streetsRoadSurface) : null } };
}
// The compiler reports candidates as advisory diagnostics. Lift them into their
// own payload with a stable index so the map can label them "#1, #2, ..." and
// the inspector can offer a ready-to-paste cluster配置.
// Append one detected cluster to the hand-authored area config. The candidate
// must still be present in the latest compile, so a stale browser tab cannot
// write a cluster that no longer exists. The edited config is validated by the
// real loader before it replaces the file: an invalid write would break every
// later command, and the file is git-tracked so a bad accept stays revertible.
function addJunctionCluster(configPath, body, compiled) {
  const index = Number(body?.index);
  if (!Number.isInteger(index)) throw new Error("请求缺少候选编号 index。");
  const candidate = junctionCandidates(compiled).find((item) => item.index === index);
  if (!candidate) throw new Error(`候选 #${index} 不在最新一次编译结果里，请刷新页面后重试。`);
  const raw = readJson(configPath);
  const templates = raw.nativeRoad?.junctionTemplates;
  if (!templates) throw new Error("区域配置缺少 nativeRoad.junctionTemplates，请先手工建立该节点。");
  const clusters = Array.isArray(templates.clusters) ? templates.clusters : [];
  const taken = new Set(clusters.flatMap((cluster) => (cluster.nodeIds || []).map(String)));
  const clash = candidate.nodeIds.filter((nodeId) => taken.has(String(nodeId)));
  if (clash.length) throw new Error(`节点 ${clash.join("、")} 已属于其他复杂路口配置。`);
  const id = uniqueClusterId(`cluster-${candidate.nodeIds[0]}`, new Set(clusters.map((cluster) => cluster.id)));
  const cluster = {
    id,
    template: candidate.template,
    coreRadiusMeters: candidate.coreRadiusMeters,
    cornerRadiusMeters: 12,
    outerRadiusExtraMeters: 18,
    nodeIds: candidate.nodeIds.map(String),
  };
  const next = { ...raw, nativeRoad: { ...raw.nativeRoad, junctionTemplates: { ...templates, enabled: true, clusters: [...clusters, cluster] } } };
  const staging = `${configPath}.candidate-${process.pid}.json`;
  fs.writeFileSync(staging, `${JSON.stringify(next, null, 2)}\n`);
  try {
    readAreaConfig(staging, { repoRoot });
  } catch (error) {
    fs.unlinkSync(staging);
    throw new Error(`写入后的配置无法通过校验，已放弃：${error.message}`);
  }
  fs.unlinkSync(staging);
  writeJsonAtomic(configPath, next);
  return cluster;
}
function uniqueClusterId(base, taken) {
  if (!taken.has(base)) return base;
  for (let suffix = 2; suffix < 100; suffix += 1) if (!taken.has(`${base}-${suffix}`)) return `${base}-${suffix}`;
  throw new Error("无法生成唯一的 cluster id。");
}

function junctionCandidates(compiled) {
  return (compiled?.diagnostics || [])
    .filter((item) => item.rule === "complex-junction-candidate" && item.suggestedCluster)
    .sort((first, second) => second.suggestedCluster.nodeCount - first.suggestedCluster.nodeCount || first.suggestedCluster.diameterMeters - second.suggestedCluster.diameterMeters)
    .map((item, index) => ({ index: index + 1, id: item.id, message: item.message, coordinate: item.geometry?.coordinates || null, ...item.suggestedCluster }));
}

function compileFresh(configPath) {
  try {
    return execFileSync(process.execPath, [path.join(repoRoot, "scripts", "compile-native-roads.js"), "--config", configPath], {
      cwd: repoRoot,
      encoding: "utf8",
      stdio: ["ignore", "pipe", "pipe"],
    });
  } catch (error) {
    const detail = String(error.stderr || error.stdout || error.message || "native compilation failed").trim();
    throw new Error(`Native road compilation failed: ${detail}`);
  }
}
function readJunctionReference(file) {
  if (!fs.existsSync(file)) throw new Error(`Junction reference not found: ${file}`);
  const converted = convertGeoJson(JSON.parse(fs.readFileSync(file, "utf8")));
  return { source: file, coordinateSystem: "GCJ-02", converted };
}
function readCompiled(area) { return readJson(path.join(area.outputs.nativeRoadDir, "compiled.json")); }
function readJson(file) { return JSON.parse(fs.readFileSync(file, "utf8")); }
function readLayer(file) { return fs.existsSync(file) ? readJson(file) : { type: "FeatureCollection", features: [] }; }
function readBody(request) { return new Promise((resolve, reject) => { let body = ""; request.setEncoding("utf8"); request.on("data", (part) => { body += part; if (body.length > 1024 * 1024) request.destroy(); }); request.on("end", () => { try { resolve(JSON.parse(body)); } catch (_) { reject(new Error("Request body must be JSON.")); } }); request.on("error", reject); }); }
function sendFile(response, file, type) { response.writeHead(200, { "Content-Type": type, "Cache-Control": "no-store" }); fs.createReadStream(file).pipe(response); }
function sendVendorFile(response, pathname) {
  const match = /^\/vendor\/(ol|rbush|quickselect)\/(.+)$/.exec(pathname);
  if (!match) return sendJson(response, 404, { error: "Not found" });
  const root = path.join(repoRoot, "node_modules", match[1]);
  const file = path.resolve(root, match[2]);
  if (!file.startsWith(`${root}${path.sep}`) || !fs.existsSync(file) || !fs.statSync(file).isFile()) return sendJson(response, 404, { error: "Not found" });
  return sendFile(response, file, file.endsWith(".css") ? "text/css; charset=utf-8" : "text/javascript; charset=utf-8");
}
function sendJson(response, status, value) { response.writeHead(status, { "Content-Type": "application/json; charset=utf-8", "Cache-Control": "no-store" }); response.end(`${JSON.stringify(value)}\n`); }
if (require.main === module) main();