122 lines
3.3 KiB
Python
122 lines
3.3 KiB
Python
"""Class representing ocserv 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_size_counter import get_storage_usage
|
|
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
|
|
from selfprivacy_api.utils.block_devices import BlockDevice
|
|
from selfprivacy_api.services.ocserv.icon import OCSERV_ICON
|
|
import selfprivacy_api.utils.network as network_utils
|
|
|
|
|
|
class Ocserv(Service):
|
|
"""Class representing ocserv service."""
|
|
|
|
@staticmethod
|
|
def get_id() -> str:
|
|
return "ocserv"
|
|
|
|
@staticmethod
|
|
def get_display_name() -> str:
|
|
return "OpenConnect VPN"
|
|
|
|
@staticmethod
|
|
def get_description() -> str:
|
|
return "OpenConnect VPN to connect your devices and access the internet."
|
|
|
|
@staticmethod
|
|
def get_svg_icon() -> str:
|
|
return base64.b64encode(OCSERV_ICON.encode("utf-8")).decode("utf-8")
|
|
|
|
@staticmethod
|
|
def get_url() -> typing.Optional[str]:
|
|
"""Return service url."""
|
|
return None
|
|
|
|
@staticmethod
|
|
def is_movable() -> bool:
|
|
return False
|
|
|
|
@staticmethod
|
|
def is_required() -> bool:
|
|
return False
|
|
|
|
@staticmethod
|
|
def is_enabled() -> bool:
|
|
with ReadUserData() as user_data:
|
|
return user_data.get("ocserv", {}).get("enable", False)
|
|
|
|
@staticmethod
|
|
def get_status() -> ServiceStatus:
|
|
return get_service_status("ocserv.service")
|
|
|
|
@staticmethod
|
|
def enable():
|
|
with WriteUserData() as user_data:
|
|
if "ocserv" not in user_data:
|
|
user_data["ocserv"] = {}
|
|
user_data["ocserv"]["enable"] = True
|
|
|
|
@staticmethod
|
|
def disable():
|
|
with WriteUserData() as user_data:
|
|
if "ocserv" not in user_data:
|
|
user_data["ocserv"] = {}
|
|
user_data["ocserv"]["enable"] = False
|
|
|
|
@staticmethod
|
|
def stop():
|
|
subprocess.run(["systemctl", "stop", "ocserv.service"])
|
|
|
|
@staticmethod
|
|
def start():
|
|
subprocess.run(["systemctl", "start", "ocserv.service"])
|
|
|
|
@staticmethod
|
|
def restart():
|
|
subprocess.run(["systemctl", "restart", "ocserv.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_location() -> str:
|
|
return "sda1"
|
|
|
|
@staticmethod
|
|
def get_dns_records() -> typing.List[ServiceDnsRecord]:
|
|
return [
|
|
ServiceDnsRecord(
|
|
type="A",
|
|
name="vpn",
|
|
content=network_utils.get_ip4(),
|
|
ttl=3600,
|
|
),
|
|
ServiceDnsRecord(
|
|
type="AAAA",
|
|
name="vpn",
|
|
content=network_utils.get_ip6(),
|
|
ttl=3600,
|
|
),
|
|
]
|
|
|
|
@staticmethod
|
|
def get_storage_usage() -> int:
|
|
return 0
|
|
|
|
def move_to_volume(self, volume: BlockDevice) -> Job:
|
|
raise NotImplementedError("ocserv service is not movable")
|