ispmail_userctl.py

#!/usr/bin/python3 -EsWerror

# MIT License
#
# Copyright (c) 2019-2021 Christian Göttsche
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.


######################
#                    #
# Configuration      #
#                    #
######################

# whether to use bcrypt passwords or sha512-crypt
USE_BCRYPT: bool = False


import sys
import curses
import curses.ascii
from enum import Enum
from signal import signal, SIGWINCH
import os
import struct
from fcntl import ioctl
from termios import TIOCGWINSZ
import re
from abc import ABC, abstractmethod
from dataclasses import dataclass
from typing import Any, Callable


if not sys.stdout.isatty() or not sys.stdin.isatty():
    print('ISPMail userctl is based on curses and needs a real tty to work!')
    sys.exit(1)


NOTE: str = '\033[34;1m[*]\033[0m'
SUCC: str = '\033[32;1m[!]\033[0m'
WARN: str = '\033[33;1m[!]\033[0m'
ERR: str = '\033[31;1m[!]\033[0m'


try:
    import MySQLdb
    from MySQLdb import cursors
    from MySQLdb import connections
except ImportError:
    print(ERR + ' No MySQLdb python module found!')
    print(NOTE + '     On Debian install python3-mysqldb')
    sys.exit(1)


if USE_BCRYPT:
    try:
        import bcrypt
    except ImportError:
        print(ERR + ' No bcrypt python module found!')
        print(NOTE + '     On Debian install python3-bcrypt')
        sys.exit(1)


def fmt_yellow(msg: str) -> str:
    return '\033[1;33m%s\033[0m' % msg


def format_quota(quota: float) -> str:
    if quota == 0:
        return 'unlimited'

    if quota < 1000:
        return f'{quota:.2f} bytes'

    quota /= 1000
    if quota < 1000:
        return f'{quota:.2f} KB'

    quota /= 1000
    if quota < 1000:
        return f'{quota:.2f} MB'

    quota /= 1000
    return f'{quota:.2f} GB'


def parse_quota(quota_raw: str) -> float:
    match = re.match(r'([0-9.,]+)\s*(\w+)?', quota_raw)
    if not match or not match[1]:
        raise ValueError(f'invalid quota: \'{quota_raw}\'')

    amount = float(match[1])

    if not match[2]:
        return amount

    quantifier = match[2].casefold()

    if quantifier == 'kb':
        return 1000 * amount
    if quantifier == 'mb':
        return 1000 * 1000 * amount
    if quantifier == 'gb':
        return 1000 * 1000 * 1000 * amount

    raise ValueError(f'invalid quota quantifier: \'{quantifier}\'')


@dataclass
class DBDomain:
    identifier: str
    name: str


@dataclass
class DBUser:
    identifier: str
    domain_id: int
    email: str
    quota: float


@dataclass
class DBAlias:
    identifier: str
    domain_id: int
    source: str
    destination: str


DB_CURSOR: MySQLdb.cursors.Cursor = None
DB_CONNECTION: MySQLdb.connections.Connection = None


def db_get_domains() -> list[DBDomain]:
    DB_CURSOR.execute('SELECT id, name FROM virtual_domains ORDER BY name;')

    return [DBDomain(row[0], row[1]) for row in DB_CURSOR.fetchall()]


def db_create_domain(name: str) -> None:
    DB_CURSOR.execute('INSERT INTO virtual_domains (name) VALUES (%s);', (name,))


def db_delete_domain(domain: DBDomain) -> None:
    DB_CURSOR.execute('DELETE FROM virtual_domains where id = %s;', (domain.identifier,))


def db_get_users(domain: DBDomain | None = None) -> list[DBUser]:
    if domain:
        DB_CURSOR.execute('SELECT id, domain_id, email, quota FROM virtual_users WHERE domain_id = %s ORDER BY email;', (domain.identifier,))
    else:
        DB_CURSOR.execute('SELECT id, domain_id, email, quota FROM virtual_users ORDER BY domain_id, email;')

    return [DBUser(row[0], row[1], row[2], row[3]) for row in DB_CURSOR.fetchall()]


def db_create_user(domain: DBDomain, email: str, password: str, quota: float) -> None:
    if USE_BCRYPT:
        hashed_pw = bcrypt.hashpw(password.encode('UTF-8'), bcrypt.gensalt())
        DB_CURSOR.execute('INSERT INTO virtual_users (domain_id, email, password, quota) VALUES ( %s, %s, CONCAT("{BLF-CRYPT}", %s), %s);', (domain.identifier, email, hashed_pw, quota,))
    else:
        DB_CURSOR.execute('INSERT INTO virtual_users (domain_id, email, password, quota) VALUES ( %s, %s, CONCAT("{SHA512-CRYPT}", ENCRYPT (%s, CONCAT("$6$", SHA(RAND())))), %s);', (domain.identifier, email, password, quota,))


