import bs4
import itertools
import time
import requests
import configparser
import urllib.parse
import mechanize
import pushover
import datetime
import os

def split_notify(string):
    values = string.split(':')
    return (values[0], values[1:])


def main():
    while True:
        pushover.init(os.environ['PUSHOVER_KEY'])
        if 'HEALTHCHECK_URL' in os.environ:
            requests.get(f"{os.environ['HEALTHCHECK_URL']}/start")
        allinfo = []
        users = list(map(lambda x: x.split(':', 1), os.environ['BIB_USERS'].split(',')))
        notify = dict(map(split_notify, os.environ['NOTIFY_USERS'].split(',')))
        for user, pwd in users:
            allinfo += check(user, pwd, notify.get(user, []))
        if 'HEALTHCHECK_URL' in os.environ:
            requests.post(f"{os.environ['HEALTHCHECK_URL']}", data='\n'.join(allinfo).encode('utf8'))
        if 'RUN_FOREVER' in os.environ and os.environ['RUN_FOREVER'] == 'False':
            break

        now = datetime.datetime.utcnow
        to = (now() + datetime.timedelta(days = 1)).replace(hour=6, minute=0, second=0)
        time.sleep((to-now()).seconds)

def check(username, password, notify_ids):
    br = mechanize.Browser()
    starturl = 'https://ssl.muenchen.de/aDISWeb/app?service=direct/0/Home/$DirectLink&sp=SOPAC'
    response = br.open(starturl)
    br.follow_link(text_regex=r"Anmelden")
    br.select_form('Form0')
    br['$Textfield'] = username
    br['$Textfield$0'] = password
    response = br.submit()
    br.follow_link(text_regex=r"Konto")
    try:
        response = br.follow_link(text_regex=r"Ausleihen? zeigen")
        br.select_form('Form0')
        response = br.submit(name='textButton$0', label='Alle verlängern')
        lentlist = bs4.BeautifulSoup(response.read(), 'html.parser')
        table = lentlist.select('table[class="rTable_table"]')[0]
        allinfo = []
        for entry in table.tbody.select('tr'):
            info = list(map(lambda x: str(x.text).strip(), entry.select('td')))
            date = datetime.datetime.strptime(info[1], '%d.%m.%Y')
            delta = date - datetime.datetime.now()
            allinfo.append(str(info))
            if delta.days <= 10 or delta.days == 20 or delta.days == 15:
                for client in itertools.join(notify_ids, os.environ['PUSHOVER_CLIENTS'].split(',')):
                    pushover.Client(client).send_message(f'Bitte an {info[3]} denken, Abgabe {info[3]} - {username}', title="Erinnerung")
    except (StopIteration, mechanize._mechanize.LinkNotFoundError) as e:
        for client in itertools.join(notify_ids, os.environ['PUSHOVER_CLIENTS'].split(',')):
            pushover.Client(client).send_message(f'nichts ausgeliehen {username} ({e})')
        return []
    return allinfo

if __name__ == "__main__":
    main()