| 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"] |
|
|
|
|
| |
| |
| |
| 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 |
|
|
