Onionr/onionr/static-data/default-plugins/cliui/main.py

120 lines
4.0 KiB
Python
Raw Normal View History

2018-08-28 04:45:31 +00:00
'''
2019-06-20 07:59:32 +00:00
Onionr - Private P2P Communication
2018-08-28 04:45:31 +00:00
This is an interactive menu-driven CLI interface for Onionr
'''
'''
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/>.
'''
# Imports some useful libraries
2019-03-04 22:29:44 +00:00
import threading, time, uuid, subprocess, sys
import config, logger
2018-08-28 04:45:31 +00:00
from onionrblockapi import Block
2019-03-04 22:29:44 +00:00
import onionrplugins
from onionrutils import localcommand
2018-08-28 04:45:31 +00:00
plugin_name = 'cliui'
PLUGIN_VERSION = '0.0.1'
class OnionrCLIUI:
def __init__(self, apiInst):
self.api = apiInst
2019-03-04 22:29:44 +00:00
self.shutdown = False
self.running = 'undetermined'
enabled = onionrplugins.get_enabled_plugins()
self.mail_enabled = 'pms' in enabled
self.flow_enabled = 'flow' in enabled
def subCommand(self, command, args=None):
try:
if args != None:
subprocess.call(['./onionr.py', command, args])
else:
subprocess.call(['./onionr.py', command])
except KeyboardInterrupt:
pass
2019-03-04 22:29:44 +00:00
def isRunning(self):
while not self.shutdown:
2019-07-20 00:41:15 +00:00
if localcommand.local_command('ping', maxWait=5) == 'pong!':
2019-03-04 22:29:44 +00:00
self.running = 'Yes'
else:
self.running = 'No'
time.sleep(5)
2018-09-01 20:38:44 +00:00
def refresh(self):
print('\n' * 80 + logger.colors.reset)
2018-09-01 20:38:44 +00:00
2018-08-28 04:45:31 +00:00
def start(self):
'''Main CLI UI interface menu'''
showMenu = True
2018-09-05 04:06:17 +00:00
choice = ''
2019-03-04 22:29:44 +00:00
threading.Thread(target=self.isRunning).start()
2018-09-02 02:59:03 +00:00
while showMenu:
2019-03-04 22:29:44 +00:00
print('Onionr\n------')
print('''Daemon Running: ''' + self.running + '''
1. Flow (Anonymous public shout box, use at your own risk)
2. Mail (Secure email-like service)
3. File Sharing
4. Quit (Does not shutdown daemon)
''')
try:
choice = input(">").strip().lower()
2018-09-01 20:38:44 +00:00
except (KeyboardInterrupt, EOFError):
choice = "quit"
if choice in ("flow", "1"):
2019-03-04 22:29:44 +00:00
if self.flow_enabled:
self.subCommand("flow")
else:
2019-07-22 23:59:48 +00:00
logger.warn('flow plugin is not enabled')
elif choice in ("2", "mail"):
2019-03-04 22:29:44 +00:00
if self.mail_enabled:
self.subCommand("mail")
else:
2019-07-22 23:59:48 +00:00
logger.warn('mail plugin not enabled')
2018-09-02 02:59:03 +00:00
elif choice in ("3", "file sharing", "file"):
try:
filename = input("Enter full path to file: ").strip()
except (EOFError, KeyboardInterrupt) as e:
pass
else:
if len(filename.strip()) > 0:
self.subCommand("addfile", filename)
elif choice in ("4", "quit"):
showMenu = False
2019-03-04 22:29:44 +00:00
self.shutdown = True
2018-09-01 20:38:44 +00:00
elif choice == "":
pass
else:
logger.error("Invalid choice", terminal=True)
return
2018-08-28 04:45:31 +00:00
def on_init(api, data = None):
'''
This event is called after Onionr is initialized, but before the command
inputted is executed. Could be called when daemon is starting or when
just the client is running.
'''
# Doing this makes it so that the other functions can access the api object
# by simply referencing the variable `pluginapi`.
pluginapi = api
ui = OnionrCLIUI(api)
api.commands.register('interactive', ui.start)
api.commands.register_help('interactive', 'Open the CLI interface')
return