From eb2c476b288dc0fba7ac6413251fd156f8323319 Mon Sep 17 00:00:00 2001
From: Mr Chen <midoks@163.com>
Date: Wed, 1 May 2024 20:33:36 +0800
Subject: [PATCH] update

---
 plugins/mongodb/config/mongodb.sql |  16 ++
 plugins/mongodb/index.html         |  15 +-
 plugins/mongodb/index.py           | 228 ++++++++++++++++++++++++-----
 plugins/mongodb/js/mongodb.js      |  59 ++++++++
 4 files changed, 282 insertions(+), 36 deletions(-)
 create mode 100644 plugins/mongodb/config/mongodb.sql

diff --git a/plugins/mongodb/config/mongodb.sql b/plugins/mongodb/config/mongodb.sql
new file mode 100644
index 000000000..2ad30f232
--- /dev/null
+++ b/plugins/mongodb/config/mongodb.sql
@@ -0,0 +1,16 @@
+CREATE TABLE IF NOT EXISTS `config` (
+  `id` INTEGER PRIMARY KEY AUTOINCREMENT,
+  `mg_root` TEXT
+);
+
+INSERT INTO `config` (`id`, `mg_root`) VALUES (1, 'mg_root');
+
+CREATE TABLE IF NOT EXISTS `users` (
+  `id` INTEGER PRIMARY KEY AUTOINCREMENT,
+  `username` TEXT,
+  `password` TEXT,
+  `role` TEXT DEFAULT '',
+  `ps` TEXT,
+  `addtime` TEXT
+);
+
diff --git a/plugins/mongodb/index.html b/plugins/mongodb/index.html
index 68bd81ae6..2bfac6ae7 100755
--- a/plugins/mongodb/index.html
+++ b/plugins/mongodb/index.html
@@ -6,7 +6,15 @@
     white-space: nowrap;
     display: inline-block;
     vertical-align: middle;
-}	
+}
+
+.bingfa .bt-input-text {
+    width: 200px;
+}
+
+.inlineBlock {
+    display: inline-block;
+}
 </style>
 
 <div class="bt-form">
@@ -14,7 +22,8 @@
 		<div class="bt-w-menu">
 			<p class="bgw" onclick="pluginService('mongodb');">服务</p>
 			<p onclick="pluginInitD('mongodb');">自启动</p>
-			<p onclick="pluginConfig('mongodb');">配置修改</p>
+			<p onclick="mongoSetConfig();">配置修改</p>
+			<p onclick="pluginConfig('mongodb');">配置文件</p>
 			<p onclick="mongoStatus();">负载状态</p>
 			<p onclick="mongoDocStatus();">文档状态</p>
 			<p onclick="mongoReplStatus();">复制状态</p>
@@ -25,6 +34,8 @@
 		</div>
 	</div>
 </div>
+
+<!-- <script type="text/javascript" src="/plugins/file?name=mongodb&f=js/mongodb.js"></script> -->
 <script type="text/javascript">
 //mongodb  start
 resetPluginWinHeight(530);
diff --git a/plugins/mongodb/index.py b/plugins/mongodb/index.py
index 2faa3efa6..e5dccc2a3 100755
--- a/plugins/mongodb/index.py
+++ b/plugins/mongodb/index.py
@@ -7,6 +7,7 @@ import time
 import re
 import json
 import datetime
+import yaml
 
 sys.path.append(os.getcwd() + "/class/core")
 import mw
@@ -46,6 +47,47 @@ def getConfTpl():
     path = getPluginDir() + "/config/mongodb.conf"
     return path
 
+def getConfigData():
+    cfg = getConf()
+    config_data = mw.readFile(cfg)
+    try:
+        config = yaml.safe_load(config_data)
+    except:
+        config = {
+            "systemLog": {
+                "destination": "file",
+                "logAppend": True,
+                "path": mw.getServerDir()+"/mongodb/log/mongodb.log"
+            },
+            "storage": {
+                "dbPath": mw.getServerDir()+"/mongodb/data",
+                "directoryPerDB": True,
+                "journal": {
+                    "enabled": True
+                }
+            },
+            "processManagement": {
+                "fork": True,
+                "pidFilePath": mw.getServerDir()+"/mongodb/log/mongodb.pid"
+            },
+            "net": {
+                "port": 27017,
+                "bindIp": "0.0.0.0"
+            },
+            "security": {
+                "authorization": "enabled",
+                "javascriptEnabled": False
+            }
+        }
+    return config
+
+def setConfig(config_data):
+    cfg = getConf()
+    try:
+        mw.writeFile(cfg, yaml.safe_dump(config_data))
+    except:
+        return False
+    return True
 
 def getInitDTpl():
     path = getPluginDir() + "/init.d/" + getPluginName() + ".tpl"
