"""Class representing Bitwarden service""" import base64 import subprocess import typing from selfprivacy_api.jobs import Job, Jobs from selfprivacy_api.services.generic_service_mover import FolderMoveNames, move_service from selfprivacy_api.services.generic_status_getter import get_service_status from selfprivacy_api.services.service import Service, ServiceDnsRecord, ServiceStatus from selfprivacy_api.utils import ReadUserData, WriteUserData, get_domain from selfprivacy_api.utils.block_devices import BlockDevice import selfprivacy_api.utils.network as network_utils from selfprivacy_api.services.bitwarden.icon import BITWARDEN_ICON class Bitwarden(Service): """Class representing Bitwarden service.""" @staticmethod def get_id() -> str: """Return service id.""" return "bitwarden" @staticmethod def get_display_name() -> str: """Return service display name.""" return "Bitwarden" @staticmethod def get_description() -> str: """Return service description.""" return "Bitwarden is a password manager." @staticmethod def get_svg_icon() -> str: """Read SVG icon from file and return it as base64 encoded string.""" return base64.b64encode(BITWARDEN_ICON.encode("utf-8")).decode("utf-8") @staticmethod def get_user() -> str: return "vaultwarden" @staticmethod def get_url() -> typing.Optional[str]: """Return service url.""" domain = get_domain() return f"https://password.{domain}" @staticmethod def is_movable() -> bool: return True @staticmethod def is_required() -> bool: return False @staticmethod def get_backup_description() -> str: return "Password database, encryption certificate and attachments." @staticmethod def is_enabled() -> bool: with ReadUserData() as user_data: return ( user_data.get("modules", {}).get("bitwarden", {}).get("enable", False) ) @staticmethod def get_status() -> ServiceStatus: """ Return Bitwarden status from systemd. Use command return code to determine status. Return code 0 means service is running. Return code 1 or 2 means service is in error stat. Return code 3 means service is stopped. Return code 4 means service is off. """ return get_service_status("vaultwarden.service") @staticmethod def enable(): """Enable Bitwarden service.""" with WriteUserData() as user_data: if "modules" not in user_data: user_data["modules"] = {} if "bitwarden" not in user_data["modules"]: user_data["modules"]["bitwarden"] = {} user_data["modules"]["bitwarden"]["enable"] = True @staticmethod def disable(): """Disable Bitwarden service.""" with WriteUserData() as user_data: if "modules" not in user_data: user_data["modules"] = {} if "bitwarden" not in user_data["modules"]: user_data["modules"]["bitwarden"] = {} user_data["modules"]["bitwarden"]["enable"] = False @staticmethod def stop(): subprocess.run(["systemctl", "stop", "vaultwarden.service"]) @staticmethod def start(): subprocess.run(["systemctl", "start", "vaultwarden.service"]) @staticmethod def restart(): subprocess.run(["systemctl", "restart", "vaultwarden.service"]) @staticmethod def get_configuration(): return {} @staticmethod def set_configuration(config_items): return super().set_configuration(config_items) @staticmethod def get_logs(): return "" @staticmethod def get_folders() -> typing.List[str]: return ["/var/lib/bitwarden", "/var/lib/bitwarden_rs"] @staticmethod def get_dns_records() -> typing.List[ServiceDnsRecord]: """Return list of DNS records for Bitwarden service.""" return [ ServiceDnsRecord( type="A", name="password", content=network_utils.get_ip4(), ttl=3600, display_name="Bitwarden", ), ServiceDnsRecord( type="AAAA", name="password", content=network_utils.get_ip6(), ttl=3600, display_name="Bitwarden (IPv6)", ), ] def move_to_volume(self, volume: BlockDevice) -> Job: job = Jobs.add( type_id="services.bitwarden.move", name="Move Bitwarden", description=f"Moving Bitwarden data to {volume.name}", ) move_service( self, volume, job, FolderMoveNames.default_foldermoves(self), "bitwarden", ) return job