Initial YakPanel commit

mod/base/web_conf/default_site.py

@@ -0,0 +1,136 @@
+import json
+import os
+import re
+from typing import Optional, Tuple
+from .util import listen_ipv6, write_file, read_file, service_reload
+
+
+def check_default():
+    vhost_path = "/www/server/panel/vhost"
+    nginx = vhost_path + '/nginx'
+    httpd = vhost_path + '/apache'
+    httpd_default = '''<VirtualHost *:80>
+    ServerAdmin webmaster@example.com
+    DocumentRoot "/www/server/apache/htdocs"
+    ServerName bt.default.com
+    <Directory "/www/server/apache/htdocs">
+        SetOutputFilter DEFLATE
+        Options FollowSymLinks
+        AllowOverride All
+        Order allow,deny
+        Allow from all
+        DirectoryIndex index.html
+    </Directory>
+</VirtualHost>
+'''
+
+    listen_ipv6_str = ''
+    if listen_ipv6():
+        listen_ipv6_str = "\n    listen [::]:80;"
+
+    nginx_default = '''server
+{
+    listen 80;%s
+    server_name _;
+    index index.html;
+    root /www/server/nginx/html;
+}''' % listen_ipv6_str
+
+    if not os.path.exists(httpd + '/0.default.conf') and not os.path.exists(httpd + '/default.conf'):
+        write_file(httpd + '/0.default.conf', httpd_default)
+    if not os.path.exists(nginx + '/0.default.conf') and not os.path.exists(nginx + '/default.conf'):
+        write_file(nginx + '/0.default.conf', nginx_default)
+
+
+def get_default_site() -> Tuple[Optional[str], Optional[str]]:
+    panel_path = "/www/server/panel"
+
+    old_ds_file = panel_path + "/data/defaultSite.pl"
+    new_ds_file = panel_path + "/data/mod_default_site.pl"
+    if os.path.exists(old_ds_file) and not os.path.exists(new_ds_file):
+        write_file(new_ds_file, json.dumps({
+            "name": read_file(old_ds_file).strip(),
+            "prefix": ''
+        }))
+
+    res = read_file(new_ds_file)
+    if not isinstance(res, str):
+        return None, None
+    data = json.loads(res)
+    return data["name"], data["prefix"]
+
+
+# site_name 传递None的时候，表示将默认站点设置给关闭
+# prefix 表示配置文件前缀， 如 "net_", 默认为空字符串
+# domain 站点的域名 如： "www.sss.com:8456"
+def set_default_site(site_name: Optional[str], prefix="", domain: str = None) -> Optional[str]:
+    # 清理旧的
+    old_default_name, old_prefix = get_default_site()
+    panel_path = "/www/server/panel"
+    default_site_save = panel_path + '/data/mod_default_site.pl'
+    if old_default_name:
+        ng_conf_file = os.path.join(panel_path, "vhost/nginx/{}{}.conf".format(old_prefix, old_default_name))
+        old_conf = read_file(ng_conf_file)
+        if isinstance(old_conf, str):
+            rep_listen_ds = re.compile(r"listen\s+.*default_server.*;")
+            new_conf_list = []
+            start_idx = 0
+            for tmp_res in rep_listen_ds.finditer(old_conf):
+                new_conf_list.append(old_conf[start_idx: tmp_res.start()])
+                new_conf_list.append(tmp_res.group().replace("default_server", ""))
+                start_idx = tmp_res.end()
+
+            new_conf_list.append(old_conf[start_idx:])
+
+            write_file(ng_conf_file, "".join(new_conf_list))
+
+        path = '/www/server/apache/htdocs/.htaccess'
+        if os.path.exists(path):
+            os.remove(path)
+
+    if site_name is None:
+        write_file(default_site_save, json.dumps({
+            "name": None,
+            "prefix": None
+        }))
+        service_reload()
+        return
+
+    # 处理新的
+    ap_path = '/www/server/apache/htdocs'
+    if os.path.exists(ap_path):
+        conf = '''<IfModule mod_rewrite.c>
+  RewriteEngine on
+  RewriteCond %{{HTTP_HOST}} !^127.0.0.1 [NC]
+  RewriteRule (.*) http://{}/$1 [L]
+</IfModule>'''.format(domain)
+
+        write_file(ap_path + '/.htaccess', conf)
+
+    ng_conf_file = os.path.join(panel_path, "vhost/nginx/{}{}.conf".format(prefix, site_name))
+    ng_conf = read_file(ng_conf_file)
+    if isinstance(ng_conf, str):
+        rep_listen = re.compile(r"listen[^;]*;")
+        new_conf_list = []
+
+        start_idx = 0
+        for tmp_res in rep_listen.finditer(ng_conf):
+            new_conf_list.append(ng_conf[start_idx: tmp_res.start()])
+            print(tmp_res.group())
+            if tmp_res.group().find("default_server") == -1:
+                new_conf_list.append(tmp_res.group()[:-1] + " default_server;")
+            else:
+                new_conf_list.append(tmp_res.group())
+            start_idx = tmp_res.end()
+
+        new_conf_list.append(ng_conf[start_idx:])
+
+        write_file(ng_conf_file, "".join(new_conf_list))
+
+    write_file(default_site_save, json.dumps({
+        "name": site_name,
+        "prefix": prefix
+    }))
+
+    service_reload()
+    return