def db_update_password(user: DBUser, password: str) -> None:
    if USE_BCRYPT:
        hashed_pw = bcrypt.hashpw(password.encode('UTF-8'), bcrypt.gensalt())
        DB_CURSOR.execute('UPDATE virtual_users SET password=CONCAT("{BLF-CRYPT}", %s) WHERE id = %s;', (hashed_pw, user.identifier,))
    else:
        DB_CURSOR.execute('UPDATE virtual_users SET password=CONCAT("{SHA512-CRYPT}", ENCRYPT (%s, CONCAT("$6$", SHA(RAND())))) WHERE id = %s;', (password, user.identifier,))


def db_update_quota(user: DBUser, quota: float) -> None:
    DB_CURSOR.execute('UPDATE virtual_users SET quota=%s WHERE id = %s;', (quota, user.identifier,))


def db_delete_user(user: DBUser) -> None:
    DB_CURSOR.execute('DELETE FROM virtual_users WHERE id = %s;', (user.identifier,))


def db_get_aliases(domain: DBDomain | None = None) -> list[DBAlias]:
    if domain:
        DB_CURSOR.execute('SELECT id, domain_id, source, destination FROM virtual_aliases WHERE domain_id = %s ORDER BY source, destination;', (domain.identifier,))
    else:
        DB_CURSOR.execute('SELECT id, domain_id, source, destination FROM virtual_aliases ORDER BY source, destination;')

    return [DBAlias(row[0], row[1], row[2], row[3]) for row in DB_CURSOR.fetchall()]


def db_create_alias(domain: DBDomain, source: str, destination: str) -> None:
    DB_CURSOR.execute('INSERT INTO virtual_aliases (domain_id, source, destination) VALUES (%s, %s, %s);', (domain.identifier, source, destination,))


def db_delete_alias(alias: DBAlias) -> None:
    DB_CURSOR.execute('DELETE FROM virtual_aliases WHERE id = %s;', (alias.identifier,))


class GuiObject(ABC):

    @abstractmethod
    def resize(self, lines: int, cols: int) -> None:
        ...

    @abstractmethod
    def draw(self) -> None:
        ...

    @abstractmethod
    def run(self) -> None | Any:
        ...


class GuiManager(GuiObject, ABC):

    @abstractmethod
    def add(self, child: GuiObject) -> None:
        ...

    @abstractmethod
    def remove(self, child: GuiObject) -> None:
        ...


class Note(GuiObject):

    def __init__(self, parent: GuiManager, screen: curses.window, title: str, top_title: str, text: str, continue_text: str = 'ok'):
        self.window = screen.derwin(0, 0)
        self.window.keypad(True)

        self.parent = parent
        self.full_title = '%s -> %s' % (top_title, title,) if top_title else title
        self.text = text
        self.continue_text = continue_text

    def resize(self, lines: int, cols: int) -> None:
        self.window.resize(lines, cols)

    def draw(self) -> None:
        self.window.clear()

        self.window.addstr(1, 10, self.full_title, curses.A_BOLD)

        self.window.addstr(3, 1, self.text)

        self.window.addstr(5, 3, self.continue_text, curses.A_REVERSE)

        self.window.noutrefresh()

    def run(self) -> None:
        self.parent.add(self)

        while True:
            self.draw()
            curses.doupdate()

            key = self.window.getch()

            if key in [curses.KEY_ENTER, ord('\n'), ord('q'), ord('Q')]:
                break

        self.parent.remove(self)


class ConfirmResult(Enum):
    OPTA = 1
    OPTB = 2
    OPTNONE = 3


class Confirm(GuiObject):

    def __init__(self, parent: GuiManager, screen: curses.window, title: str, top_title: str, text: str, opta_text: str, optb_text: str):
        self.window = screen.derwin(0, 0)
        self.window.keypad(True)

        self.parent = parent
        self.full_title = '%s -> %s' % (top_title, title,) if top_title else title
        self.text = text
        self.opta_text = opta_text
        self.optb_text = optb_text
        self.opta_active = True

    def resize(self, lines: int, cols: int) -> None:
        self.window.resize(lines, cols)

    def draw(self) -> None:
        self.window.clear()

        self.window.addstr(1, 10, self.full_title, curses.A_BOLD)

        self.window.addstr(3, 1, self.text)

        if self.opta_active:
            self.window.addstr(5, 3, self.opta_text, curses.A_REVERSE)
            self.window.addstr(7, 3, self.optb_text, curses.A_NORMAL)
        else:
            self.window.addstr(5, 3, self.opta_text, curses.A_NORMAL)
            self.window.addstr(7, 3, self.optb_text, curses.A_REVERSE)

        self.window.noutrefresh()

    def run(self) -> ConfirmResult:
        self.parent.add(self)

        opt_return = ConfirmResult.OPTNONE

        while True:
            self.draw()
            curses.doupdate()

            key = self.window.getch()

            if key in [curses.KEY_ENTER, ord('\n')]:
                if self.opta_active:
                    opt_return = ConfirmResult.OPTA
                else:
                    opt_return = ConfirmResult.OPTB
                break

            elif key == curses.KEY_UP:
                self.opta_active = True

            elif key == curses.KEY_DOWN:
                self.opta_active = False

            elif key in [ord('q'), ord('Q')]:
                break

        self.parent.remove(self)

        return opt_return


