46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
import base64
|
|
import hashlib
|
|
import hmac
|
|
from time import time
|
|
|
|
from chalicelib.core import assist
|
|
from chalicelib.utils import helper
|
|
from chalicelib.utils.helper import environ
|
|
|
|
|
|
def __get_secret():
|
|
return environ["assist_secret"] if environ["assist_secret"] is not None and len(
|
|
environ["assist_secret"]) > 0 else None
|
|
|
|
|
|
def get_temporary_credentials():
|
|
secret = __get_secret()
|
|
if secret is None:
|
|
return {"errors": ["secret not defined"]}
|
|
user = helper.generate_salt()
|
|
ttl = int(environ.get("assist_ttl", 48)) * 3600
|
|
timestamp = int(time()) + ttl
|
|
username = str(timestamp) + ':' + user
|
|
dig = hmac.new(bytes(secret, 'utf-8'), bytes(username, 'utf-8'), hashlib.sha1)
|
|
dig = dig.digest()
|
|
credential = base64.b64encode(dig).decode()
|
|
return {'username': username, 'credential': credential}
|
|
|
|
|
|
def get_full_config():
|
|
servers = assist.get_ice_servers()
|
|
if servers is None:
|
|
return None
|
|
servers = servers.split("|")
|
|
credentials = get_temporary_credentials()
|
|
if __get_secret() is not None:
|
|
servers = [{"url": s.split(",")[0], **credentials} for s in servers]
|
|
else:
|
|
for i in range(len(servers)):
|
|
s = servers[i].split("|")
|
|
if len(s) == 3:
|
|
servers[i] = {"url": s[0], "username": s[1], "credential": s[2]}
|
|
else:
|
|
servers[i] = {"url": s[0]}
|
|
|
|
return servers
|