improved project structuring

onionr/onionr.py

@@ -0,0 +1,142 @@
+#!/usr/bin/env python3
+'''
+    Onionr - P2P Microblogging Platform & Social network. 
+
+    Onionr is the name for both the protocol and the original/reference software.
+
+    Run with 'help' for usage.
+'''
+'''
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+'''
+import sys, os, configparser, base64, random, getpass, shutil, subprocess, requests
+import gui, api, colors, core
+from onionrutils import OnionrUtils
+from colors import Colors
+
+class Onionr:
+    def __init__(self):
+        '''Main Onionr class. This is for the CLI program, and does not handle much of the logic.
+        In general, external programs and plugins should not use this class.
+
+        '''
+        if os.path.exists('dev-enabled'):
+            print('DEVELOPMENT MODE ENABLED (THIS IS LESS SECURE!)')
+            self._developmentMode = True
+        else:
+            self._developmentMode = False
+
+        colors = Colors()
+
+        self.onionrCore = core.Core()
+        self.onionrUtils = OnionrUtils()
+
+        # Get configuration and Handle commands
+        
+        self.debug = False # Whole application debugging
+        try:
+            os.chdir(sys.path[0])
+        except FileNotFoundError:
+            pass
+
+        if os.path.exists('data-encrypted.dat'):
+            while True:
+                print('Enter password to decrypt:')
+                password = getpass.getpass()
+                result = self.onionrCore.dataDirDecrypt(password)
+                if os.path.exists('data/'):
+                    break
+                else:
+                    print('Failed to decrypt: ' + result[1])
+        else:
+            if not os.path.exists('data/'):
+                os.mkdir('data/')
+        
+        if not os.path.exists('data/peers.db'):
+            self.onionrCore.createPeerDB()
+            pass
+
+        # Get configuration
+        self.config = configparser.ConfigParser()
+        if os.path.exists('data/config.ini'):
+            self.config.read('data/config.ini')
+        else:
+            # Generate default config
+            # Hostname should only be set if different from 127.x.x.x. Important for DNS rebinding attack prevention.
+            if self.debug:
+                randomPort = 8080
+            else:
+                randomPort = random.randint(1024, 65535)
+            self.config['CLIENT'] = {'CLIENT HMAC': base64.b64encode(os.urandom(32)).decode('utf-8'), 'PORT': randomPort, 'API VERSION': '0.0.0'}
+            with open('data/config.ini', 'w') as configfile:
+                self.config.write(configfile)
+        command = ''
+        try:
+            command = sys.argv[1].lower()
+        except IndexError:
+            command = ''
+        finally:
+            if command == 'start':
+                if os.path.exists('.onionr-lock'):
+                    self.onionrUtils.printErr('Cannot start. Daemon is already running, or it did not exit cleanly.\n(if you are sure that there is not a daemon running, delete .onionr-lock & try again).')
+                else:
+                    if not self.debug and not self._developmentMode:
+                        lockFile = open('.onionr-lock', 'w')
+                        lockFile.write('')
+                        lockFile.close()
+                    self.daemon()
+                    if not self.debug and not self._developmentMode:
+                        os.remove('.onionr-lock')
+            elif command == 'stop':
+                self.killDaemon()
+            elif command == 'stats':
+                self.showStats()
+            elif command == 'help' or command == '--help':
+                self.showHelp()
+            elif command == '':
+                print('Do', sys.argv[0], ' --help for Onionr help.')
+            else:
+                print(colors.RED, 'Invalid Command', colors.RESET)
+                
+        if not self._developmentMode:
+            encryptionPassword = self.onionrUtils.getPassword('Enter password to encrypt directory.')
+            self.onionrCore.dataDirEncrypt(encryptionPassword)
+            shutil.rmtree('data/')
+        return
+    def daemon(self):
+        ''' Start the Onionr communication daemon
+        '''
+        if not os.environ.get("WERKZEUG_RUN_MAIN") == "true":
+            subprocess.Popen(["./communicator.py", "run"])
+            print('Started communicator')
+        api.API(self.config, self.debug)
+        return
+    def killDaemon(self):
+        '''Shutdown the Onionr Daemon'''
+        print('Killing the running daemon')
+        try:
+            self.onionrUtils.localCommand('shutdown')
+        except requests.exceptions.ConnectionError:
+            pass
+        else:
+            self.onionrCore.daemonQueueAdd('shutdown')
+        return
+    def showStats(self):
+        '''Display statistics and exit'''
+        return
+    def showHelp(self):
+        '''Show help for Onionr'''
+        return
+
+Onionr()