#!/usr/bin/env bash
# OC-OUTPUT: unified end-of-run summary for the legacy engine. Same format as
# loki-ts/src/util/run_summary.ts (Result, Files, Cost, Elapsed, Receipt, next).
# Sourced by autonomy/run.sh; defines functions only.
#
#   loki_run_summary RESULT_CODE [START_SHA]
# Honors NO_COLOR and LOKI_SUMMARY_JSON=1 (machine summary). LOKI_QUIET=1 is applied by the
# caller's log threshold (banner noise off); the summary itself always prints.

loki_run_summary() {
    # Print once per process even if two exit paths both call it.
    [ "${_LOKI_SUMMARY_PRINTED:-0}" = "1" ] && return 0
    _LOKI_SUMMARY_PRINTED=1
    local code="${1:-0}" start_sha="${2:-${_LOKI_RUN_START_SHA:-}}"
    local loki_dir="${LOKI_DIR:-${TARGET_DIR:-.}/.loki}"
    local result="done-unverified" files="" ins="" del="" shortstat="" cost="" receipt="" next=""
    local elapsed=$(( $(date +%s) - ${SESSION_START_EPOCH:-$(date +%s)} ))

    # Result: verified only when a proof exists and the run exited 0.
    local rid="" proof=""
    if [ -s "$loki_dir/state/last-proof-id.txt" ]; then rid="$(head -n1 "$loki_dir/state/last-proof-id.txt" 2>/dev/null | tr -d '[:space:]')"; fi
    case "$rid" in *[!A-Za-z0-9._-]*) rid="" ;; esac
    [ -n "$rid" ] && [ -f "$loki_dir/proofs/$rid/proof.json" ] && proof="$loki_dir/proofs/$rid/proof.json"
    case "$code" in
        0) if [ -n "$proof" ]; then result="verified"; else result="done-unverified"; fi ;;
        130|143) result="stopped" ;;
        *) result="failed" ;;
    esac
    [ -f "$loki_dir/STOPPED" ] && result="stopped"
    [ -n "$proof" ] && receipt="$proof"

    # Files changed vs run-start SHA (excluding .loki and .git).
    if [ -n "$start_sha" ]; then
        local range="${start_sha}..HEAD"
        # LOKI_SUMMARY_WORKTREE=1 (loki edit): compare against the working tree, edits are uncommitted.
        [ "${LOKI_SUMMARY_WORKTREE:-0}" = "1" ] && range="${start_sha}"
        shortstat="$( (cd "${TARGET_DIR:-.}" && git diff --shortstat "$range" -- . ':(exclude).loki/' ':(exclude).git/') 2>/dev/null || true )"
        files="$(printf '%s\n' "$shortstat" | grep -oE '[0-9]+ file' | grep -oE '[0-9]+' | head -1)"
        ins="$(printf '%s\n' "$shortstat" | grep -oE '[0-9]+ insertion' | grep -oE '[0-9]+' | head -1)"
        del="$(printf '%s\n' "$shortstat" | grep -oE '[0-9]+ deletion' | grep -oE '[0-9]+' | head -1)"
        [ -z "$files" ] && files=0
        [ -z "$ins" ] && ins=0
        [ -z "$del" ] && del=0
    fi

    # Cost: legacy cost comes from the context tracker (token estimate), so it is always marked estimated.
    if [ -f "$loki_dir/context/tracking.json" ]; then
        cost="$(LOKI_TRK="$loki_dir/context/tracking.json" python3 -E -c "
import json, os
try:
    c = json.load(open(os.environ['LOKI_TRK'])).get('totals', {}).get('total_cost_usd')
    print('' if c is None else '%.2f' % float(c))
except Exception:
    print('')
" 2>/dev/null || true)"
    fi

    # LOKI_SUMMARY_COST: exact cost from the caller (loki edit reads it from the provider).
    local cost_exact=false
    if [ -n "${LOKI_SUMMARY_COST:-}" ]; then cost="$LOKI_SUMMARY_COST"; cost_exact=true; fi

    case "$result" in
        verified) if [ "${files:-0}" != "0" ] && [ -n "$files" ]; then next="loki diff"; else next="loki status"; fi ;;
        done-unverified) next="loki verify" ;;
        stopped) next="loki status" ;;
        *) if [ -n "$files" ] && [ "$files" != "0" ]; then next="loki undo"; else next="loki doctor"; fi ;;
    esac

    # REQTRACE: requirements with no passing check, from .loki/state/requirements-map.json (written by the Bun runner and engine10).
    local unc=""
    if [ -s "$loki_dir/state/requirements-map.json" ]; then
        unc="$(LOKI_RM="$loki_dir/state/requirements-map.json" python3 -I -c "
import json, os
try:
    m = json.load(open(os.environ['LOKI_RM']))
    n = sum(1 for r in m.get('requirements', []) if r.get('status') == 'uncovered')
    print(n if n else '')
except Exception:
    print('')
" 2>/dev/null || true)"
    fi
    local next_h="$next"
    [ -n "$unc" ] && next_h="${next}; ${unc} requirement(s) uncovered, see: loki plan --requirements"

    local mm=$((elapsed / 60)) ss=$((elapsed % 60)) elapsed_h
    if [ "$elapsed" -lt 60 ]; then elapsed_h="${elapsed}s"; else elapsed_h="${mm}m$(printf '%02d' "$ss")s"; fi

    if [ "${LOKI_SUMMARY_JSON:-0}" = "1" ]; then
        printf '{"result":"%s","files_changed":%s,"insertions":%s,"deletions":%s,"cost_usd":%s,"cost_estimated":%s,"elapsed_s":%s,"receipt":%s,"next":"%s"}\n' \
            "$result" "${files:-null}" "${ins:-null}" "${del:-null}" "${cost:-null}" "$([ "$cost_exact" = true ] && echo false || echo true)" "$elapsed" \
            "$([ -n "$receipt" ] && printf '"%s"' "$receipt" || printf 'null')" "$next"
        return 0
    fi

    local c0="" c1=""
    if [ -t 1 ] && [ -z "${NO_COLOR+x}" ]; then
        c1=$'\033[0m'
        case "$result" in verified) c0=$'\033[32m' ;; done-unverified) c0=$'\033[33m' ;; *) c0=$'\033[31m' ;; esac
    fi
    local files_h="unknown" cost_h="not measured"
    [ -n "$files" ] && files_h="${files} changed (+${ins} -${del})"
    if [ -n "$cost" ]; then
        if [ "$cost_exact" = true ]; then cost_h="\$${cost}"; else cost_h="\$${cost} (estimated)"; fi
    fi
    printf '\n'
    printf 'Result:   %s%s%s\n' "$c0" "$result" "$c1"
    printf 'Files:    %s\n' "$files_h"
    printf 'Cost:     %s\n' "$cost_h"
    printf 'Elapsed:  %s\n' "$elapsed_h"
    [ -n "$receipt" ] && printf 'Receipt:  %s\n' "$receipt"
    printf 'next:     %s  (or: loki chat --attach)\n' "$next_h"
}