selfprivacy-rest-api/selfprivacy_api/migrations/add_roundcube.py

from selfprivacy_api.migrations.migration import Migration

from selfprivacy_api.services.flake_service_manager import FlakeServiceManager
from selfprivacy_api.utils import ReadUserData, WriteUserData


class AddRoundcube(Migration):
    """Adds the Roundcube if it is not present."""

    def get_migration_name(self) -> str:
        return "add_roundcube"

    def get_migration_description(self) -> str:
        return "Adds the Roundcube if it is not present."

    def is_migration_needed(self) -> bool:
        with FlakeServiceManager() as manager:
            if "roundcube" not in manager.services:
                return True
        with ReadUserData() as data:
            if "roundcube" not in data["modules"]:
                return True
        return False

    def migrate(self) -> None:
        with FlakeServiceManager() as manager:
            if "roundcube" not in manager.services:
                manager.services[
                    "roundcube"
                ] = "git+https://git.selfprivacy.org/SelfPrivacy/selfprivacy-nixos-config.git?ref=flakes&dir=sp-modules/roundcube"
        with WriteUserData() as data:
            if "roundcube" not in data["modules"]:
                data["modules"]["roundcube"] = {
                    "enable": False,
                    "subdomain": "roundcube",
                }