srevinsaju/guiscrcpy

"""
GUISCRCPY by srevinsaju
Get it on : https://github.com/srevinsaju/guiscrcpy
Licensed under GNU Public License

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""

import logging
import os
import shlex
import shutil
import sys
import subprocess
from ..logging import make_logger

from qtpy.QtGui import QPixmap
from colorama import Fore
from qtpy import QtCore
from qtpy.QtWidgets import QMessageBox
from ..platform.platform import System

environment = System()

COLORS = {
    "g": Fore.GREEN,
    "rst": Fore.RESET,
    "y": Fore.YELLOW,
    "r": Fore.RED,
    "b": Fore.BLUE,
    "B": Fore.LIGHTBLUE_EX,
    "x": Fore.LIGHTBLACK_EX,
}


def log(*args, **kwargs):
    if os.getenv("GUISCRCPY_DEBUG", False):
        print(*args, **kwargs)
    else:
        logging.debug(str(args))


def decode_process(process):
    try:
        output = process.stdout.readlines()
        for i in range(len(output)):
            output[i] = output[i].decode("utf-8")
    except NameError:
        logging.error("No stdout in process.")
        output = ""
    return output


def check_existence(paths, filename="", directory=True, path=False):
    for i in paths:
        j = os.path.expanduser(i)
        if os.path.exists(j):  # directory exists
            if directory and os.path.isdir(j):
                return [j]
            else:
                if environment.system() == "Windows":
                    append = ".exe"
                else:
                    append = ""

                if (isinstance(filename, list)) or (isinstance(filename, tuple)):
                    for exe in filename:
                        if os.path.exists(os.path.join(j, exe + append)):
                            return [os.path.join(j, exe + append)]
                else:

                    if os.path.exists(os.path.join(j, filename + append)):
                        return [os.path.join(j, filename + append)]
        else:
            logging.debug("{} doesn't exist".format(i))

    if path:
        new_paths = os.getenv("PATH").split(os.pathsep)
        found_path = check_existence(
            new_paths, filename=filename, directory=directory, path=False
        )
        if found_path:
            return found_path + ["path"]
        else:
            return False
    else:
        return False


def get_self():
    """
    Returns the path to the running executable depending on the conditions
    :return:
    :rtype:
    """
    if os.getenv("APPIMAGE"):
        # Running from AppImage
        return os.getenv("APPIMAGE")
    elif getattr(sys, "frozen", False):
        # running in precompiled bundle
        return sys.executable
    elif shutil.which("guiscrcpy"):
        # guiscrcpy is added to PATH
        return shutil.which("guiscrcpy")
    elif any([os.path.exists(os.path.join(x, "guiscrcpy")) for x in sys.path]) and any(
        [sys.executable.endswith(py) for py in ["python", "python3"]]
    ):
        # guiscrcpy is installed as a pip package, but not added to PATH
        return "{py} -m guiscrcpy".format(py=sys.executable)
    elif os.getenv("SNAP"):
        # running from SNAP
        return "/snap/bin/guiscrcpy"
    raise RuntimeError(
        "Could not detect if guiscrcpy was run from " "snap, appimage or python wheel"
    )


def format_colors(string, **kwargs):
    return string.format(**kwargs, **COLORS)


def show_message_box(text, info_text="", buttons=QMessageBox.Ok):
    """
    Shows a message box
    :param text:
    :param info_text:
    :param buttons:
    :return:
    """
    message_box = QMessageBox()
    try:
        message_box.setIconPixmap(
            QPixmap(":/res/ui/guiscrcpy_logo.png").scaledToHeight(100)
        )
    except Exception as e:
        print(
            "WARN: {e}: loading guiscrcpy "
            "message box pixmap failed. "
            "Ignoring".format(e=e)
        )
    message_box.setText("<b>{}</b>".format(text))
    message_box.setTextFormat(QtCore.Qt.RichText)
    message_box.setInformativeText(info_text)
    message_box.setStandardButtons(buttons)
    return message_box


process_logger = make_logger("process")


def open_process(*args, **kwargs):
    if (
        environment.system() == "Windows"
        and sys.version_info.major >= 3
        and sys.version_info.minor >= 7
    ):
        kwargs["creationflags"] = subprocess.CREATE_NO_WINDOW
    proc = subprocess.Popen(*args, **kwargs)
    process_logger.debug("Invoking process: {}".format(proc.args))
    return proc