55 行
1.5 KiB
Python
55 行
1.5 KiB
Python
|
import hashlib
|
||
|
|
import base64
|
||
|
|
from collections import OrderedDict
|
||
|
|
from math import inf
|
||
|
|
|
||
|
|
from quart import current_app
|
||
|
|
|
||
|
|
CONFIG = current_app.config
|
||
|
|
|
||
|
|
def generate_token_hash(token):
|
||
|
|
parts = CONFIG['SECRET_KEY'] + b'token-hash\0' + token.encode()
|
||
|
|
digest = hashlib.sha256(parts).digest()
|
||
|
|
return base64.b32encode(digest)[:26].lower().decode()
|
||
|
|
|
||
|
|
def generate_user(secret, token, broadcaster, timestamp):
|
||
|
|
return {
|
||
|
|
'token': token,
|
||
|
|
'token_hash': generate_token_hash(token),
|
||
|
|
'broadcaster': broadcaster,
|
||
|
|
'name': None,
|
||
|
|
'color': '#c7007f',
|
||
|
|
'tripcode': None,
|
||
|
|
'notices': OrderedDict(),
|
||
|
|
'seen': {
|
||
|
|
'first': timestamp,
|
||
|
|
'last': timestamp,
|
||
|
|
},
|
||
|
|
'watching_last': -inf,
|
||
|
|
}
|
||
|
|
|
||
|
|
def get_default_name(user):
|
||
|
|
return (
|
||
|
|
CONFIG['DEFAULT_HOST_NAME']
|
||
|
|
if user['broadcaster'] else
|
||
|
|
CONFIG['DEFAULT_ANON_NAME']
|
||
|
|
)
|
||
|
|
|
||
|
|
def is_watching(timestamp, user):
|
||
|
|
return user['watching_last'] >= timestamp - CONFIG['THRESHOLD_IDLE']
|
||
|
|
|
||
|
|
def is_idle(timestamp, user):
|
||
|
|
return is_present(timestamp, user) and not is_watching(timestamp, user)
|
||
|
|
|
||
|
|
def is_present(timestamp, user):
|
||
|
|
return user['seen']['last'] >= timestamp - CONFIG['THRESHOLD_ABSENT']
|
||
|
|
|
||
|
|
def is_absent(timestamp, user):
|
||
|
|
return not is_present(timestamp, user)
|
||
|
|
|
||
|
|
def is_visible(timestamp, messages, user):
|
||
|
|
has_visible_messages = any(
|
||
|
|
message['token'] == user['token'] for message in messages
|
||
|
|
)
|
||
|
|
return has_visible_messages or is_present(timestamp, user)
|