mirror of
https://github.com/LibreTranslate/LibreTranslate.git
synced 2024-11-21 15:31:00 +00:00
121 lines
3.3 KiB
Python
121 lines
3.3 KiB
Python
import base64
|
|
import random
|
|
import string
|
|
from functools import lru_cache
|
|
|
|
from libretranslate.storage import get_storage
|
|
|
|
|
|
def to_base(n, b):
|
|
if n == 0:
|
|
return 0
|
|
if n < 0:
|
|
sign = -1
|
|
else:
|
|
sign = 1
|
|
|
|
n *= sign
|
|
digits = []
|
|
while n:
|
|
digits.append(str(n % b))
|
|
n //= b
|
|
return int(''.join(digits[::-1])) * sign
|
|
|
|
@lru_cache(maxsize=4)
|
|
def obfuscate(input_str):
|
|
encoded = [ord(ch) for ch in input_str]
|
|
ops = ['+', '-', '*', '']
|
|
parts = []
|
|
|
|
for c in encoded:
|
|
num = random.randint(1, 100)
|
|
op = random.choice(ops)
|
|
if op == '+':
|
|
v = c + num
|
|
op = '-'
|
|
elif op == '-':
|
|
v = c - num
|
|
op = '+'
|
|
if random.randint(0, 1) == 0:
|
|
op = '+false+'
|
|
elif op == '*':
|
|
v = c * num
|
|
op = '/'
|
|
if random.randint(0, 1) == 0:
|
|
op = '/**\\/*//'
|
|
|
|
use_dec = random.randint(0, 1) == 0
|
|
base = random.randint(4, 7)
|
|
|
|
if op == '':
|
|
if use_dec:
|
|
parts.append(f'_({c})')
|
|
else:
|
|
parts.append(f'_(p({to_base(c, base)},{base}))')
|
|
else:
|
|
if use_dec:
|
|
parts.append(f'_({v}{op}{num})')
|
|
else:
|
|
parts.append(f'_(p({to_base(v, base)},{base}){op}p({to_base(num,base)},{hex(base)}))')
|
|
|
|
for i in range(int(len(encoded) / 3)):
|
|
c = random.randint(1, 100)
|
|
parts.insert(random.randint(0, len(parts)), f"_(/*_({c})*/)")
|
|
for i in range(int(len(encoded) / 3)):
|
|
parts.insert(random.randint(0, len(parts)), f"\n[]\n")
|
|
|
|
code = '(_=String.fromCharCode,p=parseInt,' + '+'.join(parts) + ')'
|
|
return code
|
|
|
|
def generate_secret():
|
|
return ''.join(random.choices(string.ascii_uppercase + string.digits, k=7))
|
|
|
|
def rotate_secrets():
|
|
s = get_storage()
|
|
secret_1 = s.get_str("secret_1")
|
|
s.set_str("secret_0", secret_1)
|
|
s.set_str("secret_1", generate_secret())
|
|
|
|
def secret_match(secret):
|
|
s = get_storage()
|
|
return secret == s.get_str("secret_0") or secret == s.get_str("secret_1")
|
|
|
|
def secret_bogus_match(secret):
|
|
if random.randint(0, 1) == 0:
|
|
return secret == get_bogus_secret()
|
|
return False
|
|
|
|
def get_current_secret():
|
|
return get_storage().get_str("secret_1")
|
|
|
|
def get_current_secret_b64():
|
|
return base64.b64encode(get_current_secret().encode("utf-8")).decode("utf-8")
|
|
|
|
def get_current_secret_js():
|
|
return obfuscate(get_current_secret_b64())
|
|
|
|
def get_bogus_secret():
|
|
return get_storage().get_str("secret_bogus")
|
|
|
|
def get_bogus_secret_b64():
|
|
return base64.b64encode(get_bogus_secret().encode("utf-8")).decode("utf-8")
|
|
|
|
def get_bogus_secret_js():
|
|
return obfuscate(get_bogus_secret_b64())
|
|
|
|
@lru_cache(maxsize=1)
|
|
def get_emoji():
|
|
return random.choice(["😂", "🤪", "😜", "🤣", "😹", "🐒", "🙈", "🤡", "🥸", "😆", "🥴", "🐸", "🐤", "🐒🙊", "👀", "💩", "🤯", "😛", "🤥", "👻"])
|
|
|
|
def setup(args):
|
|
if args.api_keys and args.require_api_key_secret:
|
|
s = get_storage()
|
|
|
|
if not s.exists("secret_0"):
|
|
s.set_str("secret_0", generate_secret())
|
|
|
|
if not s.exists("secret_1"):
|
|
s.set_str("secret_1", generate_secret())
|
|
|
|
if not s.exists("secret_bogus"):
|
|
s.set_str("secret_bogus", generate_secret())
|