Update index.py

plugins/ollama/index.py

@@ -29,8 +29,6 @@ class App:
 
     def __init__(self):
         self.__setupPath = self.getServerDir()
-        self.__cfg = self.__setupPath + '/nezha.cfg'
-        self.__agent_cfg = self.__setupPath + '/agent.cfg'
 
     def getArgs(self):
         args = sys.argv[3:]
@@ -54,22 +52,9 @@ class App:
                 return (False, mw.returnJson(False, '参数:(' + ck[i] + ')没有!'))
         return (True, mw.returnJson(True, 'ok'))
 
-    def __release_port(self, port):
-        from collections import namedtuple
-        try:
-            import firewall_api
-            firewall_api.firewall_api().addAcceptPortArgs(port, 'nezha', 'port')
-            return port
-        except Exception as e:
-            return "Release failed {}".format(e)
-
-    def openPort(self):
-        for i in ["9527", "5555"]:
-            self.__release_port(i)
-        return True
 
     def getPluginName(self):
-        return 'nezha'
+        return 'ollama'
 
     def getPluginDir(self):
         return mw.getPluginDir() + '/' + self.getPluginName()
@@ -77,14 +62,6 @@ class App:
     def getServerDir(self):
         return mw.getServerDir() + '/' + self.getPluginName()
 
