"""Recover a stopped shape checkpoint using only current drawing adapters."""
import argparse
import copy
import json
import os
from pathlib import Path
import sys
import tempfile
import time
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / 'src' / 'backend'))
import server


def recover(job):
    if job.get('status') not in ('failed', 'cancelled') or job.get('result'):
        raise ValueError('only stopped shape checkpoints without a final result')
    request = job['requestBody']
    if request.get('concreteApproved'):
        raise ValueError('approved/rebar models are outside this repair')
    partial = job['partialResult']
    original = partial['exteriorEvidence']
    exterior = copy.deepcopy(original)
    drawing = copy.deepcopy(partial['drawingEvidence'])
    analysis = server.rehydrate_drawing_set_analysis(request['analysis'])
    exterior = server.normalize_direct_dxf_wing_footprints(analysis, exterior, drawing)
    exterior = server.enforce_wing_haunch_object_contract(exterior)
    exterior = server.align_wing_root_bottoms_to_side_walls(exterior, analysis)
    exterior = server.enforce_wing_haunch_object_contract(exterior)
    audit = exterior['wing_haunch_object_contract_audit']
    if not audit['pass'] or audit['haunch_count'] != audit['wing_count'] or any(
            r['status'] != 'resolved' for r in audit['contact_rows']):
        raise ValueError('wing/haunch geometry or contact still unresolved')
    before = {c['id']: c for c in original['components']}
    after = {c['id']: c for c in exterior['components']}
    for key, value in before.items():
        if value.get('kind') not in ('wingwall', 'wing_wall', 'wingwall_haunch'):
            assert after.get(key) == value, 'unrelated concrete changed'
    assert exterior.get('rebar_groups') == original.get('rebar_groups')
    checkpoint = server.exterior_feedback_snapshot(drawing, exterior)
    result = dict(summary='날개벽 외곽 및 접합헌치 복구 · 전체 형상 검토 대기',
                  category_id='box_culvert', concrete_components=copy.deepcopy(exterior['components']),
                  rebar_groups=copy.deepcopy(exterior.get('rebar_groups') or []),
                  drawingEvidence=checkpoint['drawingEvidence'],
                  unresolved=copy.deepcopy(exterior.get('unresolved') or []),
                  ready_for_geometry=False, awaiting_concrete_confirmation=True,
                  workflowStage='shape', wingGeometryRecoveryAudit=audit)
    updated = copy.deepcopy(job)
    updated['previousFailureDiagnostic'] = updated.pop('failureDiagnostic', None)
    updated.update(status='awaiting_feedback', error=None, verified=False, progress=100,
                   stage='날개벽·접합헌치 복구 완료 · 전체 형상 검토 대기',
                   result=result, partialResult=checkpoint, geometryRevision=time.time_ns(),
                   pipelineVersion=server.MODEL_PIPELINE_VERSION,
                   sourceContext=server.model_job_source_context(request))
    return updated


def main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('job', type=Path)
    parser.add_argument('--apply', action='store_true')
    args = parser.parse_args()
    server.require_modeling_rules()
    target = args.job.resolve()
    raw = target.read_bytes()
    updated = recover(json.loads(raw))
    print(json.dumps(updated['result']['wingGeometryRecoveryAudit'], ensure_ascii=False))
    if not args.apply:
        return
    if target.read_bytes() != raw:
        raise ValueError('concurrent job update')
    backup = target.with_name(target.stem + f'.before-wing-recovery-{time.time_ns()}.json')
    with backup.open('xb') as stream:
        stream.write(raw)
    with tempfile.NamedTemporaryFile(mode='w', dir=target.parent, delete=False) as stream:
        json.dump(updated, stream, ensure_ascii=False, allow_nan=False)
        temp = stream.name
    if target.read_bytes() != raw:
        raise ValueError('concurrent update before commit')
    os.replace(temp, target)
    print(json.dumps({'saved': str(target), 'backup': str(backup)}))


if __name__ == '__main__':
    main()