From 084f67b10f83a38318effbfa4823c34f2c1b2fe2 Mon Sep 17 00:00:00 2001
From: merlink <merlink@pc1.fritz.box>
Date: Tue, 6 Aug 2013 13:23:56 +0200
Subject: [PATCH] Created Object for controlling bitmessage deamon

---
 src/bitmessagemain.py | 139 ++++++++++++++++++++++++------------------
 1 file changed, 80 insertions(+), 59 deletions(-)

diff --git a/src/bitmessagemain.py b/src/bitmessagemain.py
index 04ff3071..b39bdc96 100644
--- a/src/bitmessagemain.py
+++ b/src/bitmessagemain.py
@@ -770,79 +770,100 @@ if shared.useVeryEasyProofOfWorkForTesting:
     shared.networkDefaultPayloadLengthExtraBytes = int(
         shared.networkDefaultPayloadLengthExtraBytes / 7000)
 
-def main(deamon=False):
-    # is the application already running?  If yes then exit.
-    thisapp = singleton.singleinstance()
+class Main:
+    def start(self, deamon=False):
+        # is the application already running?  If yes then exit.
+        thisapp = singleton.singleinstance()
 
-    signal.signal(signal.SIGINT, helper_generic.signal_handler)
-    # signal.signal(signal.SIGINT, signal.SIG_DFL)
+        signal.signal(signal.SIGINT, helper_generic.signal_handler)
+        # signal.signal(signal.SIGINT, signal.SIG_DFL)
 
-    helper_bootstrap.knownNodes()
-    # Start the address generation thread
-    addressGeneratorThread = addressGenerator()
-    addressGeneratorThread.daemon = True  # close the main program even if there are threads left
-    addressGeneratorThread.start()
+        helper_bootstrap.knownNodes()
+        # Start the address generation thread
+        addressGeneratorThread = addressGenerator()
+        addressGeneratorThread.daemon = True  # close the main program even if there are threads left
+        addressGeneratorThread.start()
 
-    # Start the thread that calculates POWs
-    singleWorkerThread = singleWorker()
-    singleWorkerThread.daemon = True  # close the main program even if there are threads left
-    singleWorkerThread.start()
+        # Start the thread that calculates POWs
+        singleWorkerThread = singleWorker()
+        singleWorkerThread.daemon = True  # close the main program even if there are threads left
+        singleWorkerThread.start()
 
-    # Start the SQL thread
-    sqlLookup = sqlThread()
-    sqlLookup.daemon = False  # DON'T close the main program even if there are threads left. The closeEvent should command this thread to exit gracefully.
-    sqlLookup.start()
+        # Start the SQL thread
+        sqlLookup = sqlThread()
+        sqlLookup.daemon = False  # DON'T close the main program even if there are threads left. The closeEvent should command this thread to exit gracefully.
+        sqlLookup.start()
 
-    # Start the cleanerThread
-    singleCleanerThread = singleCleaner()
-    singleCleanerThread.daemon = True  # close the main program even if there are threads left
-    singleCleanerThread.start()
+        # Start the cleanerThread
+        singleCleanerThread = singleCleaner()
+        singleCleanerThread.daemon = True  # close the main program even if there are threads left
+        singleCleanerThread.start()
 
-    shared.reloadMyAddressHashes()
-    shared.reloadBroadcastSendersForWhichImWatching()
+        shared.reloadMyAddressHashes()
+        shared.reloadBroadcastSendersForWhichImWatching()
 
-    if shared.safeConfigGetBoolean('bitmessagesettings', 'apienabled'):
-        try:
-            apiNotifyPath = shared.config.get(
-                'bitmessagesettings', 'apinotifypath')
-        except:
-            apiNotifyPath = ''
-        if apiNotifyPath != '':
-            with shared.printLock:
-                print 'Trying to call', apiNotifyPath
+        if shared.safeConfigGetBoolean('bitmessagesettings', 'apienabled'):
+            try:
+                apiNotifyPath = shared.config.get(
+                    'bitmessagesettings', 'apinotifypath')
+            except:
+                apiNotifyPath = ''
+            if apiNotifyPath != '':
+                with shared.printLock:
+                    print 'Trying to call', apiNotifyPath
 
-            call([apiNotifyPath, "startingUp"])
-        singleAPIThread = singleAPI()
-        singleAPIThread.daemon = True  # close the main program even if there are threads left
-        singleAPIThread.start()
+                call([apiNotifyPath, "startingUp"])
+            singleAPIThread = singleAPI()
+            singleAPIThread.daemon = True  # close the main program even if there are threads left
+            singleAPIThread.start()
 
-    connectToStream(1)
+        connectToStream(1)
 
-    singleListenerThread = singleListener()
-    singleListenerThread.setup(selfInitiatedConnections)
-    singleListenerThread.daemon = True  # close the main program even if there are threads left
-    singleListenerThread.start()
+        singleListenerThread = singleListener()
+        singleListenerThread.setup(selfInitiatedConnections)
+        singleListenerThread.daemon = True  # close the main program even if there are threads left
+        singleListenerThread.start()
 
-    if deamon == False and shared.safeConfigGetBoolean('bitmessagesettings', 'daemon') == False:
-        try:
-            from PyQt4 import QtCore, QtGui
-        except Exception as err:
-            print 'PyBitmessage requires PyQt unless you want to run it as a daemon and interact with it using the API. You can download PyQt from http://www.riverbankcomputing.com/software/pyqt/download   or by searching Google for \'PyQt Download\'. If you want to run in daemon mode, see https://bitmessage.org/wiki/Daemon'
-            print 'Error message:', err
-            os._exit(0)
+        if deamon == False and shared.safeConfigGetBoolean('bitmessagesettings', 'daemon') == False:
+            try:
+                from PyQt4 import QtCore, QtGui
+            except Exception as err:
+                print 'PyBitmessage requires PyQt unless you want to run it as a daemon and interact with it using the API. You can download PyQt from http://www.riverbankcomputing.com/software/pyqt/download   or by searching Google for \'PyQt Download\'. If you want to run in daemon mode, see https://bitmessage.org/wiki/Daemon'
+                print 'Error message:', err
+                os._exit(0)
 
-        import bitmessageqt
-        bitmessageqt.run()
-    else:
-        shared.config.remove_option('bitmessagesettings', 'dontconnect')
+            import bitmessageqt
+            bitmessageqt.run()
+        else:
+            shared.config.remove_option('bitmessagesettings', 'dontconnect')
+
+            if deamon:
+                with shared.printLock:
+                    print 'Running as a daemon. The main program should exit this thread.'
+            else:
+                with shared.printLock:
+                    print 'Running as a daemon. You can use Ctrl+C to exit.'
+                while True:
+                    time.sleep(20)
+            
+    def stop(self):
         with shared.printLock:
-            print 'Running as a daemon. You can use Ctrl+C to exit.'
-
-        while True:
-            time.sleep(20)
-
+            print 'Stopping Bitmessage Deamon.'
+        shared.doCleanShutdown()
+        
+        
+    def getApiAddress(self):
+        if not shared.safeConfigGetBoolean('bitmessagesettings', 'apienabled'):
+            return None
+            
+        address = shared.config.get('bitmessagesettings', 'apiinterface')
+        port = shared.config.getint('bitmessagesettings', 'apiport')
+        return {'address':address,'port':port}
+            
 if __name__ == "__main__":
-    main()
+    mainprogram = Main()
+    mainprogram.start()
+
     
 # So far, the creation of and management of the Bitmessage protocol and this
 # client is a one-man operation. Bitcoin tips are quite appreciated.