class Select(GuiObject):

    def __init__(self, parent: GuiManager, screen: curses.window, title: str, top_title: str, items: list[Any]):
        self.window = screen.derwin(0, 0)
        self.window.keypad(True)

        self.parent = parent
        self.full_title = '%s -> %s' % (top_title, title,) if top_title else title
        self.position = 0
        self.items = items
        self.items.append(('Return to %s' % top_title, None))

        self.pad = curses.newpad(len(self.items) + 4, screen.getmaxyx()[1] - 2)
        self.pad.bkgd(screen.getbkgd())

    def _navigate(self, num: int) -> None:
        self.position += num
        if self.position < 0:
            self.position = 0
        elif self.position >= len(self.items):
            self.position = len(self.items) - 1

    def resize(self, lines: int, cols: int) -> None:
        self.window.resize(lines, cols)

    def draw(self) -> None:
        self.window.clear()
        self.window.noutrefresh()
        self.pad.clear()
        self.pad.addstr(1, 10, self.full_title, curses.A_BOLD)

        if len(self.items) > 1:
            for index, item in enumerate(self.items):
                mode = curses.A_REVERSE if index == self.position else curses.A_NORMAL

                # last 'return' item
                if index == len(self.items) - 1:
                    self.pad.addstr(index + 4, 1, item[0], mode)
                else:
                    msg = '%d. %s' % (index + 1, item[0])
                    self.pad.addstr(index + 3, 1, msg, mode)
        else:
            self.pad.addstr(3, 1, 'No entry to select')
            self.pad.addstr(5, 1, self.items[0][0], curses.A_REVERSE)

        padpos = self.position
        if (self.window.getmaxyx()[0] - 5) > len(self.items) - padpos:
            padpos -= (self.window.getmaxyx()[0] - 5) - (len(self.items) - padpos)

        self.pad.refresh(padpos, 0, self.window.getbegyx()[0], self.window.getbegyx()[1], self.window.getbegyx()[0] + self.window.getmaxyx()[0] - 1, self.window.getbegyx()[1] + self.window.getmaxyx()[1] - 1)

    def run(self) -> Any:
        self.parent.add(self)

        ret = None

        while True:
            self.draw()
            curses.doupdate()

            key = self.window.getch()

            if key in [curses.KEY_ENTER, ord('\n')]:
                if self.position != len(self.items) - 1:
                    ret = self.items[self.position][1]
                break

            elif key == curses.KEY_UP:
                self._navigate(-1)
            elif key == curses.KEY_DOWN:
                self._navigate(1)
            elif key == curses.KEY_NPAGE:
                self._navigate(15)
            elif key == curses.KEY_PPAGE:
                self._navigate(-15)

            elif key in [ord('q'), ord('Q')]:
                break

        self.parent.remove(self)

        return ret


