From 0b83b3b8ebcca7675b7160c7d4e449dcdfa5af3e Mon Sep 17 00:00:00 2001 From: Pavel Kirilin <win10@list.ru> Date: Mon, 20 Jan 2020 20:23:30 +0400 Subject: [PATCH] Docker support: Description: - Added docker useful commands. Docker commands: * ls * clean * stop * rm * up All documentation available with command: 'meme admin docker'. Signed-off-by: Pavel Kirilin <win10@list.ru> --- src/actions/__init__.py | 1 + src/actions/basic_actions.py | 8 +-- src/actions/docker.py | 133 +++++++++++++++++++++++++++++++++++ src/utils/server_utils.py | 13 +++- 4 files changed, 147 insertions(+), 8 deletions(-) create mode 100644 src/actions/docker.py diff --git a/src/actions/__init__.py b/src/actions/__init__.py index 3827707..f689a29 100644 --- a/src/actions/__init__.py +++ b/src/actions/__init__.py @@ -1,4 +1,5 @@ from .basic_actions import * +from .docker import * from .interactive_session import * from .permissions import * from .server_management import * diff --git a/src/actions/basic_actions.py b/src/actions/basic_actions.py index fa6be86..5c35909 100644 --- a/src/actions/basic_actions.py +++ b/src/actions/basic_actions.py @@ -20,7 +20,7 @@ async def overall_help(message: Message, state=None): '* meme admin permissions - Manage servers permissions\n\n' # '* meme admin commands - Manage stored commands\n\n' '* meme admin exec {alias} {command} - Execute single command\n\n' - # '* meme admin docker - Manage docker on remote server\n\n' + '* meme admin docker - Manage docker on remote server\n\n' '* meme admin stats - See server statistics\n\n' '```', parse_mode=ParseMode.MARKDOWN @@ -33,11 +33,7 @@ async def overall_help(message: Message, state=None): ('command', r'(.*)') ]) async def bot_run_code(message: Message, alias: str, command: str, state=None): - server = await get_server_by_alias(message, alias) - if server.server_permissions.value < ServerPermissions.RCE.value \ - and str(message.from_user.id) != server.server_admin: - raise Exception('You do not have permission for remote code execution.' - ' Please ask user who added this server to change permission level.') + server = await get_server_by_alias(message, alias, ServerPermissions.RCE) results = chunks(await run_ssh_command(server, command), 4095) for res in results: await message.reply(f'```\n{res}```', parse_mode=ParseMode.MARKDOWN) diff --git a/src/actions/docker.py b/src/actions/docker.py new file mode 100644 index 0000000..7dfbb57 --- /dev/null +++ b/src/actions/docker.py @@ -0,0 +1,133 @@ +import json + +from aiogram.types import Message, ParseMode + +from src.models.server import ServerPermissions +from src.utils import chunks +from src.utils.decorators import ( + bot_action +) +from src.utils.server_utils import get_server_by_alias +from src.utils.ssh import run_ssh_command + + +@bot_action(r'meme admin docker( help)?$') +async def overall_help(message: Message, state=None): + await message.reply( + '```\n' + '* ls {alias} - List all running containers\n\n' + '* clean {alias} - Remove exited containers\n\n' + '* stop {alias} {containers} - stop one or more containers\n\n' + '* rm {alias} {containers} - remove one or more containers\n\n' + '* up {alias} {containers} - run one or more stopped containers\n\n' + '```', + parse_mode=ParseMode.MARKDOWN + ) + + +@bot_action(r'meme admin docker ls', + params=[ + ('alias', r'[\w\d]+') + ]) +async def docker_ls(message: Message, alias, state=None): + server = await get_server_by_alias(message, alias, ServerPermissions.DOCKER) + containers_ids_list = await run_ssh_command(server, 'docker ps -aq') + if not containers_ids_list: + await message.reply('* No running containers was found *', parse_mode=ParseMode.MARKDOWN) + return + ids = ' '.join(containers_ids_list.splitlines()) + response = await run_ssh_command(server, f'docker inspect {ids}') + inspect_info = json.loads(response) + formatted_containers = "" + total = 0 + for container in inspect_info: + total += 1 + formatted_containers += ( + '```\n' + f'ID: {container["Id"][:12]}\n' + f'name: {container["Name"]}\n' + f'status: {container["State"]["Status"]}\n' + f'networks: {", ".join(container["NetworkSettings"]["Networks"].keys())}\n' + '```\n') + message_text = ('* Docker containers: *\n' + f'*Total:* {total}\n\n' + f'{formatted_containers}') + + results = chunks(message_text, 4095) + for res in results: + await message.reply(res, parse_mode=ParseMode.MARKDOWN) + + +@bot_action(r'meme admin docker clean', + params=[ + ('alias', r'[\w\d]+') + ]) +async def docker_clean(message: Message, alias, state=None): + server = await get_server_by_alias(message, alias, ServerPermissions.DOCKER) + output = await run_ssh_command(server, 'docker ps -a | grep Exit | cut -d ' ' -f 1 | xargs docker rm') + total = len(output.splitlines()) + await message.reply( + '* Removed containers: *\n\n' + f'*Total:* {total}' + '```\n' + f'{output}' + '```', + parse_mode=ParseMode.MARKDOWN + ) + + +@bot_action(r'meme admin docker rm', + params=[ + ('alias', r'[\w\d]+'), + ('containers', r'.*') + ]) +async def docker_rm(message: Message, alias, containers, state=None): + server = await get_server_by_alias(message, alias, ServerPermissions.DOCKER) + output = await run_ssh_command(server, f'docker rm {containers}') + total = len(output.splitlines()) + await message.reply( + '* Removed containers: *\n\n' + f'*Total:* {total}' + '```\n' + f'{output}' + '```', + parse_mode=ParseMode.MARKDOWN + ) + + +@bot_action(r'meme admin docker stop', + params=[ + ('alias', r'[\w\d]+'), + ('containers', r'.*') + ]) +async def docker_rm(message: Message, alias, containers, state=None): + server = await get_server_by_alias(message, alias, ServerPermissions.DOCKER) + output = await run_ssh_command(server, f'docker stop {containers}') + total = len(output.splitlines()) + await message.reply( + '* Stopped containers: *\n\n' + f'*Total:* {total}' + '```\n' + f'{output}' + '```', + parse_mode=ParseMode.MARKDOWN + ) + + +@bot_action(r'meme admin docker up', + params=[ + ('alias', r'[\w\d]+'), + ('containers', r'.*') + ]) +async def docker_rm(message: Message, alias, containers, state=None): + server = await get_server_by_alias(message, alias, ServerPermissions.DOCKER) + output = await run_ssh_command(server, f'docker start {containers}') + total = len(output.splitlines()) + await message.reply( + '* Upped containers: *\n\n' + f'*Total:* {total}' + '```\n' + f'{output}' + '```', + parse_mode=ParseMode.MARKDOWN + ) diff --git a/src/utils/server_utils.py b/src/utils/server_utils.py index 1ebb429..50ce0d8 100644 --- a/src/utils/server_utils.py +++ b/src/utils/server_utils.py @@ -2,10 +2,13 @@ from aiogram.types import Message from src.models import Server from src.models.crud.server_crud import fn_get_server +from src.models.server import ServerPermissions from src.settings import settings -async def get_server_by_alias(message: Message, server_alias: str) -> Server: +async def get_server_by_alias(message: Message, + server_alias: str, + minimal_permission: ServerPermissions = ServerPermissions.INFO) -> Server: found_servers = await fn_get_server(settings.engine, chat_id=str(message.chat.id), server_alias=server_alias) @@ -19,4 +22,10 @@ async def get_server_by_alias(message: Message, server_alias: str) -> Server: ) else: raise Exception('No server linked to this chat.') - return found_servers[0] + server = found_servers[0] + if server.server_permissions.value < minimal_permission.value \ + and str(message.from_user.id) != server.server_admin: + raise Exception('You do not have permission for that action.\n' + f'Minimal permission level is `{minimal_permission.name}`\n' + f'Contact server administrator to change server permissions.') + return server -- GitLab