137 lines
4.5 KiB
Python
137 lines
4.5 KiB
Python
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
|