mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2026-06-17 02:11:44 +08:00
refactoring
This commit is contained in:
Devaev Maxim
20
kvmd/apps/kvmd/api/__init__.py
Normal file
20
kvmd/apps/kvmd/api/__init__.py
Normal file
@@ -0,0 +1,20 @@
|
||||
# ========================================================================== #
|
||||
# #
|
||||
# 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/>. #
|
||||
# #
|
||||
# ========================================================================== #
|
||||
64
kvmd/apps/kvmd/api/atx.py
Normal file
64
kvmd/apps/kvmd/api/atx.py
Normal file
@@ -0,0 +1,64 @@
|
||||
# ========================================================================== #
|
||||
# #
|
||||
# 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 aiohttp.web
|
||||
|
||||
from ....plugins.atx import BaseAtx
|
||||
|
||||
from ....validators.kvm import valid_atx_power_action
|
||||
from ....validators.kvm import valid_atx_button
|
||||
|
||||
from ..http import exposed_http
|
||||
from ..http import make_json_response
|
||||
|
||||
|
||||
# =====
|
||||
class AtxApi:
|
||||
def __init__(self, atx: BaseAtx) -> None:
|
||||
self.__atx = atx
|
||||
|
||||
# =====
|
||||
|
||||
@exposed_http("GET", "/atx")
|
||||
async def __state_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
return make_json_response(self.__atx.get_state())
|
||||
|
||||
@exposed_http("POST", "/atx/power")
|
||||
async def __power_handler(self, request: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
action = valid_atx_power_action(request.query.get("action"))
|
||||
processing = await ({
|
||||
"on": self.__atx.power_on,
|
||||
"off": self.__atx.power_off,
|
||||
"off_hard": self.__atx.power_off_hard,
|
||||
"reset_hard": self.__atx.power_reset_hard,
|
||||
}[action])()
|
||||
return make_json_response({"processing": processing})
|
||||
|
||||
@exposed_http("POST", "/atx/click")
|
||||
async def __click_handler(self, request: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
button = valid_atx_button(request.query.get("button"))
|
||||
await ({
|
||||
"power": self.__atx.click_power,
|
||||
"power_long": self.__atx.click_power_long,
|
||||
"reset": self.__atx.click_reset,
|
||||
}[button])()
|
||||
return make_json_response()
|
||||
93
kvmd/apps/kvmd/api/hid.py
Normal file
93
kvmd/apps/kvmd/api/hid.py
Normal file
@@ -0,0 +1,93 @@
|
||||
# ========================================================================== #
|
||||
# #
|
||||
# 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 Dict
|
||||
|
||||
import aiohttp.web
|
||||
|
||||
from ....plugins.hid import BaseHid
|
||||
|
||||
from ....validators.basic import valid_bool
|
||||
|
||||
from ....validators.kvm import valid_hid_key
|
||||
from ....validators.kvm import valid_hid_mouse_move
|
||||
from ....validators.kvm import valid_hid_mouse_button
|
||||
from ....validators.kvm import valid_hid_mouse_wheel
|
||||
|
||||
from ..http import exposed_http
|
||||
from ..http import exposed_ws
|
||||
from ..http import make_json_response
|
||||
|
||||
|
||||
# =====
|
||||
class HidApi:
|
||||
def __init__(self, hid: BaseHid) -> None:
|
||||
self.__hid = hid
|
||||
|
||||
# =====
|
||||
|
||||
@exposed_http("GET", "/hid/state")
|
||||
async def __state_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
return make_json_response(self.__hid.get_state())
|
||||
|
||||
@exposed_http("GET", "/hid/reset")
|
||||
async def __reset_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
await self.__hid.reset()
|
||||
return make_json_response()
|
||||
|
||||
# =====
|
||||
|
||||
@exposed_ws("key")
|
||||
async def __ws_key_handler(self, _: aiohttp.web.WebSocketResponse, event: Dict) -> None:
|
||||
try:
|
||||
key = valid_hid_key(event["key"])
|
||||
state = valid_bool(event["state"])
|
||||
except Exception:
|
||||
return
|
||||
await self.__hid.send_key_event(key, state)
|
||||
|
||||
@exposed_ws("mouse_button")
|
||||
async def __ws_mouse_button_handler(self, _: aiohttp.web.WebSocketResponse, event: Dict) -> None:
|
||||
try:
|
||||
button = valid_hid_mouse_button(event["button"])
|
||||
state = valid_bool(event["state"])
|
||||
except Exception:
|
||||
return
|
||||
await self.__hid.send_mouse_button_event(button, state)
|
||||
|
||||
@exposed_ws("mouse_move")
|
||||
async def __ws_mouse_move_handler(self, _: aiohttp.web.WebSocketResponse, event: Dict) -> None:
|
||||
try:
|
||||
to_x = valid_hid_mouse_move(event["to"]["x"])
|
||||
to_y = valid_hid_mouse_move(event["to"]["y"])
|
||||
except Exception:
|
||||
return
|
||||
await self.__hid.send_mouse_move_event(to_x, to_y)
|
||||
|
||||
@exposed_ws("mouse_wheel")
|
||||
async def __ws_mouse_wheel_handler(self, _: aiohttp.web.WebSocketResponse, event: Dict) -> None:
|
||||
try:
|
||||
delta_x = valid_hid_mouse_wheel(event["delta"]["x"])
|
||||
delta_y = valid_hid_mouse_wheel(event["delta"]["y"])
|
||||
except Exception:
|
||||
return
|
||||
await self.__hid.send_mouse_wheel_event(delta_x, delta_y)
|
||||
57
kvmd/apps/kvmd/api/log.py
Normal file
57
kvmd/apps/kvmd/api/log.py
Normal file
@@ -0,0 +1,57 @@
|
||||
# ========================================================================== #
|
||||
# #
|
||||
# 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 Optional
|
||||
|
||||
import aiohttp.web
|
||||
|
||||
from ....validators.basic import valid_bool
|
||||
|
||||
from ....validators.kvm import valid_log_seek
|
||||
|
||||
from ..logreader import LogReader
|
||||
|
||||
from ..http import exposed_http
|
||||
|
||||
|
||||
# =====
|
||||
class LogApi:
|
||||
def __init__(self, log_reader: LogReader) -> None:
|
||||
self.__log_reader = log_reader
|
||||
|
||||
# =====
|
||||
|
||||
@exposed_http("GET", "/log")
|
||||
async def __log_handler(self, request: aiohttp.web.Request) -> aiohttp.web.StreamResponse:
|
||||
seek = valid_log_seek(request.query.get("seek", "0"))
|
||||
follow = valid_bool(request.query.get("follow", "false"))
|
||||
response: Optional[aiohttp.web.StreamResponse] = None
|
||||
async for record in self.__log_reader.poll_log(seek, follow):
|
||||
if response is None:
|
||||
response = aiohttp.web.StreamResponse(status=200, reason="OK", headers={"Content-Type": "text/plain"})
|
||||
await response.prepare(request)
|
||||
await response.write(("[%s %s] --- %s" % (
|
||||
record["dt"].strftime("%Y-%m-%d %H:%M:%S"),
|
||||
record["service"],
|
||||
record["msg"],
|
||||
)).encode("utf-8") + b"\r\n")
|
||||
return response
|
||||
106
kvmd/apps/kvmd/api/msd.py
Normal file
106
kvmd/apps/kvmd/api/msd.py
Normal file
@@ -0,0 +1,106 @@
|
||||
# ========================================================================== #
|
||||
# #
|
||||
# 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 aiohttp
|
||||
import aiohttp.web
|
||||
|
||||
from ....logging import get_logger
|
||||
|
||||
from ....plugins.msd import BaseMsd
|
||||
|
||||
from ....validators.basic import valid_bool
|
||||
|
||||
from ....validators.kvm import valid_msd_image_name
|
||||
|
||||
from ..http import exposed_http
|
||||
from ..http import make_json_response
|
||||
from ..http import get_multipart_field
|
||||
|
||||
|
||||
# ======
|
||||
class MsdApi:
|
||||
def __init__(self, msd: BaseMsd, sync_chunk_size: int) -> None:
|
||||
self.__msd = msd
|
||||
self.__sync_chunk_size = sync_chunk_size
|
||||
|
||||
# =====
|
||||
|
||||
@exposed_http("GET", "/msd")
|
||||
async def __state_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
return make_json_response(await self.__msd.get_state())
|
||||
|
||||
@exposed_http("POST", "/msd/set_params")
|
||||
async def __set_params_handler(self, request: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
params = {
|
||||
key: validator(request.query.get(param))
|
||||
for (param, key, validator) in [
|
||||
("image", "name", (lambda arg: str(arg).strip() and valid_msd_image_name(arg))),
|
||||
("cdrom", "cdrom", valid_bool),
|
||||
]
|
||||
if request.query.get(param) is not None
|
||||
}
|
||||
await self.__msd.set_params(**params) # type: ignore
|
||||
return make_json_response()
|
||||
|
||||
@exposed_http("POST", "/msd/connect")
|
||||
async def __connect_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
await self.__msd.connect()
|
||||
return make_json_response()
|
||||
|
||||
@exposed_http("POST", "/msd/disconnect")
|
||||
async def __disconnect_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
await self.__msd.disconnect()
|
||||
return make_json_response()
|
||||
|
||||
@exposed_http("POST", "/msd/write")
|
||||
async def __write_handler(self, request: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
logger = get_logger(0)
|
||||
reader = await request.multipart()
|
||||
name = ""
|
||||
written = 0
|
||||
try:
|
||||
name_field = await get_multipart_field(reader, "image")
|
||||
name = valid_msd_image_name((await name_field.read()).decode("utf-8"))
|
||||
|
||||
data_field = await get_multipart_field(reader, "data")
|
||||
|
||||
async with self.__msd.write_image(name):
|
||||
logger.info("Writing image %r to MSD ...", name)
|
||||
while True:
|
||||
chunk = await data_field.read_chunk(self.__sync_chunk_size)
|
||||
if not chunk:
|
||||
break
|
||||
written = await self.__msd.write_image_chunk(chunk)
|
||||
finally:
|
||||
if written != 0:
|
||||
logger.info("Written image %r with size=%d bytes to MSD", name, written)
|
||||
return make_json_response({"image": {"name": name, "size": written}})
|
||||
|
||||
@exposed_http("POST", "/msd/remove")
|
||||
async def __remove_handler(self, request: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
await self.__msd.remove(valid_msd_image_name(request.query.get("image")))
|
||||
return make_json_response()
|
||||
|
||||
@exposed_http("POST", "/msd/reset")
|
||||
async def __reset_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
await self.__msd.reset()
|
||||
return make_json_response()
|
||||
45
kvmd/apps/kvmd/api/wol.py
Normal file
45
kvmd/apps/kvmd/api/wol.py
Normal file
@@ -0,0 +1,45 @@
|
||||
# ========================================================================== #
|
||||
# #
|
||||
# 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 aiohttp.web
|
||||
|
||||
from ..wol import WakeOnLan
|
||||
|
||||
from ..http import exposed_http
|
||||
from ..http import make_json_response
|
||||
|
||||
|
||||
# =====
|
||||
class WolApi:
|
||||
def __init__(self, wol: WakeOnLan) -> None:
|
||||
self.__wol = wol
|
||||
|
||||
# =====
|
||||
|
||||
@exposed_http("GET", "/wol")
|
||||
async def __wol_state_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
return make_json_response(self.__wol.get_state())
|
||||
|
||||
@exposed_http("POST", "/wol/wakeup")
|
||||
async def __wol_wakeup_handler(self, _: aiohttp.web.Request) -> aiohttp.web.Response:
|
||||
await self.__wol.wakeup()
|
||||
return make_json_response()
|
||||
Reference in New Issue
Block a user