from tldextract import extract

from app.models.bridges import Bridge
from app.models.mirrors import Origin, Proxy


def mirror_sites():
    return {
        "version": "2.0",
        "sites": [{
            "main_domain":  x.description[len("proxy:"):].replace("www.", "") if x.description.startswith("proxy:") else x.domain_name.replace("www.", ""),
            "available_alternatives": [
                                          {
                                              "proto": "tor" if ".onion" in a.url else "https",
                                              "type": "eotk" if ".onion" in a.url else "mirror",
                                              "created_at": str(a.added),
                                              "updated_at": str(a.updated),
                                              "url": a.url
                                          } for a in x.mirrors if not a.deprecated and not a.destroyed
                                      ] + [
                                          {
                                              "proto": "https",
                                              "type": "mirror",
                                              "created_at": str(a.added),
                                              "updated_at": str(a.updated),
                                              "url": a.url
                                          } for a in x.proxies if
                                          a.url is not None and not a.deprecated and not a.destroyed and a.provider == "cloudfront"
                                      ]} for x in Origin.query.order_by(Origin.domain_name).all() if x.destroyed is None
        ]
    }


def bridgelines():
    return {
        "version": "1.0",
        "bridgelines": [
            b.bridgeline for b in Bridge.query.filter(
                Bridge.destroyed == None,
                Bridge.bridgeline != None
            )
        ]
    }


def mirror_mapping():
    return {
        d.url.lstrip("https://"): {
            "origin_domain": d.origin.domain_name,
            "origin_domain_normalized": d.origin.domain_name.lstrip("www."),
            "origin_domain_root": extract(d.origin.domain_name).registered_domain
        } for d in Proxy.query.all() if d.url is not None
    }