mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-12 09:10:30 +08:00
106 lines
4.1 KiB
Python
106 lines
4.1 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/>. #
|
|
# #
|
|
# ========================================================================== #
|
|
|
|
|
|
from typing import List
|
|
from typing import Optional
|
|
|
|
from ...logging import get_logger
|
|
|
|
from ...plugins.hid import get_hid_class
|
|
from ...plugins.atx import get_atx_class
|
|
from ...plugins.msd import get_msd_class
|
|
|
|
from .. import init
|
|
|
|
from .auth import AuthManager
|
|
from .info import InfoManager
|
|
from .logreader import LogReader
|
|
from .wol import WakeOnLan
|
|
from .ugpio import UserGpio
|
|
from .streamer import Streamer
|
|
from .snapshoter import Snapshoter
|
|
from .server import KvmdServer
|
|
|
|
|
|
# =====
|
|
def main(argv: Optional[List[str]]=None) -> None:
|
|
config = init(
|
|
prog="kvmd",
|
|
description="The main Pi-KVM daemon",
|
|
argv=argv,
|
|
check_run=True,
|
|
load_auth=True,
|
|
load_hid=True,
|
|
load_atx=True,
|
|
load_msd=True,
|
|
load_gpio=True,
|
|
)[2]
|
|
|
|
msd_kwargs = config.kvmd.msd._unpack(ignore=["type"])
|
|
if config.kvmd.msd.type == "otg":
|
|
msd_kwargs["gadget"] = config.otg.gadget # XXX: Small crutch to pass gadget name to the plugin
|
|
|
|
hid_kwargs = config.kvmd.hid._unpack(ignore=["type", "keymap"])
|
|
if config.kvmd.hid.type == "otg":
|
|
hid_kwargs["udc"] = config.otg.udc # XXX: Small crutch to pass UDC to the plugin
|
|
|
|
global_config = config
|
|
config = config.kvmd
|
|
|
|
hid = get_hid_class(config.hid.type)(**hid_kwargs)
|
|
streamer = Streamer(**config.streamer._unpack(ignore=["forever"]))
|
|
|
|
KvmdServer(
|
|
auth_manager=AuthManager(
|
|
internal_type=config.auth.internal.type,
|
|
internal_kwargs=config.auth.internal._unpack(ignore=["type", "force_users"]),
|
|
external_type=config.auth.external.type,
|
|
external_kwargs=(config.auth.external._unpack(ignore=["type"]) if config.auth.external.type else {}),
|
|
force_internal_users=config.auth.internal.force_users,
|
|
enabled=config.auth.enabled,
|
|
),
|
|
info_manager=InfoManager(global_config),
|
|
log_reader=LogReader(),
|
|
wol=WakeOnLan(**config.wol._unpack()),
|
|
user_gpio=UserGpio(config.gpio),
|
|
|
|
hid=hid,
|
|
atx=get_atx_class(config.atx.type)(**config.atx._unpack(ignore=["type"])),
|
|
msd=get_msd_class(config.msd.type)(**msd_kwargs),
|
|
streamer=streamer,
|
|
|
|
snapshoter=Snapshoter(
|
|
hid=hid,
|
|
streamer=streamer,
|
|
**config.snapshot._unpack(),
|
|
),
|
|
|
|
heartbeat=config.server.heartbeat,
|
|
sync_chunk_size=config.server.sync_chunk_size,
|
|
|
|
keymap_path=config.hid.keymap,
|
|
|
|
stream_forever=config.streamer.forever,
|
|
).run(**config.server._unpack(ignore=["heartbeat", "sync_chunk_size"]))
|
|
|
|
get_logger(0).info("Bye-bye")
|