| |
| |
| |
| |
| |
| |
| |
| |
| import json |
| import os |
| |
| |
| |
| import sys |
| import time |
| import datetime |
|
|
| if "/www/server/panel/class" not in sys.path: |
| sys.path.insert(0, "/www/server/panel/class") |
|
|
| os.chdir("/www/server/panel") |
| import public |
|
|
| if "/www/server/panel" not in sys.path: |
| sys.path.insert(0, "/www/server/panel") |
| from mod.project.nodejs.base import NodeJs |
|
|
|
|
| class main(NodeJs): |
|
|
| def __init__(self): |
| super(main, self).__init__() |
|
|
| |
| def create_project(self, get): |
| ''' |
| @name 创建项目 |
| @param get: dict_obj {} |
| get.project_type string 项目类型 nodejs/pm2/general 必传 |
| get.project_name string 项目名称 my_project 必传 |
| get.nodejs_version string node版本 v20.15.0 必传 |
| get.project_file string 项目启动文件 /www/wwwroot/my_project/server.js 必传 |
| get.project_cwd string 项目路径 /www/wwwroot/my_project 必传 |
| get.project_args string 启动参数 --debug 非必传 |
| get.env string 环境变量 key=value\nkey=value\n... 非必传 |
| get.run_user string 运行用户 www/root/... 必传 |
| get.port string 端口 4001 非必传 |
| get.release_firewall bool 是否放行防火墙 True/False 非必传 |
| get.is_power_on bool 是否开机启动 True/False 非必传 |
| get.max_memory_limit int 最大内存限制 4096 非必传 |
| get.bind_extranet bool 是否绑定外网 True/False 依赖于get.port 非必传 |
| get.domains list 域名列表 ["www.bt.cn", "bt.cn", ...] 非必传 |
| get.project_ps string 备注 ps 非必传 |
| ''' |
| self.set_self_get(get) |
| self.set_def_name(get.def_name) |
| get.project_cwd = get.get("project_cwd", None) |
| get.project_file = get.get("project_file", None) |
| if get.project_cwd is None: |
| self.ws_err_exit(False, 'project_cwd参数不能为空', code=2) |
| if not os.path.exists(get.project_cwd): |
| self.ws_err_exit(False, '{} 指定项目目录不存在'.format(get.project_cwd), code=2) |
| if not os.path.isdir(get.project_cwd): |
| self.ws_err_exit(False, '{} 指定项目目录不是一个目录'.format(get.project_cwd), code=2) |
| if get.project_file is None: |
| self.ws_err_exit(False, 'project_file参数不能为空', code=2) |
| if not os.path.exists(get.project_file): |
| self.ws_err_exit(False, '{} 指定项目启动文件不存在'.format(get.project_file), code=2) |
| if not os.path.isfile(get.project_file): |
| self.ws_err_exit(False, '{} 指定项目启动文件不是一个文件'.format(get.project_file), code=2) |
| get.project_name = get.get("project_name", None) |
| if get.project_name is None: |
| self.ws_err_exit(False, 'project_name参数不能为空', code=2) |
| get.project_name = public.xssencode2(get.project_name) |
| get.project_type = get.get("project_type", None) |
| if get.project_type is None: |
| self.ws_err_exit(False, 'project_type参数不能为空', code=2) |
| if get.project_type != "general": |
| self.ws_err_exit(False, '此模型仅支持general项目', code=2) |
| get.project_args = get.get("project_args", "") |
| get.run_user = get.get("run_user", "www") |
| get.env = get.get("env", "") |
| if get.env != "": |
| get.env = get.env.split("\n") |
| for env in get.env: |
| if not "=" in env: |
| self.ws_err_exit(False, "环境变量: {} 格式错误，请重新输入例如：key=value".format(env), code=2) |
| self.get_run_env(get) |
|
|
| get.nodejs_version = get.get("nodejs_version", None) |
| if get.nodejs_version is None: |
| self.ws_err_exit(False, 'nodejs_version参数不能为空', code=2) |
| get.release_firewall = get.get("release_firewall", False) |
| get.is_power_on = get.get("is_power_on", True) |
| get.max_memory_limit = get.get("max_memory_limit", 4096) |
|
|
| get._ws.send(json.dumps(self.wsResult(True, "正在构造启动脚本...", code=0))) |
| self.structure_start_script(get) |
| get._ws.send(json.dumps(self.wsResult(True, "启动脚本构造完成.", code=0))) |
| get._ws.send(json.dumps(self.wsResult(True, "正在写入项目必要配置文件...", code=1))) |
| project_id = self.create_site(get) |
| if project_id is None: |
| self.ws_err_exit(False, '创建网站失败，无法正常写入数据库，请尝试重新添加！', code=2) |
| self.set_config(get.project_name) |
| get._ws.send(json.dumps(self.wsResult(True, "配置文件写入完成.\r\n正在启动项目...", code=1))) |
| start_result = self.start_project(get) |
| if not start_result["status"]: |
| self.ws_err_exit(False, start_result["msg"] if "msg" in start_result else start_result["error_msg"], code=5) |
| get._ws.send(json.dumps(self.wsResult(True, "项目创建成功!", code=-1))) |
| get._ws.close() |
|
|
| |
| def structure_start_script(self, get): |
| ''' |
| @name 构造传统项目的启动脚本 |
| ''' |
| last_env = self.get_last_env(get.nodejs_version) |
| node_bin = self.get_node_bin(get.nodejs_version) |
| env = "\n".join(["export {}".format(x) for x in get.env.split('\n')]) |
| command = '''{last_env} |
| {run_env} |
| export NODE_PROJECT_NAME="{project_name}" |
| cd {project_cwd} |
| nohup {node_bin} {project_file} {project_args} &>> {log_file} & |
| echo $! > {pid_file} |
| '''.format( |
| last_env=last_env, |
| run_env=env, |
| project_cwd=get.project_cwd, |
| node_bin=node_bin, |
| project_file=get.project_file, |
| project_args=get.project_args, |
| log_file="{}/{}.log".format(self.node_logs_path, get.project_name), |
| pid_file="{}/{}.pid".format(self.node_pid_path, get.project_name), |
| project_name=get.project_name |
| ) |
|
|
| public.writeFile(os.path.join(get.project_cwd, "{}_start.sh".format(get.project_name)), command) |
| get.project_script = command |
|
|
| |
| def start_project(self, get): |
| ''' |
| @name 启动项目 |
| ''' |
| pid_file = "{}/{}.pid".format(self.node_pid_path, get.project_name) |
| if os.path.exists(pid_file): |
| self.stop_project(get) |
|
|
| project_find = self.get_project_find(get.project_name) |
| if not project_find: return public.returnResult(False, "项目不存在", code=5) |
|
|
| project_find = self.get_project_find(get.project_name) |
| if project_find['edate'] != "0000-00-00" and project_find['edate'] < datetime.datetime.today().strftime("%Y-%m-%d"): |
| return public.returnResult(False, "当前项目已过期，请重新设置项目到期时间", code=5) |
|
|
| self._update_project(get.project_name, project_find) |
| if not os.path.exists(project_find['path']): |
| error_msg = '启动失败，Nodejs项目{}，运行目录{}不存在!'.format(get.project_name, project_find['path']) |
| public.WriteLog(self.log_name, error_msg) |
| return public.returnResult(False, error_msg, code=5) |
|
|
| |
| nodejs_version = project_find['project_config']['nodejs_version'] |
| node_bin = self.get_node_bin(nodejs_version) |
| project_script = project_find['project_config']['project_script'].strip().replace(' ', ' ') |
| log_file = "{}/{}.log".format(project_find['project_config']["log_path"], project_find["name"]) |
| if not project_script: return public.returnResult(False, "未配置启动脚本", code=5) |
|
|
| last_env = self.get_last_env(nodejs_version, project_find['path']) |
| public.writeFile(log_file, '') |
| public.ExecShell('chmod 777 {}'.format(log_file)) |
| |
| if os.path.exists(project_script): |
| start_cmd = '''{last_env} |
| export NODE_PROJECT_NAME="{project_name}" |
| cd {project_cwd} |
| nohup {node_bin} {project_script} &>> {log_file} & |
| echo $! > {pid_file} |
| '''.format( |
| project_cwd=project_find['path'], |
| node_bin=node_bin, |
| project_script=project_script, |
| log_file=log_file, |
| pid_file=pid_file, |
| last_env=last_env, |
| project_name=get.project_name |
| ) |
| else: |
| get.nodejs_version = project_find['project_config']['nodejs_version'] |
| get.env = project_find['project_config']['env'] |
| get.project_cwd = project_find['project_config']['project_cwd'] |
| get.project_file = project_find['project_config']['project_file'] |
| get.project_args = project_find['project_config']['project_args'] |
| self.structure_start_script(get) |
| start_cmd = get.project_script |
| script_file = "{}/{}.sh".format(self.node_run_scripts, get.project_name) |
| |
| public.writeFile(script_file, start_cmd) |
| if os.path.exists(pid_file): os.remove(pid_file) |
|
|
| |
| public.ExecShell("chown -R {user}:{user} {project_cwd}".format( |
| user=project_find['project_config']['run_user'], |
| project_cwd=project_find['path'])) |
| public.ExecShell("chown -R www:www {}/vhost".format(self.nodejs_path)) |
| public.ExecShell("chmod 755 {} {} {}".format(self.nodejs_path, public.get_setup_path(), '/www')) |
| public.set_own(script_file, project_find['project_config']['run_user'], |
| project_find['project_config']['run_user']) |
| public.set_mode(script_file, 755) |
|
|
| |
| p = public.ExecShell("bash {}".format(script_file), |
| user=project_find['project_config']['run_user'], |
| env=os.environ.copy()) |
|
|
| time.sleep(1) |
| n = 0 |
| while n < 5: |
| if self.get_project_state_by_cwd(get.project_name): break |
| n += 1 |
| if not os.path.exists(pid_file): |
| p = '\n'.join(p) |
| public.writeFile(log_file, p, "a+") |
| if p.find('[Errno 0]') != -1: |
| if os.path.exists('{}/bt_security'.format(public.get_plugin_path())): |
| return public.returnResult(False, '启动命令被【堡塔防入侵】拦截，请关闭{}用户的防护'.format( |
| project_find['project_config']['run_user']), code=5) |
| return public.returnResult(False, '启动命令被未知安全软件拦截，请检查安装软件日志', code=5) |
| return public.returnResult(False, |
| '启动失败{}<script>setTimeout(function(){{$(".layui-layer-msg").css("width","800px");}},100)</script>'.format( |
| p), code=5) |
|
|
| if p[-1]: |
| public.returnResult(False, '启动失败{}'.format(p[-1]), code=5) |
|
|
| |
| try: |
| pid = int(public.readFile(pid_file)) |
| except: |
| return public.returnResult(False, '启动失败{}'.format(public.GetNumLines(log_file, 20)), code=5) |
| pids = self.get_project_pids(pid=pid) |
| if not pids: |
| if os.path.exists(pid_file): os.remove(pid_file) |
| return public.returnResult(False, '启动失败{}'.format(public.GetNumLines(log_file, 20)), code=5) |
|
|
| self.start_by_user(project_find["id"]) |
| return public.returnResult(True, '启动成功') |
|
|
| |
| def stop_project(self, get): |
| ''' |
| @name 停止项目 |
| @author hwliang<2021-08-09> |
| @param get<dict_obj>{ |
| project_name: string<项目名称> |
| } |
| @return dict |
| ''' |
| project_find = self.get_project_find(get.project_name) |
| if not project_find: return public.returnResult(False, '项目不存在', code=5) |
| project_find = self.get_project_find(get.project_name) |
| if project_find['edate'] != "0000-00-00" and project_find['edate'] < datetime.datetime.today().strftime("%Y-%m-%d"): |
| return public.returnResult(False, '当前项目已过期，请重新设置项目到期时间', code=5) |
| project_script = project_find['project_config']['project_script'].strip().replace(' ', ' ') |
| pid_file = "{}/{}.pid".format(self.node_pid_path, get.project_name) |
| if project_script.find('pm2 start') != -1: |
| nodejs_version = project_find['project_config']['nodejs_version'] |
| last_env = self.get_last_env(nodejs_version, project_find['path']) |
| project_script = project_script.replace('pm2 start', 'pm2 stop') |
| public.ExecShell('''{} |
| cd {} |
| {}'''.format(last_env, project_find['path'], project_script)) |
| else: |
| pid_file = "{}/{}.pid".format(self.node_pid_path, get.project_name) |
| if not os.path.exists(pid_file): return public.returnResult(False, '项目未启动', code=5) |
| data = public.readFile(pid_file) |
| if isinstance(data, str) and data: |
| pid = int(data) |
| pids = self.get_project_pids(pid=pid) |
| else: |
| return public.returnResult(False, '项目未启动', code=5) |
| if not pids: return public.returnResult(False, '项目未启动', code=5) |
| self.kill_pids(pids=pids) |
| if os.path.exists(pid_file): os.remove(pid_file) |
| time.sleep(0.5) |
| pids = self.get_project_state_by_cwd(get.project_name) |
| if pids: self.kill_pids(pids=pids) |
|
|
| self.stop_by_user(project_find["id"]) |
| return public.returnResult(True, '停止成功') |
|
|
| |
| def restart_project(self, get): |
| ''' |
| @name 重启项目 |
| @author hwliang<2021-08-09> |
| @param get<dict_obj>{ |
| project_name: string<项目名称> |
| } |
| @return dict |
| ''' |
| project_find = self.get_project_find(get.project_name) |
| if project_find: |
| if project_find['edate'] != "0000-00-00" and project_find['edate'] < datetime.datetime.today().strftime("%Y-%m-%d"): |
| return public.returnResult(False, '当前项目已过期，请重新设置项目到期时间', code=5) |
| res = self.stop_project(get) |
| if not res['status']: return res |
| res = self.start_project(get) |
| if not res['status']: return res |
| return public.returnResult(True, '重启成功') |
|
|
