feat: Add script and just command to reset failed orders

justfile

@@ -416,3 +416,8 @@ bento-setup:
 job-status job_id:
     #!/usr/bin/env bash
     ./scripts/job_status.sh {{job_id}}
+
+# Reset an order
+reset-order order_id:
+    #!/usr/bin/env bash
+    ./scripts/reset-order.sh {{order_id}}

scripts/reset-order.sh

@@ -0,0 +1,55 @@
+#!/bin/bash
+
+# This script performs a full reset of a failed order using the ORDER ID. It will:
+# 1. Find the corresponding job_id (proof_id) from the SQLite database (broker.db).
+# 2. Delete the job, tasks, and task dependencies from the PostgreSQL database (taskdb).
+# 3. Reset the order's status to 'PendingProving' in the SQLite database.
+
+set -euo pipefail
+
+ORDER_ID_FRAGMENT="$1"
+if [ -z "$ORDER_ID_FRAGMENT" ]; then
+  echo "Usage: $0 <order_id_fragment>"
+  echo "You can provide a partial or full order ID."
+  exit 1
+fi
+
+echo "--- [Step 1/3] Finding Job ID (proof_id) for Order fragment: ${ORDER_ID_FRAGMENT} ---"
+
+# The sqlite3 command to find the proof_id (job_id)
+SQLITE_FIND_SQL="SELECT json_extract(data, '\$.proof_id') FROM orders WHERE id LIKE '%${ORDER_ID_FRAGMENT}%';"
+JOB_ID=$(docker run --rm -i -v bento_broker-data:/db nouchka/sqlite3 /db/broker.db "${SQLITE_FIND_SQL}")
+
+if [ -z "$JOB_ID" ] || [ "$JOB_ID" == "null" ]; then
+  echo "Error: No order found with an ID fragment matching '${ORDER_ID_FRAGMENT}' or the order does not have a proof_id."
+  exit 1
+fi
+
+echo "Found Job ID: ${JOB_ID}"
+echo ""
+
+echo "--- [Step 2/3] Resetting PostgreSQL data for Job ID: ${JOB_ID} ---"
+PG_USER="${POSTGRES_USER:-worker}"
+PG_DB="${POSTGRES_DB:-taskdb}"
+PG_SQL="DELETE FROM public.task_deps WHERE job_id = '${JOB_ID}'; DELETE FROM public.tasks WHERE job_id = '${JOB_ID}'; DELETE FROM public.jobs WHERE id = '${JOB_ID}';"
+
+PG_RESULT=$(docker compose exec -T postgres psql -U "${PG_USER}" -d "${PG_DB}" -c "${PG_SQL}")
+echo "PostgreSQL cleanup complete."
+echo ""
+
+echo "--- [Step 3/3] Resetting SQLite order status to PendingProving ---"
+# Use the original Order ID fragment to update the correct order
+SQLITE_RESET_SQL="
+UPDATE orders
+SET data = json_set(
+               json_set(data, '\$.status', 'PendingProving'),
+               '\$.proof_id', NULL
+           )
+WHERE id LIKE '%${ORDER_ID_FRAGMENT}%';
+SELECT 'SQLite: Order status reset for ' || changes() || ' order(s).';
+"
+SQLITE_RESULT=$(docker run --rm -i -v bento_broker-data:/db nouchka/sqlite3 /db/broker.db "${SQLITE_RESET_SQL}")
+echo "${SQLITE_RESULT}"
+
+echo ""
+echo "--- Reset complete. The broker should now pick up the order for proving. ---"