@@ -53,11 +95,22 @@ def getInitDTpl():
 
 
 def getConfPort():
-    file = getConf()
-    content = mw.readFile(file)
-    rep = 'port\s*=\s*(.*)'
-    tmp = re.search(rep, content)
-    return tmp.groups()[0].strip()
+    data = getConfigData()
+    return data['net']['port']
+    # file = getConf()
+    # content = mw.readFile(file)
+    # rep = 'port\s*=\s*(.*)'
+    # tmp = re.search(rep, content)
+    # return tmp.groups()[0].strip()
+
+def getConfAuth():
+    data = getConfigData()
+    return data['security']['authorization']
+    # file = getConf()
+    # content = mw.readFile(file)
+    # rep = 'auth\s*=\s*(.*)'
+    # tmp = re.search(rep, content)
+    # return tmp.groups()[0].strip()
 
 def getArgs():
     args = sys.argv[2:]
@@ -84,6 +137,61 @@ def status():
         return 'stop'
     return 'start'
 
+def pSqliteDb(dbname='users'):
+    file = getServerDir() + '/mongodb.db'
+    name = 'mongodb'
+
+    sql_file = getPluginDir() + '/config/mongodb.sql'
+    import_sql = mw.readFile(sql_file)
+    # print(sql_file,import_sql)
+    md5_sql = mw.md5(import_sql)
+
+    import_sign = False
+    save_md5_file = getServerDir() + '/import_mongodb.md5'
+    if os.path.exists(save_md5_file):
+        save_md5_sql = mw.readFile(save_md5_file)
+        if save_md5_sql != md5_sql:
+            import_sign = True
+            mw.writeFile(save_md5_file, md5_sql)
+    else:
+        mw.writeFile(save_md5_file, md5_sql)
+
+    if not os.path.exists(file) or import_sql:
+        conn = mw.M(dbname).dbPos(getServerDir(), name)
+        csql_list = import_sql.split(';')
+        for index in range(len(csql_list)):
+            conn.execute(csql_list[index], ())
+
+    conn = mw.M(dbname).dbPos(getServerDir(), name)
+    return conn
+
+def mongdbClient():
+    import pymongo
+    port = getConfPort()
+    auth = getConfAuth()
+    mg_root = pSqliteDb('config').where('id=?', (1,)).getField('mg_root')
+
+    # print(auth)
+    if auth == 'disabled':
+        client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True)
+    else:
+        client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True,username='root',password=mg_root)
+    return client
+
+def mongdbClientWithPass():
+    import pymongo
+    port = getConfPort()
+    auth = getConfAuth()
+    # print(auth)
+    mg_root = pSqliteDb('config').where('id=?', (1,)).getField('mg_root')
+    if auth == 'disabled':
+        client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True)
+    else:
+        uri = "mongodb://root:"+mg_root+"@127.0.0.1:"+str(port)
+        # print(uri)
+        client = pymongo.MongoClient(uri)
+    return client
+
 
 def initDreplace():
 
@@ -132,6 +240,40 @@ def initDreplace():
 
     return file_bin
 
+def initUserRoot():
+    # client = mongdbClient()
+    # db = client.admin
+
+    client_pass = mongdbClientWithPass()
+    listDbs = client_pass.admin.command({"listDatabases": 1})
+    print(listDbs)
+
+    print(client_pass.list_database_names());
+    exit(0)
+
+    # db.command("updateUser", "root", pwd=mg_pass, roles=db_all_rules)
+    # db_all_rules = [
+    #     {'role': 'root', 'db': 'admin'},
+    #     {'role': 'clusterAdmin', 'db': 'admin'},
+    #     {'role': 'readAnyDatabase', 'db': 'admin'},
+    #     {'role': 'readWriteAnyDatabase', 'db': 'admin'},
+    #     {'role': 'userAdminAnyDatabase', 'db': 'admin'},
+    #     {'role': 'dbAdminAnyDatabase', 'db': 'admin'},
+    #     {'role': 'userAdmin', 'db': 'admin'},
+    #     {'role': 'dbAdmin', 'db': 'admin'}
+    # ]
+
+    # mg_pass = mw.getRandomString(10)
+    # print(mg_pass)
+    # try:
+    #     db.command("createUser", "root", pwd=mg_pass, roles=db_all_rules)
+    # except Exception as e:
+    #     db.command('dropUser','root')
+    #     db.command("createUser", "root", pwd=mg_pass, roles=db_all_rules)
+        
+    # pSqliteDb('config').where('id=?', (1,)).save('mg_root',(mg_pass,))
+    return True
+
 
 def mgOp(method):
     file = initDreplace()
