"""Own OMEMO messages coming back from the archive (MAM) are shown decrypted, like when edited:
their local id is m_…, the browser stored the plaintext under the id it sent them with."""
import asyncio
import os
import sqlite3
import subprocess
from playwright.async_api import async_playwright
from browser import login, BASE, OUT
from browser_edit import send, edit_last, bubbles, open_chat

D = os.path.dirname(os.path.abspath(__file__))


async def main():
    async with async_playwright() as p:
        browser = await p.chromium.launch()
        actx, alice, aerr = await login(browser, "alice@localhost", "alicepass", "historyalice")
        await alice.wait_for_timeout(4000)
        await open_chat(alice, "bob@localhost")
        await alice.evaluate("ChatOmemo.enableContactState('bob@localhost', false)")
        await alice.wait_for_timeout(5000)
        await send(alice, "Lịch sử: bản gốc")
        await edit_last(alice, "Lịch sử: đã sửa")
        await send(alice, "Lịch sử: tin thứ hai")

        bctx, bob, berr = await login(browser, "bob@localhost", "bobpass", "historybob")
        await bob.wait_for_timeout(4000)
        await open_chat(bob, "alice@localhost")
        await bob.evaluate("ChatOmemo.enableContactState('alice@localhost', false)")
        await bob.wait_for_timeout(5000)
        await send(bob, "Lịch sử: trả lời của bob")
        await alice.wait_for_timeout(3000)
        print("before, alice:", await bubbles(alice, 3))
        await bctx.close()

        # The server forgets the discussion, the history comes back from MAM
        subprocess.run(["kill", open(f"{D}/server.pid").read().strip()])
        await alice.wait_for_timeout(1000)
        db = sqlite3.connect(f"{D}/data/movim.sqlite")
        for table in ("messages", "mam_earliests"):
            db.execute(f"DELETE FROM {table} WHERE user_id = 'alice@localhost'")
        db.commit()
        db.close()
        subprocess.run([f"{D}/run.sh"])

        # Same browser context: the decrypted messages of IndexedDB are kept
        page = await actx.new_page()
        await page.goto(BASE + "/login")
        await page.wait_for_timeout(3000)
        if "/login" in page.url:
            await page.wait_for_selector("#login_widget input[type=submit]:not([disabled])", timeout=15000)
            await page.fill("input#username", "alice@localhost")
            await page.fill("input#password", "alicepass")
            await page.click("#login_widget input[type=submit]")
            await page.wait_for_url("**/chat**", timeout=30000)
        await page.wait_for_timeout(4000)
        await open_chat(page, "bob@localhost")
        await page.wait_for_timeout(4000)
        print("ids:", await page.evaluate(
            "[...document.querySelectorAll('#chat_widget .bubble > div.message')].slice(-3).map(m => m.id)"))
        print("from history, alice:", await bubbles(page, 3))
        await page.screenshot(path=f"{OUT}/history_alice.png")

        for e in aerr:
            print(e)
        await browser.close()

asyncio.run(main())