44 行
1.4 KiB
Python
44 行
1.4 KiB
Python
# SPDX-FileCopyrightText: 2022 n9k <https://gitler.moe/ninya9k>
|
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
import json
|
|
|
|
from anonstream.control.exceptions import CommandFailed
|
|
from anonstream.control.spec import Spec, NoParse
|
|
from anonstream.control.spec.common import Str, End, ArgsJsonString
|
|
from anonstream.control.spec.utils import get_item, json_dumps_contiguous
|
|
from anonstream.stream import get_stream_title, set_stream_title
|
|
|
|
async def cmd_title_help():
|
|
normal = ['title', 'help']
|
|
response = (
|
|
'Usage: title {show | set TITLE}\n'
|
|
'Commands:\n'
|
|
' title [show].......show the stream title\n'
|
|
' title set TITLE....set the stream title to TITLE\n'
|
|
'Definitions:\n'
|
|
' TITLE..............a json string, whitespace must be \\uXXXX-escaped\n'
|
|
)
|
|
return normal, response
|
|
|
|
async def cmd_title_show():
|
|
normal = ['title', 'show']
|
|
response = json.dumps(await get_stream_title()) + '\n'
|
|
return normal, response
|
|
|
|
async def cmd_title_set(title):
|
|
try:
|
|
await set_stream_title(title)
|
|
except OSError as e:
|
|
raise CommandFailed(f'could not set title: {e}') from e
|
|
normal = ['title', 'set', json_dumps_contiguous(title)]
|
|
response = ''
|
|
return normal, response
|
|
|
|
SPEC = Str({
|
|
None: End(cmd_title_show),
|
|
'help': End(cmd_title_help),
|
|
'show': End(cmd_title_show),
|
|
'set': ArgsJsonString(End(cmd_title_set)),
|
|
})
|