class SingleInput(GuiObject):

    def __init__(self, parent: GuiManager, screen: curses.window, title: str, top_title: str, text: str, input_visible: bool):
        self.window = screen.derwin(0, 0)
        self.window.keypad(True)

        self.parent = parent
        self.top_title = top_title
        self.full_title = '%s -> %s' % (top_title, title,) if top_title else title
        self.text = text.splitlines()
        self.text_lines = len(self.text) - 1
        self.input_visible = input_visible
        self.input_string = ''
        self.input_active = True

        self.pad = curses.newpad(1, screen.getmaxyx()[1] - 2)
        self.pad.bkgd(curses.color_pair(4))
        self.input_size = self.window.getmaxyx()[1] - 9


    def resize(self, lines: int, cols: int) -> None:
        self.window.resize(lines, cols)
        self.input_size = self.window.getmaxyx()[1] - 9

    def draw(self) -> None:
        self.window.clear()

        self.window.addstr(1, 10, self.full_title, curses.A_BOLD)

        for idx, line in enumerate(self.text):
            self.window.addstr(3 + idx, 1, line)

        self.window.addstr(7 + self.text_lines, 1, 'Return to %s' % self.top_title, curses.A_NORMAL if self.input_active else curses.A_REVERSE)

        self.window.noutrefresh()

        self.pad.clear()

        txt = self.input_string if self.input_visible else '*' * len(self.input_string)
        if len(txt) >= self.input_size - 1:
            txt = txt[-(self.input_size - 1):]

        self.pad.addstr(0, 0, '> ', curses.color_pair(2))
        self.pad.addstr(0, 2, txt, curses.color_pair(3))

        self.pad.refresh(0, 0,
                         self.window.getbegyx()[0] + 5 + self.text_lines, self.window.getbegyx()[1] + 4,
                         self.window.getbegyx()[0] + 5 + self.text_lines + 1, self.window.getbegyx()[1] + self.window.getmaxyx()[1] - 4)


    def run(self) -> None | str:
        self.parent.add(self)

        while True:
            self.draw()
            curses.doupdate()

            if self.input_active:
                curses.curs_set(1)
            self.pad.move(0, min(2 + len(self.input_string), self.input_size))
            self.pad.clrtoeol()
            key = self.window.getch()
            curses.curs_set(0)

            if key in [curses.KEY_ENTER, ord('\n')]:
                break

            elif key == curses.KEY_UP:
                self.input_active = True

            elif key == curses.KEY_DOWN:
                self.input_active = False

            elif self.input_active and curses.ascii.isgraph(key):
                self.input_string += chr(key)

            elif self.input_active and key in [curses.KEY_BACKSPACE, ord('\b')]:
                if self.input_string:
                    self.input_string = self.input_string[:-1]

        self.parent.remove(self)

        return self.input_string if self.input_active else None


class Info(GuiObject):

    def __init__(self, parent: GuiManager, screen: curses.window, title: str, top_title: str, info: str):
        self.window = screen.derwin(0, 0)
        self.window.keypad(True)

        self.pad = curses.newpad(5 + info.count('\n'), screen.getmaxyx()[1] - 2)
        self.pad.bkgd(screen.getbkgd())

        self.parent = parent
        self.top_title = top_title
        self.full_title = '%s -> %s' % (top_title, title,) if top_title else title
        self.info = info
        self.pos = 0
        self.size = 0

    def _navigate(self, num: int) -> None:
        self.pos += num
        if self.pos < 0:
            self.pos = 0
        if self.pos > self.size - self.window.getmaxyx()[0] + 4:
            self.pos = self.size - self.window.getmaxyx()[0] + 4

    def resize(self, lines: int, cols: int) -> None:
        self.window.resize(lines, cols)

    def draw(self) -> None:
        self.window.clear()
        self.window.noutrefresh()
        self.pad.clear()
        self.pad.addstr(0, 9, self.full_title, curses.A_BOLD)
        self.pad.addstr(2, 0, self.info)
        self.pad.addstr('\n\nReturn to %s' % self.top_title, curses.A_REVERSE)
        self.pad.refresh(self.pos, 0, self.window.getbegyx()[0] + 1, self.window.getbegyx()[1] + 1, self.window.getbegyx()[0] + self.window.getmaxyx()[0] - 3, self.window.getbegyx()[1] + self.window.getmaxyx()[1] - 3)
        self.size = self.pad.getyx()[0]

    def run(self) -> None:
        self.parent.add(self)

        while True:
            self.draw()
            curses.doupdate()

            key = self.window.getch()

            if key in [curses.KEY_ENTER, ord('\n'), ord('q'), ord('Q')]:
                break
            elif key == curses.KEY_UP:
                self._navigate(-1)
            elif key == curses.KEY_DOWN:
                self._navigate(1)
            elif key == curses.KEY_NPAGE:
                self._navigate(15)
            elif key == curses.KEY_PPAGE:
                self._navigate(-15)

        self.parent.remove(self)


MenuItemType = list[tuple[str, Callable[..., None | bool]]]


