ninya9k
/
anonstream
1
1
フォーク 0
コード イシュー プルリクエスト プロジェクト リリース アクティビティ

Control socket: view and delete eyes

このコミットが含まれているのは:
n9k 2022-06-14 08:53:37 +00:00
親 70c5836ed0
コミット 0cb2f226d7
3個のファイルの変更、61行の追加、25行の削除
統計情報を表示 Patchファイルをダウンロード Diffファイルをダウンロード すべてのファイルを展開 すべてのファイルを折り畳む

1
anonstream/control/commands/__init__.py
ファイルの表示

@ -37,6 +37,7 @@ METHOD_COMMAND_FUNCTIONS = {
'attr': command_user_attr,
'get': command_user_get,
'set': command_user_set,
'eyes': command_user_eyes,
},
}

26
anonstream/control/commands/help.py
ファイルの表示

@ -7,19 +7,19 @@ async def command_help(args):
response = (
'Usage: METHOD [COMMAND | help]\n'
'Examples:\n'
' help.......................show this help message\n'
' exit.......................close the control connection\n'
' title [show]...............show the stream title\n'
' title set TITLE............set the stream title\n'
' user [show]................show a list of users\n'
' user attr USER.............set an attribute of a user\n'
' user get USER ATTR.........set an attribute of a user\n'
' user set USER ATTR VALUE...set an attribute of a user\n'
#' user kick USERS [FAREWELL].kick users\n'
#' user eyes USER [show]......show a list of active video responses\n'
#' user eyes USER delete IDS..kill a set of video responses\n'
#' chat show MESSAGES.........show a list of messages\n'
' chat delete SEQS...........delete a set of messages\n'
' help...........................show this help message\n'
' exit...........................close the control connection\n'
' title [show]...................show the stream title\n'
' title set TITLE................set the stream title\n'
' user [show]....................show a list of users\n'
' user attr USER.................set an attribute of a user\n'
' user get USER ATTR.............set an attribute of a user\n'
' user set USER ATTR VALUE.......set an attribute of a user\n'
#' user kick USERS [FAREWELL].....kick users\n'
' user eyes USER [show]..........show a list of active video responses\n'
' user eyes USER delete EYES_ID..end a video response\n'
#' chat show MESSAGES.............show a list of messages\n'
' chat delete SEQS...............delete a set of messages\n'
)
case [*garbage]:
raise Garbage(garbage)

59
anonstream/control/commands/user.py
ファイルの表示

@ -17,16 +17,19 @@ async def command_user_help(args):
response = (
'Usage: user [show | attr USER | get USER ATTR | set USER ATTR VALUE]\n'
'Commands:\n'
' user [show]...........show all users\' tokens\n'
' user attr USER........show names of a user\'s attributes\n'
' user get USER ATTR....show an attribute of a user\n'
' user set USER ATTR....set an attribute of a user\n'
' user [show].......................show all users\' tokens\n'
' user attr USER....................show names of a user\'s attributes\n'
' user get USER ATTR................show an attribute of a user\n'
' user set USER ATTR................set an attribute of a user\n'
' user eyes USER [show].............show a user\'s active video responses\n'
' user eyes USER delete EYES_ID.....end a video response to a user\n'
'Definitions:\n'
#' USER..................={token TOKEN | hash HASH}\n'
' USER..................=token TOKEN\n'
' TOKEN.................a token\n'
#' HASH..................a token hash\n'
' ATTR..................a user attribute, re:[a-z0-9_]+\n'
#' USER..............................={token TOKEN | hash HASH}\n'
' USER..............................=token TOKEN\n'
' TOKEN..............................a token\n'
#' HASH..............................a token hash\n'
' ATTR...............................a user attribute, re:[a-z0-9_]+\n'
' EYES_ID............................a user\'s eyes_id, base 10 integer\n'
)
case [*garbage]:
raise Garbage(garbage)
@ -49,7 +52,7 @@ async def command_user_attr(args):
try:
token = json.loads(token_json)
except json.JSONDecodeError:
raise BadArgument('could not decode token as json')
raise BadArgument('could not decode TOKEN as json')
try:
user = USERS_BY_TOKEN[token]
except KeyError:
@ -66,7 +69,7 @@ async def command_user_get(args):
try:
token = json.loads(token_json)
except json.JSONDecodeError:
raise BadArgument('could not decode token as json')
raise BadArgument('could not decode TOKEN as json')
try:
user = USERS_BY_TOKEN[token]
except KeyError:
@ -93,7 +96,7 @@ async def command_user_set(args):
try:
token = json.loads(token_json)
except json.JSONDecodeError:
raise BadArgument('could not decode token as json')
raise BadArgument('could not decode TOKEN as json')
try:
user = USERS_BY_TOKEN[token]
except KeyError:
@ -116,3 +119,35 @@ async def command_user_set(args):
case [*garbage]:
raise Garbage(garbage)
return normal_options, response
async def command_user_eyes(args):
match args:
case ['token', token_json, *subargs]:
try:
token = json.loads(token_json)
except json.JSONDecodeError:
raise BadArgument('could not decode TOKEN as json')
try:
user = USERS_BY_TOKEN[token]
except KeyError:
raise Failed(f"no user exists with token {token!r}, try 'user show'")
match subargs:
case [] | ['show']:
normal_options = ['eyes', 'token', json_dumps_contiguous(token), 'show']
response = json.dumps(user['eyes']['current']) + '\n'
case ['delete', eyes_id_json]:
try:
eyes_id = json.loads(eyes_id_json)
except json.JSONDecodeError:
raise BadArgument('could not decode EYES_ID as json')
try:
user['eyes']['current'].pop(eyes_id)
except KeyError:
pass
normal_options = ['eyes', 'token', json_dumps_contiguous(token), 'delete', json_dumps_contiguous(eyes_id)]
response = ''
case []:
raise Incomplete
case [*garbage]:
raise Garbage(garbage)
return normal_options, response