import os
import re
import shutil
import tempfile
import unittest
from unittest import mock

from patchbay_web import create_app

ED25519 = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINJ2M1DCOTYJkGHaPO2iMGqF7isZXmwqN5HxtLP5+xq1 test@host"


class AppCase(unittest.TestCase):
    """Fresh config + database per test; mails are captured instead of sent."""

    def setUp(self):
        self.dir = tempfile.mkdtemp(prefix="pbtest.")
        self.conf_path = os.path.join(self.dir, "patchbay.conf")
        with open(self.conf_path, "w") as f:
            f.write("# test config\nRole = target\n"
                    f"Database = {self.dir}/patchbay.db\nRunDir = {self.dir}\n"
                    "SysopUser = admin\nSysopEmail = admin@example.org\n"
                    "SysopPassword = correct horse battery\nMailHost = smtp.invalid\n")
        self.mails = []
        patcher = mock.patch("patchbay_web.mailer.send",
                             side_effect=lambda conf, to, subj, body: self.mails.append((to, body)))
        patcher.start()
        self.addCleanup(patcher.stop)
        self.app = create_app(self.conf_path, testing=True)
        self.client = self.app.test_client()

    def tearDown(self):
        shutil.rmtree(self.dir)

    def csrf(self):
        page = self.client.get("/login").get_data(as_text=True)
        m = re.search(r'name="csrf-token" content="([^"]+)"', page)
        return m.group(1)

    def login(self, user="admin", password="correct horse battery"):
        tok = self.csrf()
        r = self.client.post("/login", data={"username": user, "password": password, "csrf_token": tok})
        if r.status_code != 302:
            return r
        code = re.search(r"code is: (\d{6})", self.mails[-1][1]).group(1)
        page = self.client.get("/verify").get_data(as_text=True)
        tok = re.search(r'name="csrf-token" content="([^"]+)"', page).group(1)
        return self.client.post("/verify", data={"code": code, "csrf_token": tok})

    def token_from(self, path):
        page = self.client.get(path).get_data(as_text=True)
        return re.search(r'name="csrf-token" content="([^"]+)"', page).group(1)

    def token(self):
        return self.token_from("/settings")