class Menu(GuiManager):

    def __init__(self, parent: GuiManager, screen: curses.window, title: str, top_title: str | None, items: MenuItemType, *args: Any):
        self.window = screen.derwin(0, 0)
        self.window.keypad(True)

        self.parent = parent
        self.position = 0
        self.items = items
        if top_title:
            self.items.append(('Return to %s' % top_title, os.abort))
        else:
            self.items.append(('Exit and Save Changes', os.abort))
        self.full_title = '%s -> %s' % (top_title, title,) if top_title else title
        self.screen = screen
        self.args = args
        self.children: set[GuiObject] = set()

    def _navigate(self, num: int) -> None:
        self.position += num
        if self.position < 0:
            self.position = 0
        elif self.position >= len(self.items):
            self.position = len(self.items) - 1

    def add(self, child: GuiObject) -> None:
        self.children.add(child)

    def remove(self, child: GuiObject) -> None:
        self.children.remove(child)

    def resize(self, lines: int, cols: int) -> None:
        self.window.resize(lines, cols)

        for child in self.children:
            child.resize(lines, cols)

    def draw(self) -> None:
        self.window.clear()

        self.window.addstr(1, 10, self.full_title, curses.A_BOLD)

        for index, item in enumerate(self.items):
            mode = curses.A_REVERSE if index == self.position else curses.A_NORMAL

            msg = '%d. %s' % (index + 1, item[0])
            self.window.addstr(index + 3, 1, msg, mode)

        self.window.noutrefresh()

        for child in self.children:
            child.draw()

    def run(self) -> None:
        self.parent.add(self)

        while True:
            self.draw()
            curses.doupdate()

            key = self.window.getch()

            if key in [curses.KEY_ENTER, ord('\n')]:
                if self.position == len(self.items) - 1:
                    break
                else:
                    do_exit = self.items[self.position][1](self, self.screen, self.full_title, *self.args)
                    if do_exit:
                        break

            elif key in map(ord, map(str, range(1, len(self.items) + 1))):
                idx = int(chr(key)) - 1
                if idx == len(self.items) - 1:
                    break
                else:
                    do_exit = self.items[idx][1](self, self.screen, self.full_title, *self.args)
                    if do_exit:
                        break

            elif key == curses.KEY_UP:
                self._navigate(-1)

            elif key == curses.KEY_DOWN:
                self._navigate(1)

            elif key in [ord('q'), ord('Q')]:
                break

        self.parent.remove(self)


def domain_overview_win(parent: GuiManager, window: curses.window, top_title: str) -> None:
    domains = db_get_domains()
    text = 'Found %d domain(s):\n\n' % len(domains)
    for domain in domains:
        text += '\t%s\n' % domain.name

    handle = Info(parent, window, 'Domain Overview', top_title, text)
    handle.run()


def full_overview_win(parent: GuiManager, window: curses.window, top_title: str) -> None:
    domains = db_get_domains()
    users = db_get_users()
    aliases = db_get_aliases()
    text = 'Found %d domain(s):\n\n' % len(domains)
    for domain in domains:
        text += '\t%s\n' % domain.name
    text += '\nFound %d user(s):\n\n' % len(users)
    for user in users:
        text += '\t%s  --  %s quota\n' % (user.email, format_quota(user.quota))
    text += '\nFound %d alias(es):\n\n' % len(aliases)
    aliases.sort(key=lambda alias: alias.source)
    prev_source = None
    for alias in aliases:
        if any(alias.destination == user.email for user in users):
            foreign_msg = '  (internal destination email)'
        else:
            foreign_msg = '  (foreign destination email)'

        if prev_source == alias.source:
            text += '\t  -> %s%s\n' % (alias.destination, foreign_msg)
        else:
            text += '\t%s\n\t  -> %s%s\n' % (alias.source, alias.destination, foreign_msg)

        prev_source = alias.source

    handle = Info(parent, window, 'Full Overview', top_title, text)
    handle.run()


def domain_add_win(parent: GuiManager, window: curses.window, top_title: str) -> None:
    handle0 = SingleInput(parent, window, 'Add Domain', top_title, 'Enter the new domain name:', True)
    domain_name = handle0.run()
    if domain_name:
        if domain_name in [domain.name for domain in db_get_domains()]:
            handle1 = Note(parent, window, 'Add Domain Failed', top_title, 'Could not add domain \'%s\': domainname already exists.' % domain_name)
            handle1.run()
        else:
            db_create_domain(domain_name)
            handle1 = Note(parent, window, 'Add Domain Successful', top_title, 'Domain \'%s\' successfully added.' % domain_name)
            handle1.run()


def domain_selection_win(parent: GuiManager, window: curses.window, top_title: str) -> None:
    handle0 = Select(parent, window, 'Select Domain to manage', top_title, [(domain.name, domain) for domain in db_get_domains()])
    domain = handle0.run()
    if not domain:
        return

    menu_items: MenuItemType = [
        ('List users and aliases', domain_list_usersaliases_win),
        ('Change password of an user', domain_change_pw_win),
        ('Change quota of an user', domain_change_quota_win),
        ('Add user', domain_add_user_win),
        ('Add alias', domain_add_alias_win),
        ('Delete user', domain_delete_user_win),
        ('Delete alias', domain_delete_alias_win),
        ('Delete domain', domain_delete_confirm_win)
        ]
    handle1 = Menu(parent, window, 'Manage Domain \'%s\'' % domain.name, top_title, menu_items, domain)
    handle1.run()


