import schema
import aq_utils
import os

run_id = os.environ.get('BATCH_ID')
if not run_id:
    raise Exception("BATCH_ID NOT SET")


def setup():
    global regions_dict
    regions_dict = aq_utils.load("")[0]


def extract_event_id(string, key, delimiter):
    values = string.split(",")
    for value in values:
        if key in value:
            return value.split(delimiter)[1]
    return ""


def dat_hnd(rec: any) -> any:
    # ! tenant_id field
    rec['query']['tid'] = rec['query'].get('tid') or "tid"
    rec['runid'] = run_id

    rec['page_url'] = {}
    rec['page_url']['query'] = {"pqs": rec['query'].get('pqs')} if rec['query'].get('pqs') not in [None, "na", "NA"] else {}
    rec['page_url']['host'] = rec['query'].get('dmn') or rec.get('domain', "")
    rec['page_url']['path'] = rec['query'].get('pn') or ""
    rec['page_url']['scheme'] = ''
    
    rec['page_referer_url'] = {
      "scheme": "",
      "host": rec['query'].get('rd') or rec.get('referer'),
      "path": rec['query'].get('rpn') or "",
      "query":{"rqs": rec['query'].get('rqs')} if rec['query'].get('rqs') not in [None, "na", "NA"] else {}
    }
    
    country_code = rec['country_code']
    rec['global_region'] = "Unknown"
    for region, countries in regions_dict.items():
        if country_code in countries:
            rec['global_region'] = region
            break

    rec['query']['pevid'] = rec['query'].get("pevid") or extract_event_id(rec['aqfer_var'], "z_evid", "=")
    rec['query']['src'] = rec['query'].get('src') or "aUT"
    
    return [rec]


def schema_handler_v1(sch: schema.Schema) -> (schema.Schema, any):
    setup()
    url_schema = aq_utils.load_schema("amdp_schemas-v6.2.yaml", "url_struct")
    schema.add_fields(sch, schema.Field(name="global_region", schema=schema.StringSchema()))
    schema.add_fields(sch, schema.Field(name="runid", schema=schema.StringSchema()))
    schema.add_fields(sch, schema.Field(name="page_url", schema=url_schema))
    schema.add_fields(sch, schema.Field(name="page_referer_url", schema=url_schema))
    return sch, dat_hnd