From 513402235e6b408034b7b6c216cf8cf545e8477c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Asbj=C3=B8rn=20Riddervold?= <aridder@pm.me>
Date: Fri, 5 Sep 2025 21:05:24 +0200
Subject: [PATCH] feat: Add just command for resetting an order

---
 justfile               |  5 ++++
 scripts/reset-order.sh | 55 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 60 insertions(+)
 create mode 100755 scripts/reset-order.sh

diff --git a/justfile b/justfile
index a823fb1f5..6f7d57c54 100644
--- a/justfile
+++ b/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}}
diff --git a/scripts/reset-order.sh b/scripts/reset-order.sh
new file mode 100755
index 000000000..4408b0385
--- /dev/null
+++ b/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. ---"
\ No newline at end of file