def domain_list_usersaliases_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> None:
    users = db_get_users(domain)
    aliases = db_get_aliases(domain)
    text = '\nFound %d user(s):\n\n' % len(users)
    for user in users:
        text += '\t%s  --  %s quota\n' % (user.email, format_quota(user.quota))
    text += '\nFound %d alias(es):\n\n' % len(aliases)
    aliases.sort(key=lambda alias: alias.source)
    prev_source = None
    for alias in aliases:
        if any(alias.destination == user.email for user in users):
            foreign_msg = '  (internal destination email)'
        else:
            foreign_msg = '  (foreign destination email)'

        if prev_source == alias.source:
            text += '\t  -> %s%s\n' % (alias.destination, foreign_msg)
        else:
            text += '\t%s\n\t  -> %s%s\n' % (alias.source, alias.destination, foreign_msg)

        prev_source = alias.source

    handle = Info(parent, window, 'List of users and aliases', top_title, text)
    handle.run()


def domain_delete_user_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> None:
    handle0 = Select(parent, window, 'Select user to delete', top_title, [(user.email, user) for user in db_get_users(domain)])
    user = handle0.run()
    if not user:
        return

    handle1 = Confirm(parent, window, 'Delete User', top_title, 'Do you want to delete the user \'%s\'?' % user.email, 'no', 'yes')
    result = handle1.run()
    if result == ConfirmResult.OPTB:
        db_delete_user(user)


def domain_delete_alias_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> None:
    handle0 = Select(parent, window, 'Select alias to delete', top_title, [('%s  ->  %s' %(alias.source, alias.destination,), alias) for alias in db_get_aliases(domain)])
    alias = handle0.run()
    if not alias:
        return

    handle1 = Confirm(parent, window, 'Delete Alias', top_title, 'Do you want to delete the alias \'%s\' (to \'%s\')?' % (alias.source, alias.destination,), 'no', 'yes')
    result = handle1.run()
    if result == ConfirmResult.OPTB:
        db_delete_alias(alias)


def domain_change_pw_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> None:
    handle0 = Select(parent, window, 'Select user for password change', top_title, [(user.email, user) for user in db_get_users(domain)])
    user = handle0.run()
    if not user:
        return

    handle1 = SingleInput(parent, window, 'Password for user \'%s\' (1/2)' % user.email, top_title, 'Enter the new password:', False)
    pw1 = handle1.run()
    handle2 = SingleInput(parent, window, 'Password for user \'%s\' (2/2)' % user.email, top_title, 'Enter the new password again:', False)
    pw2 = handle2.run()

    if pw1 and pw1 == pw2:
        db_update_password(user, pw1)
        handle3 = Note(parent, window, 'Password Changed', top_title, 'Password for user \'%s\' successfully changed.' % user.email)
        handle3.run()
    else:
        handle3 = Note(parent, window, 'Password Changed Failed', top_title, 'Could not change password for user \'%s\': different new passwords.' % user.email)
        handle3.run()


def domain_change_quota_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> None:
    handle0 = Select(parent, window, 'Select user for quota change', top_title, [(user.email, user) for user in db_get_users(domain)])
    user = handle0.run()
    if not user:
        return

    handle1 = SingleInput(parent, window, 'Quota for user \'%s\'' % user.email, top_title, 'Old quota: %s\n\nEnter the new quota amount (e.g. 10MB or 0 for unlimited):' % format_quota(user.quota), True)
    quota_raw = handle1.run()
    if not quota_raw:
        return

    try:
        quota_parsed = parse_quota(quota_raw)
    except ValueError as err:
        handle2 = Note(parent, window, 'Quota Changed Failed', top_title, f'Could not change quota for user \'{user.email}\': {err}')
        handle2.run()
        return

    db_update_quota(user, quota_parsed)
    handle2 = Note(parent, window, 'Quota Changed', top_title, 'Quota for user \'%s\' successfully changed to %s.' % (user.email, format_quota(quota_parsed),))
    handle2.run()