-    def getInitdConfTpl(self):
-        path = self.getPluginDir() + "/init.d/nezha.tpl"
-        return path
-
-    def getInitdAgentConfTpl(self):
-        path = self.getPluginDir() + "/init.d/nezha-agent.tpl"
-        return path
-
     def getHomeDir(self):
         if mw.isAppleSystem():
             user = mw.execShell(
@@ -102,20 +79,12 @@ class App:
             return 'root'
 
     def status(self):
-        cmd = "ps -ef|grep " + self.getPluginName() + \
+        cmd = "ps -ef|grep " + self.getPluginName() + " |grep -v grep | grep -v python | awk '{print $2}'"
-            " |grep -v grep | grep -v nezha-agent | grep -v python | awk '{print $2}'"
         data = mw.execShell(cmd)
         if data[0] == '':
             return "stop"
         return 'start'
 
-    def status_agent(self):
-        cmd = "ps -ef | grep nezha-agent | grep -v grep | grep -v python | awk '{print $2}'"
-        data = mw.execShell(cmd)
-        if data[0] == '':
-            return 'stop'
-        return 'start'
-
     def contentReplace(self, content):
 
         service_path = mw.getServerDir()
@@ -145,15 +114,15 @@ class App:
             mw.execShell('chmod +x ' + file_bin)
 
         # systemd
-        systemDir = mw.systemdCfgDir()
+        # systemDir = mw.systemdCfgDir()
-        systemService = systemDir + '/nezha.service'
+        # systemService = systemDir + '/nezha.service'
-        systemServiceTpl = self.getPluginDir() + '/init.d/nezha.service.tpl'
+        # systemServiceTpl = self.getPluginDir() + '/init.d/nezha.service.tpl'
-        if os.path.exists(systemDir) and not os.path.exists(systemService):
+        # if os.path.exists(systemDir) and not os.path.exists(systemService):
-            service_path = mw.getServerDir()
+        #     service_path = mw.getServerDir()
-            se_content = mw.readFile(systemServiceTpl)
+        #     se_content = mw.readFile(systemServiceTpl)
-            se_content = self.contentReplace(se_content)
+        #     se_content = self.contentReplace(se_content)
-            mw.writeFile(systemService, se_content)
+        #     mw.writeFile(systemService, se_content)
-            mw.execShell('systemctl daemon-reload')
+        #     mw.execShell('systemctl daemon-reload')
 
         return file_bin
 
@@ -166,39 +135,11 @@ class App:
 
         return content
 
-    def initDAgent(self):
-        file_tpl = self.getInitdAgentConfTpl()
-
-        initD_path = self.getServerDir() + '/init.d'
-        if not os.path.exists(initD_path):
-            os.mkdir(initD_path)
-
-        file_agent_bin = initD_path + '/nezha-agent'
-
-        content = mw.readFile(file_tpl)
-        content = self.contentReplace(content)
-        content = self.contentAgentReplace(content)
-        mw.writeFile(file_agent_bin, content)
-        mw.execShell('chmod +x ' + file_agent_bin)
-
-        # systemd
-        sysDir = mw.systemdCfgDir()
-        sysService = sysDir + '/nezha-agent.service'
-        sysServiceTpl = self.getPluginDir() + '/init.d/nezha-agent.service.tpl'
-        service_path = mw.getServerDir()
-        content = mw.readFile(sysServiceTpl)
-        content = self.contentReplace(content)
-        content = self.contentAgentReplace(content)
-        mw.writeFile(sysService, content)
-        mw.execShell('systemctl daemon-reload')
-
-        return file_agent_bin
 
     def init_cfg(self):
         self.initDreplace()
-        self.initDAgent()
 
-    def imOp(self, method):
+    def oaOp(self, method):
 
         file = self.initDreplace()
 
@@ -215,26 +156,21 @@ class App:
         return data[0]
 
     def start(self):
-        return self.imOp('start')
+        return self.oaOp('start')
 
     def stop(self):
-        return self.imOp('stop')
+        return self.oaOp('stop')
 
     def restart(self):
-        return self.imOp('restart')
+        return self.oaOp('restart')
 
     def reload(self):
-        return self.imOp('reload')
+        return self.oaOp('reload')
 
     def agOp(self, method):
-        file = self.initDAgent()
-
-        path = self.__agent_cfg
-        if not os.path.exists(path):
-            return '请先设置Agent配置!'
 
         if not mw.isAppleSystem():
-            cmd = 'systemctl {} {}'.format(method, 'nezha-agent')
+            cmd = 'systemctl {} {}'.format(method, self.getPluginName())
             data = mw.execShell(cmd)
             if data[1] == '':
                 return 'ok'
@@ -258,101 +194,23 @@ class App:
         return self.agentOp('reload')
 
     def initd_status(self):
-        cmd = 'systemctl status nezha | grep loaded | grep "enabled;"'
+        cmd = 'systemctl status '+self.getPluginName()+' | grep loaded | grep "enabled;"'
         data = mw.execShell(cmd)
         if data[0] == '':
             return 'fail'
         return 'ok'
 
     def initd_install(self):
-        mw.execShell('systemctl enable nezha')
+        mw.execShell('systemctl enable '+self.getPluginName())
         return 'ok'
 
     def initd_uninstall(self):
-        mw.execShell('systemctl disable nezha')
+        mw.execShell('systemctl disable '+self.getPluginName())
-        return 'ok'
-
-    def initd_status_agent(self):
-        cmd = 'systemctl status nezha-agent | grep loaded | grep "enabled;"'
-        data = mw.execShell(cmd)
-        if data[0] == '':
-            return 'fail'
-        return 'ok'
-
-    def initd_install_agent(self):
-        mw.execShell('systemctl enable nezha-agent')
-        return 'ok'
-
-    def initd_uninstall_agent(self):
-        mw.execShell('systemctl disable nezha-agent')
         return 'ok'
 
     def conf(self):
         return self.getServerDir() + '/dashboard/data/config.yaml'
 
-    def nezha_cfg(self):
-        path = self.__cfg
-        if not os.path.exists(path):
-            d = {}
-            cmd_un = 'cd ' + self.getServerDir() + '/dashboard && ./nezha conf -u ""'
-            mw.execShell(cmd_un)
-            td = mw.execShell(cmd_un)
-            d['username'] = td[0].strip()
-
-            pwd = mw.getRandomString(16)
-            cmd_pwd = 'cd ' + self.getServerDir() + '/dashboard && ./nezha conf -u "" -p ' + pwd
-            td = mw.execShell(cmd_pwd)
-            d['password'] = pwd
-
-            mw.writeFile(path, mw.enDoubleCrypt('nezha', mw.getJson(d)))
-
-        info = mw.readFile(path)
-        info = mw.deDoubleCrypt('nezha', info)
-
-        info = json.loads(info)
-        return mw.returnJson(True, 'ok', info)
-
-    def nezha_save_cfg(self):
-        args = self.getArgs()
-        data = self.checkArgs(args, ['username', 'password'])
-        if not data[0]:
-            return data[1]
-        path = self.__cfg
-
-        cmd = 'cd ' + self.getServerDir() + '/dashboard && ./nezha conf -su "' + \
-            args['username'] + '" -p ' + args['password']
-        # print(cmd)
-        t = mw.execShell(cmd)
-        # print(t)
-        mw.writeFile(path, mw.enDoubleCrypt('nezha', mw.getJson(args)))
-        return mw.returnJson(True, '修改成功!')
-
-    def get_agent_cfg(self):
-        path = self.__agent_cfg
-        info = mw.readFile(path)
-        info = mw.deDoubleCrypt('agent', info)
-        info = json.loads(info)
-        return info
-
-    def agent_cfg(self):
-        path = self.__agent_cfg
-        if not os.path.exists(path):
-            d = {}
-            d['host'] = '127.0.0.1:5555'
-            d['secret'] = 'secret'
-            mw.writeFile(path, mw.enDoubleCrypt('agent', mw.getJson(d)))
-
-        info = self.get_agent_cfg()
-        return mw.returnJson(True, 'ok', info)
-
-    def agent_save_cfg(self):
-        args = self.getArgs()
-        data = self.checkArgs(args, ['host', 'secret'])
-        if not data[0]:
-            return data[1]
-        path = self.__agent_cfg
-        mw.writeFile(path, mw.enDoubleCrypt('agent', mw.getJson(args)))
-        return mw.returnJson(True, '修改成功!')
 
     def run_log(self):
         return self.getServerDir() + '/logs/nezha.log'