From c9927c3fb99ea4a9218f31e08f7ac36f95b66dec Mon Sep 17 00:00:00 2001
From: midoks <midoks@163.com>
Date: Tue, 14 Mar 2023 00:09:30 +0800
Subject: [PATCH] =?UTF-8?q?tgbot=E5=8A=A0=E5=85=A5=E6=89=A9=E5=B1=95?=
 =?UTF-8?q?=E6=9C=BA=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 plugins/tgbot/index.html                    |  2 +-
 plugins/tgbot/index.py                      | 85 ++++++++++++++++++---
 plugins/tgbot/js/tgbot.js                   | 80 ++++++++++---------
 plugins/tgbot/startup/extend/receive_faq.py | 14 ++++
 plugins/tgbot/startup/tgbot.py              | 69 ++++++++++++-----
 5 files changed, 179 insertions(+), 71 deletions(-)
 create mode 100644 plugins/tgbot/startup/extend/receive_faq.py

diff --git a/plugins/tgbot/index.html b/plugins/tgbot/index.html
index b0020e579..973239f62 100755
--- a/plugins/tgbot/index.html
+++ b/plugins/tgbot/index.html
@@ -19,7 +19,7 @@
 }
 </style>
 <script type="text/javascript">
-resetPluginWinHeight(300);
+resetPluginWinHeight(350);
 $.getScript( "/plugins/file?name=tgbot&f=js/tgbot.js", function(){
 	pluginService('tgbot');
 });
diff --git a/plugins/tgbot/index.py b/plugins/tgbot/index.py
index aebb79b1a..ba1b86124 100755
--- a/plugins/tgbot/index.py
+++ b/plugins/tgbot/index.py
@@ -58,10 +58,7 @@ def getExtCfg():
 
 def writeExtCfg(data):
     cfg_path = getServerDir() + "/extend.cfg"
-    if not os.path.exists(cfg_path):
-        mw.writeFile(cfg_path, '{}')
-    t = mw.readFile(cfg_path)
-    return json.loads(t)
+    return mw.writeFile(cfg_path, json.dumps(data))
 
 
 def getInitDTpl():
@@ -147,9 +144,9 @@ def tbOp(method):
         return data[1]
 
     data = mw.execShell(file + ' ' + method)
-    print(data)
-    # if data[1] == '':
-    #     return 'ok'
+    # print(data)
+    if data[1] == '':
+        return 'ok'
     return 'ok'
 
 
@@ -243,7 +240,21 @@ def uninstallPreInspection():
     return "请手动删除<br/> rm -rf {}".format(getServerDir())
 
 
+def getExtCfgByName(name):
+    elist = getExtCfg()
+    for x in elist:
+        if x['name'] == name:
+            return x
+    return None
+
+
 def botExtList():
+
+    args = getArgs()
+    data_args = checkArgs(args, ['p'])
+    if not data_args[0]:
+        return data_args[1]
+
     ext_path = getServerDir() + '/extend'
     if not os.path.exists(ext_path):
         return mw.returnJson(False, 'ok', [])
@@ -254,9 +265,63 @@ def botExtList():
         if e.endswith('py'):
             elist.append(e)
 
-    # extList = getExtCfg()
+    page = int(args['p'])
+    page_size = 5
+
+    make_ext_list = []
+    for ex in elist:
+        tmp = {}
+        tmp['name'] = ex
+        edata = getExtCfgByName(ex)
+        if edata:
+            tmp['status'] = edata['status']
+        else:
+            tmp['status'] = 'stop'
+
+        tmp['tag'] = ex.split('_')[0]
+        make_ext_list.append(tmp)
+
+    writeExtCfg(make_ext_list)
+    dlist_sum = len(make_ext_list)
+
+    page_start = int((page - 1) * page_size)
+    page_end = page_start + page_size
+
+    if page_end >= dlist_sum:
+        ret_data = make_ext_list[page_start:]
+    else:
+        ret_data = make_ext_list[page_start:page_end]
+
+    data = {}
+    data['data'] = ret_data
+    data['args'] = args
+    data['list'] = mw.getPage(
+        {'count': dlist_sum, 'p': page, 'row': page_size, 'tojs': 'botExtListP'})
+
+    return mw.returnJson(True, 'ok', data)
+
+
+def setExtStatus():
+    args = getArgs()
+    data_args = checkArgs(args, ['name', 'status'])
+    if not data_args[0]:
+        return data_args[1]
+
+    elist = getExtCfg()
+    name = args['name']
+    status = args['status']
+    for x in range(len(elist)):
+        if elist[x]['name'] == name:
+            elist[x]['status'] = status
+            break
+
+    writeExtCfg(elist)
+
+    action = '开启'
+    if status == 'stop':
+        action = '关闭'
 