def domain_add_user_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> None:
    handle0 = SingleInput(parent, window, 'Add User (1/4)', top_title, 'Enter the new username (the domain \'@%s\' will be appended):' % domain.name, True)
    user_name = handle0.run()
    if not user_name or user_name.find('@') != -1:
        handle1 = Note(parent, window, 'Add User Failed', top_title, 'Could not add new user: invalid username.')
        handle1.run()
        return

    if user_name + '@' + domain.name in [user.email for user in db_get_users(domain)]:
        handle2 = Note(parent, window, 'User Add Failed', top_title, 'Could not add user \'%s@%s\': username already exists.' % (user_name, domain.name,))
        handle2.run()
        return

    handle3 = SingleInput(parent, window, 'Add User (2/4)', top_title, 'Enter the new password:', False)
    pw1 = handle3.run()
    handle4 = SingleInput(parent, window, 'Add User (3/4)', top_title, 'Enter the new password again:', False)
    pw2 = handle4.run()

    if not pw1 or not pw2 or pw1 != pw2:
        handle5 = Note(parent, window, 'User Add Failed', top_title, 'Could not add new user \'%s@%s\': different new passwords.' % (user_name, domain.name,))
        handle5.run()
        return

    handle6 = SingleInput(parent, window, 'Add User (4/4)', top_title, 'Enter the new quota amount (e.g. 10MB or 0 for unlimited):', True)
    quota_raw = handle6.run()
    if quota_raw is None:
        return

    try:
        quota_parsed = parse_quota(quota_raw)
    except ValueError as err:
        handle7 = Note(parent, window, 'User Add Failed', top_title, f'Could not add quota for user \'{user_name}@{domain.name}\': {err}')
        handle7.run()
        return

    db_create_user(domain, user_name + '@%s' % domain.name, pw1, quota_parsed)
    handle8 = Note(parent, window, 'User Added Successful', top_title, 'User \'%s@%s\' successfully added.' % (user_name, domain.name,))
    handle8.run()


def domain_add_alias_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> None:
    handle0 = SingleInput(parent, window, 'Add Alias (1/2)', top_title, 'Enter the new alias source (the domain \'@%s\' will be appended):' % domain.name, True)
    source = handle0.run()
    if not source or source.find('@') != -1:
        handle1 = Note(parent, window, 'Add Alias Failed', top_title, 'Could not add new alias: invalid source.')
        handle1.run()
        return

    handle2 = SingleInput(parent, window, 'Add Alias (2/2)', top_title, 'Enter the new alias destination (supply the full address, with domain):', True)
    destination = handle2.run()
    if not destination or destination.find('@') == -1:
        handle3 = Note(parent, window, 'Add Alias Failed', top_title, 'Could not add new alias: invalid destination.')
        handle3.run()
        return

    if any(alias.source == source + '@' + domain.name and alias.destination == destination for alias in db_get_aliases(domain)):
        handle4 = Note(parent, window, 'Add Alias Failed', top_title, 'Could not add new alias: alias already exists.')
        handle4.run()
        return

    db_create_alias(domain, source + '@%s' % domain.name, destination)
    handle5 = Note(parent, window, 'Alias Added', top_title, 'Alias \'%s@%s\' to \'%s\' successfully added.' % (source, domain.name, destination,))
    handle5.run()


def domain_delete_confirm_win(parent: GuiManager, window: curses.window, top_title: str, domain: DBDomain) -> bool:
    handle = Confirm(parent, window, 'Delete Domain', top_title, 'Do you want to delete the domain \'%s\'?' % domain.name, 'no', 'yes')
    result = handle.run()

    if result == ConfirmResult.OPTB:
        db_delete_domain(domain)
        return True

    return False


def discard_changes_win(parent: GuiManager, window: curses.window, top_title: str) -> None:
    handle = Confirm(parent, window, 'Discard Changes', top_title, 'Do you want to discard all changes?', 'no', 'yes')
    if handle.run() == ConfirmResult.OPTB:
        DB_CONNECTION.rollback()


def save_changes_win(parent: GuiManager, window: curses.window, top_title: str) -> None:
    handle = Confirm(parent, window, 'Save Changes', top_title, 'Do you want to save all changes?', 'no', 'yes')
    if handle.run() == ConfirmResult.OPTB:
        DB_CONNECTION.commit()


# From https://groups.google.com/forum/#!msg/comp.lang.python/CpUszNNXUQM/QADpl11Z-nAJ
def getheightwidth() -> tuple[int, int]:
    """ getwidth() -> (int, int)

    Return the height and width of the console in characters """
    try:
        return int(os.environ["LINES"]), int(os.environ["COLUMNS"])
    except KeyError:
        height, width = struct.unpack("hhhh", ioctl(0, TIOCGWINSZ, "\000"*8))[0:2]  # type: ignore
        if not height:
            return 25, 80
        return height, width


