Initial YakPanel commit

class/msg/sms_msg.py

@@ -0,0 +1,218 @@
+#coding: utf-8
+# +-------------------------------------------------------------------
+# | YakPanel
+# +-------------------------------------------------------------------
+# | Copyright (c) 2015-2020 YakPanel(www.yakpanel.com) All rights reserved.
+# +-------------------------------------------------------------------
+# | Author: 沐落 <cjx@yakpanel.com>
+# | Author: lx
+# | 消息通道邮箱模块
+# +-------------------------------------------------------------------
+
+import os, sys, public, base64, json, re
+import sys, os
+panelPath = "/www/server/panel"
+os.chdir(panelPath)
+sys.path.insert(0,panelPath + "/class/")
+import public
+
+class sms_msg:
+
+    _APIURL = 'http://www.yakpanel.com/api/wmsg';
+    __UPATH = panelPath + '/data/userInfo.json';
+    conf_path = panelPath + '/data/sms_main.json'
+
+        #构造方法
+    def __init__(self):
+        self.setupPath = public.GetConfigValue('setup_path')
+        pdata = {}
+        data = {}
+        if os.path.exists(self.__UPATH):
+            try:
+                self.__userInfo = json.loads(public.readFile(self.__UPATH));
+
+                if self.__userInfo:
+                    pdata['access_key'] = self.__userInfo['access_key'];
+                    data['secret_key'] = self.__userInfo['secret_key'];
+            except :
+                self.__userInfo = None
+        else:
+            pdata['access_key'] = 'test'
+            data['secret_key'] = '123456'
+
+        pdata['data'] = data
+        self.__PDATA = pdata
+        self._APIURL = public.GetConfigValue('home').rstrip('/') + '/api/wmsg'
+        self.__module_name = self.__class__.__name__.replace('_msg','')
+
+
+    def get_version_info(self,get):
+        """
+        获取版本信息
+        """
+        data = {}
+        data['ps'] = 'YakPanel 短信消息通道，用于接收面板消息推送'
+        data['version'] = '1.1'
+        data['date'] = '2022-08-02'
+        data['author'] = 'YakPanel'
+        data['title'] = '短信'
+        data['help'] = 'http://www.yakpanel.com'
+        return data
+
+    def get_config(self,get):
+        result = {}
+        data = {}
+        skey = 'sms_count_{}'.format(public.get_user_info()['username'])
+        try:
+            from YakPanel import cache
+            result = cache.get(skey)
+        except:
+            cache = None
+        if not result:
+            result = self.request('get_user_sms')
+            if cache: cache.set(skey,result,3600)
+        try:
+            data = json.loads(public.readFile(self.conf_path))
+        except :pass
+
+        for key in data.keys():
+            result[key] = data[key]
+        return result
+
+
+    def is_strong_password(self,password):
+        """判断密码复杂度是否安全
+        非弱口令标准：长度大于等于9，分别包含数字、小写。
+        @return: True/False
+        @author: linxiao<2020-9-19>
+        """
+        if len(password) < 6:return False
+
+        import re
+        digit_reg = "[0-9]"  # 匹配数字 +1
+        lower_case_letters_reg = "[a-z]"  # 匹配小写字母 +1
+        special_characters_reg = r"((?=[\x21-\x7e]+)[^A-Za-z0-9])"  # 匹配特殊字符 +1
+
+        regs = [digit_reg,lower_case_letters_reg,special_characters_reg]
+        grade = 0
+        for reg in regs:
+            if re.search(reg, password):
+                grade += 1
+
+        if grade >= 2 or (grade == 1 and len(password) >= 9):
+            return True
+        return False
+
+    def __check_auth_path(self):
+
+        auth = public.readFile('data/admin_path.pl')
+        if not auth: return False
+
+        slist = ['/','/123456','/admin123','/111111','/bt','/login','/cloudtencent','/tencentcloud','/admin','/admin888','/test']
+        if auth in slist: return False
+
+        if not self.is_strong_password(auth.strip('/')):
+            return False
+        return True
+
+    def set_config(self,get):
+
+        data = {}
+        try:
+            data = json.loads(public.readFile(self.conf_path))
+        except :pass
+
+        if 'login' in get:
+            is_login = int(get['login'])
+            if is_login and not self.__check_auth_path(): return public.returnMsg(False, public.lang("Security entry complexity is not enough <br>1, length must not be less than 9 <br>2, English + digital combination."))
+            data['login'] = is_login
+
+        public.writeFile(self.conf_path,json.dumps(data));
+        return public.returnMsg(True, public.lang("operate successfully!"))
+
+    """
+    @发送短信
+    @sm_type 预警类型, ssl_end|YakPanel SSL到期提醒
+    @sm_args 预警参数
+    """
+    def send_msg(self,sm_type = None,sm_args = None):
+
+        s_type = sm_type
+        title = 'YakPanel 告警提醒'
+        tmps = sm_type.split('|')
+        if len(tmps) >= 2:
+            s_type = tmps[0]
+            title = tmps[1]
+
+        self.__PDATA['data']['sm_type'] = s_type
+        self.__PDATA['data']['sm_args'] = sm_args
+        result = self.request('send_msg')
+
+        try:
+
+            res = {}
+            uinfo = public.get_user_info()
+            u_key = '{}****{}'.format(uinfo['username'][0:3],uinfo['username'][-3:])
+
+            res[u_key] = 0
+            if result['status']:
+                res[u_key] = 1
+
+                skey = 'sms_count_{}'.format(public.get_user_info()['username'])
+                try:
+                    from YakPanel import cache
+                except:
+                    cache = None
+                if not result:
+                    result = self.request('get_user_sms')
+                    if cache: cache.set(skey,result,3600)
+
+            public.write_push_log(self.__module_name,title,res)
+        except:pass
+
+        return result
+
+    def canonical_data(self, args):
+        """规范数据内容
+
+        Args:
+            args(dict): 消息原始参数
+
+        Returns:
+            new args: 替换后的消息参数
+        """
+
+        if not type(args) == dict: return args
+        new_args = {}
+        for param, value in args.items():
+            if type(value) != str:
+                new_str = str(value)
+            else:
+                new_str = value.replace(".", "_").replace("+", "＋")
+            new_args[param] = new_str
+        return new_args
+
+    def push_data(self,data):
+        sm_args = self.canonical_data(data['sm_args'])
+        return self.send_msg(data['sm_type'],sm_args)
+
+    #发送请求
+    def request(self,dname):
+
+        pdata = {}
+        pdata['access_key'] = self.__PDATA['access_key']
+        pdata['data'] = json.dumps(self.__PDATA['data'])
+        try:
+            result = public.httpPost(self._APIURL + '/' + dname,pdata)
+            result = json.loads(result)
+            # print("发送result:")
+            # print(result)
+            return result
+        except Exception as e:
+            # print("短信发送异常:")
+            # print(e)
+            return public.returnMsg(False,public.get_error_info())
+
+    def uninstall(self):
+        if os.path.exists(self.conf_path):
+            os.remove(self.conf_path)