import io
import os
import subprocess

from PyQt5.QtWidgets import QInputDialog, QLineEdit

from bauh.view.qt.dialog import show_error


def is_root():
    return os.getuid() == 0


def ask_root_password(locale_keys: dict):

    dialog_pwd = QInputDialog()
    dialog_pwd.setInputMode(QInputDialog.TextInput)
    dialog_pwd.setTextEchoMode(QLineEdit.Password)
    dialog_pwd.setWindowTitle(locale_keys['popup.root.title'])
    dialog_pwd.setLabelText(locale_keys['popup.root.password'] + ':')
    dialog_pwd.setCancelButtonText(locale_keys['popup.button.cancel'])
    dialog_pwd.resize(400, 200)

    ok = dialog_pwd.exec_()

    if ok:
        if not validate_password(dialog_pwd.textValue()):
            show_error(title=locale_keys['popup.root.bad_password.title'],
                       body=locale_keys['popup.root.bad_password.body'])
            ok = False

    return dialog_pwd.textValue(), ok


def validate_password(password: str) -> bool:
    proc = subprocess.Popen('sudo -k && echo {} | sudo -S whoami'.format(password),
                            shell=True,
                            stdout=subprocess.PIPE,
                            stderr=subprocess.DEVNULL,
                            bufsize=-1)
    stream = os._wrap_close(io.TextIOWrapper(proc.stdout), proc)
    res = stream.read()
    stream.close()

    return bool(res.strip())