@@ -144,6 +286,7 @@ def mgOp(method):
 
     data = mw.execShell('systemctl ' + method + ' ' + getPluginName())
     if data[1] == '':
+        initUserRoot()
         return 'ok'
     return 'fail'
 
@@ -169,14 +312,19 @@ def restart():
     return mgOp('restart')
 
 
+def getConfig():
+    d = getConfigData()
+    return mw.returnJson(True,'ok',d)
+
+def saveConfig():
+    d = getConfigData()
+    return mw.returnJson(True,'保持成功')
+
 def runInfo():
     '''
     cd /www/server/mdserver-web && source bin/activate && python3 /www/server/mdserver-web/plugins/mongodb/index.py run_info
     '''
-    import pymongo
-    
-    port = getConfPort()
-    client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True)
+    client = mongdbClient()
     db = client.admin
     serverStatus = db.command('serverStatus')
 
@@ -196,11 +344,8 @@ def runInfo():
     return mw.getJson(result)
 
 
-def runDocInfo():
-    import pymongo
-    
-    port = getConfPort()
-    client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True)
+def runDocInfo():    
+    client = mongdbClient()
     db = client.admin
     serverStatus = db.command('serverStatus')
 
@@ -221,10 +366,7 @@ def runDocInfo():
     return mw.getJson(result)
 
 def runReplInfo():
-    import pymongo
-    
-    port = getConfPort()
-    client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True)
+    client = mongdbClient()
     db = client.admin
     serverStatus = db.command('serverStatus')
 
@@ -251,6 +393,7 @@ def runReplInfo():
     
     return mw.returnJson(True, 'OK', result)
 
+
 def testData():
     '''
     cd /www/server/mdserver-web && source bin/activate && python3 /www/server/mdserver-web/plugins/mongodb/index.py test_data
@@ -258,8 +401,8 @@ def testData():
     import pymongo
     from pymongo import ReadPreference
     
-    port = getConfPort()
-    client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True)
+    client = mongdbClient()
+
     db = client.test
     col = db["demo"]
 
@@ -269,29 +412,43 @@ def testData():
     print(x)
 
 
+
 def test():
+    initUserRoot()
     '''
+
     cd /www/server/mdserver-web && source bin/activate && python3 /www/server/mdserver-web/plugins/mongodb/index.py test
