Download bt-source/panel/class/projectModel/nodejsModel.py from SZFSPC/apple: direct link, hf CLI and curl.
- Browser
- Download file 112 kB
-
https://huggingface.co/spaces/SZFSPC/apple/resolve/main/bt-source/panel/class/projectModel/nodejsModel.py
- Command line
-
hf download hf://spaces/SZFSPC/apple/bt-source/panel/class/projectModel/nodejsModel.py
-
curl -L -o nodejsModel.py https://huggingface.co/spaces/SZFSPC/apple/resolve/main/bt-source/panel/class/projectModel/nodejsModel.py
112 kB
| # coding: utf-8 | |
| # ------------------------------------------------------------------- | |
| # 宝塔Linux面板 | |
| # ------------------------------------------------------------------- | |
| # Copyright (c) 2015-2099 宝塔软件(http://bt.cn) All rights reserved. | |
| # ------------------------------------------------------------------- | |
| # Author: hwliang <hwl@bt.cn> | |
| # ------------------------------------------------------------------- | |
| # ------------------------------ | |
| # node.js模型 | |
| # ------------------------------ | |
| import os, sys, re, json, shutil, psutil, time | |
| from typing import Optional | |
| import datetime | |
| from projectModel.base import projectBase | |
| import public | |
| try: | |
| from BTPanel import cache | |
| except: | |
| pass | |
| class main(projectBase): | |
| _panel_path = public.get_panel_path() | |
| _nodejs_plugin_path = public.get_plugin_path('nodejs') | |
| _nodejs_path = '{}/nodejs'.format(public.get_setup_path()) | |
| _log_name = '项目管理' | |
| _npm_exec_log = '{}/logs/npm-exec.log'.format(_panel_path) | |
| _node_pid_path = '{}/vhost/pids'.format(_nodejs_path) | |
| _node_logs = '{}/vhost/logs'.format(_nodejs_path) | |
| _node_logs_path = "/www/wwwlogs/nodejs" | |
| _node_run_scripts = '{}/vhost/scripts'.format(_nodejs_path) | |
| _pids = None | |
| _vhost_path = '{}/vhost'.format(_panel_path) | |
| _www_home = '/home/www' | |
| __log_split_script_py = public.get_panel_path() + '/script/run_log_split.py' | |
| def __init__(self): | |
| if not os.path.exists(self._node_run_scripts): | |
| os.makedirs(self._node_run_scripts, 493) | |
| if not os.path.exists(self._node_pid_path): | |
| os.makedirs(self._node_pid_path, 493) | |
| if not os.path.exists(self._node_logs_path): | |
| os.makedirs(self._node_logs_path, 493) | |
| if not os.path.exists(self._www_home): | |
| os.makedirs(self._www_home, 493) | |
| public.set_own(self._www_home, 'www') | |
| def get_exec_logs(self,get): | |
| ''' | |
| @name 获取执行日志 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj> | |
| @return string | |
| ''' | |
| if not os.path.exists(self._npm_exec_log): return public.returnMsg(False, 'NODE_NOT_EXISTS') | |
| return public.GetNumLines(self._npm_exec_log, 20) | |
| def get_project_list(self, get): | |
| ''' | |
| @name 获取项目列表 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| if not 'p' in get: get.p = 1 | |
| if not 'limit' in get: get.limit = 20 | |
| if not 'callback' in get: get.callback = '' | |
| if not 'order' in get: get.order = 'id desc' | |
| type_id = None | |
| if "type_id" in get: | |
| try: | |
| type_id = int(get.type_id) | |
| except: | |
| type_id = None | |
| if 'search' in get: | |
| get.project_name = get.search.strip() | |
| search = "%{}%".format(get.project_name) | |
| if type_id is None: | |
| count = public.M('sites').where('project_type=? AND (name LIKE ? OR ps LIKE ?)', ('Node', search, search)).count() | |
| data = public.get_page(count, int(get.p), int(get.limit), get.callback) | |
| data['data'] = public.M('sites').where('project_type=? AND (name LIKE ? OR ps LIKE ?)', ('Node', search, search)).limit(data['shift'] + ',' + data['row']).order(get.order).select() | |
| else: | |
| count = public.M('sites').where('project_type=? AND (name LIKE ? OR ps LIKE ?) AND type_id = ?', | |
| ('Node', search, search, type_id)).count() | |
| data = public.get_page(count, int(get.p), int(get.limit), get.callback) | |
| data['data'] = public.M('sites').where('project_type=? AND (name LIKE ? OR ps LIKE ?) AND type_id = ?', ('Node', search, search, type_id)).limit(data['shift'] + ',' + data['row']).order(get.order).select() | |
| else: | |
| if type_id is None: | |
| count = public.M('sites').where('project_type=?', 'Node').count() | |
| data = public.get_page(count, int(get.p), int(get.limit), get.callback) | |
| data['data'] = public.M('sites').where('project_type=?', 'Node').limit(data['shift'] + ',' + data['row']).order(get.order).select() | |
| else: | |
| count = public.M('sites').where('project_type=? AND type_id = ?', ('Node', type_id)).count() | |
| data = public.get_page(count, int(get.p), int(get.limit), get.callback) | |
| data['data'] = public.M('sites').where('project_type=? AND type_id = ?', ('Node', type_id)).limit(data['shift'] + ',' + data['row']).order(get.order).select() | |
| if isinstance(data["data"], str) and data["data"].startswith("error"): | |
| raise public.PanelError("数据库查询错误:" + data["data"]) | |
| for i in range(len(data['data'])): | |
| data['data'][i] = self.get_project_stat(data['data'][i]) | |
| return data | |
| def get_ssl_end_date(self,project_name): | |
| ''' | |
| @name 获取SSL信息 | |
| @author hwliang<2021-08-09> | |
| @param project_name <string> 项目名称 | |
| @return dict | |
| ''' | |
| import data | |
| return data.data().get_site_ssl_info('node_{}'.format(project_name)) | |
| def is_install_nodejs(self,get): | |
| ''' | |
| @name 是否安装nodejs版本管理器 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj> 请求数据 | |
| @return bool | |
| ''' | |
| return os.path.exists(self._nodejs_plugin_path) | |
| def get_nodejs_version(self,get): | |
| ''' | |
| @name 获取已安装的nodejs版本 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj> 请求数据 | |
| @return list | |
| ''' | |
| nodejs_list = [] | |
| if not os.path.exists(self._nodejs_path): return nodejs_list | |
| for v in os.listdir(self._nodejs_path): | |
| if v[0] != 'v' or v.find('.') == -1: continue | |
| node_path = os.path.join(self._nodejs_path,v) | |
| node_bin = '{}/bin/node'.format(node_path) | |
| if not os.path.exists(node_bin): | |
| if os.path.exists(node_path + '/bin'): | |
| public.ExecShell('rm -rf {}'.format(node_path)) | |
| continue | |
| nodejs_list.append(v) | |
| return nodejs_list | |
| def get_run_list(self,get): | |
| ''' | |
| @name 获取node项目启动列表 | |
| @author hwliang<2021-08-10> | |
| @param get<dict_obj>{ | |
| project_cwd: string<项目目录> | |
| } | |
| ''' | |
| project_cwd = get.project_cwd.strip() | |
| if not os.path.exists(project_cwd): return public.return_error('项目目录不存在!') | |
| package_file = '{}/package.json'.format(project_cwd) | |
| if not os.path.exists(package_file): return {} #public.return_error('没有在项目目录中找到package.json配置文件!') | |
| try: | |
| package_info = json.loads(public.readFile(package_file)) | |
| except: | |
| return {} | |
| if not 'scripts' in package_info: return {} # public.return_error('没有在项目配置文件package.json中找到scripts配置项!') | |
| if not package_info['scripts']: return {} # public.return_error('没有找到可用的启动选项!') | |
| return package_info['scripts'] | |
| def get_npm_bin(self,nodejs_version): | |
| ''' | |
| @name 获取指定node版本的npm路径 | |
| @author hwliang<2021-08-10> | |
| @param nodejs_version<string> nodejs版本 | |
| @return string | |
| ''' | |
| npm_path = '{}/{}/bin/npm'.format(self._nodejs_path,nodejs_version) | |
| if not os.path.exists(npm_path): return False | |
| return npm_path | |
| def get_yarn_bin(self,nodejs_version): | |
| ''' | |
| @name 获取指定node版本的yarn路径 | |
| @author hwliang<2021-08-28> | |
| @param nodejs_version<string> nodejs版本 | |
| @return string | |
| ''' | |
| yarn_path = '{}/{}/bin/yarn'.format(self._nodejs_path,nodejs_version) | |
| if not os.path.exists(yarn_path): | |
| from plugin.nodejs.nodejs_main import nodejs_main | |
| # nodejs_main().install_module({"version": nodejs_version, "module": "yarn"}) | |
| nodejs_main().install_module(public.to_dict_obj({"version": nodejs_version, "module": "yarn"})) | |
| if not os.path.exists(yarn_path): | |
| return False | |
| return yarn_path | |
| def get_pnpm_bin(self, nodejs_version): | |
| ''' | |
| @name 获取指定node版本的yarn路径 | |
| @author 包子<2021-08-28> | |
| @param nodejs_version<string> nodejs版本 | |
| @return string | |
| ''' | |
| pnpm_path = '{}/{}/bin/pnpm'.format(self._nodejs_path, nodejs_version) | |
| if not os.path.exists(pnpm_path): | |
| if not os.path.exists('/www/server/panel/plugin/nodejs/nodejs_main.py'): | |
| return False | |
| if '/www/server/panel' not in sys.path: | |
| sys.path.insert(0, '/www/server/panel') | |
| from plugin.nodejs.nodejs_main import nodejs_main | |
| _v = int(nodejs_version.split(".", 1)[0][1:]) | |
| if 12 <= _v < 14: | |
| _pnpm = "pnpm@6" | |
| elif 14 <= _v < 16: | |
| _pnpm = "pnpm@7" | |
| elif _v >= 16: | |
| _pnpm = "pnpm@8" | |
| else: | |
| return False | |
| nodejs_main().install_module({"version": nodejs_version, "module": _pnpm}) | |
| if not os.path.exists(pnpm_path): | |
| return False | |
| return pnpm_path | |
| def get_node_bin(self,nodejs_version): | |
| ''' | |
| @name 获取指定node版本的node路径 | |
| @author hwliang<2021-08-10> | |
| @param nodejs_version<string> nodejs版本 | |
| @return string | |
| ''' | |
| node_path = '{}/{}/bin/node'.format(self._nodejs_path,nodejs_version) | |
| if not os.path.exists(node_path): return False | |
| return node_path | |
| def get_last_env(self,nodejs_version,project_cwd = None): | |
| ''' | |
| @name 获取前置环境变量 | |
| @author hwliang<2021-08-25> | |
| @param nodejs_version<string> Node版本 | |
| @return string | |
| ''' | |
| nodejs_bin_path = '{}/{}/bin'.format(self._nodejs_path,nodejs_version) | |
| if project_cwd: | |
| _bin = '{}/node_modules/.bin'.format(project_cwd) | |
| if os.path.exists(_bin): | |
| nodejs_bin_path = _bin + ':' + nodejs_bin_path | |
| last_env = '''PATH={nodejs_bin_path}:/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin | |
| export PATH | |
| '''.format(nodejs_bin_path = nodejs_bin_path) | |
| return last_env | |
| def install_packages(self, get): | |
| ''' | |
| @name 安装指定项目的依赖包 | |
| @author hwliang<2021-08-10> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| return dict | |
| ''' | |
| if not getattr(get, "from_create", False): | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: return public.return_error('指定项目不存在!') | |
| else: | |
| project_find = { | |
| "name": get.project_name, | |
| "path": get.project_cwd, | |
| "project_config": {"nodejs_version":get.nodejs_version} | |
| } | |
| if not os.path.exists(project_find['path']): return public.return_error('项目目录不存在!') | |
| package_file = '{}/package.json'.format(project_find['path']) | |
| if not os.path.exists(package_file): return public.return_error('没有在项目目录中找到package.json配置文件!') | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| pkg_manager = project_find['project_config']["pkg_manager"] if "pkg_manager" in project_find['project_config'] else "null" | |
| if "pkg_manager" in get and get.pkg_manager in ("npm", "pnpm", "yarn", "null"): | |
| pkg_manager = get.pkg_manager | |
| package_lock_file = '{}/package-lock.json'.format(project_find['path']) | |
| node_modules_path = '{}/node_modules'.format(project_find['path']) | |
| # 已经安装过的依赖包的情况下,可能存在不同node版本导致的问题,可能需要重新构建依赖包 | |
| rebuild = False | |
| if os.path.exists(package_lock_file) and os.path.exists(node_modules_path): | |
| rebuild = True | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| yarn_bin = self.get_yarn_bin(nodejs_version) | |
| pnmp_bin = self.get_pnpm_bin(nodejs_version) | |
| if not npm_bin and not yarn_bin and not pnmp_bin: | |
| return public.return_error('指定nodejs版本不存在!') | |
| registry_name, registry_url = self.test_registry_url() | |
| public.writeFile(self._npm_exec_log, "正在检查镜像源连接情况...\n", "a") | |
| if not registry_name and not registry_url: | |
| return public.return_error('无法连接任何镜像源! 请检查网络连接情况!') | |
| public.writeFile(self._npm_exec_log, "发现可用镜像源【{}】\n".format(registry_name),"a") | |
| set_registry_sh = "%s config set registry {}".format(registry_url) | |
| public.writeFile(self._npm_exec_log, "正在安装依赖包...\n", "a") | |
| if yarn_bin and pkg_manager in ("yarn","null"): | |
| if os.path.exists(package_lock_file): | |
| os.remove(package_lock_file) | |
| public.ExecShell("{} && {} config set strict-ssl false".format(set_registry_sh % yarn_bin, yarn_bin)) | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} install &>> {};echo '|-Successify --- 命令已执行! ---' > {}".format(project_find['path'],yarn_bin,self._npm_exec_log,self._npm_exec_log)) | |
| elif pnmp_bin and pkg_manager in ("pnpm","null"): | |
| public.ExecShell("{} && {} config set strict-ssl false".format(set_registry_sh % pnmp_bin, pnmp_bin)) | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} install &>> {};echo '|-Successify --- 命令已执行! ---' > {}".format(project_find['path'],pnmp_bin,self._npm_exec_log,self._npm_exec_log)) | |
| elif npm_bin and pkg_manager in ("npm","null"): | |
| public.ExecShell("{} && {} config set strict-ssl false".format(set_registry_sh % npm_bin, npm_bin)) | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} install &>> {};echo '|-Successify --- 命令已执行! ---' > {}".format(project_find['path'],npm_bin,self._npm_exec_log, self._npm_exec_log)) | |
| else: | |
| _v = int(nodejs_version.split(".", 1)[0][1:]) | |
| if _v < 12 and pkg_manager == "pnpm": | |
| return public.return_error("pnpm不支持Nodejs12以下的版本") | |
| return public.return_error('没有指定的包管理器,无法安装依赖包!') | |
| public.WriteLog(self._log_name, 'Node项目:{}, 安装依赖包完成!'.format(project_find['name'])) | |
| if rebuild: # 重新构建已安装模块? | |
| self.rebuild_project(get.project_name) | |
| return public.return_data(True, '安装依赖包成功!') | |
| def test_registry_url(): | |
| from mod.project.nodejs.utils import test_registry_url | |
| return test_registry_url() | |
| def update_packages(self,get): | |
| ''' | |
| @name 更新指定项目的依赖包 | |
| @author hwliang<2021-08-10> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| return dict | |
| ''' | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: return public.return_error('指定项目不存在!') | |
| if not os.path.exists(project_find['path']): return public.return_error('项目目录不存在!') | |
| package_file = '{}/package.json'.format(project_find['path']) | |
| if not os.path.exists(package_file): return public.return_error('没有在项目目录中找到package.json配置文件!') | |
| package_lock_file = '{}/package-lock.json'.format(project_find['path']) | |
| if not os.path.exists(package_lock_file): return public.return_error('请先安装依赖包!') | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| if not npm_bin: | |
| return public.return_error('指定nodejs版本不存在!') | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} update &> {}".format(project_find['path'],npm_bin,self._npm_exec_log)) | |
| public.WriteLog(self._log_name, '项目[{}]更新所有依赖包'.format(get.project_name)) | |
| return public.return_data(True,'依赖包更新成功!') | |
| def reinstall_packages(self,get): | |
| ''' | |
| @name 重新安装指定项目的依赖包 | |
| @author hwliang<2021-08-10> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| return dict | |
| ''' | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: return public.return_error('指定项目不存在!') | |
| if not os.path.exists(project_find['path']): return public.return_error('项目目录不存在!') | |
| package_file = '{}/package.json'.format(project_find['path']) | |
| if not os.path.exists(package_file): return public.return_error('没有在项目目录中找到package.json配置文件!') | |
| package_lock_file = '{}/package-lock.json'.format(project_find['path']) | |
| if os.path.exists(package_lock_file): os.remove(package_lock_file) | |
| package_path = '{}/node_modules' | |
| if os.path.exists(package_path): shutil.rmtree(package_path) | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| if not npm_bin: | |
| return public.return_error('指定nodejs版本不存在!') | |
| public.WriteLog(self._log_name,'Node项目:{},已重装所有依赖包') | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} install &> {}".format(project_find['path'],npm_bin,self._npm_exec_log)) | |
| return public.return_data(True,'依赖包重装成功!') | |
| def get_project_modules(self,get): | |
| ''' | |
| @name 获取指定项目的依赖包列表 | |
| @author hwliang<2021-08-10> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| project_cwd: string<项目目录> 可选 | |
| } | |
| return list | |
| ''' | |
| if not 'project_cwd' in get: | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: return public.return_error('指定项目不存在!') | |
| project_cwd = project_find['path'] | |
| else: | |
| project_cwd = get.project_cwd | |
| mod_path = os.path.join(project_cwd,'node_modules') | |
| modules = [] | |
| if not os.path.exists(mod_path): return modules | |
| for mod_name in os.listdir(mod_path): | |
| try: | |
| mod_pack_file = os.path.join(mod_path,mod_name,'package.json') | |
| if not os.path.exists(mod_pack_file): continue | |
| mod_pack_info = json.loads(public.readFile(mod_pack_file)) | |
| pack_info = { | |
| "name": mod_name, | |
| "version": mod_pack_info['version'], | |
| "description":mod_pack_info['description'], | |
| "license": mod_pack_info['license'] if 'license' in mod_pack_info else 'NULL', | |
| "homepage": mod_pack_info['homepage'] if 'homepage' in mod_pack_info else 'NULL', | |
| } | |
| modules.append(pack_info) | |
| except: | |
| continue | |
| return modules | |
| def install_module(self,get): | |
| ''' | |
| @name 安装指定模块 | |
| @author hwliang<2021-08-10> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| mod_name: string<模块名称> | |
| } | |
| @return dict | |
| ''' | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: return public.return_error('指定项目不存在!') | |
| project_cwd = project_find['path'] | |
| mod_name = get.mod_name | |
| filename = '{}/node_modules/{}/package.json'.format(project_cwd,mod_name) | |
| if os.path.exists(filename): return public.return_error('指定模块已经安装过了!') | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| pkg_manager = project_find['project_config']["pkg_manager"] if "pkg_manager" in project_find['project_config'] else "null" | |
| if "pkg_manager" in get and get.pkg_manager in ("npm", "pnpm", "yarn", "null"): | |
| pkg_manager = get.pkg_manager | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| yarn_bin = self.get_yarn_bin(nodejs_version) | |
| pnpm_bin = self.get_pnpm_bin(nodejs_version) | |
| if not npm_bin and not yarn_bin and not pnpm_bin: | |
| return public.return_error('指定nodejs版本不存在!') | |
| if yarn_bin and pkg_manager in ("yarn", "null"): | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} add {} &> {}".format(project_find['path'],yarn_bin,mod_name,self._npm_exec_log)) | |
| elif npm_bin and pkg_manager in ("npm", "null"): | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} install {} &> {}".format(project_find['path'],npm_bin,mod_name,self._npm_exec_log)) | |
| elif pnpm_bin and pkg_manager in ("pnpm", "null"): | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} install {} &> {}".format(project_find['path'], pnpm_bin,mod_name, self._npm_exec_log)) | |
| else: | |
| _v = int(nodejs_version.split(".", 1)[0][1:]) | |
| if _v < 12 and pkg_manager == "pnpm": | |
| return public.return_error("pnpm不支持Nodejs12以下的版本") | |
| return public.return_error("没有指定的包管理器,无法安装依赖包!") | |
| if not os.path.exists(filename): return public.return_error('指定模块安装失败!') | |
| public.WriteLog(self._log_name,'Node项目{} , {}模块安装完成!'.format(get.project_name,mod_name)) | |
| return public.return_data(True,'安装成功!') | |
| def uninstall_module(self,get): | |
| ''' | |
| @name 卸载指定模块 | |
| @author hwliang<2021-04-08> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| mod_name: string<模块名称> | |
| } | |
| @return dict | |
| ''' | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: return public.return_error('指定项目不存在!') | |
| project_cwd = project_find['path'] | |
| mod_name = get.mod_name | |
| filename = '{}/node_modules/{}/package.json'.format(project_cwd,mod_name) | |
| if not os.path.exists(filename): return public.return_error('指定模块未安装!') | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| pkg_manager = project_find['project_config']["pkg_manager"] if "pkg_manager" in project_find['project_config'] else "null" | |
| if "pkg_manager" in get and get.pkg_manager in ("npm", "pnpm", "yarn", "null"): | |
| pkg_manager = get.pkg_manager | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| yarn_bin = self.get_yarn_bin(nodejs_version) | |
| pnpm_bin = self.get_pnpm_bin(nodejs_version) | |
| if not npm_bin and not yarn_bin and not pnpm_bin: | |
| return public.return_error('指定nodejs版本不存在!') | |
| if yarn_bin and pkg_manager in ("null", "yarn"): | |
| result = public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} remove {}".format(project_find['path'],yarn_bin,mod_name)) | |
| elif npm_bin and pkg_manager in ("null", "npm"): | |
| result = public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} uninstall {}".format(project_find['path'],npm_bin,mod_name)) | |
| elif pnpm_bin and pkg_manager in ("null", "pnpm"): | |
| result = public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} uninstall {}".format(project_find['path'],pnpm_bin,mod_name)) | |
| else: | |
| _v = int(nodejs_version.split(".", 1)[0][1:]) | |
| if _v < 12 and pkg_manager == "pnpm": | |
| return public.return_error("pnpm不支持Nodejs12以下的版本") | |
| return public.return_error("没有指定的包管理器,无法卸载依赖包!") | |
| if os.path.exists(filename): | |
| result = "\n".join(result) | |
| if result.find('looking for funding') != -1: | |
| return public.return_error("此模块被其它已安装模块依赖,无法卸载!") | |
| return public.return_error("无法卸载此模块!") | |
| public.WriteLog(self._log_name,'Node项目{} , {}模块卸载完成!'.format(get.project_name,mod_name)) | |
| return public.return_data(True,'模块卸载成功!') | |
| def upgrade_module(self,get): | |
| ''' | |
| @name 更新指定模块 | |
| @author hwliang<2021-08-10> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| mod_name: string<模块名称> | |
| } | |
| @return dict | |
| ''' | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: return public.return_error('指定项目不存在!') | |
| project_cwd = project_find['path'] | |
| mod_name = get.mod_name | |
| filename = '{}/node_modules/{}/package.json'.format(project_cwd,mod_name) | |
| if not os.path.exists(filename): return public.return_error('指定模块未安装!') | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| if not npm_bin: | |
| return public.return_error('指定nodejs版本不存在!') | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} update {} &> {}".format(project_find['path'],npm_bin,mod_name,self._npm_exec_log)) | |
| public.WriteLog(self._log_name,'Node项目{} , {}模块更新完成!'.format(get.project_name,mod_name)) | |
| return public.return_data(True,'模块更新成功!') | |
| def create_project(self,get): | |
| ''' | |
| @name 创建新的项目 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| project_cwd: string<项目目录> | |
| project_script: string<项目脚本> | |
| project_ps: string<项目备注信息> | |
| bind_extranet: int<是否绑定外网> 1:是 0:否 | |
| domains: list<域名列表> ["domain1:80","domain2:80"] // 在bind_extranet=1时,需要填写 | |
| is_power_on: int<是否开机启动> 1:是 0:否 | |
| run_user: string<运行用户> | |
| max_memory_limit: int<最大内存限制> // 超出此值项目将被强制重启 | |
| nodejs_version: string<nodejs版本> | |
| pkg_manager:string["npm", "pnpm", "yarn"], 选择包管理工具 | |
| } | |
| @return dict | |
| ''' | |
| if not isinstance(get,public.dict_obj): return public.return_error('参数类型错误,需要dict_obj对像') | |
| if not self.is_install_nodejs(get): | |
| return public.return_error('请先安装nodejs版本管理器,并安装至少1个node.js版本') | |
| if not 'project_script' in get: | |
| return public.return_error("请选择【启动选项】") | |
| project_name = get.project_name.strip() | |
| if not re.match("^\w+$",project_name): | |
| return public.return_error('项目名称格式不正确,支持字母、数字、下划线,表达式: ^[0-9A-Za-z_]$') | |
| if public.M('sites').where('name=?',(get.project_name,)).count(): | |
| return public.return_error('指定项目名称已存在: {}'.format(get.project_name)) | |
| get.project_cwd = get.project_cwd.strip() | |
| if get.project_cwd[-1] == '/': | |
| get.project_cwd = get.project_cwd[:-1] | |
| if not os.path.exists(get.project_cwd): | |
| return public.return_error('项目目录不存在: {}'.format(get.project_cwd)) | |
| # 端口占用检测 | |
| if self.check_port_is_used(get.get('port/port')): | |
| return public.return_error('指定端口已被其它应用占用,请修改您的项目配置使用其它端口, 端口: {}'.format(get.port)) | |
| domains = [] | |
| if get.bind_extranet == 1: | |
| domains = get.domains | |
| if not public.is_apache_nginx(): return public.return_error('需要安装Nginx或Apache才能使用外网映射功能') | |
| for domain in domains: | |
| if "[" in domain and "]" in domain: # IPv6格式特殊处理 | |
| if "]:" in domain: | |
| domain_arr = domain.rsplit(":", 1) | |
| else: | |
| domain_arr = [domain] | |
| else: | |
| domain_arr = domain.split(':') | |
| if public.M('domain').where('name=?',domain_arr[0]).count(): | |
| return public.return_error('指定域名已存在: {}'.format(domain)) | |
| if "pkg_manager" not in get or get.pkg_manager not in ("npm", "pnpm", "yarn"): | |
| return public.return_error('可选的包管理器为: 【npm, pnpm, yarn】') | |
| else: | |
| get.from_create = True | |
| res = self.install_packages(get) | |
| if not res["status"]: | |
| return res | |
| pdata = { | |
| 'name': get.project_name, | |
| 'path': get.project_cwd, | |
| 'ps': get.project_ps, | |
| 'status':1, | |
| 'type_id':0, | |
| 'project_type': 'Node', | |
| 'project_config': json.dumps( | |
| { | |
| 'project_name': get.project_name, | |
| 'project_cwd': get.project_cwd, | |
| 'project_script': get.project_script, | |
| 'bind_extranet': get.bind_extranet, | |
| 'domains': [], | |
| 'is_power_on': get.is_power_on, | |
| 'run_user': get.run_user, | |
| 'max_memory_limit': get.max_memory_limit, | |
| 'nodejs_version': get.nodejs_version, | |
| 'port': int(get.port), | |
| 'log_path': self._node_logs_path, | |
| 'pkg_manager': get.pkg_manager | |
| } | |
| ), | |
| 'addtime': public.getDate() | |
| } | |
| project_id = public.M('sites').insert(pdata) | |
| if get.bind_extranet == 1: | |
| format_domains = [] | |
| for domain in domains: | |
| if domain.find(':') == -1: domain += ':80' | |
| format_domains.append(domain) | |
| get.domains = format_domains | |
| self.project_add_domain(get) | |
| self.set_config(get.project_name) | |
| public.WriteLog(self._log_name,'添加Node.js项目{}'.format(get.project_name)) | |
| self.start_project(get) | |
| flag, tip = self._release_firewall(get) | |
| msg = '添加项目成功' + ("" if flag else tip) | |
| return public.return_data(True, msg, project_id) | |
| def modify_project(self,get): | |
| ''' | |
| @name 修改指定项目 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| project_cwd: string<项目目录> | |
| project_script: string<项目脚本> | |
| project_ps: string<项目备注信息> | |
| is_power_on: int<是否开机启动> 1:是 0:否 | |
| run_user: string<运行用户> | |
| max_memory_limit: int<最大内存限制> // 超出此值项目将被强制重启 | |
| nodejs_version: string<nodejs版本> | |
| } | |
| @return dict | |
| ''' | |
| if not isinstance(get,public.dict_obj): return public.return_error('参数类型错误,需要dict_obj对像') | |
| if not self.is_install_nodejs(get): | |
| return public.return_error('请先安装nodejs版本管理器,并安装至少1个node.js版本') | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: | |
| return public.return_error('指定项目不存在: {}'.format(get.project_name)) | |
| if not os.path.exists(get.project_cwd): | |
| return public.return_error('项目目录不存在: {}'.format(get.project_cwd)) | |
| rebuild = False | |
| if hasattr(get,'port'): | |
| if int(project_find['project_config']['port']) != int(get.port): | |
| if self.check_port_is_used(get.get('port/port'),True): | |
| return public.return_error('指定端口已被其它应用占用,请修改您的项目配置使用其它端口, 端口: {}'.format(get.port)) | |
| project_find['project_config']['port'] = int(get.port) | |
| if hasattr(get,'project_cwd'): | |
| if get.project_cwd[-1] == '/': | |
| get.project_cwd = get.project_cwd[:-1] | |
| project_find['project_config']['project_cwd'] = get.project_cwd | |
| if hasattr(get,'project_script'): | |
| if not get.project_script.strip(): | |
| return public.return_error('启动命令不能为空') | |
| project_find['project_config']['project_script'] = get.project_script.strip() | |
| if hasattr(get,'is_power_on'): project_find['project_config']['is_power_on'] = get.is_power_on | |
| if hasattr(get, 'pkg_manager') and get.pkg_manager in ("npm", "pnpm", "yarn"): | |
| project_find['project_config']['pkg_manager'] = get.pkg_manager | |
| if hasattr(get,'run_user'): project_find['project_config']['run_user'] = get.run_user | |
| if hasattr(get,'max_memory_limit'): project_find['project_config']['max_memory_limit'] = get.max_memory_limit | |
| if hasattr(get,'nodejs_version'): | |
| if project_find['project_config']['nodejs_version'] != get.nodejs_version: | |
| rebuild = True | |
| project_find['project_config']['nodejs_version'] = get.nodejs_version | |
| if "pkg_manager" in project_find['project_config'] and project_find['project_config']['pkg_manager'] == "pnpm": | |
| _v = int(project_find['project_config']['nodejs_version'].split(".", 1)[0][1:]) | |
| if _v < 12: | |
| return public.return_error("pnpm不支持Nodejs12以下的版本") | |
| pdata = { | |
| 'path': get.project_cwd, | |
| 'ps': get.project_ps, | |
| 'project_config': json.dumps(project_find['project_config']) | |
| } | |
| public.M('sites').where('name=?',(get.project_name,)).update(pdata) | |
| self.set_config(get.project_name) | |
| public.WriteLog(self._log_name,'修改Node.js项目{}'.format(get.project_name)) | |
| if rebuild: | |
| self.rebuild_project(get.project_name) | |
| return public.return_data(True,'修改项目成功') | |
| def rebuild_project(self,project_name): | |
| ''' | |
| @name 重新构建指定项目 | |
| @author hwliang<2021-08-26> | |
| @param project_name: string<项目名称> | |
| @return bool | |
| ''' | |
| project_find = self.get_project_find(project_name) | |
| if not project_find: return False | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| public.ExecShell(self.get_last_env(nodejs_version) + "cd {} && {} rebuild &>> {}".format(project_find['path'],npm_bin,self._npm_exec_log)) | |
| return True | |
| def remove_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.return_error('指定项目不存在: {}'.format(get.project_name)) | |
| self.stop_project(get) | |
| self.clear_config(get.project_name) | |
| public.M('domain').where('pid=?',(project_find['id'],)).delete() | |
| public.M('sites').where('name=?',(get.project_name,)).delete() | |
| pid_file = "{}/{}.pid".format(self._node_pid_path,get.project_name) | |
| if os.path.exists(pid_file): os.remove(pid_file) | |
| script_file = '{}/{}.sh'.format(self._node_run_scripts,get.project_name) | |
| if os.path.exists(script_file): os.remove(script_file) | |
| if "log_path" not in project_find['project_config']: | |
| log_file = "{}/{}.log".format(self._node_logs, project_find["name"]) | |
| else: | |
| log_file = "{}/{}.log".format(project_find['project_config']["log_path"], project_find["name"]) | |
| if os.path.exists(log_file): os.remove(log_file) | |
| self.del_crontab(get.project_name.strip()) | |
| from mod.base.web_conf import remove_sites_service_config | |
| remove_sites_service_config(get.project_name, "node_") | |
| public.WriteLog(self._log_name,'删除Node.js项目{}'.format(get.project_name)) | |
| return public.return_data(True,'删除项目成功') | |
| def project_get_domain(self,get): | |
| ''' | |
| @name 获取指定项目的域名列表 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| project_id = public.M('sites').where('name=?', (get.project_name,)).getField('id') | |
| if not project_id: | |
| return public.return_data(False, '站点查询失败') | |
| domains = public.M('domain').where('pid=?', (project_id,)).order('id desc').select() | |
| # project_find = self.get_project_find(get.project_name) | |
| # if not project_find: | |
| # return public.return_data(False, '站点查询失败') | |
| # if len(domains) != len(project_find['project_config']['domains']): | |
| # public.M('domain').where('pid=?',(project_id,)).delete() | |
| # if not project_find: return [] | |
| # for d in project_find['project_config']['domains']: | |
| # domain = {} | |
| # arr = d.split(':') | |
| # if len(arr) < 2: arr.append(80) | |
| # domain['name'] = arr[0] | |
| # domain['port'] = int(arr[1]) | |
| # domain['pid'] = project_id | |
| # domain['addtime'] = public.getDate() | |
| # public.M('domain').insert(domain) | |
| # if project_find['project_config']['domains']: | |
| # domains = public.M('domain').where('pid=?',(project_id,)).select() | |
| return domains | |
| def project_add_domain(self,get): | |
| ''' | |
| @name 为指定项目添加域名 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| domains: list<域名列表> | |
| } | |
| @return dict | |
| ''' | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: | |
| return public.return_error('指定项目不存在',data='') | |
| project_id = project_find['id'] | |
| domains = get.domains | |
| flag = False | |
| res_domains = [] | |
| for domain in domains: | |
| domain = domain.strip() | |
| if not domain: continue | |
| if "[" in domain and "]" in domain: # IPv6格式特殊处理 | |
| if "]:" in domain: | |
| domain_arr = domain.rsplit(":", 1) | |
| else: | |
| domain_arr = [domain] | |
| else: | |
| domain_arr = domain.split(':') | |
| domain_arr[0] = self.check_domain(domain_arr[0]) | |
| if domain_arr[0] is False: | |
| res_domains.append({"name": domain, "status": False, "msg": '域名格式错误'}) | |
| continue | |
| if len(domain_arr) == 1: | |
| domain_arr.append("") | |
| if domain_arr[1] == "": | |
| domain_arr[1] = 80 | |
| domain += ':80' | |
| try: | |
| if not (0 < int(domain_arr[1]) < 65535): | |
| res_domains.append({"name": domain, "status": False, "msg": '域名格式错误'}) | |
| continue | |
| except ValueError: | |
| res_domains.append({"name": domain, "status": False, "msg": '域名格式错误'}) | |
| continue | |
| if not public.M('domain').where('name=? and port =? ',(domain_arr[0],domain_arr[1])).count(): | |
| public.M('domain').add('name,pid,port,addtime',(domain_arr[0],project_id,domain_arr[1],public.getDate())) | |
| if not domain in project_find['project_config']['domains']: | |
| project_find['project_config']['domains'].append(domain) | |
| public.WriteLog(self._log_name,'成功添加域名{}到项目{}'.format(domain,get.project_name)) | |
| res_domains.append({"name": domain_arr[0], "status": True, "msg": '添加成功'}) | |
| flag = True | |
| else: | |
| public.WriteLog(self._log_name,'添加域名错误,域名{}已存在'.format(domain)) | |
| res_domains.append({"name": domain_arr[0], "status": False, "msg": '添加错误,域名{}已存在'.format(domain)}) | |
| if flag: | |
| public.M('sites').where('id=?',(project_id,)).save('project_config',json.dumps(project_find['project_config'])) | |
| self.set_config(get.project_name) | |
| return self._ckeck_add_domain(get.project_name, res_domains) | |
| def project_remove_domain(self,get): | |
| ''' | |
| @name 为指定项目删除域名 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| domain: string<域名> | |
| } | |
| @return dict | |
| ''' | |
| project_find = self.get_project_find(get.project_name) | |
| if not project_find: | |
| return public.returnResult(False, '指定项目不存在', code=2) | |
| domain_arr = get.domain.rsplit(":", 1) | |
| if len(domain_arr) == 1: | |
| domain_arr.append(80) | |
| project_id = public.M('sites').where('name=?',(get.project_name,)).getField('id') | |
| if (len(public.M('domain').where('pid=?', (project_find['id'],)).select()) == 1 and | |
| project_find["project_config"]["bind_extranet"] == 1): | |
| return public.returnResult(False, '项目至少需要一个域名', code=2) | |
| domain_id = public.M('domain').where('name=? AND port=? AND pid=?',(domain_arr[0],domain_arr[1], project_id)).getField('id') | |
| if not domain_id: | |
| return public.returnResult(False, '指定域名不存在', code=2) | |
| public.M('domain').where('id=?',(domain_id,)).delete() | |
| if get.domain in project_find['project_config']['domains']: | |
| project_find['project_config']['domains'].remove(get.domain) | |
| if get.domain+":80" in project_find['project_config']['domains']: | |
| project_find['project_config']['domains'].remove(get.domain + ":80") | |
| public.M('sites').where('id=?',(project_id,)).save('project_config',json.dumps(project_find['project_config'])) | |
| public.WriteLog(self._log_name,'从项目:{},删除域名{}'.format(get.project_name,get.domain)) | |
| self.set_config(get.project_name) | |
| return public.returnResult(True, '删除域名成功') | |
| def bind_extranet(self,get): | |
| ''' | |
| @name 绑定外网 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| res_msg = self._check_webserver() | |
| if res_msg: | |
| return public.returnResult(False, msg=res_msg, data=res_msg, code=2) | |
| if not public.is_apache_nginx(): | |
| return public.returnResult(False, msg='需要安装Nginx或Apache才能使用外网映射功能', data='需要安装Nginx或Apache才能使用外网映射功能', code=2) | |
| project_name = get.project_name.strip() | |
| project_find = self.get_project_find(project_name) | |
| if not project_find: | |
| return public.returnResult(False, msg='项目不存在', data='项目不存在', code=2) | |
| if not project_find['project_config']['domains']: | |
| return public.returnResult(False, msg='请先到【域名管理】选项中至少添加一个域名', data='请先到【域名管理】选项中至少添加一个域名', code=2) | |
| project_find['project_config']['bind_extranet'] = 1 | |
| public.M('sites').where("id=?",(project_find['id'],)).setField('project_config',json.dumps(project_find['project_config'])) | |
| res = self.set_config(project_name) | |
| if not res["status"]: | |
| # 开启失败则关闭外网映射标识 | |
| project_find['project_config']['bind_extranet'] = 0 | |
| public.M('sites').where("id=?", (project_find['id'],)).setField('project_config',json.dumps(project_find['project_config'])) | |
| return public.returnResult(False, msg=res["msg"], data=res["msg"], code=2) | |
| public.WriteLog(self._log_name, 'Node项目{}, 开启外网映射'.format(project_name)) | |
| return public.returnResult(True, msg='开启外网映射成功', data='开启外网映射成功', code=0) | |
| def set_config(self,project_name): | |
| ''' | |
| @name 设置项目配置 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return bool | |
| ''' | |
| project_find = self.get_project_find(project_name) | |
| if not project_find: return False | |
| if not project_find['project_config']: return False | |
| if not project_find['project_config']['bind_extranet']: return False | |
| if not project_find['project_config']['domains']: return False | |
| res = self.set_nginx_config(project_find) | |
| if not res["status"]: | |
| return res | |
| self.set_apache_config(project_find) | |
| public.serviceReload() | |
| return public.returnResult(True) | |
| def clear_config(self,project_name): | |
| ''' | |
| @name 清除项目配置 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return bool | |
| ''' | |
| project_find = self.get_project_find(project_name) | |
| if not project_find: return False | |
| self.clear_nginx_config(project_find) | |
| self.clear_apache_config(project_find) | |
| public.serviceReload() | |
| return True | |
| def clear_apache_config(self,project_find): | |
| ''' | |
| @name 清除apache配置 | |
| @author hwliang<2021-08-09> | |
| @param project_find: dict<项目信息> | |
| @return bool | |
| ''' | |
| project_name = project_find['name'] | |
| config_file = "{}/apache/node_{}.conf".format(self._vhost_path,project_name) | |
| if os.path.exists(config_file): | |
| os.remove(config_file) | |
| return True | |
| def clear_nginx_config(self,project_find): | |
| ''' | |
| @name 清除nginx配置 | |
| @author hwliang<2021-08-09> | |
| @param project_find: dict<项目信息> | |
| @return bool | |
| ''' | |
| project_name = project_find['name'] | |
| config_file = "{}/nginx/node_{}.conf".format(self._vhost_path,project_name) | |
| if os.path.exists(config_file): | |
| os.remove(config_file) | |
| rewrite_file = "{panel_path}/vhost/rewrite/node_{project_name}.conf".format(panel_path = self._panel_path,project_name = project_name) | |
| if os.path.exists(rewrite_file): | |
| os.remove(rewrite_file) | |
| return True | |
| def set_nginx_config(self,project_find): | |
| ''' | |
| @name 设置Nginx配置 | |
| @author hwliang<2021-08-09> | |
| @param project_find: dict<项目信息> | |
| @return bool | |
| ''' | |
| project_name = project_find['name'] | |
| ports = set() | |
| domains = set() | |
| for d in project_find['project_config']['domains']: | |
| domain_tmp = d.rsplit(":", 1) | |
| if len(domain_tmp) == 1: domain_tmp.append(80) | |
| if not int(domain_tmp[1]) in ports: | |
| ports.add(int(domain_tmp[1])) | |
| if not domain_tmp[0] in domains: | |
| domains.add(domain_tmp[0]) | |
| listen_ipv6 = public.listen_ipv6() | |
| is_ssl, is_force_ssl = self.exists_nginx_ssl(project_name) | |
| listen_ports_list = [] | |
| for p in ports: | |
| listen_ports_list.append(" listen {};".format(p)) | |
| if listen_ipv6: | |
| listen_ports_list.append(" listen [::]:{};".format(p)) | |
| ssl_config = '' | |
| if is_ssl: | |
| http3_header = "" | |
| if self.is_nginx_http3(): | |
| http3_header = '''\n add_header Alt-Svc 'quic=":443"; h3=":443"; h3-29=":443"; h3-27=":443";h3-25=":443"; h3-T050=":443"; h3-Q050=":443";h3-Q049=":443";h3-Q048=":443"; h3-Q046=":443"; h3-Q043=":443"';''' | |
| http3_header += "\n quic_retry on;\n quic_gso on;" | |
| if self.ng_ssl_early_data_enabled(): | |
| http3_header += "\n ssl_early_data on;" | |
| nginx_ver = public.nginx_version() | |
| if nginx_ver: | |
| port_str = ["443"] | |
| if listen_ipv6: | |
| port_str.append("[::]:443") | |
| use_http2_on = False | |
| for p in port_str: | |
| listen_str = " listen {} ssl".format(p) | |
| if nginx_ver < [1, 9, 5]: | |
| listen_str += ";" | |
| elif [1, 9, 5] <= nginx_ver < [1, 25, 1]: | |
| listen_str += " http2;" | |
| else: # >= [1, 25, 1] | |
| listen_str += ";" | |
| use_http2_on = True | |
| listen_ports_list.append(listen_str) | |
| if self.is_nginx_http3(): | |
| listen_ports_list.append(" listen {} quic;".format(p)) | |
| if use_http2_on: | |
| listen_ports_list.append(" http2 on;") | |
| if self.is_nginx_http3(): | |
| listen_ports_list.append(" http3 on;") | |
| else: | |
| listen_ports_list.append(" listen 443 ssl;") | |
| ssl_config = '''ssl_certificate {vhost_path}/cert/{priject_name}/fullchain.pem; | |
| ssl_certificate_key {vhost_path}/cert/{priject_name}/privkey.pem; | |
| ssl_protocols TLSv1.1 TLSv1.2 TLSv1.3; | |
| ssl_ciphers EECDH+CHACHA20:EECDH+CHACHA20-draft:EECDH+AES128:RSA+AES128:EECDH+AES256:RSA+AES256:EECDH+3DES:RSA+3DES:!MD5; | |
| ssl_prefer_server_ciphers on; | |
| ssl_session_cache shared:SSL:10m; | |
| ssl_session_timeout 10m; | |
| add_header Strict-Transport-Security "max-age=31536000";{http3_header} | |
| error_page 497 https://$host$request_uri;'''.format(vhost_path = self._vhost_path,priject_name = project_name, http3_header=http3_header) | |
| if is_force_ssl: | |
| ssl_config += ''' | |
| #HTTP_TO_HTTPS_START | |
| if ($server_port !~ 443){ | |
| rewrite ^(/.*)$ https://$host$1 permanent; | |
| } | |
| #HTTP_TO_HTTPS_END''' | |
| config_file = "{}/nginx/node_{}.conf".format(self._vhost_path,project_name) | |
| template_file = "{}/template/nginx/node_http.conf".format(self._vhost_path) | |
| listen_ports = "\n".join(listen_ports_list).strip() | |
| config_body = public.readFile(template_file) | |
| args = public.dict_obj() | |
| args.project_name = project_name | |
| project_info = self.get_project_info(args) | |
| old_port = self._get_nginx_proxy_port(config_file) | |
| proxy_port = project_find['project_config']['port'] if not project_find['project_config']['port'] is None else None | |
| if proxy_port is None: | |
| if not project_info["listen"]: | |
| if not old_port: | |
| return public.returnResult(False, "无法获取监听的端口,请输入监听端口后再开启外网映射", code=5) | |
| else: | |
| proxy_port = old_port | |
| else: | |
| proxy_port = project_info['listen'][0] | |
| mut_config = { | |
| "site_path": project_find['path'], | |
| "domains": ' '.join(domains), | |
| "url": 'http://127.0.0.1:{}'.format(proxy_port), | |
| "ssl_config": ssl_config, | |
| "listen_ports": listen_ports | |
| } | |
| config_body = config_body.format( | |
| site_path=mut_config["site_path"], | |
| domains=mut_config["domains"], | |
| project_name=project_name, | |
| panel_path=self._panel_path, | |
| log_path=public.get_logs_path(), | |
| url=mut_config["url"], | |
| host='$host', | |
| listen_ports=listen_ports, | |
| ssl_config=ssl_config | |
| ) | |
| # # 恢复旧的SSL配置 | |
| # ssl_config = self.get_nginx_ssl_config(project_name) | |
| # if ssl_config: | |
| # config_body.replace('#error_page 404/404.html;',ssl_config) | |
| rewrite_file = "{panel_path}/vhost/rewrite/node_{project_name}.conf".format(panel_path = self._panel_path,project_name = project_name) | |
| if not os.path.exists(rewrite_file): | |
| public.writeFile(rewrite_file,'# 请将伪静态规则或自定义NGINX配置填写到此处\n') | |
| if not os.path.exists("/www/server/panel/vhost/nginx/well-known"): | |
| os.makedirs("/www/server/panel/vhost/nginx/well-known", 0o600) | |
| apply_check = "{}/vhost/nginx/well-known/{}.conf".format(self._panel_path, project_name) | |
| from mod.base.web_conf import ng_ext | |
| config_body = ng_ext.set_extension_by_config(project_name, config_body) | |
| if not os.path.exists(apply_check): | |
| public.writeFile(apply_check, '') | |
| if not os.path.exists(config_file): | |
| public.writeFile(config_file, config_body) | |
| else: | |
| if not self._replace_nginx_conf(config_file, mut_config): | |
| public.writeFile(config_file, config_body) | |
| return public.returnResult(True) | |
| def _replace_nginx_conf(config_file, mut_config: dict) -> bool: | |
| """尝试替换""" | |
| data:str = public.readFile(config_file) | |
| tab_spc = " " | |
| rep_list = [ | |
| ( | |
| r"([ \f\r\t\v]*listen[^;\n]*;\n(\s*http2\s+on\s*;[^\n]*\n)?)+", | |
| mut_config["listen_ports"] + "\n" | |
| ), | |
| ( | |
| r"[ \f\r\t\v]*root [ \f\r\t\v]*/[^;\n]*;", | |
| " root {};".format(mut_config["site_path"]) | |
| ), | |
| ( | |
| r"[ \f\r\t\v]*server_name [ \f\r\t\v]*[^\n;]*;", | |
| " server_name {};".format(mut_config["domains"]) | |
| ), | |
| ( | |
| r"[ \f\r\t\v]*location */ *\{ *\n *proxy_pass[^\n;]*;\n *proxy_set_header *Host", | |
| "{}location / {{\n{}proxy_pass {};\n{}proxy_set_header Host".format( | |
| tab_spc, tab_spc*2, mut_config["url"], tab_spc*2,) | |
| ), | |
| ( | |
| "[ \f\r\t\v]*#SSL-START(.*\n){2,15}[ \f\r\t\v]*#SSL-END", | |
| "{}#SSL-START SSL相关配置\n{}#error_page 404/404.html;\n{}{}\n{}#SSL-END".format( | |
| tab_spc, tab_spc, tab_spc, mut_config["ssl_config"], tab_spc) | |
| ) | |
| ] | |
| for rep, info in rep_list: | |
| if re.search(rep, data): | |
| data = re.sub(rep, info, data, 1) | |
| else: | |
| return False | |
| public.writeFile(config_file, data) | |
| return True | |
| def get_nginx_ssl_config(self,project_name): | |
| ''' | |
| @name 获取项目Nginx SSL配置 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return string | |
| ''' | |
| result = '' | |
| config_file = "{}/nginx/node_{}".format(self._vhost_path,project_name) | |
| if not os.path.exists(config_file): | |
| return result | |
| config_body = public.readFile(config_file) | |
| if not config_body: | |
| return result | |
| if config_body.find('ssl_certificate') == -1: | |
| return result | |
| ssl_body = re.search("#SSL-START(.|\n)+#SSL-END",config_body) | |
| if not ssl_body: return result | |
| result = ssl_body.group() | |
| return result | |
| def exists_nginx_ssl(self,project_name): | |
| ''' | |
| @name 判断项目是否配置Nginx SSL配置 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return tuple | |
| ''' | |
| config_file = "{}/nginx/node_{}.conf".format(public.get_vhost_path(),project_name) | |
| if not os.path.exists(config_file): | |
| return False,False | |
| config_body = public.readFile(config_file) | |
| if not config_body: | |
| return False,False | |
| is_ssl,is_force_ssl = False,False | |
| if config_body.find('ssl_certificate') != -1: | |
| is_ssl = True | |
| if config_body.find('HTTP_TO_HTTPS_START') != -1: | |
| is_force_ssl = True | |
| return is_ssl,is_force_ssl | |
| def exists_apache_ssl(self,project_name): | |
| ''' | |
| @name 判断项目是否配置Apache SSL配置 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return bool | |
| ''' | |
| config_file = "{}/apache/node_{}.conf".format(public.get_vhost_path(),project_name) | |
| if not os.path.exists(config_file): | |
| return False,False | |
| config_body = public.readFile(config_file) | |
| if not config_body: | |
| return False,False | |
| is_ssl,is_force_ssl = False,False | |
| if config_body.find('SSLCertificateFile') != -1: | |
| is_ssl = True | |
| if config_body.find('HTTP_TO_HTTPS_START') != -1: | |
| is_force_ssl = True | |
| return is_ssl,is_force_ssl | |
| def set_apache_config(self,project_find): | |
| ''' | |
| @name 设置Apache配置 | |
| @author hwliang<2021-08-09> | |
| @param project_find: dict<项目信息> | |
| @return bool | |
| ''' | |
| project_name = project_find['name'] | |
| # 处理域名和端口 | |
| ports = set() | |
| domains = set() | |
| for d in project_find['project_config']['domains']: | |
| domain_tmp = d.rsplit(":", 1) | |
| if len(domain_tmp) == 1: domain_tmp.append(80) | |
| if not int(domain_tmp[1]) in ports: | |
| ports.add(int(domain_tmp[1])) | |
| if not domain_tmp[0] in domains: | |
| domains.add(domain_tmp[0]) | |
| config_file = "{}/apache/node_{}.conf".format(self._vhost_path,project_name) | |
| template_file = "{}/template/apache/node_http.conf".format(self._vhost_path) | |
| config_body = public.readFile(template_file) | |
| apache_config_body = '' | |
| # 旧的配置文件是否配置SSL | |
| is_ssl,is_force_ssl = self.exists_apache_ssl(project_name) | |
| if is_ssl: | |
| if not 443 in ports: ports.add(443) | |
| from panelSite import panelSite | |
| s = panelSite() | |
| # 根据端口列表生成配置 | |
| for p in ports: | |
| # 生成SSL配置 | |
| ssl_config = '' | |
| if p == 443 and is_ssl: | |
| ssl_key_file = "{vhost_path}/cert/{project_name}/privkey.pem".format(project_name = project_name,vhost_path = public.get_vhost_path()) | |
| if not os.path.exists(ssl_key_file): continue # 不存在证书文件则跳过 | |
| ssl_config = '''#SSL | |
| SSLEngine On | |
| SSLCertificateFile {vhost_path}/cert/{project_name}/fullchain.pem | |
| SSLCertificateKeyFile {vhost_path}/cert/{project_name}/privkey.pem | |
| SSLCipherSuite EECDH+CHACHA20:EECDH+CHACHA20-draft:EECDH+AES128:RSA+AES128:EECDH+AES256:RSA+AES256:EECDH+3DES:RSA+3DES:!MD5 | |
| SSLProtocol All -SSLv2 -SSLv3 -TLSv1 | |
| SSLHonorCipherOrder On'''.format(project_name = project_name,vhost_path = public.get_vhost_path()) | |
| else: | |
| if is_force_ssl: | |
| ssl_config = '''#HTTP_TO_HTTPS_START | |
| <IfModule mod_rewrite.c> | |
| RewriteEngine on | |
| RewriteCond %{SERVER_PORT} !^443$ | |
| RewriteRule (.*) https://%{SERVER_NAME}$1 [L,R=301] | |
| </IfModule> | |
| #HTTP_TO_HTTPS_END''' | |
| # 生成vhost主体配置 | |
| args = public.dict_obj() | |
| args.project_name = project_name | |
| project_info = self.get_project_info(args) | |
| proxy_port = project_find['project_config']['port'] if not project_find['project_config']['port'] is None else None | |
| if proxy_port is None: | |
| if not project_info["listen_ok"]: | |
| return public.returnResult(False, "无法获取监听的端口,请输入监听端口后再开启外网映射", code=5) | |
| proxy_port = (project_info['listen'] or " ")[0] | |
| apache_config_body += config_body.format( | |
| site_path = project_find['path'], | |
| server_name = '{}.{}'.format(p,project_name), | |
| domains = ' '.join(domains), | |
| log_path = public.get_logs_path(), | |
| server_admin = 'admin@{}'.format(project_name), | |
| url = 'http://127.0.0.1:{}'.format(proxy_port), | |
| port = p, | |
| ssl_config = ssl_config, | |
| project_name = project_name | |
| ) | |
| apache_config_body += "\n" | |
| # 添加端口到主配置文件 | |
| if not p in [80]: | |
| s.apacheAddPort(p) | |
| # 写.htaccess | |
| rewrite_file = "{}/.htaccess".format(project_find['path']) | |
| if not os.path.exists(rewrite_file): public.writeFile(rewrite_file,'# 请将伪静态规则或自定义Apache配置填写到此处\n') | |
| from mod.base.web_conf import ap_ext | |
| apache_config_body = ap_ext.set_extension_by_config(project_name, apache_config_body) | |
| # 写配置文件 | |
| public.writeFile(config_file,apache_config_body) | |
| return True | |
| def unbind_extranet(self,get): | |
| ''' | |
| @name 解绑外网 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| project_name = get.project_name.strip() | |
| self.clear_config(project_name) | |
| public.serviceReload() | |
| project_find = self.get_project_find(project_name) | |
| project_find['project_config']['bind_extranet'] = 0 | |
| public.M('sites').where("id=?",(project_find['id'],)).setField('project_config',json.dumps(project_find['project_config'])) | |
| public.WriteLog(self._log_name,'Node项目{}, 关闭外网映射'.format(project_name)) | |
| return public.return_data(True,'关闭成功') | |
| def get_project_pids(self,get = None,pid = None): | |
| ''' | |
| @name 获取项目进程pid列表 | |
| @author hwliang<2021-08-10> | |
| @param pid: string<项目pid> | |
| @return list | |
| ''' | |
| if get: pid = int(get.pid) | |
| if not self._pids: self._pids = psutil.pids() | |
| project_pids = [] | |
| for i in self._pids: | |
| try: | |
| p = psutil.Process(i) | |
| if p.status() == "zombie": | |
| continue | |
| if p.ppid() == pid: | |
| if i in project_pids: | |
| continue | |
| project_pids.append(i) | |
| except: | |
| continue | |
| other_pids = [] | |
| for i in project_pids: | |
| other_pids += self.get_project_pids(pid=i) | |
| if os.path.exists('/proc/{}'.format(pid)): | |
| project_pids.append(pid) | |
| all_pids = list(set(project_pids + other_pids)) | |
| if not all_pids: | |
| all_pids = self.get_other_pids(pid) | |
| return sorted(all_pids) | |
| def get_other_pids(self,pid): | |
| ''' | |
| @name 获取其他进程pid列表 | |
| @author hwliang<2021-08-10> | |
| @param pid: string<项目pid> | |
| @return list | |
| ''' | |
| project_name = None | |
| for pid_name in os.listdir(self._node_pid_path): | |
| pid_file = '{}/{}'.format(self._node_pid_path,pid_name) | |
| #s_pid = int(public.readFile(pid_file)) | |
| data = public.readFile(pid_file) | |
| if isinstance(data,str): | |
| try: | |
| s_pid = int(data) | |
| except: | |
| return [] | |
| else: | |
| return [] | |
| if pid == s_pid: | |
| project_name = pid_name[:-4] | |
| break | |
| project_find = self.get_project_find(project_name) | |
| if not project_find: return [] | |
| if not self._pids: self._pids = psutil.pids() | |
| all_pids = [] | |
| for i in self._pids: | |
| try: | |
| p = psutil.Process(i) | |
| if p.cwd() == project_find['path']: | |
| pname = p.name() | |
| if pname in ['node','npm','pm2','yarn'] or pname.find('node ') == 0: | |
| cmdline = ','.join(p.cmdline()) | |
| if cmdline.find('God Daemon') != -1:continue | |
| env_list = p.environ() | |
| if 'name' in env_list: | |
| if not env_list['name'] == project_name: continue | |
| if 'NODE_PROJECT_NAME' in env_list: | |
| if not env_list['NODE_PROJECT_NAME'] == project_name: continue | |
| all_pids.append(i) | |
| except: continue | |
| return all_pids | |
| def get_project_state_by_cwd(self,project_name): | |
| ''' | |
| @name 通过cwd获取项目状态 | |
| @author hwliang<2022-01-17> | |
| @param project_name<string> 项目名称 | |
| @return bool or list | |
| ''' | |
| project_find = self.get_project_find(project_name) | |
| self._pids = psutil.pids() | |
| if not project_find: return [] | |
| all_pids = [] | |
| for i in self._pids: | |
| try: | |
| p = psutil.Process(i) | |
| if p.cwd() == project_find['path']: | |
| pname = p.name() | |
| if pname in ['node','npm','pm2','yarn'] or pname.find('node ') == 0: | |
| cmdline = ','.join(p.cmdline()) | |
| if cmdline.find('God Daemon') != -1:continue | |
| env_list = p.environ() | |
| if 'name' in env_list: | |
| if not env_list['name'] == project_name: continue | |
| if 'NODE_PROJECT_NAME' in env_list: | |
| if not env_list['NODE_PROJECT_NAME'] == project_name: continue | |
| all_pids.append(i) | |
| except: continue | |
| if all_pids: | |
| pid_file = "{}/{}.pid".format(self._node_pid_path,project_name) | |
| public.writeFile(pid_file,str(all_pids[0])) | |
| return all_pids | |
| return False | |
| def kill_pids(self,get=None,pids = None): | |
| ''' | |
| @name 结束进程列表 | |
| @author hwliang<2021-08-10> | |
| @param pids: string<进程pid列表> | |
| @return dict | |
| ''' | |
| if get: pids = get.pids | |
| if not pids: return public.return_data(True, '没有进程') | |
| pids = sorted(pids,reverse=True) | |
| for i in pids: | |
| try: | |
| p = psutil.Process(i) | |
| p.terminate() | |
| except: | |
| pass | |
| return public.return_data(True, '进程已全部结束') | |
| def start_project(self,get): | |
| ''' | |
| @name 启动项目 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| 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.return_error('项目不存在') | |
| 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.return_error('当前项目已过期,请重新设置项目到期时间') | |
| 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.return_error(error_msg) | |
| # 是否安装依赖模块? | |
| package_file = "{}/package.json".format(project_find['path']) | |
| package_info = {} | |
| if os.path.exists(package_file): | |
| node_modules_path = "{}/node_modules".format(project_find['path']) | |
| if not os.path.exists(node_modules_path): | |
| return public.return_error('请先到模块管理中点击【一键安装项目模块】来安装模块依赖!') | |
| try: | |
| package_info = json.loads(public.readFile(package_file)) | |
| except: | |
| package_info = {} | |
| if not package_info: package_info['scripts'] = {} | |
| if 'scripts' not in package_info: package_info['scripts'] = {} | |
| try: | |
| scripts_keys = package_info['scripts'].keys() | |
| except: | |
| scripts_keys = [] | |
| # 前置准备 | |
| nodejs_version = project_find['project_config']['nodejs_version'] | |
| node_bin = self.get_node_bin(nodejs_version) | |
| npm_bin = self.get_npm_bin(nodejs_version) | |
| project_script = project_find['project_config']['project_script'].strip().replace(' ',' ') | |
| if project_script[:3] == 'pm2': # PM2启动方式处理 | |
| project_script = project_script.replace('pm2 ','pm2 -u {} -n {} '.format(project_find['project_config']['run_user'],get.project_name)) | |
| project_find['project_config']['run_user'] = 'root' | |
| log_file = "{}/{}.log".format(project_find['project_config']["log_path"], project_find["name"]) | |
| if not project_script: return public.return_error('未配置启动脚本') | |
| 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 | |
| ) | |
| elif project_script in scripts_keys: | |
| start_cmd = '''{last_env} | |
| export NODE_PROJECT_NAME="{project_name}" | |
| cd {project_cwd} | |
| nohup {npm_bin} run {project_script} &>> {log_file} & | |
| echo $! > {pid_file} | |
| '''.format( | |
| project_cwd = project_find['path'], | |
| npm_bin = npm_bin, | |
| project_script = project_script, | |
| pid_file = pid_file, | |
| log_file = log_file, | |
| last_env = last_env, | |
| project_name = get.project_name | |
| ) | |
| else: | |
| start_cmd = '''{last_env} | |
| export NODE_PROJECT_NAME="{project_name}" | |
| cd {project_cwd} | |
| nohup {project_script} &>> {log_file} & | |
| echo $! > {pid_file} | |
| '''.format( | |
| project_cwd = project_find['path'], | |
| project_script = project_script, | |
| pid_file = pid_file, | |
| log_file = log_file, | |
| last_env = last_env, | |
| project_name = get.project_name | |
| ) | |
| 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.return_error('启动命令被【堡塔防入侵】拦截,请关闭{}用户的防护'.format(project_find['project_config']['run_user'])) | |
| return public.return_error('启动命令被未知安全软件拦截,请检查安装软件日志') | |
| return public.returnMsg(False,'启动失败{}<script>setTimeout(function(){{$(".layui-layer-msg").css("width","800px");}},100)</script>'.format(p)) | |
| # 获取PID | |
| try: | |
| pid = int(public.readFile(pid_file)) | |
| except: | |
| return public.return_error('启动失败{}'.format(public.GetNumLines(log_file,20))) | |
| pids = self.get_project_pids(pid=pid) | |
| if not pids: | |
| if os.path.exists(pid_file): os.remove(pid_file) | |
| return public.return_error('启动失败{}'.format(public.GetNumLines(log_file,20))) | |
| self.start_by_user(project_find["id"]) | |
| return public.return_data(True, '启动成功', pids) | |
| 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.return_error('项目不存在') | |
| 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.return_error('当前项目已过期,请重新设置项目到期时间') | |
| 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: # 处理PM2启动的项目 | |
| 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.return_error('项目未启动') | |
| data = public.readFile(pid_file) | |
| if isinstance(data,str) and data: | |
| pid = int(data) | |
| pids = self.get_project_pids(pid=pid) | |
| else: | |
| return public.return_error('项目未启动') | |
| if not pids: return public.return_error('项目未启动') | |
| 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.return_data(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['edate'] != "0000-00-00" and project_find['edate'] < datetime.datetime.today().strftime("%Y-%m-%d"): | |
| return public.return_error('当前项目已过期,请重新设置项目到期时间') | |
| res = self.stop_project(get) | |
| # if not res['status']: return res | |
| res = self.start_project(get) | |
| if not res['status']: return res | |
| return public.return_data(True, '重启成功') | |
| # xss 防御 | |
| def xsssec(self,text): | |
| return text.replace('<', '<').replace('>', '>') | |
| def get_project_log(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.returnMsg(False,'项目不存在') | |
| if "log_path" not in project_find['project_config']: | |
| log_file = "{}/{}.log".format(self._node_logs, project_find["name"]) | |
| else: | |
| log_file = "{}/{}.log".format(project_find['project_config']["log_path"], project_find["name"]) | |
| if not os.path.exists(log_file): return public.return_error('日志文件不存在') | |
| return { | |
| "status" : True, | |
| "path": log_file.rsplit("/", 1)[0], | |
| "data" : self.xsssec(public.GetNumLines(log_file, 3000)), | |
| "szie": public.to_size(os.path.getsize(log_file)) | |
| } | |
| def get_project_load_info(self,get = None,project_name = None): | |
| ''' | |
| @name 获取项目负载信息 | |
| @author hwliang<2021-08-12> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| if get: project_name = get.project_name.strip() | |
| load_info = {} | |
| pid_file = "{}/{}.pid".format(self._node_pid_path,project_name) | |
| if not os.path.exists(pid_file): return load_info | |
| data = public.readFile(pid_file) | |
| if isinstance(data,str) and data: | |
| pid = int(data) | |
| pids = self.get_project_pids(pid=pid) | |
| else: | |
| return load_info | |
| if not pids: return load_info | |
| for i in pids: | |
| process_info = self.get_process_info_by_pid(i) | |
| if process_info: load_info[i] = process_info | |
| return load_info | |
| def get_pm2_load_info(self,get = None,project_name = None): | |
| ''' | |
| @name 获取项目负载信息 | |
| @author hwliang<2021-08-12> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| if get: project_name = get.project_name.strip() | |
| load_info = {} | |
| pids = [] | |
| from mod.project.nodejs.pm2Mod import main | |
| pm2_list = main().get_jlist() | |
| for pl in pm2_list: | |
| env_projectname = "" | |
| if "pm2_env" in pl and "NODE_PROJECT_NAME" in pl["pm2_env"]: | |
| env_projectname = pl["pm2_env"]["NODE_PROJECT_NAME"] | |
| if pl["name"] == project_name or env_projectname == project_name: | |
| if not "pid" in pl: continue | |
| pids.append(pl["pid"]) | |
| if not pids: return load_info | |
| for i in pids: | |
| process_info = self.get_process_info_by_pid(i) | |
| if process_info: load_info[i] = process_info | |
| return load_info | |
| def get_pm2_load_info_new(self,pids: list): | |
| ''' | |
| @name 获取项目负载信息 | |
| @author hwliang<2021-08-12> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| load_info = {} | |
| for i in pids: | |
| process_info = self.get_process_info_by_pid(i) | |
| if process_info: load_info[i] = process_info | |
| return load_info | |
| def object_to_dict(self,obj): | |
| ''' | |
| @name 将对象转换为字典 | |
| @author hwliang<2021-08-09> | |
| @param obj<object> | |
| @return dict | |
| ''' | |
| result = {} | |
| for name in dir(obj): | |
| value = getattr(obj, name) | |
| if not name.startswith('__') and not callable(value) and not name.startswith('_'): result[name] = value | |
| return result | |
| def list_to_dict(self,data): | |
| ''' | |
| @name 将列表转换为字典 | |
| @author hwliang<2021-08-09> | |
| @param data<list> | |
| @return dict | |
| ''' | |
| result = [] | |
| for s in data: | |
| result.append(self.object_to_dict(s)) | |
| return result | |
| def get_connects(self,pid): | |
| ''' | |
| @name 获取进程连接信息 | |
| @author hwliang<2021-08-09> | |
| @param pid<int> | |
| @return dict | |
| ''' | |
| connects = 0 | |
| try: | |
| if pid == 1: return connects | |
| tp = '/proc/' + str(pid) + '/fd/' | |
| if not os.path.exists(tp): return connects | |
| for d in os.listdir(tp): | |
| fname = tp + d | |
| if os.path.islink(fname): | |
| l = os.readlink(fname) | |
| if l.find('socket:') != -1: connects += 1 | |
| except:pass | |
| return connects | |
| def format_connections(self,connects): | |
| ''' | |
| @name 获取进程网络连接信息 | |
| @author hwliang<2021-08-09> | |
| @param connects<pconn> | |
| @return list | |
| ''' | |
| result = [] | |
| for i in connects: | |
| raddr = i.raddr | |
| if not i.raddr: | |
| raddr = ('',0) | |
| laddr = i.laddr | |
| if not i.laddr: | |
| laddr = ('',0) | |
| result.append({ | |
| "fd": i.fd, | |
| "family": i.family, | |
| "local_addr": laddr[0], | |
| "local_port": laddr[1], | |
| "client_addr": raddr[0], | |
| "client_rport": raddr[1], | |
| "status": i.status | |
| }) | |
| return result | |
| def get_process_info_by_pid(self,pid): | |
| ''' | |
| @name 获取进程信息 | |
| @author hwliang<2021-08-12> | |
| @param pid: int<进程id> | |
| @return dict | |
| ''' | |
| process_info = {} | |
| try: | |
| if not os.path.exists('/proc/{}'.format(pid)): return process_info | |
| p = psutil.Process(pid) | |
| status_ps = {'sleeping':'睡眠','running':'活动'} | |
| with p.oneshot(): | |
| p_mem = p.memory_full_info() | |
| if p_mem.uss + p_mem.rss + p_mem.pss + p_mem.data == 0: return process_info | |
| p_state = p.status() | |
| if p_state in status_ps: p_state = status_ps[p_state] | |
| # process_info['exe'] = p.exe() | |
| process_info['name'] = p.name() | |
| process_info['pid'] = pid | |
| process_info['ppid'] = p.ppid() | |
| process_info['create_time'] = int(p.create_time()) | |
| process_info['status'] = p_state | |
| process_info['user'] = p.username() | |
| process_info['memory_used'] = p_mem.uss | |
| process_info['cpu_percent'] = self.get_cpu_precent(p) | |
| process_info['io_write_bytes'],process_info['io_read_bytes'] = self.get_io_speed(p) | |
| process_info['connections'] = self.format_connections(p.connections()) | |
| process_info['connects'] = self.get_connects(pid) | |
| process_info['open_files'] = self.list_to_dict(p.open_files()) | |
| process_info['threads'] = p.num_threads() | |
| process_info['exe'] = ' '.join(p.cmdline()) | |
| return process_info | |
| except: | |
| return process_info | |
| def get_io_speed(self,p): | |
| ''' | |
| @name 获取磁盘IO速度 | |
| @author hwliang<2021-08-12> | |
| @param p: Process<进程对像> | |
| @return list | |
| ''' | |
| skey = "io_speed_{}".format(p.pid) | |
| old_pio = cache.get(skey) | |
| if not hasattr(p,'io_counters'): return 0,0 | |
| pio = p.io_counters() | |
| if not old_pio: | |
| cache.set(skey,[pio,time.time()],3600) | |
| # time.sleep(0.1) | |
| old_pio = cache.get(skey) | |
| pio = p.io_counters() | |
| old_write_bytes = old_pio[0].write_bytes | |
| old_read_bytes = old_pio[0].read_bytes | |
| old_time = old_pio[1] | |
| new_time = time.time() | |
| write_bytes = pio.write_bytes | |
| read_bytes = pio.read_bytes | |
| cache.set(skey,[pio,new_time],3600) | |
| write_speed = int((write_bytes - old_write_bytes) / (new_time - old_time)) | |
| read_speed = int((read_bytes - old_read_bytes) / (new_time - old_time)) | |
| return write_speed,read_speed | |
| def get_cpu_precent(self,p): | |
| ''' | |
| @name 获取进程cpu使用率 | |
| @author hwliang<2021-08-09> | |
| @param p: Process<进程对像> | |
| @return dict | |
| ''' | |
| skey = "cpu_pre_{}".format(p.pid) | |
| old_cpu_times = cache.get(skey) | |
| process_cpu_time = self.get_process_cpu_time(p.cpu_times()) | |
| if not old_cpu_times: | |
| cache.set(skey,[process_cpu_time,time.time()],3600) | |
| # time.sleep(0.1) | |
| old_cpu_times = cache.get(skey) | |
| process_cpu_time = self.get_process_cpu_time(p.cpu_times()) | |
| old_process_cpu_time = old_cpu_times[0] | |
| old_time = old_cpu_times[1] | |
| new_time = time.time() | |
| cache.set(skey,[process_cpu_time,new_time],3600) | |
| percent = round(100.00 * (process_cpu_time - old_process_cpu_time) / (new_time - old_time) / psutil.cpu_count(),2) | |
| return percent | |
| def get_process_cpu_time(self,cpu_times): | |
| cpu_time = 0.00 | |
| for s in cpu_times: cpu_time += s | |
| return cpu_time | |
| def get_project_run_state(self,get = None,project_name = None): | |
| ''' | |
| @name 获取项目运行状态 | |
| @author hwliang<2021-08-12> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @param project_name<string> 项目名称 | |
| @return bool | |
| ''' | |
| if get: project_name = get.project_name.strip() | |
| pid_file = "{}/{}.pid".format(self._node_pid_path,project_name) | |
| if not os.path.exists(pid_file): return False | |
| data=public.readFile(pid_file) | |
| if isinstance(data,str) and data: | |
| try: | |
| pid = int(data) | |
| pids = self.get_project_pids(pid=pid) | |
| except Exception: | |
| return self.get_project_state_by_cwd(project_name) | |
| else: | |
| return self.get_project_state_by_cwd(project_name) | |
| if not pids: return self.get_project_state_by_cwd(project_name) | |
| return True | |
| def get_project_find(self,project_name): | |
| ''' | |
| @name 获取指定项目配置 | |
| @author hwliang<2021-08-09> | |
| @param project_name<string> 项目名称 | |
| @return dict | |
| ''' | |
| project_info = public.M('sites').where('project_type=? AND name=?',('Node',project_name)).find() | |
| if isinstance(project_info, str): | |
| raise public.PanelError('数据库查询错误:'+ project_info) | |
| if not project_info: return False | |
| project_info['project_config'] = json.loads(project_info['project_config']) | |
| return project_info | |
| def get_project_info(self,get): | |
| ''' | |
| @name 获取指定项目信息 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| } | |
| @return dict | |
| ''' | |
| project_info = public.M('sites').where('project_type=? AND name=?',('Node',get.project_name)).find() | |
| if not project_info: return public.return_error('指定项目不存在!') | |
| project_info = self.get_project_stat(project_info) | |
| return project_info | |
| def get_project_stat(self,project_info): | |
| ''' | |
| @name 获取项目状态信息 | |
| @author hwliang<2021-08-09> | |
| @param project_info<dict> 项目信息 | |
| @return list | |
| ''' | |
| project_info['project_config'] = json.loads(project_info['project_config']) | |
| project_info['load_info'] = {} | |
| if ("pm2_name" in project_info['project_config'] and | |
| "project_type" in project_info['project_config'] and | |
| project_info['project_config']["project_type"] == "pm2") or "pm2" in project_info['project_config']['project_script']: | |
| from mod.project.nodejs.pm2Mod import main | |
| pm2_list = main().get_jlist() | |
| project_info['run'] = False | |
| pm2_name = project_info['project_config'].get("pm2_name", project_info['name']) | |
| pm2_pids = [] | |
| for pm2_info in pm2_list: | |
| if pm2_info.get("name") in ("pm2-sysmonit", "pm2-logrotate"): | |
| continue | |
| env_projectname = "" | |
| if "pm2_env" in pm2_info and "NODE_PROJECT_NAME" in pm2_info["pm2_env"]: | |
| env_projectname = pm2_info["pm2_env"]["NODE_PROJECT_NAME"] | |
| if pm2_name != pm2_info.get("name") and pm2_name != env_projectname: | |
| continue | |
| project_info['run'] = pm2_info.get("pm2_env").get("status") == "online" | |
| if project_info['run']: | |
| pm2_pids.append(pm2_info.get("pid")) | |
| break | |
| project_info['load_info'] = self.get_pm2_load_info_new(pm2_pids) | |
| else: | |
| project_info['run'] = self.get_project_run_state(project_name=project_info['name']) | |
| if project_info['run']: | |
| project_info['load_info'] = self.get_project_load_info(project_name=project_info['name']) | |
| project_info['ssl'] = self.get_ssl_end_date(project_name=project_info['name']) | |
| project_info['listen'] = [] | |
| project_info['listen_ok'] = True | |
| if project_info['load_info']: | |
| for pid in project_info['load_info'].keys(): | |
| if not 'connections' in project_info['load_info'][pid]: | |
| project_info['load_info'][pid]['connections'] = [] | |
| for conn in project_info['load_info'][pid]['connections']: | |
| if not conn['status'] == 'LISTEN': continue | |
| if not conn['local_port'] in project_info['listen']: | |
| project_info['listen'].append(conn['local_port']) | |
| if project_info['listen']: | |
| if project_info['project_config']['port'] is None: | |
| project_info['project_config']['port'] = project_info['listen'][0] | |
| project_info['listen_ok'] = project_info['project_config']['port'] in project_info['listen'] | |
| return project_info | |
| def get_project_state(self,project_name): | |
| ''' | |
| @name 获取项目状态 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return dict | |
| ''' | |
| project_info = public.M('sites').where('project_type=? AND name=?',('Node',project_name)).find() | |
| if not project_info: return False | |
| return project_info['status'] | |
| def get_project_listen(self,project_name): | |
| ''' | |
| @name 获取项目监听端口 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return dict | |
| ''' | |
| project_config = json.loads(public.M('sites').where('name=?',project_name).getField('project_config')) | |
| if 'listen_port' in project_config: return project_config['listen_port'] | |
| return False | |
| def set_project_listen(self,get): | |
| ''' | |
| @name 设置项目监听端口(请设置与实际端口相符的,仅在自动获取不正确时使用) | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| port: int<端口> | |
| } | |
| @return dict | |
| ''' | |
| project_config = json.loads(public.M('sites').where('name=?',get.project_name).getField('project_config')) | |
| project_config['listen_port'] = get.port | |
| public.M('sites').where('name=?',get.project_name).save('project_config',json.dumps(project_config)) | |
| public.WriteLog(self._log_name, '修改项目['+get.project_name+']的端口为为['+get.port+']') | |
| return public.return_data(True,'设置成功') | |
| def set_project_nodejs_version(self,get): | |
| ''' | |
| @name 设置nodejs版本 | |
| @author hwliang<2021-08-09> | |
| @param get<dict_obj>{ | |
| project_name: string<项目名称> | |
| nodejs_version: string<nodejs版本> | |
| } | |
| @return dict | |
| ''' | |
| project_config = json.loads(public.M('sites').where('name=?',get.project_name).getField('project_config')) | |
| project_config['nodejs_version'] = get.nodejs_version | |
| public.M('sites').where('name=?',get.project_name).save('project_config',json.dumps(project_config)) | |
| public.WriteLog(self._log_name, '修改项目['+get.project_name+']的nodejs版本为['+get.nodejs_version+']') | |
| return public.return_data(True,'设置成功') | |
| def get_project_nodejs_version(self,project_name): | |
| ''' | |
| @name 获取nodejs版本 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return string | |
| ''' | |
| project_config = json.loads(public.M('sites').where('name=?',project_name).getField('project_config')) | |
| if 'nodejs_version' in project_config: return project_config['nodejs_version'] | |
| return False | |
| def check_port_is_used(self,port,sock=False): | |
| ''' | |
| @name 检查端口是否被占用 | |
| @author hwliang<2021-08-09> | |
| @param port: int<端口> | |
| @return bool | |
| ''' | |
| if not isinstance(port,int): port = int(port) | |
| if port == 0: return False | |
| project_list = public.M('sites').where('status=? AND project_type=?',(1,'Node')).field('name,path,project_config').select() | |
| for project_find in project_list: | |
| project_config = json.loads(project_find['project_config']) | |
| if not 'port' in project_config: continue | |
| try: | |
| if int(project_config['port']) == port: | |
| return True | |
| except: | |
| continue | |
| if sock: return False | |
| return public.check_tcp('127.0.0.1',port) | |
| def get_project_run_state_byaotu(self,project_name): | |
| ''' | |
| @name 获取项目运行状态 | |
| @author hwliang<2021-08-09> | |
| @param project_name: string<项目名称> | |
| @return dict | |
| ''' | |
| pid_file = "{}/{}.pid".format(self._node_pid_path,project_name) | |
| if not os.path.exists(pid_file): return False | |
| pid = public.readFile(pid_file) | |
| pids = self.get_project_pids(pid=pid) | |
| if not pids: return False | |
| return True | |
| def auto_run(self): | |
| ''' | |
| @name 自动启动所有项目 | |
| @author hwliang<2021-08-09> | |
| @return bool | |
| ''' | |
| project_list = public.M('sites').where('project_type=?',('Node',)).field('name,path,project_config').select() | |
| get= public.dict_obj() | |
| success_count = 0 | |
| error_count = 0 | |
| for project_find in project_list: | |
| try: | |
| project_config = json.loads(project_find['project_config']) | |
| if project_config['is_power_on'] in [0,False,'0',None]: continue | |
| project_name = project_find['name'] | |
| project_state = self.get_project_run_state(project_name=project_name) | |
| if not project_state: | |
| get.project_name = project_name | |
| result = self.start_project(get) | |
| if not result['status']: | |
| error_count += 1 | |
| error_msg = '自动启动Nodej项目['+project_name+']失败!' | |
| public.WriteLog(self._log_name, error_msg) | |
| else: | |
| success_count += 1 | |
| success_msg = '自动启动Nodej项目['+project_name+']成功!' | |
| public.WriteLog(self._log_name, success_msg) | |
| except: | |
| error_count += 1 | |
| if (success_count + error_count) < 1: return False | |
| dene_msg = '共需要启动{}个Nodejs项目,成功{}个,失败{}个'.format(success_count + error_count,success_count,error_count) | |
| public.WriteLog(self._log_name, dene_msg) | |
| return True | |
| def change_log_path(self, get): | |
| """"修改日志文件地址 | |
| @author baozi <202-03-13> | |
| @param: | |
| get ( dict ): 请求: 包含项目名称和新的路径 | |
| @return | |
| """ | |
| project_info = self.get_project_find(get.project_name.strip()) | |
| if not project_info: return public.returnMsg(False,'项目不存在') | |
| new_log_path = get.path.strip() if "path" in get else None | |
| if not new_log_path or new_log_path[0] != "/": | |
| return public.returnMsg(False, "路径设置错误") | |
| if new_log_path[-1] == "/": new_log_path = new_log_path[:-1] | |
| if not os.path.exists(new_log_path): | |
| os.makedirs(new_log_path, mode=0o777) | |
| project_info['project_config']['log_path'] = new_log_path | |
| pdata = { | |
| 'name': project_info["name"], | |
| 'project_config': json.dumps(project_info['project_config']) | |
| } | |
| public.M('sites').where('name=?',(get.project_name.strip(),)).update(pdata) | |
| #重启项目 | |
| #return self.restart_project(get) | |
| res = self.stop_project(get) | |
| res = self.start_project(get) | |
| public.WriteLog(self._log_name,'Nodejs项目{}, 修改日志路径成功'.format(get.project_name)) | |
| return public.returnMsg(True, "项目日志路径修改成功") | |
| def for_split(self, logsplit, project): | |
| """日志切割方法调用 | |
| @author baozi <202-03-20> | |
| @param: | |
| logsplit ( LogSplit ): 日志切割方法,传入 pjanme:项目名称 sfile:日志文件路径 log_prefix:产生的日志文件前缀 | |
| project ( dict ): 项目内容 | |
| @return | |
| """ | |
| log_file = "{}/{}.log".format(project['project_config']["log_path"], project["name"]) | |
| logsplit(project["name"], log_file, project["name"]) | |
| #————————————— | |
| # 日志切割 | | |
| #————————————— | |
| def del_crontab(self, name): | |
| """ | |
| @name 删除项目日志切割任务 | |
| @auther hezhihong<2022-10-31> | |
| @return | |
| """ | |
| cron_name = f'[勿删]Node项目[{name}]运行日志切割' | |
| cron_path = public.GetConfigValue('setup_path') + '/cron/' | |
| cron_list = public.M('crontab').where("name=?", (cron_name, )).select() | |
| if cron_list: | |
| for i in cron_list: | |
| if not i: continue | |
| cron_echo = public.M('crontab').where("id=?", (i['id'], )).getField('echo') | |
| args = {"id": i['id']} | |
| import crontab | |
| crontab.crontab().DelCrontab(args) | |
| del_cron_file = cron_path + cron_echo | |
| public.ExecShell("crontab -u root -l| grep -v '{}'|crontab -u root -".format(del_cron_file)) | |
| def add_crontab(self, name, log_conf, python_path): | |
| """ | |
| @name 构造站点运行日志切割任务 | |
| """ | |
| cron_name = f'[勿删]Node项目[{name}]运行日志切割' | |
| if not public.M('crontab').where('name=?',(cron_name, )).count(): | |
| cmd = '{pyenv} {script_path} {name}'.format( | |
| pyenv = python_path, | |
| script_path = self.__log_split_script_py, | |
| name = name | |
| ) | |
| args = { | |
| "name": cron_name, | |
| "type": 'day' if log_conf["log_size"] == 0 else "minute-n", | |
| "where1": "" if log_conf["log_size"] == 0 else log_conf["minute"], | |
| "hour": log_conf["hour"], | |
| "minute": log_conf["minute"], | |
| "sName": name, | |
| "sType": 'toShell', | |
| "notice": '0', | |
| "notice_channel": '', | |
| "save": str(log_conf["num"]), | |
| "save_local": '1', | |
| "backupTo": '', | |
| "sBody": cmd, | |
| "urladdress": '' | |
| } | |
| import crontab | |
| res = crontab.crontab().AddCrontab(args) | |
| if res and "id" in res.keys(): | |
| return True, "新建任务成功" | |
| return False, res["msg"] | |
| return True | |
| def change_cronta(self, name, log_conf): | |
| """ | |
| @name 更改站点运行日志切割任务 | |
| """ | |
| python_path = "/www/server/panel/pyenv/bin/python3" | |
| if not python_path: return False | |
| cronInfo = public.M('crontab').where('name=?',(f'[勿删]Node项目[{name}]运行日志切割', )).find() | |
| if not cronInfo: | |
| return self.add_crontab(name, log_conf, python_path) | |
| import crontab | |
| recrontabMode = crontab.crontab() | |
| id = cronInfo['id'] | |
| del(cronInfo['id']) | |
| del(cronInfo['addtime']) | |
| cronInfo['sBody'] = '{pyenv} {script_path} {name}'.format( | |
| pyenv = python_path, | |
| script_path = self.__log_split_script_py, | |
| name = name | |
| ) | |
| cronInfo['where_hour'] = log_conf['hour'] | |
| cronInfo['where_minute'] = log_conf['minute'] | |
| cronInfo['save'] = log_conf['num'] | |
| cronInfo['type'] = 'day' if log_conf["log_size"] == 0 else "minute-n" | |
| cronInfo['where1'] = '' if log_conf["log_size"] == 0 else log_conf['minute'] | |
| columns = 'where_hour,where_minute,sBody,save,type,where1' | |
| values = (cronInfo['where_hour'],cronInfo['where_minute'],cronInfo['sBody'], cronInfo['save'], cronInfo['type'],cronInfo['where1']) | |
| recrontabMode.remove_for_crond(cronInfo['echo']) | |
| if cronInfo['status'] == 0: return False, '当前任务处于停止状态,请开启任务后再修改!' | |
| sync_res=recrontabMode.sync_to_crond(cronInfo) | |
| if not sync_res['status']: | |
| return False,sync_res['msg'] | |
| public.M('crontab').where('id=?',(id,)).save(columns,values) | |
| public.WriteLog('计划任务','修改计划任务['+cronInfo['name']+']成功') | |
| return True, '修改成功' | |
| def mamger_log_split(self,get): | |
| """管理日志切割任务 | |
| @author baozi <202-02-27> | |
| @param: | |
| get ( dict ): 包含name, mode, hour, minute | |
| @return | |
| """ | |
| name = get.name.strip() | |
| project = self.get_project_find(name) | |
| if not project: | |
| return public.returnMsg(False, "没有该项目,请尝试刷新页面") | |
| try: | |
| _compress = False | |
| _log_size = float(get.log_size) if float(get.log_size)>=0 else 0 | |
| _hour = get.hour.strip() if 0 <= int(get.hour) < 24 else "2" | |
| _minute = get.minute.strip() if 0 <= int(get.minute) < 60 else '0' | |
| _num = int(get.num) if 0 < int(get.num) <= 1800 else 180 | |
| if "compress" in get: | |
| _compress = int(get.compress) == 1 | |
| except (ValueError, AttributeError): | |
| _log_size = 0 | |
| _hour = "2" | |
| _minute = "0" | |
| _num = 180 | |
| _compress = False | |
| if _log_size != 0: | |
| _log_size = _log_size * 1024 * 1024 | |
| _hour = 0 | |
| _minute = 5 | |
| log_conf = { | |
| "log_size": _log_size, | |
| "hour": _hour, | |
| "minute": _minute, | |
| "num": _num, | |
| "compress": _compress | |
| } | |
| flag, msg = self.change_cronta(name, log_conf) | |
| if flag: | |
| conf_path = '{}/data/run_log_split.conf'.format(public.get_panel_path()) | |
| if os.path.exists(conf_path): | |
| try: | |
| data = json.loads(public.readFile(conf_path)) | |
| except: | |
| data = {} | |
| else: | |
| data = {} | |
| data[name] = { | |
| "stype": "size" if bool(_log_size) else "day", | |
| "log_size": _log_size, | |
| "limit": _num, | |
| "compress": _compress | |
| } | |
| public.writeFile(conf_path, json.dumps(data)) | |
| project["project_config"]["log_conf"] = log_conf | |
| pdata = { | |
| "project_config":json.dumps(project["project_config"]) | |
| } | |
| public.M('sites').where('name=?',(name,)).update(pdata) | |
| return public.returnMsg(flag, msg) | |
| def set_log_split(self,get): | |
| """设置日志计划任务状态 | |
| @author baozi <202-02-27> | |
| @param: | |
| get ( dict ): 包含项目名称name | |
| @return msg : 操作结果 | |
| """ | |
| name = get.name.strip() | |
| project_conf = self.get_project_find(name) | |
| if not project_conf: | |
| return public.returnMsg(False, "没有该项目,请尝试刷新页面") | |
| cronInfo = public.M('crontab').where('name=?',(f'[勿删]Node项目[{name}]运行日志切割', )).find() | |
| if not cronInfo: | |
| return public.returnMsg(False, "该项目没有设置运行日志的切割任务") | |
| status_msg = ['停用','启用'] | |
| status = 1 | |
| import crontab | |
| recrontabMode = crontab.crontab() | |
| if cronInfo['status'] == status: | |
| status = 0 | |
| recrontabMode.remove_for_crond(cronInfo['echo']) | |
| else: | |
| cronInfo['status'] = 1 | |
| sync_res=recrontabMode.sync_to_crond(cronInfo) | |
| if not sync_res['status']: | |
| return public.returnMsg(False, sync_res['msg']) | |
| public.M('crontab').where('id=?',(cronInfo["id"],)).setField('status',status) | |
| public.WriteLog('计划任务','修改计划任务['+cronInfo['name']+']状态为['+status_msg[status]+']') | |
| return public.returnMsg(True,'设置成功') | |
| def get_log_split(self,get): | |
| """获取站点的日志切割任务 | |
| @author baozi <202-02-27> | |
| @param: | |
| get ( dict ): name | |
| @return msg : 操作结果 | |
| """ | |
| name = get.name.strip() | |
| project_conf = self.get_project_find(name) | |
| if not project_conf: | |
| return public.returnMsg(False, "没有该项目,请尝试刷新页面") | |
| if self._check_old(project_conf): | |
| return {"status": False, "msg": "更新版本后需要重启项目,才能开启运行日志切割任务,建议您找一个合适的时间重启项目", "is_old": True} | |
| cronInfo = public.M('crontab').where('name=?',(f'[勿删]Node项目[{name}]运行日志切割', )).find() | |
| if not cronInfo: | |
| return public.returnMsg(False, "该项目没有设置运行日志的切割任务") | |
| if "log_conf" not in project_conf["project_config"]: | |
| return public.returnMsg(False, "日志切割配置丢失,请尝试重新设置") | |
| res = project_conf["project_config"]["log_conf"] | |
| res["status"] = cronInfo["status"] | |
| return {"status": True,"data": res} | |
| def _update_project(self, project_name, project_info): | |
| # 检查是否需要更新 | |
| # 移动日志文件 | |
| # 保存 | |
| target_file = self._node_logs_path + "/" + project_name + ".log" | |
| if "log_path" in project_info['project_config']: | |
| return | |
| log_file = "{}/{}.log".format(self._node_logs, project_name) | |
| if os.path.exists(log_file): | |
| self._move_logs(log_file, target_file) | |
| if not os.path.exists(target_file): | |
| return | |
| else: | |
| os.remove(log_file) | |
| project_info['project_config']["log_path"] = self._node_logs_path | |
| pdata = { | |
| 'name':project_name, | |
| 'project_config': json.dumps(project_info['project_config']) | |
| } | |
| public.M('sites').where('name=?',(project_name,)).update(pdata) | |
| def _move_logs(self, s_file, target_file): | |
| if os.path.getsize(s_file) > 3145928 : | |
| res = public.GetNumLines(s_file, 3000) | |
| public.WriteFile(target_file, res) | |
| else: | |
| shutil.copyfile(s_file, target_file) | |
| def _check_old(self, project_info): | |
| if not "log_path" in project_info['project_config']: | |
| return True | |
| def _ckeck_add_domain(self, site_name, domains): | |
| from panelSite import panelSite | |
| ssl_data = panelSite().GetSSL(type("get", tuple(), {"siteName": site_name})()) | |
| if not ssl_data["status"] or not ssl_data.get("cert_data", {}).get("dns", None): | |
| return {"domains": domains} | |
| domain_rep = [] | |
| for i in ssl_data["cert_data"]["dns"]: | |
| if i.startswith("*"): | |
| _rep = "^[^\.]+\." + i[2:].replace(".", "\.") | |
| else: | |
| _rep = "^" + i.replace(".", "\.") | |
| domain_rep.append(_rep) | |
| no_ssl = [] | |
| for domain in domains: | |
| if not domain["status"]: continue | |
| for _rep in domain_rep: | |
| if re.search(_rep, domain["name"]): | |
| break | |
| else: | |
| no_ssl.append(domain["name"]) | |
| if no_ssl: | |
| return { | |
| "domains": domains, | |
| "not_ssl": no_ssl, | |
| "tip": "本站点已启用SSL证书,但本次添加的域名:{},无法匹配当前证书,如有需求,请重新申请证书。".format(str(no_ssl)) | |
| } | |
| return {"domains": domains} | |
| def get_project_status(self, project_id): | |
| # 仅使用在项目停止告警中 | |
| project_info = public.M('sites').where('project_type=? AND id=?', ('Node', project_id)).find() | |
| if not project_info: | |
| return None, project_info["name"] | |
| if self.is_stop_by_user(project_id): | |
| return True, project_info["name"] | |
| res = self.get_project_run_state(project_name=project_info['name']) | |
| return res, project_info["name"] | |
| def _get_nginx_proxy_port(config_file: str) -> Optional[int]: | |
| conf_data = public.readFile(config_file) | |
| if conf_data or not isinstance(conf_data, str): | |
| return None | |
| from urllib import parse | |
| regexp = re.compile(r'(?P<sign>#.*)?[ \t\r\v\f]*location\s+/\s+{\s+proxy_pass\s+(?P<proxy_url>[^;])+;') | |
| for tmp in regexp.finditer(conf_data): | |
| if tmp.group('sign'): # 注释的配置 | |
| continue | |
| if not tmp.group('proxy_url'): | |
| continue | |
| url = parse.urlparse(tmp.group('proxy_url')) | |
| if url.port: | |
| return int(url.port) | |
| return None | |