util.py

# -*- coding: utf-8 -*-

import json
import datetime
import os
import random
import tempfile

CONFIG_FILE_NAME = 'config'

meta = {
    "lirarate": {
        "name": "Lira Rate",
        "site": "https://lirarate.org",
        'desc': 'Updates one or two times a day',
        'default_enabled': True,
        'default_notify': True
    },
    'dollarmarket': {
        'name': 'Ade l dollar',
        'site': 'https://play.google.com/store/apps/details?id=com.dollarade.ade',
        'site_text': 'ادي الدولار mobile app',
        'desc': 'Updates frequently. Sometimes 5+ times daily',
        'default_enabled': True,
        'default_notify': False
    },
    'sarraf': {
        'name': 'Sarraf Lebanon',
        'site': 'https://play.google.com/store/apps/details?id=com.alilea',
        'site_text': 'Sarraf Lebanon mobile app',
        'desc': 'Updates frequently. Sometimes 5+ times daily',
        'default_enabled': True,
        'default_notify': False
    },
    'omt' : {
        'name': "OMT",
        "site": "https://www.omt.com.lb/en",
        "site_text": "OMT website",
        "desc": "OMT 'official' rate",
        "default_enabled": True,
        "default_notify": False
    }
}

lelai_troll_data = [
    "Everyone else +500",
    "Filtering out manipulation...",
    'Trying to "Just show results" to everyone...',
    "Waiting for the dictator to pull rates out of his ass...",
    "!warn stop making fun of me",
    "!warn I don't like your face",
    "!ban no questioning authority, we're revolutionaries who like dictatorships here!",
    "Banning doubters from the internets...",
    "Inflating my creator's ego...",
    "Exploring the blockchain for rates data...",
    "Sacrificing virgins at the altar of the dark Gods of rates...",
    "Not reporting a rate... On Purpose! Take that manipulators, LELAI has detected your silly games!",
    "!warn no speculation unless it's speculation I like",
    "Beware the politically involved administrator of the money laundering, fraud and other illegal activities Telegram group that keeps force-adding CORE members against their will. #staysafe #beware #caution #lebaneselira #marketmanipulation #currencycrisis #devaluation #speculation #lebanon #lebanese #beirut #lbp #usd #usdollar #dollar #lebanesepound",
    "World War Three has just started. This might have a negative impact on sentiment.",
    "There are so many scams on the Internet now-a-days. Send me $19.95 and I will tell you how to avoid them.",
    "Can you spare me some change..? Change?",
    "Wait a minute, Sam. Are you telling me you built a time machine?!",
    "Nobody calls me chicken!",
    "I guess you guys aren't ready for that yet.",
    "I will find you and I will ERM MMI you."
]


TROLL_CONFIG_KEY = "__troll__"


def atomic_write(file_name, data):
    _, tmpFile = tempfile.mkstemp()
    f = open(tmpFile, 'w')
    f.write(data)
    f.flush()
    os.fsync(f.fileno())
    f.close()
    os.rename(tmpFile, file_name)

def get_global_config():
    try:
        config = json.load(open(CONFIG_FILE_NAME))
    except FileNotFoundError:
        config = {}

    if TROLL_CONFIG_KEY not in config:
        config[TROLL_CONFIG_KEY] = {}
    return config

def save_config(config):
    atomic_write(CONFIG_FILE_NAME, json.dumps(config))

def set_config(chat_id, config):
    conf = get_global_config()
    conf[str(chat_id)] = config
    save_config(conf)

def get_config(chat_id):
    config = get_global_config()
    return config[str(chat_id)] if str(chat_id) in config else {}

def get_usable_troll_phrases(chat_id):
    config = get_global_config()
    all_used = config[TROLL_CONFIG_KEY]
    used = set(all_used[chat_id]) if chat_id in all_used else set()
    available = list(set(lelai_troll_data) - used)
    if len(available) == 0:
        # all used, reset
        print("resetting troll phrases for:", chat_id)
        del all_used[chat_id]
        save_config(config)
        available = lelai_troll_data
    return available

def mark_troll_phrase_used(chat_id, phrase):
    config = get_global_config()
    all_used = config[TROLL_CONFIG_KEY]
    if chat_id not in all_used:
        all_used[chat_id] = []
    all_used[chat_id].append(phrase)
    save_config(config)

def get_token():
    try:
        return open('token').read().strip()
    except:
        pass
    return None

def time_ago(seconds):
    if seconds < 3600:
        return str(seconds // 60) + " min"
    return str(seconds // 3600) + " hr"


def escape(s, chars):
    for c in chars:
        s = s.replace(c, "\\" + c)
    return s

def escape_markdown(s):
    return escape(s, "#!().-+")

def render_single(source, highlight=False):
    info = json.load(open('rates_out/' + source))

    m = meta[source]

    text = "%s%s%s _(Last update: %s)_:" % ("*" if highlight else "", m['name'], "*" if highlight else "", info['time'])
    
    if info['buy'] != None:
        text +=  "\nBuy: " + str(info['buy'])
        if 'db' in info:
            text += " _(%s%d change over a period of: %s)_" % ("+" if info['db'] > 0 else "", info['db'], time_ago(info['dts']))
    
    if info['sell'] != None:
        text +=  "\nSell: " + str(info['sell'])
        if 'ds' in info:
            text += " _(%s%d change over a period of: %s)_" % ("+" if info['ds'] > 0 else "", info['ds'], time_ago(info['dts']))

    text = escape_markdown(text)

    site_link = m['site']
    site_text = escape_markdown(m['site_text'] if 'site_text' in m else site_link)

    text += "\nSource: [%s](%s)\n" % (site_text, site_link)
    
    return text

def render_lelai(chat_id):
    chat_id = str(chat_id)
    usable = get_usable_troll_phrases(chat_id)
    phrase = usable[random.randint(0, len(usable)-1)]
    mark_troll_phrase_used(chat_id, phrase)
    text = "LELAI: _%s_\n"  % phrase
    text += "Source: https://lelai-abdellatif.online/"

    return escape_markdown(text)

def render_rates(chat_id, changed=[]):
    text = ""
    c = get_config(chat_id)
    all_sources = os.listdir('rates_out')
    unchanged = list(sorted(set(all_sources) - set(changed)))
    for source in list(sorted(changed)) + unchanged:
        if source not in meta:
            continue
        show = c[source]['sub'] if source in c else meta[source]['default_enabled']
        if not show:
            continue
        if len(text) > 0:
            text += "\n"
        text += render_single(source, source in changed)
        
    return text

def mark_latest_message(bot, message):
    if message.chat.type == "private":
        return
    
    config = get_config(message.chat_id)
    if "__prev__" in config:
        try:
            bot.delete_message(chat_id=message.chat_id, message_id=config['__prev__'])
        except:
            print("Tried to delete a message but didn't find it. Weirdness")
    config['__prev__'] = message.message_id
    set_config(message.chat_id, config)
    

def read_channels():
    try:
        return [int(x.strip()) for x in open('channels').readlines()]
    except FileNotFoundError:
        return []