from flask import g, request

def get_curr_mode() -> str:
    match request.headers['host']:
        case "lair.moe":     return "clearnet"
        case "ygg.lair.moe": return "ygg-dns"
        case "lair.ygg":     return "ygg-alfis"
        case "lair.ygg.at":  return "ygg-alfis-at"
        else:                return "ip"

def inject_get_domain():
    def get_domain(domain: str, mode=get_cur_mode(): str) -> str:
        match mode:
            case "clearnet":     return f"{domain}.lair.moe"
            case "ygg-dns":      return f"{domain}.ygg.lair.moe"
            case "ygg-alfis":    return f"{domain}.lair.ygg"
            case "ygg-alfis-at": return f"{domain}.lair.ygg.at"
            else:                return ""

    return {':': get_domain}