Onionr/onionr/onionrcommands/parser/__init__.py

102 lines
3.8 KiB
Python
Raw Normal View History

2019-07-30 18:13:10 +00:00
'''
Onionr - Private P2P Communication
This module loads in the Onionr arguments and their help messages
'''
'''
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/>.
'''
2019-07-31 05:10:28 +00:00
import sys
2019-07-30 18:13:10 +00:00
from etc import onionrvalues
2019-07-31 05:10:28 +00:00
import logger, onionrexceptions
2019-08-05 00:34:57 +00:00
import onionrplugins
from onionrplugins import onionrpluginapi
2019-07-31 05:10:28 +00:00
from . import arguments, recommend
2019-08-05 00:34:57 +00:00
plugin_command = lambda cmd: 'on_%s_cmd' % (cmd,)
def register_plugin_commands(cmd)->bool:
plugin_cmd = plugin_command(cmd)
2019-08-05 00:34:57 +00:00
for pl in onionrplugins.get_enabled_plugins():
pl = onionrplugins.get_plugin(pl)
if hasattr(pl, plugin_cmd):
getattr(pl, plugin_cmd)(onionrpluginapi.PluginAPI)
2019-08-05 00:34:57 +00:00
return True
return False
2019-08-05 00:34:57 +00:00
2019-07-30 18:13:10 +00:00
def register():
2019-08-30 00:37:39 +00:00
"""Registers commands and handles help command processing"""
2019-08-29 22:17:47 +00:00
def get_help_message(cmd: str, default: str = 'No help available for this command'):
2019-08-30 00:37:39 +00:00
"""Return help message for a given command, supports plugin commands"""
pl_cmd = plugin_command(cmd)
for pl in onionrplugins.get_enabled_plugins():
pl = onionrplugins.get_plugin(pl)
if hasattr(pl, pl_cmd):
try:
return getattr(pl, pl_cmd).onionr_help
except AttributeError:
pass
2019-08-29 22:17:47 +00:00
for i in arguments.get_arguments():
for alias in i:
try:
return arguments.get_help(cmd)
except AttributeError:
pass
2019-08-30 00:37:39 +00:00
return default # Return the help string
2019-08-29 22:17:47 +00:00
PROGRAM_NAME = "onionr"
2019-08-30 00:37:39 +00:00
# Get the command
2019-08-02 23:00:04 +00:00
try:
cmd = sys.argv[1]
except IndexError:
2019-08-30 00:37:39 +00:00
logger.debug("Detected Onionr run with no commands specified")
return
is_help_cmd = False
if cmd.replace('--', '').lower() == 'help': is_help_cmd = True
try:
arguments.get_func(cmd)()
except onionrexceptions.NotFound:
if not register_plugin_commands(cmd) and not is_help_cmd:
recommend.recommend()
sys.exit(3)
2019-08-29 22:17:47 +00:00
if is_help_cmd:
2019-08-29 22:17:47 +00:00
try:
sys.argv[2]
except IndexError:
for i in arguments.get_arguments():
logger.info('%s <%s>: %s' % (PROGRAM_NAME, '/'.join(i), get_help_message(i[0])), terminal=True)
for pl in onionrplugins.get_enabled_plugins():
pl = onionrplugins.get_plugin(pl)
if hasattr(pl, 'ONIONR_COMMANDS'):
print('')
try:
logger.info('%s commands:' % (pl.plugin_name,), terminal=True)
except AttributeError:
logger.info('%s commands:' % (pl.__name__,), terminal=True)
for plugin_cmd in pl.ONIONR_COMMANDS:
logger.info('%s %s: %s' % (PROGRAM_NAME, plugin_cmd, get_help_message(plugin_cmd)), terminal=True)
print('')
2019-08-29 22:17:47 +00:00
else:
try:
logger.info('%s %s: %s' % (PROGRAM_NAME, sys.argv[2], get_help_message(sys.argv[2])), terminal=True)
except KeyError:
logger.error('%s: command does not exist.' % [sys.argv[2]], terminal=True)
sys.exit(3)
return