mirror of https://github.com/conqp/rcon
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.6 KiB
86 lines
2.6 KiB
"""RCON client CLI."""
|
|
|
|
from argparse import ArgumentParser, Namespace
|
|
from getpass import getpass
|
|
from logging import DEBUG, INFO, basicConfig, getLogger
|
|
from socket import timeout
|
|
from sys import exit # pylint: disable=W0622
|
|
from typing import Tuple
|
|
|
|
from rcon.exceptions import InvalidConfig
|
|
from rcon.config import Config, servers
|
|
from rcon.exceptions import RequestIdMismatch, WrongPassword
|
|
from rcon.proto import Client
|
|
|
|
|
|
__all__ = ['get_credentials', 'main']
|
|
|
|
|
|
LOGGER = getLogger('rconclt')
|
|
LOG_FORMAT = '[%(levelname)s] %(name)s: %(message)s'
|
|
|
|
|
|
def get_args() -> Namespace:
|
|
"""Parses and returns the CLI arguments."""
|
|
|
|
parser = ArgumentParser(description='A Minecraft RCON client.')
|
|
parser.add_argument('server', help='the server to connect to')
|
|
parser.add_argument('-t', '--timeout', type=float, metavar='seconds',
|
|
help='connection timeout in seconds')
|
|
parser.add_argument('-d', '--debug', action='store_true',
|
|
help='print additional debug information')
|
|
parser.add_argument('command', help='command to execute on the server')
|
|
parser.add_argument('argument', nargs='*', default=(),
|
|
help='arguments for the command')
|
|
return parser.parse_args()
|
|
|
|
|
|
def get_credentials(server: str) -> Tuple[str, int, str]:
|
|
"""Get the credentials for a server from the respective server name."""
|
|
|
|
try:
|
|
host, port, passwd = Config.from_string(server)
|
|
except InvalidConfig:
|
|
try:
|
|
host, port, passwd = servers()[server]
|
|
except KeyError:
|
|
LOGGER.error('No such server: %s.', server)
|
|
exit(2)
|
|
|
|
if passwd is None:
|
|
try:
|
|
passwd = getpass('Password: ')
|
|
except (KeyboardInterrupt, EOFError):
|
|
print()
|
|
LOGGER.error('Aborted by user.')
|
|
exit(1)
|
|
|
|
return (host, port, passwd)
|
|
|
|
|
|
def main():
|
|
"""Runs the RCON client."""
|
|
|
|
args = get_args()
|
|
log_level = DEBUG if args.debug else INFO
|
|
basicConfig(level=log_level, format=LOG_FORMAT)
|
|
host, port, passwd = get_credentials(args.server)
|
|
|
|
try:
|
|
with Client(host, port, timeout=args.timeout) as client:
|
|
client.login(passwd)
|
|
text = client.run(args.command, *args.argument)
|
|
except ConnectionRefusedError:
|
|
LOGGER.error('Connection refused.')
|
|
exit(3)
|
|
except timeout:
|
|
LOGGER.error('Connection timeout.')
|
|
exit(4)
|
|
except RequestIdMismatch:
|
|
LOGGER.error('Unexpected request ID mismatch.')
|
|
exit(5)
|
|
except WrongPassword:
|
|
LOGGER.error('Wrong password.')
|
|
exit(6)
|
|
else:
|
|
print(text, flush=True)
|
|
|