Initial YakPanel commit

mod/base/web_conf/util.py

@@ -0,0 +1,200 @@
+import os
+import sys
+from typing import Optional, Tuple, Callable
+
+if "/www/server/panel/class" not in sys.path:
+    sys.path.insert(0, "/www/server/panel/class")
+
+import public
+
+
+def webserver() -> Optional[str]:
+    if os.path.exists('/www/server/nginx/sbin/nginx'):
+        web_server = 'nginx'
+    elif os.path.exists('/www/server/apache/bin/apachectl'):
+        web_server = 'apache'
+    elif os.path.exists('/usr/local/lsws/bin/lswsctrl'):
+        web_server = 'openlitespeed'
+    else:
+        web_server = None
+    return web_server
+
+
+def check_server_config() -> Optional[str]:
+    w_s = webserver()
+    setup_path = "/www/server"
+    if w_s == 'nginx':
+        shell_str = (
+            "ulimit -n 8192; "
+            "{setup_path}/nginx/sbin/nginx -t -c {setup_path}/nginx/conf/nginx.conf"
+        ).format(setup_path=setup_path)
+        result: Tuple[str, str] = public.ExecShell(shell_str)
+        searchStr = 'successful'
+    elif w_s == 'apache':
+        shell_str = (
+            "ulimit -n 8192; "
+            "{setup_path}/apache/bin/apachectl -t"
+        ).format(setup_path=setup_path)
+        result: Tuple[str, str] = public.ExecShell(shell_str)
+        searchStr = 'Syntax OK'
+    else:
+        return None
+    if result[1].find(searchStr) == -1:
+        public.WriteLog("TYPE_SOFT", 'CONF_CHECK_ERR', (result[1],))
+        return result[1]
+
+
+def read_file(filename, mode='r') -> Optional[str]:
+    """
+    读取文件内容
+    @filename 文件名
+    return string(bin) 若文件不存在，则返回None
+    """
+    import os
+    if not os.path.exists(filename):
+        return None
+    fp = None
+    try:
+        fp = open(filename, mode=mode)
+        f_body = fp.read()
+    except:
+        return None
+    finally:
+        if fp and not fp.closed:
+            fp.close()
+    return f_body
+
+
+def write_file(filename: str, s_body: str, mode='w+') -> bool:
+    """
+    写入文件内容
+    @filename 文件名
+    @s_body 欲写入的内容
+    return bool 若文件不存在则尝试自动创建
+    """
+    try:
+        fp = open(filename, mode=mode)
+        fp.write(s_body)
+        fp.close()
+        return True
+    except:
+        try:
+            fp = open(filename, mode=mode, encoding="utf-8")
+            fp.write(s_body)
+            fp.close()
+            return True
+        except:
+            return False
+
+
+def debug_api_warp(fn):
+    def inner(*args, **kwargs):
+        try:
+            return fn(*args, **kwargs)
+        except:
+            public.print_log(public.get_error_info())
+            return {
+
+            }
+
+    return inner
+
+
+# 重载Web服务配置
+def service_reload():
+    setup_path = "/www/server"
+    if os.path.exists('{}/nginx/sbin/nginx'.format(setup_path)):
+        result = public.ExecShell('/etc/init.d/nginx reload')
+        if result[1].find('nginx.pid') != -1:
+            public.ExecShell('pkill -9 nginx && sleep 1')
+            public.ExecShell('/etc/init.d/nginx start')
+    elif os.path.exists('{}/apache/bin/apachectl'.format(setup_path)):
+        result = public.ExecShell('/etc/init.d/httpd reload')
+    else:
+        result = public.ExecShell('rm -f /tmp/lshttpd/*.sock* && /usr/local/lsws/bin/lswsctrl restart')
+    return result
+
+
+# 防正则转译
+def pre_re_key(input_str: str) -> str:
+    re_char = ['$', '(', ')', '*', '+', '.', '[', ']', '{', '}', '?', '^', '|', '\\']
+    res = []
+    for i in input_str:
+        if i in re_char:
+            res.append("\\" + i)
+        else:
+            res.append(i)
+    return "".join(res)
+
+
+def get_log_path() -> str:
+    log_path = public.readFile("{}/data/sites_log_path.pl".format(public.get_panel_path()))
+    if isinstance(log_path, str) and os.path.isdir(log_path):
+        return log_path
+    return public.GetConfigValue('logs_path')
+
+
+
+# 2024/4/18 上午9:44 域名编码转换
+def to_puny_code(domain):
+    try:
+        try:
+            import idna
+        except:
+            os.system("btpip install idna -I")
+            import idna
+
+        import re
+        match = re.search(u"[^u\0000-u\001f]+", domain)
+        if not match:
+            return domain
+        try:
+            if domain.startswith("*."):
+                return "*." + idna.encode(domain[2:]).decode("utf8")
+            else:
+                return idna.encode(domain).decode("utf8")
+        except:
+            return domain
+    except:
+        return domain
+
+
+# 2024/4/18 下午5:48 中文路径处理
+def to_puny_code_path(path):
+    if sys.version_info[0] == 2: path = path.encode('utf-8')
+    if os.path.exists(path): return path
+    import re
+    match = re.search(u"[\x80-\xff]+", path)
+    if not match: match = re.search(u"[\u4e00-\u9fa5]+", path)
+    if not match: return path
+    npath = ''
+    for ph in path.split('/'):
+        npath += '/' + to_puny_code(ph)
+    return npath.replace('//', '/')
+
+
+
+class _DB:
+
+    def __call__(self, table: str):
+        import db
+        with db.Sql() as t:
+            t.table(table)
+            return t
+
+
+DB = _DB()
+
+GET_CLASS = public.dict_obj
+
+listen_ipv6: Callable[[], bool] = public.listen_ipv6
+
+ExecShell: Callable = public.ExecShell
+
+
+def use_http2() -> bool:
+    versionStr = public.readFile('/www/server/nginx/version.pl')
+    if isinstance(versionStr, str):
+        if versionStr.find('1.8.1') == -1:
+            return True
+    return False