+    python3 plugins/mongodb/index.py test
     '''
     # https://pymongo.readthedocs.io/en/stable/examples/high_availability.html
     import pymongo
     from pymongo import ReadPreference
     
-    port = getConfPort()
-    client = pymongo.MongoClient(host='127.0.0.1', port=int(port), directConnection=True)
-    db = client.admin
+    # client = mongdbClient()
 
-    config = {
-        '_id': 'test',
-        'members': [
-            # 'priority': 10 
-            {'_id': 0, 'host': '154.21.203.138:27017'},
-            {'_id': 1, 'host': '154.12.53.216:27017'},
-        ]
-    }
+    # db = client.admin
 
-    rsStatus = client.admin.command('replSetInitiate',config)
-    print(rsStatus)
+    # print(db['users'])
+    # r = db.command("grantRolesToUser", "root",
+    #                  roles=["root"])
+    # print(r)
+    # users_collection = db['users']
+    # print(users_collection)
+
+    # mg_pass = mw.getRandomString(10)
+    # r = db.command("createUser", "root1", pwd=mg_pass, roles=["root"])
+    # print(r)
+    # config = {
+    #     '_id': 'test',
+    #     'members': [
+    #         # 'priority': 10 
+    #         {'_id': 0, 'host': '154.21.203.138:27017'},
+    #         {'_id': 1, 'host': '154.12.53.216:27017'},
+    #     ]
+    # }
+
+    # rsStatus = client.admin.command('replSetInitiate',config)
+    # print(rsStatus)
 
     # 需要通过命令行操作
     # rs.initiate({
@@ -321,7 +478,8 @@ def test():
     # serverStatus = db.command('serverStatus')
     # print(serverStatus)
     
-    # return mw.returnJson(True, 'OK', result)
+    return mw.returnJson(True, 'OK')
+
     
 
 def initdStatus():
@@ -412,6 +570,8 @@ if __name__ == "__main__":
         print(runReplInfo())
     elif func == 'conf':
         print(getConf())
+    elif func == 'get_config':
+        print(getConfig())
     elif func == 'run_log':
         print(runLog())
     elif func == 'test':
diff --git a/plugins/mongodb/js/mongodb.js b/plugins/mongodb/js/mongodb.js
index e3b5a783a..d1d5c174b 100644
--- a/plugins/mongodb/js/mongodb.js
+++ b/plugins/mongodb/js/mongodb.js
@@ -1,3 +1,32 @@
+function mgPost(method, version, args,callback){
+    var loadT = layer.msg('正在获取...', { icon: 16, time: 0, shade: 0.3 });
+
+    var req_data = {};
+    req_data['name'] = 'mongodb';
+    req_data['func'] = method;
+    req_data['version'] = version;
+ 
+    if (typeof(args) == 'string'){
+        req_data['args'] = JSON.stringify(toArrayObject(args));
+    } else {
+        req_data['args'] = JSON.stringify(args);
+    }
+
+    $.post('/plugins/run', req_data, function(data) {
+        layer.close(loadT);
+        if (!data.status){
+            //错误展示10S
+            layer.msg(data.msg,{icon:0,time:2000,shade: [10, '#000']});
+            return;
+        }
+
+        if(typeof(callback) == 'function'){
+            callback(data);
+        }
+    },'json'); 
+}
+
+
 function mongoStatus() {
     var loadT = layer.msg('正在获取...', { icon: 16, time: 0, shade: 0.3 });
     $.post('/plugins/run', {name:'mongodb', func:'run_info'}, function(data) {
@@ -104,6 +133,36 @@ function mongoReplStatus() {
     },'json');
 }
 
+//配置修改
+function mongoSetConfig() {
+    mgPost('get_config', '','',function(data){
+        var rdata = $.parseJSON(data.data);
+        rdata = rdata.data;
+  
+     	var body_auth = '<input class="btswitch btswitch-ios" id="auth" type="checkbox"><label  style="float: left;top: -3px;" class="btswitch-btn" for="auth" onclick=""></label>';
+       
+        var body = "<div class='bingfa'>" +
+            "<p class='line'><span class='span_tit'>IP：</span><input class='bt-input-text' type='text' name='bind_ip' value='" + rdata['net']['bindIp'] + "' />，<font>监听IP请勿随意修改</font></p>" +
+            "<p class='line'><span class='span_tit'>port： </span><input class='bt-input-text' type='text' name='port' value='" + rdata['net']['port'] + "' />，<font>监听端口,一般无需修改</font></p>" +
+            "<p class='line'><span class='span_tit'>dbPath：</span><input class='bt-input-text' type='text' name='data_path' value='" + rdata['storage']['dbPath'] + "' />，<font>数据存储位置</font></p>" +
+            "<p class='line'><span class='span_tit'>path：</span><input class='bt-input-text' type='text' name='log' value='" + rdata['systemLog']['path'] + "' />，<font>日志文件位置</font></p>" +
+            "<p class='line'><span class='span_tit'>pidFilePath：</span><input class='bt-input-text' type='text' name='pid_file_path' value='" + rdata['processManagement']['pidFilePath'] + "' />，<font>PID保存路径</font></p>" +
+            "<p class='line'><span class='span_tit' style='float:left;'>安全认证：</span>"+body_auth+"</p>" +
+            "<div class='mtb15' style='padding-top: 10px;text-align: center;'>\
+            	<button class='btn btn-success btn-sm mr5' onclick='mongoSetConfig();'>刷新</button>\
+            	<button class='btn btn-success btn-sm' onclick='mongoConfigSave();'>保存</button>" +
+            "</div></div>";
+
+        // console.log(body);
+        $(".soft-man-con").html(body);
+    });
+}
+
+function mongoConfigSave(){
+	mgPost('set_config', '','',function(data){
+    });
+}
+