diff --git a/plugins/modules/mariadb.py b/plugins/modules/mariadb.py new file mode 100644 index 0000000..b1a012f --- /dev/null +++ b/plugins/modules/mariadb.py @@ -0,0 +1,104 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- + +# Copyright: (c) 2024, Luca Bilke +# MIT License (see LICENSE) + +# TODO: write ansible sections + +DOCUMENTATION = r""" +""" + +EXAMPLES = r""" +""" + +RETURN = r""" +""" + +from __future__ import annotations # pyright: ignore[reportGeneralTypeIssues] + +import shlex + +from ansible_collections.snailed.ez_compose.plugins.module_utils.common import ( + State, + run_service, + update_service, +) + + +def helper(state: State) -> State: + backup = state.module.params.get("backup", True) + database = state.module.params["database"] + username = state.module.params["username"] + password = state.module.params["password"] + root_password = state.module.params["root_password"] + name = state.module.params["name"] + project_name = state.module.params["project_name"] + + volumes = [ + { + "source": name, + "target": "/var/lib/mysql", + "type": "volume", + } + ] + + environment = { + "MARIADB_DATABASE": database, + "MARIADB_USER": username, + "MARIADB_PASSWORD": password, + } + + labels: dict[str, str] = {} + + if root_password: + environment.update( + { + "MARIADB_ROOT_PASSWORD": root_password, + } + ) + + if backup: + labels.update( + { + "docker-volume-backup.archive-pre": ( + "/bin/sh -c '" + "mysqldump --all-databases " + f"-u {shlex.quote(username)} " + f"-p {shlex.quote(password)} " + f">/backup/{shlex.quote(project_name)}.sql" + "'" + ) + } + ) + # mysqldump -psecret --all-databases > /tmp/dumps/dump.sql + volumes.append( + { + "type": "volume", + "source": f"{name}_backup", + "target": "/backup", + } + ) + + update = { + "environment": environment, + "volumes": volumes, + "labels": labels, + } + + return update_service(state, update) + + +def main(): + extra_args = { + "archive": {"type": "bool"}, + "database": {"type": "str", "required": True}, + "username": {"type": "str", "required": True}, + "password": {"type": "str", "required": True}, + "root_password": {"type": "str"}, + } + run_service(extra_args, helper) + + +if __name__ == "__main__": + main()