Probe MDI auto-draft persistence

Author: chubes4

packages/runtime-cloudflare/src/request-routing.ts

@@ -8,7 +8,7 @@ export type WorkerRequestRoute =
   | { kind: "editor-probe"; phase: EditorProbePhase }
   | { kind: "probe"; phase: string }
 
-export type EditorProbePhase = "admin" | "auto-draft" | "block-editor"
+export type EditorProbePhase = "admin" | "auto-draft" | "auto-draft-no-persist" | "block-editor"
 
 export function routeWorkerRequest(request: Request): WorkerRequestRoute {
   const phase = new URL(request.url).searchParams.get("phase")
@@ -20,6 +20,7 @@ export function routeWorkerRequest(request: Request): WorkerRequestRoute {
   if (phase === "operator-reset") return { kind: "operator-reset" }
   if (phase === "editor-probe-admin") return { kind: "editor-probe", phase: "admin" }
   if (phase === "editor-probe-auto-draft") return { kind: "editor-probe", phase: "auto-draft" }
+  if (phase === "editor-probe-auto-draft-no-persist") return { kind: "editor-probe", phase: "auto-draft-no-persist" }
   if (phase === "editor-probe-block-editor") return { kind: "editor-probe", phase: "block-editor" }
   return { kind: "probe", phase }
 }

packages/runtime-cloudflare/src/worker.ts

@@ -106,6 +106,7 @@ async function runCoordinatedEditorProbe(request: Request, env: Env, coordinator
   try {
     if (!lease.pointer) throw new Error("Canonical WordPress must be initialized before running an editor probe.")
     const runtime = await getRuntime(env, lease.pointer, new URL(request.url).origin)
+    if (phase === "auto-draft-no-persist") patchAutoDraftPersistenceProbe(runtime.php)
     patchEditorProbe(runtime.php, phase)
     const response = toFetchResponse(request, await runtime.requestHandler.request(await toPHPRequest(request)))
     await releaseLease(coordinator, request.url, lease)
@@ -125,6 +126,7 @@ function patchEditorProbe(php: PHP, phase: EditorProbePhase): void {
   const markers: Record<EditorProbePhase, string> = {
     admin: "require_once __DIR__ . '/admin.php';",
     "auto-draft": "$post_ID = $post->ID;",
+    "auto-draft-no-persist": "$post_ID = $post->ID;",
     "block-editor": "require_once ABSPATH . 'wp-admin/admin-footer.php';",
   }
   const marker = markers[phase]
@@ -135,6 +137,23 @@ function patchEditorProbe(php: PHP, phase: EditorProbePhase): void {
   php.writeFile(path, new TextEncoder().encode(`${source.slice(0, insertion)}${stop}${source.slice(insertion)}`))
 }
 
+function patchAutoDraftPersistenceProbe(php: PHP): void {
+  const path = "/wordpress/wp-content/plugins/markdown-database-integration/inc/class-wp-markdown-write-engine.php"
+  const source = new TextDecoder().decode(php.readFileAsBuffer(path))
+  const needle = `if ( $id ) {
+				$this->mark_post_dirty( (int) $id );
+			}`
+  const replacement = `if ( $id ) {
+				$rows = $this->driver->query( "SELECT post_status FROM \`{$this->prefix()}posts\` WHERE ID = " . (int) $id );
+				if ( ! is_array( $rows ) || empty( $rows ) || 'auto-draft' !== ( $rows[0]->post_status ?? '' ) ) {
+					$this->mark_post_dirty( (int) $id );
+				}
+			}`
+  const index = source.indexOf(needle)
+  if (index === -1 || index !== source.lastIndexOf(needle)) throw new Error("MDI auto-draft persistence probe marker is not unique.")
+  php.writeFile(path, new TextEncoder().encode(`${source.slice(0, index)}${replacement}${source.slice(index + needle.length)}`))
+}
+
 interface MarkdownManifestFile {
   path: string
   objectKey: string

tests/cloudflare-runtime.test.ts

@@ -45,6 +45,7 @@ test("Cloudflare routing reserves phases while the phase-less route serves WordP
   assert.deepEqual(routeWorkerRequest(new Request("https://worker.example/?phase=operator-reset")), { kind: "operator-reset" })
   assert.deepEqual(routeWorkerRequest(new Request("https://worker.example/wp-admin/post-new.php?phase=editor-probe-admin")), { kind: "editor-probe", phase: "admin" })
   assert.deepEqual(routeWorkerRequest(new Request("https://worker.example/wp-admin/post-new.php?phase=editor-probe-auto-draft")), { kind: "editor-probe", phase: "auto-draft" })
+  assert.deepEqual(routeWorkerRequest(new Request("https://worker.example/wp-admin/post-new.php?phase=editor-probe-auto-draft-no-persist")), { kind: "editor-probe", phase: "auto-draft-no-persist" })
   assert.deepEqual(routeWorkerRequest(new Request("https://worker.example/wp-admin/post-new.php?phase=editor-probe-block-editor")), { kind: "editor-probe", phase: "block-editor" })
   assert.deepEqual(routeWorkerRequest(new Request("https://worker.example/?phase=seeded-wordpress")), { kind: "probe", phase: "seeded-wordpress" })
 })
@@ -126,6 +127,7 @@ test("Cloudflare editor probes bound the authenticated editor lifecycle and disc
   assert.equal((probe.match(/await discardCachedRuntime\(\)/g) ?? []).length, 2)
   assert.match(probe, /\/wordpress\/wp-admin\/post-new\.php/)
   assert.match(probe, /memory_get_peak_usage/)
+  assert.match(probe, /'auto-draft' !== \( \$rows\[0\]->post_status/)
 })
 
 test("Cloudflare runtime packages a provenanced canonical MDI seed", async () => {