-    print(elist)
+    return mw.returnJson(True, action + '[' + name + ']扩展成功')
 
 
 def runLog():
@@ -292,6 +357,8 @@ if __name__ == "__main__":
         print(setBotConf())
     elif func == 'bot_ext_list':
         print(botExtList())
+    elif func == 'set_ext_status':
+        print(setExtStatus())
     elif func == 'run_log':
         print(runLog())
 
diff --git a/plugins/tgbot/js/tgbot.js b/plugins/tgbot/js/tgbot.js
index 0170cdcd7..30e53c214 100755
--- a/plugins/tgbot/js/tgbot.js
+++ b/plugins/tgbot/js/tgbot.js
@@ -86,7 +86,6 @@ function botExtList(){
                         <th width="20">脚本</th>\
                         <th width="120">类型</th>\
                         <th width="10">状态</th>\
-                        <th style="text-align: right;" width="50">操作</th>\
                     </tr>\
                 </thead>\
                 <tbody id="ext_list"></tbody>\
@@ -96,55 +95,54 @@ function botExtList(){
             </div>\
         </div>';
     $('.soft-man-con').html(body);
+    botExtListP(1);
+}
 
-
-    botExtListP(1)
+function setBotExtStatus(name,status){
+    appPost('set_ext_status',{'name':name,'status':status}, function(rdata){
+        var rdata = $.parseJSON(rdata.data);
+        layer.msg(rdata['msg'],);
+        showMsg(rdata['msg'], function(){
+            botExtListP(1);
+        },{icon:rdata['status']?1:2,shade: [0.3, '#000']},2000);
+    });
 }
+
 function botExtListP(p=1){
-    appPost('bot_ext_list',{'page':p}, function(rdata){
+    appPost('bot_ext_list',{'p':p}, function(rdata){
+        // console.log(rdata);
         var rdata = $.parseJSON(rdata.data);
+        // console.log(rdata);
         var tBody = '';
 
-        if (rdata.data.length == 0 ){
+        if (!rdata.status && rdata.data.length == 0 ){
             var tBody = '<tr><td colspan="4"><div style="text-align:center;">无数据</div></td></tr>';
+        } else{
+            var ldata = rdata.data.data;
+            for (var i = 0; i < ldata.length; i++) {
+                tBody += '<tr data-name="'+ldata[i]['name']+'">'
+                tBody += '<td>'+ldata[i]['name']+'</td>';
+                tBody += '<td>'+ldata[i]['tag']+'</td>';
+
+                if (ldata[i]['status'] == 'start'){
+                    tBody += '<td><span style="color:#20a53a;cursor: pointer;"  class="ext_status glyphicon glyphicon-play"></span></td>';
+                } else{
+                    tBody += '<td><span style="color:red;cursor: pointer;" class="ext_status glyphicon glyphicon-pause"></span></td>';
+                }
+                tBody +='<tr>';
+            }
         }
+        
+        $('#ext_list').html(tBody);
+        $('#ext_list_page').html(rdata.data.list);
 
-        var ldata = rdata.data;
-        for (var i = 0; i < ldata.length; i++) {
-            tBody += '<tr data-id="'+ldata[i]['name']+'">'
-            tBody += '<td>'+ldata[i]['id']+'</td>';
-            tBody += '<td>'+ldata[i]['name']+'</td>';
-
-            if (ldata[i]['status'] == 'start'){
-                tBody += '<td><span style="color:#20a53a;cursor: pointer;"  class="strategy_status glyphicon glyphicon-play"></span></td>';
-            } else{
-                tBody += '<td><span style="color:red;cursor: pointer;" class="strategy_status glyphicon glyphicon-pause"></span></td>';
+        $('#ext_list .ext_status').click(function(){
+            var name = $(this).parent().parent().data('name');
+            var status = 'stop';
+            if ($(this).hasClass('glyphicon-pause')){
+                status = 'start';
             }
-            
-            tBody += "<td style='text-align: right;'><a class='btlink restart'>重启</a> | <a class='btlink edit'>编辑</a></td>";
-            tBody +='<tr>';
-        };
-        $('#ext_list').html(tBody);
-        // $('#strategy_list_page').html(rdata.data.list);
-
-
-        // $('#strategy_list .strategy_status').click(function(){
-        //     var id = $(this).parent().parent().data('id');
-        //     var status = 'stop';
-        //     if ($(this).hasClass('glyphicon-pause')){
-        //         status = 'start';
-        //     }
-        //     setStrategyStatus(id,status);
-        // });
-
-        // $('#strategy_list .restart').click(function(){
-        //     var id = $(this).parent().parent().data('id');
-        //     setStrategyRestart(id);
-        // });
-
-        // $('#strategy_list .edit').click(function(){
-        //     var id = $(this).parent().parent().data('id');
-        //     setStrategyEdit(id);
-        // });
+            setBotExtStatus(name,status);
+        });
     });
 }
diff --git a/plugins/tgbot/startup/extend/receive_faq.py b/plugins/tgbot/startup/extend/receive_faq.py
new file mode 100644
index 000000000..031efe3e9
--- /dev/null
+++ b/plugins/tgbot/startup/extend/receive_faq.py
@@ -0,0 +1,14 @@
+import sys
+import io
+import os
+import time
+import re
+import json
+import base64
+import threading
+
+import telebot
+
+
+def run(bot, message):
+    return bot
diff --git a/plugins/tgbot/startup/tgbot.py b/plugins/tgbot/startup/tgbot.py
index b73d1dd19..12045ec42 100644
--- a/plugins/tgbot/startup/tgbot.py
+++ b/plugins/tgbot/startup/tgbot.py
@@ -28,6 +28,9 @@ def getServerDir():
     return mw.getServerDir() + '/' + getPluginName()
 
 
+sys.path.append(getServerDir() + "/extend")
+
+
 def getConfigData():
     cfg_path = getServerDir() + "/data.cfg"
     if not os.path.exists(cfg_path):
@@ -42,6 +45,24 @@ def writeConf(data):
     return True
 
 
+def getExtCfg():
+    cfg_path = getServerDir() + "/extend.cfg"
+    if not os.path.exists(cfg_path):
+        mw.writeFile(cfg_path, '{}')
+    t = mw.readFile(cfg_path)
+    return json.loads(t)
+
+
+def getStartExtCfgByTag(tag='push'):
+    # 获取开启的扩展
+    elist = getExtCfg()
+    rlist = []
+    for x in elist:
+        if x['tag'] == tag and x['status'] == 'start':
+            rlist.append(x)
+    return rlist
+
+
 def writeLog(log_str):
     if __name__ == "__main__":
         print(log_str)
@@ -60,6 +81,7 @@ while True:
     writeLog('等待输入配置,填写app_token')
     time.sleep(3)
 
+
 bot = telebot.TeleBot(cfg['bot']['app_token'])
 
 
@@ -76,9 +98,6 @@ bot.set_my_commands(
         telebot.types.BotCommand("start", "查看帮助信息"),
         telebot.types.BotCommand("mw_chat_id", "查看群组ChatID")
     ],
-    # scope=telebot.types.BotCommandScopeChat(12345678)  # use for personal command for users
-    # scope=telebot.types.BotCommandScopeAllPrivateChats()  # use for all
-    # private chats
 )
 
 
@@ -87,13 +106,8 @@ def hanle_start_help(message):
     bot.reply_to(message, "hello world")
 
 
-@bot.message_handler(commands=['mw'])
-def hanle_start_mw(message):
-    bot.reply_to(message, "我就是最靓的仔!")
-
-
 @bot.message_handler(commands=['mw_echo'])
-def hanle_start_help(message):
+def hanle_mw_echo(message):
     bot.reply_to(message, message.text)
 
 
@@ -103,9 +117,30 @@ def hanle_get_chat_id(message):
 
 
 @bot.message_handler(func=lambda message: True)
-def echo_message(message):
-    print(message)
-    # bot.reply_to(message, "拦截所有消息:" + message.text)
+def all_message(message):
+    rlist = getStartExtCfgByTag('receive')
+    for r in rlist:
+        try:
+            script = r['name'].split('.')[0]
+            __import__(script).run(bot, message)
+        except Exception as e:
+            pass
+
+
+def runBotPushTask():
+    plist = getStartExtCfgByTag('push')
+    for p in plist:
+        try:
+            script = p['name'].split('.')[0]
+            __import__(script).run(bot)
+        except Exception as e:
+            pass
+
+
+def botPush():
+    while True:
+        runBotPushTask()
+        time.sleep(3)
 
 
 def setDaemon(t):
@@ -113,19 +148,13 @@ def setDaemon(t):
         t.daemon = True
     else:
         t.setDaemon(True)
-    return t
-
-
-def botPush():
-    while True:
-        print('a')
-        time.sleep(3)
 
 if __name__ == "__main__":
 
     # 机器人推送任务
     botPushTask = threading.Thread(target=botPush)
-    botPushTask = setDaemon(botPushTask)
+    # print(botPushTask)
+    # botPushTask = setDaemon(botPushTask)
     botPushTask.start()
 
     writeLog('启动成功')