Protocol and other stuff

This commit is contained in:
Christoph Stahl 2024-01-02 10:06:37 +01:00
parent 7cf5a5137e
commit b31276d818
8 changed files with 151 additions and 26 deletions

77
Protocol.md Normal file
View file

@ -0,0 +1,77 @@
Protocol
========
This document describes the workflows of the software.
Preliminaries
-------------
- **Song**: A reference to the file containing the audio and video. Can be separated into two files (e.g. mp3+cdg) or a link (e.g. a YouTube link)
- **Source**: A collection of _songs_, that can be searched and played back from. (e.g. a folder, a s3 storage or YouTube)
- **Performer**: The person(s) doing the actual singing
- **Entry**: A _song_ together with a _performer_
- **Queue**: A list of _entries_. Once the first entry is completely played back, the next entry is played.
- **Waiting Room**: A list of _entries_. To limit one performer filling the entire _queue_, a waiting room can be configured. If so, each performer can only have one entry in the queue. Each additional entry is put in the waiting room. Once the last entry of a performer left the queue, the first entry of that performer in the waiting room is added at the end of the queue.
- **Recents**: A list of _entries_. Once an entry successfully leaves the _queue_, it is added to the recents.
- **Playback client**: Part of the software, that does the actual playback, usually hooked to a video output device like a monitor or a projector. This needs to have access to the configured sources.
- **Web client**: User facing part of the software. Used to search and add _entries_ to the _queue_. Has an admin view to manipulate the queue and the _waiting room_.
- **Room**: One specific karaoke event, consisting of one _queue_, one _recents_, up to one _waiting room_, one _playback client_ and several _web clients_. It has an identifier and a _secret_, used to authenticate as an admin.
- **Server**: Manages all _rooms_.
We will use the abbreviations _P_, _W_, and _S_ when talking about the _playback client_, _web client_ and the _server_.
Communication usually happens between P ↔ S and W ↔ S and as messages on top of web sockets, using [socket.io](https://socket.io/docs/v4/client-api/).
### Entry
Entries are regularly sent between all participants and are encoded in JSON as follows:
| Key | Type | Description | Optional |
|----------|-------|----------------------------------------------------------------------------------|------------------------------------------|
| ident | `str` | Identifier for the entry in its given source. E.g. a file name or a YouTube Link | No |
| source | `str` | Name of the source (`files`, `s3`, `youtube`, etc.) | No |
| duration | `int` | Duration of the song | No |
| title | `str` | Name of the song | No |
| artist | `str` | Artist of the original song | No |
| album | `str` | Name of the collection this song belongs to | No |
| uuid | `str` | A UUID for this entry | Yes (generated automatically if omitted) |
### Client Config
A client config specifies the knowlege the server has of a specific playback client.
| Key | Type | Description | Optional | Default |
|---------------------|------|----------------------------------------------------------------------------------------------------------|----------|--------------------------|
| server | str | URL of the server | Yes | `https://localhost:8080` |
| room | str | Identifier of the room the client wants to connect to | Yes | Generated by the server |
| secret | str | The secret for the room | No | |
| preview_duration | int | Time between songs, where a preview is shown for the next song | Yes | 3 |
| last_song | int | Unix timestamp of the last song allowed to be played | Yes | None |
| waiting_room_policy | str | `forced` if waiting room is forced, `optional` if performers are given the choice, `None` if deactivated | Yes | None |
Workflow
--------
### Connect P ↔ S
When a playback client connects to a server, it can provide a room identifier and a room secret.
If none are given, the server will generate both and send them to the client.
If the server does not know a room with that identifier, a new room is created with the given secret.
The client sends its initial configuration to the server (including an initial possible empty queue, waiting room and recent list),
and the configuration of each configured source.
If the server has already registered a room with the given identifier, if the secret is the same, the connection to the new playback client is stored and the old connection is forgotten.
The server-side config is updated with the values of the new client.
The following messages are exchanged during connection:
- **Direction:** P -> S, **Message**: `connect`, **Parameters** None
Socket.io connect
| Communication | Message | Params | Notes |
|---------------|---------------------|-------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| P → S | `connect` | -- | Socket.io connect |
| S → P | `connect` | -- | Socket.io connect |
| P → S | `register-client` | `{ queue: list[Entry], waiting_room: list[Entry], recents: list[Entry], config: Config }` | The playback client can push an initial queue, waiting_room and recents to the server. |
| S -> P | `client-registered` | `{ success: bool, room: str }` | sucess is `true` if requested room is not in use or secrets match, otherwise `false`. The server confirms the room name, if it was requested in `register-client`, otherwise a new room name is returned |
| S -> P | `state` | `{ queue: list[Entry], waiting_room: list[Entry], recents: list[Entry], config: Config}` | The server returns

View file

@ -353,22 +353,30 @@ async def handle_request_config(data: dict[str, Any]) -> None:
:type data: dict[str, Any]
:rtype: None
"""
if data["source"] in sources:
config: dict[str, Any] | list[dict[str, Any]] = await sources[data["source"]].get_config()
async def send_config(source: str, update: bool) -> None:
config: dict[str, Any] | list[dict[str, Any]] = await sources[source].get_config(update)
if isinstance(config, list):
num_chunks: int = len(config)
for current, chunk in enumerate(config):
await sio.emit(
"config-chunk",
{
"source": data["source"],
"source": source,
"config": chunk,
"number": current + 1,
"total": num_chunks,
},
)
else:
await sio.emit("config", {"source": data["source"], "config": config})
await sio.emit("config", {"source": source, "config": config})
if data["source"] in sources:
await send_config(data["source"], False)
if data["update"]:
await sources[data["source"]].get_config(True)
await sio.emit("request-resend-config", {"source": data["source"]})
def signal_handler() -> None:

View file

@ -234,7 +234,8 @@ class OptionFrame(customtkinter.CTkScrollableFrame): # type:ignore
def get_config(self) -> dict[str, Any]:
config: dict[str, Any] = {}
for name, textbox in self.string_options.items():
config[name] = textbox.get("0.0", "end").strip()
value = textbox.get("0.0", "end").strip()
config[name] = value if value else None
for name, optionmenu in self.choose_options.items():
config[name] = optionmenu.get().strip()
@ -338,7 +339,9 @@ class SyngGui(customtkinter.CTk): # type:ignore
config: dict[str, dict[str, Any]] = {"sources": {}, "config": default_config()}
try:
config["config"] |= loaded_config["config"]
for option, value in loaded_config["config"].items():
if value:
config["config"][option] = value
except (KeyError, TypeError):
print("Could not load config")
@ -400,8 +403,11 @@ class SyngGui(customtkinter.CTk): # type:ignore
self.tabs = {}
for source_name in available_sources:
source_config = {}
try:
source_config = loaded_config["sources"][source_name]
for option, value in loaded_config["sources"][source_name].items():
if value:
source_config[option] = value
except (KeyError, TypeError):
source_config = {}

View file

@ -454,8 +454,6 @@ async def handle_append_anyway(sid: str, data: dict[str, Any]) -> None:
entry.uid = data["uid"] if "uid" in data else None
print(entry)
await append_to_queue(room, entry, sid)
@ -637,9 +635,10 @@ async def handle_register_client(sid: str, data: dict[str, Any]) -> None:
- `config`, an dictionary of initial configurations
- `queue`, a list of initial entries for the queue. The entries are
encoded as a dictionary.
- `waiting_room`, a list of initial entries for the waiting room. The
entries are encoded as a dictionary.
- `recent`, a list of initial entries for the recent list. The entries
are encoded as a dictionary.
- `secret`, the secret of the room
This will register a new playback client to a specific room. If there
already exists a playback client registered for this room, this
@ -777,7 +776,7 @@ async def handle_sources(sid: str, data: dict[str, Any]) -> None:
state.client.sources_prio = data["sources"]
for name in new_sources:
await sio.emit("request-config", {"source": name}, room=sid)
await sio.emit("request-config", {"source": name, "update": True}, room=sid)
@sio.on("config-chunk")
@ -811,6 +810,32 @@ async def handle_config_chunk(sid: str, data: dict[str, Any]) -> None:
state.client.sources[data["source"]].add_to_config(data["config"])
@sio.on("request-resend-config")
async def handle_request_resend_config(sid: str, data: dict[str, Any]) -> None:
"""
Handle the "request-resend-config" message.
Clears the config for a given source and requests a resend of the config
from the playback client.
:param sid: The session id of the playback client
:type sid: str
:param data: A dictionary with the "source" (str) entry
:rtype: None
"""
async with sio.session(sid) as session:
room = session["room"]
state = clients[room]
if sid != state.sid:
return
state.client.sources[data["source"]] = available_sources[data["source"]]({})
print(f"Rerequesting {data['source']}")
await sio.emit("request-config", {"source": data["source"], "update": False}, sid)
@sio.on("config")
async def handle_config(sid: str, data: dict[str, Any]) -> None:
"""

View file

@ -19,7 +19,8 @@ class FilesSource(FileBasedSource):
source_name = "files"
config_schema = FileBasedSource.config_schema | {
"dir": (str, "Directory to index", "."),
"index_file": (str, "Index file", "files-index"),
# "index_file": (str, "Index file", str(user_cache_path("syng") / "files" / "index")),
# "recreate_index": (bool, "Recreate index file", False),
}
def __init__(self, config: dict[str, Any]):
@ -29,7 +30,7 @@ class FilesSource(FileBasedSource):
self.dir = config["dir"] if "dir" in config else "."
self.extra_mpv_arguments = ["--scale=oversample"]
async def get_file_list(self) -> list[str]:
async def get_file_list(self, update: bool = False) -> list[str]:
"""Collect all files in ``dir``, that have the correct filename extension"""
def _get_file_list() -> list[str]:

View file

@ -8,6 +8,8 @@ import os
from json import dump, load
from typing import Any, Optional, Tuple, cast
from platformdirs import user_cache_path
try:
from minio import Minio
@ -31,6 +33,7 @@ class S3Source(FileBasedSource):
- ``index_file``: If the file does not exist, saves the paths of
files from the s3 instance to this file. If it exists, loads
the list of files from this file.
-``recreate_index``, rebuild index even if it exists
"""
source_name = "s3"
@ -41,7 +44,7 @@ class S3Source(FileBasedSource):
"secure": (bool, "Use SSL", True),
"bucket": (str, "Bucket of the s3", ""),
"tmp_dir": (str, "Folder for\ntemporary download", "/tmp/syng"),
"index_file": (str, "Index file", "s3-index"),
"index_file": (str, "Index file", str(user_cache_path("syng") / "s3" / "index")),
}
def __init__(self, config: dict[str, Any]):
@ -66,7 +69,7 @@ class S3Source(FileBasedSource):
self.index_file: Optional[str] = config["index_file"] if "index_file" in config else None
self.extra_mpv_arguments = ["--scale=oversample"]
async def get_file_list(self) -> list[str]:
async def get_file_list(self, update: bool = False) -> list[str]:
"""
Return the list of files on the s3 instance, according to the extensions.
@ -78,8 +81,8 @@ class S3Source(FileBasedSource):
:rtype: list[str]
"""
def _get_file_list() -> list[str]:
if self.index_file is not None and os.path.isfile(self.index_file):
def _get_file_list(update: bool) -> list[str]:
if not update and self.index_file is not None and os.path.isfile(self.index_file):
with open(self.index_file, "r", encoding="utf8") as index_file_handle:
return cast(list[str], load(index_file_handle))
@ -89,11 +92,12 @@ class S3Source(FileBasedSource):
if self.has_correct_extension(obj.object_name)
]
if self.index_file is not None and not os.path.isfile(self.index_file):
os.makedirs(os.path.dirname(self.index_file), exist_ok=True)
with open(self.index_file, "w", encoding="utf8") as index_file_handle:
dump(file_list, index_file_handle)
return file_list
return await asyncio.to_thread(_get_file_list)
return await asyncio.to_thread(_get_file_list, update)
async def get_missing_metadata(self, entry: Entry) -> dict[str, Any]:
"""

View file

@ -145,7 +145,7 @@ class Source(ABC):
"""
args = ["--fullscreen", *options, video] + ([f"--audio-file={audio}"] if audio else [])
print(f"File is {video=} and {audio=}")
# print(f"File is {video=} and {audio=}")
mpv_process = asyncio.create_subprocess_exec(
"mpv",
@ -372,19 +372,21 @@ class Source(ABC):
splitquery = shlex.split(query)
return [element for element in data if contains_all_words(splitquery, element)]
async def get_file_list(self) -> list[str]:
async def get_file_list(self, update: bool = False) -> list[str]:
"""
Gather a list of all files belonging to the source.
This list will be send to the server. When the server searches, this
list will be searched.
:param update: If true, regenerates caches
:type: bool
:return: List of filenames belonging to the source
:rtype: list[str]
"""
return []
async def get_config(self) -> dict[str, Any] | list[dict[str, Any]]:
async def get_config(self, update: bool = False) -> dict[str, Any] | list[dict[str, Any]]:
"""
Return the part of the config, that should be send to the server.
@ -399,14 +401,16 @@ class Source(ABC):
But this can be any other values, as long as the respective source can
handle that data.
:param update: If true, forces an update of caches
:type update: bool
:return: The part of the config, that should be sended to the server.
:rtype: dict[str, Any] | list[dict[str, Any]]
"""
if not self._index:
if update or not self._index:
self._index = []
print(f"{self.source_name}: generating index")
self._index = await self.get_file_list()
print(f"{self.source_name}: done")
# print(f"{self.source_name}: generating index")
self._index = await self.get_file_list(update)
# print(f"{self.source_name}: done")
chunked = zip_longest(*[iter(self._index)] * 1000, fillvalue="")
return [{"index": list(filter(lambda x: x != "", chunk))} for chunk in chunked]

View file

@ -87,7 +87,7 @@ class YoutubeSource(Source):
}
)
async def get_config(self) -> dict[str, Any] | list[dict[str, Any]]:
async def get_config(self, update: bool = False) -> dict[str, Any] | list[dict[str, Any]]:
"""
Return the list of channels in a dictionary with key ``channels``.