diff --git a/src/actions/__init__.py b/src/actions/__init__.py
index 3827707173022015df58a51f57101e3804362773..f689a298030bdb793cabc394d600864289f1ba2e 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 fa6be86e8d8e213d2b549e29d1afc65d975e4edf..5c35909c6f12b7373d32e0bfe0bc744a5b012d6c 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 0000000000000000000000000000000000000000..7dfbb576ae8fc3c6b16db69477fa15ae364dc909
--- /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 1ebb4296eca6c78eb97af44dae47c7396a8d3802..50ce0d84c7a285d5adceb389d5f98100c7837850 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