class MainApp(GuiManager):

    header_size = 5
    header_text = 'ISPMail userctl'
    footer_size = 1
    main_margin = 2

    def __init__(self, screen: curses.window):
        lines, cols = screen.getmaxyx()
        self.header_win = screen.derwin(self.header_size, cols, 0, 0)
        self.working_win = screen.derwin(lines - self.header_size - self.footer_size, cols - 2 * self.main_margin, self.header_size, self.main_margin)
        self.working_win.bkgd(curses.color_pair(2))
        self.footer_win = screen.derwin(self.footer_size, cols, lines - self.footer_size, 0)
        self.children: set[GuiObject] = set()

        main_menu_items: MenuItemType = [
            ('List domains', domain_overview_win),
            ('List everything', full_overview_win),
            ('Add domain', domain_add_win),
            ('Manage domain', domain_selection_win),
            ('Save changes', save_changes_win),
            ('Discard changes', discard_changes_win),
            ]
        self.main_menu = Menu(self, self.working_win, 'Overview', None, main_menu_items)

    def add(self, child: GuiObject) -> None:
        self.children.add(child)

    def remove(self, child: GuiObject) -> None:
        self.children.remove(child)

    def resize(self, lines: int, cols: int) -> None:
        self.header_win.resize(self.header_size, cols)
        self.header_win.mvwin(0, 0)

        self.working_win.resize(lines - self.header_size - self.footer_size, cols - 2 * self.main_margin)
        self.working_win.mvwin(self.header_size, self.main_margin)

        self.footer_win.resize(self.footer_size, cols)
        self.footer_win.mvwin(lines - self.footer_size, 0)

        for child in self.children:
            child.resize(lines - self.header_size - self.footer_size, cols - 2 * self.main_margin)

    def draw(self) -> None:
        self.header_win.clear()

        self.header_win.addstr(int(self.header_size / 2), int(self.header_win.getmaxyx()[1] / 2) - int(len(self.header_text) / 2), self.header_text, curses.color_pair(1) | curses.A_BOLD)

        self.header_win.noutrefresh()

        self.working_win.clear()
        self.working_win.noutrefresh()

        self.footer_win.clear()
        self.footer_win.addstr(0, 7, 'Usage: (q) to return/quit, UP/DOWN to navigate')
        self.footer_win.noutrefresh()

        for child in self.children:
            child.draw()

    def run(self) -> None:
        self.draw()
        self.main_menu.run()


MAINAPP: MainApp


def resize_handler(signum: int, _: Any) -> None:
    del signum
    lines, cols = getheightwidth()
    curses.resizeterm(lines, cols)
    MAINAPP.resize(lines, cols)
    MAINAPP.draw()


def main_app(screen: curses.window) -> None:
    curses.curs_set(0)
    curses.start_color()

    curses.init_pair(1, curses.COLOR_YELLOW, curses.COLOR_BLACK)
    curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_BLUE)
    curses.init_pair(3, curses.COLOR_BLACK, curses.COLOR_GREEN)
    curses.init_pair(4, curses.COLOR_BLACK, curses.COLOR_WHITE)

    global MAINAPP
    MAINAPP = MainApp(screen)

    signal(SIGWINCH, resize_handler)

    MAINAPP.run()


def main() -> None:
    print(NOTE + ' # ISPMail userctl')

    lines, cols = getheightwidth()
    if lines < 25 or cols < 80:
        print(WARN + '   small terminal detected (%d x %d)' % (lines, cols, ))
        print(WARN + '   recommend minimum is (25 x 80)')
        print(WARN + '   app might be unstable')

    global DB_CURSOR
    global DB_CONNECTION

    try:
        DB_CONNECTION = MySQLdb.connect(
            host='localhost',
            user='root',
            # password='',
            db='mailserver',
            charset='utf8mb4')

        ## DEBUG support
        #import sqlite3
        #DB_CONNECTION = sqlite3.connect("testdb.db")

        DB_CURSOR = DB_CONNECTION.cursor()

        curses.wrapper(main_app)

    except KeyboardInterrupt:
        if DB_CURSOR:
            DB_CURSOR.close()
        if DB_CONNECTION:
            DB_CONNECTION.rollback()
            DB_CONNECTION.close()

        print(WARN + fmt_yellow(' Unsaved changes are lost!'))
        sys.exit(1)

    except MySQLdb.Error as err:
        if DB_CURSOR:
            DB_CURSOR.close()
        if DB_CONNECTION:
            DB_CONNECTION.rollback()
            DB_CONNECTION.close()

        print(ERR + ' MySQLdb error %d: %s' % (err.args[0], err.args[1],))
        print(WARN + fmt_yellow(' Unsaved changes are lost!'))
        raise

    except:
        if DB_CURSOR:
            DB_CURSOR.close()
        if DB_CONNECTION:
            DB_CONNECTION.rollback()
            DB_CONNECTION.close()

        print(ERR + ' Unexpected exception:', sys.exc_info()[1])
        print(WARN + fmt_yellow(' Unsaved changes are lost!'))
        raise

    DB_CURSOR.close()
    DB_CONNECTION.commit()
    DB_CONNECTION.close()

    print(SUCC + ' Bye..')


if __name__ == '__main__':
    main()