"""Explicit local-source recipe. Network is opt-in; no retries/mirrors/raw redistribution."""
import argparse
import hashlib
import json
import shutil
import time
import urllib.request
import urllib.error
import zipfile
from pathlib import Path
class NoRedirect(urllib.request.HTTPRedirectHandler):
    def redirect_request(self,*args):return None

SOURCES=[
    ('sparc-table1-metadata.body','https://astroweb.cwru.edu/SPARC/SPARC_Lelli2016c.mrt',28259,'5aa0501f6b0d881fa579030e315e7b5b6ef561a5bd3a07472f9929c7e5728243'),
    ('sparc-rotmod-named-member-archive.body','https://astroweb.cwru.edu/SPARC/Rotmod_LTG.zip',110737,'0a80cc90714828cc28b7dd57923576714d209f2490328c087c4a4ad607faf588'),
]
MEMBER='NGC2403_rotmod.dat'
MEMBER_SHA='9be193eccc5c363e74d04270290d831738f41ec9225a3b2c95353fd3812170b8'

def main():
    parser=argparse.ArgumentParser();source=parser.add_mutually_exclusive_group(required=True)
    source.add_argument('--from-cache',type=Path);source.add_argument('--online',action='store_true')
    parser.add_argument('--http-budget-bytes',type=int,default=0);parser.add_argument('--output',type=Path,required=True);args=parser.parse_args()
    if args.online and args.http_budget_bytes<sum(row[2] for row in SOURCES):raise ValueError('Register enough explicit body budget before online acquisition')
    args.output.mkdir(parents=True,exist_ok=False);ledger=[];started=time.monotonic()
    try:
        for filename,url,size,expected in SOURCES:
            entry={'url':url,'bytes':0,'sha256':None,'receivedHttpBodyBytes':0,'status':None,'headers':{}};ledger.append(entry)
            if args.from_cache:body=(args.from_cache/filename).read_bytes()
            else:
                request=urllib.request.Request(url,headers={'Accept-Encoding':'identity','User-Agent':'SandboxPhysics-AC39-Local-Source-Recipe/1.0'})
                payload=bytearray()
                try:
                    response=urllib.request.build_opener(NoRedirect).open(request,timeout=30)
                    with response:
                        entry['status']=response.code;entry['headers']=dict(response.headers.items())
                        if response.code!=200 or response.headers.get('Content-Length')!=str(size):raise ValueError('Changed status/body size; no redirect/retry/mirror')
                        if response.headers.get('Content-Encoding','identity')!='identity':raise ValueError('Unregistered encoded response')
                        while len(payload)<size:
                            chunk=response.read(min(65536,size-len(payload)))
                            if not chunk:break
                            payload.extend(chunk)
                except Exception as failure:
                    if isinstance(failure,urllib.error.HTTPError):
                        entry['status']=failure.code;entry['headers']=dict(failure.headers.items());payload.extend(failure.read(size))
                    if isinstance(getattr(failure,'partial',None),bytes):payload.extend(failure.partial[:size-len(payload)])
                    entry['receivedHttpBodyBytes']=len(payload);entry['bytes']=len(payload);entry['sha256']=hashlib.sha256(payload).hexdigest()
                    (args.output/filename).write_bytes(payload);raise
                body=bytes(payload);entry['receivedHttpBodyBytes']=len(body)
                (args.output/filename).write_bytes(body)
            actual=hashlib.sha256(body).hexdigest();entry['bytes']=len(body);entry['sha256']=actual
            if len(body)!=size or actual!=expected:raise ValueError('Named original identity changed; no automatic replacement')
            if args.from_cache:(args.output/filename).write_bytes(body)
        with zipfile.ZipFile(args.output/SOURCES[1][0]) as archive:
            names=archive.namelist()
            if names.count(MEMBER)!=1:raise ValueError('Missing/duplicate named member')
            body=archive.read(MEMBER)  # zipfile verifies the named member CRC.
            if hashlib.sha256(body).hexdigest()!=MEMBER_SHA:raise ValueError('Wrong named member identity')
            (args.output/MEMBER).write_bytes(body)
            result={'status':'PASS','sourceBodies':ledger,'namedMember':MEMBER,'namedMemberBytes':len(body),'namedMemberSha256':MEMBER_SHA,
                    'unusedUnextractedMembers':len(names)-1,'httpBytes':sum(row['receivedHttpBodyBytes'] for row in ledger),'cacheMode':bool(args.from_cache),
                    'newDirectory':str(args.output),'sourceFetchIsNotModelFit':True,'noRedistributionPermissionClaim':True,'retries':0,'wallSeconds':time.monotonic()-started}
    except Exception as failure:
        result={'status':'FAIL','sourceBodies':ledger,'error':type(failure).__name__+': '+str(failure),'retries':0,'noMirror':True}
        (args.output/'source-recipe-receipt.json').write_text(json.dumps(result,indent=2)+'\n');raise
    (args.output/'source-recipe-receipt.json').write_text(json.dumps(result,indent=2)+'\n');print(json.dumps(result,indent=2))

if __name__=='__main__':main()