# Handle file related helpers, be it a local file or one on the internet
# by Humans for All

import urllib.request
import urllib.parse
import debug
import toolcalls as mTC
from dataclasses import dataclass



def get_from_web(url: str, tag: str, inContentType: str, inHeaders: mTC.HttpHeaders):
    """
    Get the url specified from web.

    If passed header doesnt contain certain useful http header entries,
    some predefined defaults will be used in place. This includes User-Agent,
    Accept-Language and Accept.

    One should ideally pass the header got in the request being proxied, so as
    to help one to try mimic the real client, whose request we are proxying.
    In case a header is missing in the got request, fallback to using some
    possibly ok enough defaults.
    """
    try:
        hUA = inHeaders.get('User-Agent', None)
        if not hUA:
            hUA = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:132.0) Gecko/20100101 Firefox/132.0'
        hAL = inHeaders.get('Accept-Language', None)
        if not hAL:
            hAL = "en-US,en;q=0.9"
        hA = inHeaders.get('Accept', None)
        if not hA:
            hA = "text/html,*/*"
        headers = {
            'User-Agent': hUA,
            'Accept': hA,
            'Accept-Language': hAL
        }
        req = urllib.request.Request(url, headers=headers)
        # Get requested url
        print(f"DBUG:{tag}:Req:{req.full_url}:{req.headers}")
        with urllib.request.urlopen(req, timeout=10) as response:
            contentData = response.read()
            statusCode = response.status or 200
            statusMsg = response.msg or ""
            contentType = response.getheader('Content-Type') or inContentType
            print(f"DBUG:FM:GFW:Resp:{response.status}:{response.msg}")
            debug.dump({ 'op': 'FileMagic.GetFromWeb', 'url': req.full_url, 'req.headers': req.headers, 'resp.headers': response.headers, 'ctype': contentType }, { 'cdata': contentData })
        return mTC.TCOutResponse(True, statusCode, statusMsg, contentType, contentData)
    except Exception as exc:
        return mTC.TCOutResponse(False, 502, f"WARN:{tag}:Failed:{exc}")


def get_from_local(urlParts: urllib.parse.ParseResult, tag: str, inContentType: str):
    """
    Get the requested file from the local filesystem
    """
    try:
        fPdf = open(urlParts.path, 'rb')
        dPdf = fPdf.read()
        return mTC.TCOutResponse(True, 200, "", inContentType, dPdf)
    except Exception as exc:
        return mTC.TCOutResponse(False, 502, f"WARN:{tag}:Failed:{exc}")


def get_file(url: str, tag: str, inContentType: str, inHeaders: mTC.HttpHeaders={}):
    """
    Based on the scheme specified in the passed url,
    either get from local file system or from the web.
    """
    urlParts = urllib.parse.urlparse(url)
    if urlParts.scheme == "file":
        return get_from_local(urlParts, tag, inContentType)
    else:
        return get_from_web(url, tag, inContentType, inHeaders)