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

134 lines
4.4 KiB
Python
Raw Normal View History

2018-08-28 04:45:31 +00:00
'''
Onionr - P2P Anonymous Storage Network
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
import logger, config, threading, time, uuid, subprocess
2018-08-28 04:45:31 +00:00
from onionrblockapi import Block
plugin_name = 'cliui'
PLUGIN_VERSION = '0.0.1'
class OnionrCLIUI:
def __init__(self, apiInst):
self.api = apiInst
self.myCore = apiInst.get_core()
return
def subCommand(self, command):
try:
subprocess.run(["./onionr.py", command])
except KeyboardInterrupt:
pass
2018-09-01 20:38:44 +00:00
def refresh(self):
for i in range(100):
print('')
2018-08-28 04:45:31 +00:00
def start(self):
'''Main CLI UI interface menu'''
showMenu = True
2018-09-01 20:38:44 +00:00
isOnline = "No"
firstRun = True
2018-09-05 04:06:17 +00:00
choice = ''
2018-09-01 20:38:44 +00:00
2018-09-02 02:59:03 +00:00
if self.myCore._utils.localCommand('ping') == 'pong':
firstRun = False
while showMenu:
2018-09-01 20:38:44 +00:00
if firstRun:
print("please wait while Onionr starts...")
2018-09-09 05:12:41 +00:00
daemon = subprocess.Popen(["./onionr.py", "start"], stdin=subprocess.PIPE, stdout=subprocess.DEVNULL)
2018-09-01 20:38:44 +00:00
time.sleep(30)
firstRun = False
if self.myCore._utils.localCommand('ping') == 'pong':
isOnline = "Yes"
2018-09-02 02:59:03 +00:00
else:
isOnline = "No"
print('''
Daemon Running: ''' + isOnline + '''
2018-09-01 20:38:44 +00:00
1. Flow (Anonymous public chat, use at your own risk)
2. Mail (Secure email-like service)
3. File Sharing
4. User Settings
2018-09-01 20:38:44 +00:00
5. Start/Stop Daemon
2018-09-02 02:59:03 +00:00
6. 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"):
self.subCommand("flow")
elif choice in ("2", "mail"):
self.subCommand("mail")
2018-09-02 02:59:03 +00:00
elif choice in ("3", "file sharing", "file"):
print("Not supported yet")
elif choice in ("4", "user settings", "settings"):
try:
self.setName()
except (KeyboardInterrupt, EOFError) as e:
pass
2018-09-01 20:38:44 +00:00
elif choice in ("5", "daemon"):
if isOnline == "Yes":
print("Onionr daemon will shutdown...")
self.myCore.daemonQueueAdd('shutdown')
2018-09-02 04:55:24 +00:00
try:
daemon.kill()
except UnboundLocalError:
pass
2018-09-01 20:38:44 +00:00
else:
print("Starting Daemon...")
2018-09-02 04:55:24 +00:00
daemon = subprocess.Popen(["./onionr.py", "start"], stdin=subprocess.PIPE, stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)
2018-09-01 20:38:44 +00:00
elif choice in ("6", "quit"):
showMenu = False
2018-09-01 20:38:44 +00:00
elif choice == "":
pass
else:
print("Invalid choice")
return
def setName(self):
2018-09-02 02:59:03 +00:00
try:
name = input("Enter your name: ")
if name != "":
self.myCore.insertBlock("userInfo-" + str(uuid.uuid1()), sign=True, header='userInfo', meta={'name': name})
except KeyboardInterrupt:
pass
2018-08-28 04:45:31 +00:00
return
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