mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-15 10:40:28 +08:00
103 lines
4.7 KiB
Python
103 lines
4.7 KiB
Python
# ========================================================================== #
|
|
# #
|
|
# KVMD - The main Pi-KVM daemon. #
|
|
# #
|
|
# Copyright (C) 2018 Maxim Devaev <mdevaev@gmail.com> #
|
|
# #
|
|
# This program is free software: you can redistribute it and/or modify #
|
|
# it under the terms of the GNU General Public License as published by #
|
|
# the Free Software Foundation, either version 3 of the License, or #
|
|
# (at your option) any later version. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
|
# GNU General Public License for more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License #
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>. #
|
|
# #
|
|
# ========================================================================== #
|
|
|
|
|
|
import os
|
|
import contextlib
|
|
|
|
from typing import Dict
|
|
from typing import Optional
|
|
|
|
import dbus # pylint: disable=import-error
|
|
import dbus.exceptions
|
|
|
|
from ....logging import get_logger
|
|
|
|
from ....yamlconf import Section
|
|
from ....yamlconf.loader import load_yaml_file
|
|
|
|
from .... import aiotools
|
|
|
|
from .base import BaseInfoSubmanager
|
|
|
|
|
|
# =====
|
|
class ExtrasInfoSubmanager(BaseInfoSubmanager):
|
|
def __init__(self, global_config: Section) -> None:
|
|
self.__global_config = global_config
|
|
|
|
async def get_state(self) -> Optional[Dict]:
|
|
return (await aiotools.run_async(self.__inner_get_state))
|
|
|
|
# =====
|
|
|
|
def __inner_get_state(self) -> Optional[Dict]:
|
|
try:
|
|
extras_path = self.__global_config.kvmd.info.extras
|
|
extras: Dict[str, Dict] = {}
|
|
for app in os.listdir(extras_path):
|
|
if app[0] != "." and os.path.isdir(os.path.join(extras_path, app)):
|
|
extras[app] = load_yaml_file(os.path.join(extras_path, app, "manifest.yaml"))
|
|
self.__rewrite_app_daemon(extras[app])
|
|
self.__rewrite_app_port(extras[app])
|
|
return extras
|
|
except Exception:
|
|
get_logger(0).exception("Can't parse extras")
|
|
return None
|
|
|
|
def __rewrite_app_daemon(self, extras: Dict) -> None:
|
|
daemon = extras.get("daemon", "")
|
|
if isinstance(daemon, str) and daemon.strip():
|
|
extras["enabled"] = self.__is_daemon_enabled(daemon)
|
|
|
|
def __rewrite_app_port(self, extras: Dict) -> None:
|
|
port_path = extras.get("port", "")
|
|
if isinstance(port_path, str) and port_path.strip():
|
|
extras["port"] = 0
|
|
config = self.__global_config
|
|
for item in filter(None, map(str.strip, port_path.split("/"))):
|
|
config = getattr(config, item, None)
|
|
if isinstance(config, int):
|
|
extras["port"] = config
|
|
|
|
def __is_daemon_enabled(self, name: str) -> bool:
|
|
if not name.endswith(".service"):
|
|
name += ".service"
|
|
|
|
try:
|
|
with contextlib.closing(dbus.SystemBus()) as bus:
|
|
systemd = bus.get_object("org.freedesktop.systemd1", "/org/freedesktop/systemd1") # pylint: disable=no-member
|
|
manager = dbus.Interface(systemd, dbus_interface="org.freedesktop.systemd1.Manager")
|
|
|
|
try:
|
|
unit_proxy = bus.get_object("org.freedesktop.systemd1", manager.GetUnit(name)) # pylint: disable=no-member
|
|
unit_properties = dbus.Interface(unit_proxy, dbus_interface="org.freedesktop.DBus.Properties")
|
|
enabled = (unit_properties.Get("org.freedesktop.systemd1.Unit", "ActiveState") == "active")
|
|
except dbus.exceptions.DBusException as err:
|
|
if "NoSuchUnit" not in str(err):
|
|
raise
|
|
enabled = False
|
|
|
|
return (enabled or (manager.GetUnitFileState(name) in ["enabled", "enabled-runtime", "static", "indirect", "generated"]))
|
|
except Exception as err:
|
|
get_logger(0).error("Can't get info about the service %r: %s: %s", name, type(err).__name__, err)
|
|
return True
|