382 lines
9.4 KiB
Python
Executable File
382 lines
9.4 KiB
Python
Executable File
# coding:utf-8
|
||
|
||
import sys
|
||
import io
|
||
import os
|
||
import time
|
||
import re
|
||
|
||
web_dir = os.getcwd() + "/web"
|
||
if os.path.exists(web_dir):
|
||
sys.path.append(web_dir)
|
||
os.chdir(web_dir)
|
||
|
||
import core.mw as mw
|
||
|
||
app_debug = False
|
||
if mw.isAppleSystem():
|
||
app_debug = True
|
||
|
||
|
||
def getPluginName():
|
||
return 'gorse'
|
||
|
||
|
||
def getPluginDir():
|
||
return mw.getPluginDir() + '/' + getPluginName()
|
||
|
||
|
||
def getServerDir():
|
||
return mw.getServerDir() + '/' + getPluginName()
|
||
|
||
|
||
def getInitDFile():
|
||
current_os = mw.getOs()
|
||
if current_os == 'darwin':
|
||
return '/tmp/' + getPluginName()
|
||
|
||
if current_os.startswith('freebsd'):
|
||
return '/etc/rc.d/' + getPluginName()
|
||
|
||
return '/etc/init.d/' + getPluginName()
|
||
|
||
|
||
def getConf():
|
||
path = getServerDir() + "/gorse.toml"
|
||
return path
|
||
|
||
|
||
def getConfTpl():
|
||
path = getPluginDir() + "/config/gorse.toml"
|
||
return path
|
||
|
||
|
||
def getInitDTpl():
|
||
path = getPluginDir() + "/init.d/" + getPluginName() + ".tpl"
|
||
return path
|
||
|
||
|
||
def getArgs():
|
||
args = sys.argv[3:]
|
||
tmp = {}
|
||
args_len = len(args)
|
||
|
||
if args_len == 1:
|
||
t = args[0].strip('{').strip('}')
|
||
if t.strip() == '':
|
||
tmp = []
|
||
else:
|
||
t = t.split(':',1)
|
||
tmp[t[0]] = t[1]
|
||
tmp[t[0]] = t[1]
|
||
elif args_len > 1:
|
||
for i in range(len(args)):
|
||
t = args[i].split(':',1)
|
||
tmp[t[0]] = t[1]
|
||
return tmp
|
||
|
||
def checkArgs(data, ck=[]):
|
||
for i in range(len(ck)):
|
||
if not ck[i] in data:
|
||
return (False, mw.returnJson(False, '参数:(' + ck[i] + ')没有!'))
|
||
return (True, mw.returnJson(True, 'ok'))
|
||
|
||
def configTpl():
|
||
path = getPluginDir() + '/tpl'
|
||
pathFile = os.listdir(path)
|
||
tmp = []
|
||
for one in pathFile:
|
||
file = path + '/' + one
|
||
tmp.append(file)
|
||
return mw.getJson(tmp)
|
||
|
||
|
||
def readConfigTpl():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['file'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
content = mw.readFile(args['file'])
|
||
content = contentReplace(content)
|
||
return mw.returnJson(True, 'ok', content)
|
||
|
||
def getPidFile():
|
||
file = getConf()
|
||
content = mw.readFile(file)
|
||
rep = r'pidfile\s*(.*)'
|
||
tmp = re.search(rep, content)
|
||
return tmp.groups()[0].strip()
|
||
|
||
def status():
|
||
# pid_file = getPidFile()
|
||
# if not os.path.exists(pid_file):
|
||
# return 'stop'
|
||
|
||
cmd = "ps -ef|grep gorse |grep -v grep | grep -v python | grep -v mdserver-web | awk '{print $2}'"
|
||
data = mw.execShell(cmd)
|
||
|
||
if data[0] == '':
|
||
return 'stop'
|
||
return 'start'
|
||
|
||
|
||
def initRedisConf():
|
||
requirepass = ""
|
||
conf = mw.getServerDir() + '/redis/redis.conf'
|
||
content = mw.readFile(conf)
|
||
rep = r"^(requirepass)\s*([.0-9A-Za-z_& ~]+)"
|
||
tmp = re.search(rep, content, re.M)
|
||
if tmp:
|
||
requirepass = tmp.groups()[1]
|
||
|
||
port = "6379"
|
||
rep = r"^(port)\s*([.0-9A-Za-z_& ~]+)"
|
||
tmp = re.search(rep, content, re.M)
|
||
if tmp:
|
||
port = tmp.groups()[1]
|
||
|
||
return 'redis://:'+requirepass+'@127.0.0.1:'+port+'/3'
|
||
|
||
def contentReplace(content):
|
||
service_path = mw.getServerDir()
|
||
content = content.replace('{$ROOT_PATH}', mw.getFatherDir())
|
||
content = content.replace('{$SERVER_PATH}', service_path)
|
||
content = content.replace('{$CONFIG_ADMIN}', mw.getRandomString(6))
|
||
content = content.replace('{$CONFIG_PASS}', mw.getRandomString(10))
|
||
content = content.replace('{$CONFIG_REDIS}', initRedisConf())
|
||
return content
|
||
|
||
|
||
def initDreplace():
|
||
|
||
file_tpl = getInitDTpl()
|
||
service_path = mw.getServerDir()
|
||
|
||
initD_path = getServerDir() + '/init.d'
|
||
if not os.path.exists(initD_path):
|
||
mw.execShell("mkdir -p " + initD_path)
|
||
|
||
file_bin = initD_path + '/' + getPluginName()
|
||
|
||
# initd replace
|
||
if not os.path.exists(file_bin):
|
||
content = mw.readFile(file_tpl)
|
||
content = content.replace('{$SERVER_PATH}', service_path)
|
||
mw.writeFile(file_bin, content)
|
||
mw.execShell('chmod +x ' + file_bin)
|
||
|
||
# log
|
||
dataLog = getServerDir() + '/data'
|
||
if not os.path.exists(dataLog):
|
||
mw.execShell('chmod +x ' + file_bin)
|
||
|
||
# config replace
|
||
dst_conf = getServerDir() + '/gorse.toml'
|
||
dst_conf_init = getServerDir() + '/init.pl'
|
||
if not os.path.exists(dst_conf_init):
|
||
content = mw.readFile(getConfTpl())
|
||
content = content.replace('{$SERVER_PATH}', service_path)
|
||
content = contentReplace(content)
|
||
mw.writeFile(dst_conf, content)
|
||
mw.writeFile(dst_conf_init, 'ok')
|
||
|
||
# systemd
|
||
systemDir = mw.systemdCfgDir()
|
||
systemService = systemDir + '/' + getPluginName() + '.service'
|
||
if os.path.exists(systemDir) and not os.path.exists(systemService):
|
||
systemServiceTpl = getPluginDir() + '/init.d/' + getPluginName() + '.service.tpl'
|
||
service_path = mw.getServerDir()
|
||
content = mw.readFile(systemServiceTpl)
|
||
content = content.replace('{$SERVER_PATH}', service_path)
|
||
mw.writeFile(systemService, content)
|
||
mw.execShell('systemctl daemon-reload')
|
||
|
||
return file_bin
|
||
|
||
|
||
def gorseOp(method):
|
||
file = initDreplace()
|
||
|
||
# print(file)
|
||
|
||
current_os = mw.getOs()
|
||
if current_os == "darwin":
|
||
data = mw.execShell(file + ' ' + method)
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[1]
|
||
|
||
if current_os.startswith("freebsd"):
|
||
data = mw.execShell('service ' + getPluginName() + ' ' + method)
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[1]
|
||
|
||
data = mw.execShell('systemctl ' + method + ' ' + getPluginName())
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[1]
|
||
|
||
|
||
def start():
|
||
return gorseOp('start')
|
||
|
||
|
||
def stop():
|
||
return gorseOp('stop')
|
||
|
||
|
||
def restart():
|
||
status = gorseOp('restart')
|
||
log_file = runLog()
|
||
mw.execShell("echo '' > " + log_file)
|
||
return status
|
||
|
||
|
||
def reload():
|
||
return gorseOp('reload')
|
||
|
||
|
||
def getPort():
|
||
conf = getServerDir() + '/gorse.conf'
|
||
content = mw.readFile(conf)
|
||
|
||
rep = "^(" + 'port' + ')\\s*([.0-9A-Za-z_& ~]+)'
|
||
tmp = re.search(rep, content, re.M)
|
||
if tmp:
|
||
return tmp.groups()[1]
|
||
|
||
return '6379'
|
||
|
||
|
||
def initdStatus():
|
||
current_os = mw.getOs()
|
||
if current_os == 'darwin':
|
||
return "Apple Computer does not support"
|
||
|
||
if current_os.startswith('freebsd'):
|
||
initd_bin = getInitDFile()
|
||
if os.path.exists(initd_bin):
|
||
return 'ok'
|
||
|
||
shell_cmd = 'systemctl status ' + getPluginName() + ' | grep loaded | grep "enabled;"'
|
||
data = mw.execShell(shell_cmd)
|
||
if data[0] == '':
|
||
return 'fail'
|
||
return 'ok'
|
||
|
||
|
||
def initdInstall():
|
||
current_os = mw.getOs()
|
||
if current_os == 'darwin':
|
||
return "Apple Computer does not support"
|
||
|
||
# freebsd initd install
|
||
if current_os.startswith('freebsd'):
|
||
import shutil
|
||
source_bin = initDreplace()
|
||
initd_bin = getInitDFile()
|
||
shutil.copyfile(source_bin, initd_bin)
|
||
mw.execShell('chmod +x ' + initd_bin)
|
||
mw.execShell('sysrc ' + getPluginName() + '_enable="YES"')
|
||
return 'ok'
|
||
|
||
mw.execShell('systemctl enable ' + getPluginName())
|
||
return 'ok'
|
||
|
||
|
||
def initdUinstall():
|
||
current_os = mw.getOs()
|
||
if current_os == 'darwin':
|
||
return "Apple Computer does not support"
|
||
|
||
if current_os.startswith('freebsd'):
|
||
initd_bin = getInitDFile()
|
||
os.remove(initd_bin)
|
||
mw.execShell('sysrc ' + getPluginName() + '_enable="NO"')
|
||
return 'ok'
|
||
|
||
mw.execShell('systemctl disable ' + getPluginName())
|
||
return 'ok'
|
||
|
||
|
||
def runLog():
|
||
return getServerDir() + '/logs.pl'
|
||
|
||
def getGorseInfo():
|
||
conf_file = getConf()
|
||
content = mw.readFile(conf_file)
|
||
|
||
rdata = {}
|
||
|
||
rep = r'dashboard_user_name\s*=\s*"(.*)"'
|
||
tmp = re.search(rep, content)
|
||
tmp = re.search(rep, content, re.M)
|
||
if tmp:
|
||
rdata['dashboard_user_name'] = tmp.groups()[0]
|
||
|
||
|
||
rep = r'dashboard_password\s*=\s*"(.*)"'
|
||
tmp = re.search(rep, content)
|
||
tmp = re.search(rep, content, re.M)
|
||
if tmp:
|
||
rdata['dashboard_password'] = tmp.groups()[0]
|
||
|
||
rep = r'http_port\s*=\s*(.*)'
|
||
tmp = re.search(rep, content)
|
||
tmp = re.search(rep, content, re.M)
|
||
if tmp:
|
||
rdata['http_port'] = tmp.groups()[0]
|
||
|
||
rdata['ip'] = mw.getHostAddr()
|
||
return mw.returnJson(True,'ok', rdata)
|
||
|
||
def installPreInspection():
|
||
redis_path = mw.getServerDir() + "/redis"
|
||
if not os.path.exists(redis_path):
|
||
return "默认需要安装Redis"
|
||
|
||
mongodb_path = mw.getServerDir() + "/mongodb"
|
||
if not os.path.exists(mongodb_path):
|
||
return "默认需要安装MongoDB"
|
||
|
||
if not mw.isAppleSystem():
|
||
glibc_ver = mw.getGlibcVersion()
|
||
if float(glibc_ver) < 2.32:
|
||
return '当前libc{}过低，需要大于2.31'.format(glibc_ver)
|
||
return 'ok'
|
||
|
||
if __name__ == "__main__":
|
||
func = sys.argv[1]
|
||
if func == 'status':
|
||
print(status())
|
||
elif func == 'start':
|
||
print(start())
|
||
elif func == 'stop':
|
||
print(stop())
|
||
elif func == 'restart':
|
||
print(restart())
|
||
elif func == 'reload':
|
||
print(reload())
|
||
elif func == 'initd_status':
|
||
print(initdStatus())
|
||
elif func == 'initd_install':
|
||
print(initdInstall())
|
||
elif func == 'initd_uninstall':
|
||
print(initdUinstall())
|
||
elif func == 'install_pre_inspection':
|
||
print(installPreInspection())
|
||
elif func == 'conf':
|
||
print(getConf())
|
||
elif func == 'run_log':
|
||
print(runLog())
|
||
elif func == 'config_tpl':
|
||
print(configTpl())
|
||
elif func == 'read_config_tpl':
|
||
print(readConfigTpl())
|
||
elif func == 'info':
|
||
print(getGorseInfo())
|
||
else:
|
||
print('error')
|