from tools import *
from Utils.Config import *
from Utils.Lib import *
from AppBridge import *
# pip install beautifulsoup4 jsbeautifier cssbeautifier
from bs4 import BeautifulSoup
import jsbeautifier
import cssbeautifier

CUSTOM_MIME_MAP = {
    "application/json": ".json",
    "text/html": ".html",
    "text/css": ".css",
    "text/javascript": ".js",
    "text/plain": ".txt",
}

# ------------------- HTML 压缩 -------------------
def CompressHtml(html_str: str) -> str:
    """
    压缩 HTML 字符串（移除多余空白、注释，保留核心结构）
    :param html_str: 原始 HTML 字符串
    :return: 压缩后的 HTML 字符串
    """
    # 解析 HTML
    soup = BeautifulSoup(html_str, "html.parser")
    # 移除所有注释节点
    for comment in soup.find_all(string=lambda text: isinstance(text, str) and text.strip().startswith('<!--')):
        comment.extract()
    # 生成紧凑格式的 HTML（移除多余空白）
    compressed = soup.prettify(formatter="minimal")
    # 进一步移除换行和多余空格（保留标签内必要空格）
    compressed = re.sub(r'\n\s*', '', compressed)
    # 移除标签间多余空格（如 >  < → ><）
    compressed = re.sub(r'>\s+<', '><', compressed)
    return compressed

# ------------------- JS 压缩 -------------------
def CompressJS(js_str: str) -> str:
    """
    压缩 JS 字符串（移除空白、注释，保留语法正确性）
    :param js_str: 原始 JS 字符串
    :return: 压缩后的 JS 字符串
    """
    # jsbeautifier 反向配置：禁用所有格式化，实现压缩
    opts = jsbeautifier.default_options()
    opts.indent_size = 0          # 无缩进
    opts.space_in_empty_paren = False  # 空括号无空格
    opts.preserve_newlines = False     # 不保留换行
    opts.break_chained_methods = False # 链式方法不换行
    opts.keep_array_indentation = False # 数组不缩进
    opts.keep_function_indentation = False # 函数不缩进
    opts.space_after_anon_function = False # 匿名函数后无空格
    opts.space_before_conditional = False # 条件语句前无空格
    
    # 先美化再压缩（确保语法正确），再移除注释
    compressed = jsbeautifier.beautify(js_str, opts)
    # 移除单行注释（// ...）
    compressed = re.sub(r'//.*?$', '', compressed, flags=re.MULTILINE)
    # 移除多行注释（/* ... */）
    compressed = re.sub(r'/\*[\s\S]*?\*/', '', compressed)
    # 移除多余空白（保留语法必要空格）
    compressed = re.sub(r'\s+', ' ', compressed).strip()
    return compressed

# ------------------- CSS 压缩 -------------------
def CompressCSS(css_str: str) -> str:
    """
    压缩 CSS 字符串（移除空白、注释，保留语法正确性）
    :param css_str: 原始 CSS 字符串
    :return: 压缩后的 CSS 字符串
    """
    # cssbeautifier 反向配置：禁用所有格式化
    opts = cssbeautifier.default_options()
    opts.indent_size = 0          # 无缩进
    opts.newline_between_rules = False # 规则间不换行
    opts.space_around_selector_separator = False # 选择器分隔符无空格
    
    # 先美化再压缩，确保语法正确
    compressed = cssbeautifier.beautify(css_str, opts)
    # 移除单行注释（// ...）
    compressed = re.sub(r'//.*?$', '', compressed, flags=re.MULTILINE)
    # 移除多行注释（/* ... */）
    compressed = re.sub(r'/\*[\s\S]*?\*/', '', compressed)
    # 移除多余空白和换行
    compressed = re.sub(r'\s+', ' ', compressed).strip()
    # 移除 {}/: 前后多余空格（CSS 语法优化）
    compressed = re.sub(r'\s*{\s*', '{', compressed)
    compressed = re.sub(r'\s*}\s*', '}', compressed)
    compressed = re.sub(r'\s*:\s*', ':', compressed)
    compressed = re.sub(r'\s*;\s*', ';', compressed)
    return compressed

def IsInRule(Path, Method):
    try:
        App.url_map.match(Path, Method)
        return True
    except: return False

@asyncs
def MakeCompressLater(code, ext, sha256):
    if ext == ".html":
        openw(os.path.join("./FastTemp", sha256), CompressHtml(code))
    if ext == ".css":
        openw(os.path.join("./FastTemp", sha256), CompressCSS(code))
    if ext == ".js":
        openw(os.path.join("./FastTemp", sha256), CompressJS(code))

@App.after_request
def FastTempLoader(Response: flask.Response):
    try:
        if Response.status_code != 200: return Response
        if not Response: return Response
        if os.path.isfile(os.path.join("./FastTemp", hashlib.sha256(Response.get_data()).hexdigest())):
            return openr(os.path.join("./FastTemp", hashlib.sha256(Response.get_data()).hexdigest()))
        if IsInRule(flask.request.path, flask.request.method): return Response
        ContentType = Response.headers.get("Content-Type", "").split(";")[0].strip()
        ext = CUSTOM_MIME_MAP.get(ContentType)
        if not ext: return Response
        MakeCompressLater(Response.get_data(as_text=True), ext, sha256=hashlib.sha256(Response.get_data()).hexdigest())
        return Response
    except:
        pass
    return Response