# -----------------------------------------------------------------------------
# Copyright (c) 2005-2021, PyInstaller Development Team.
#
# Distributed under the terms of the GNU General Public License (version 2
# or later) with exception for distributing the bootloader.
#
# The full license is in the file COPYING.txt, distributed with this software.
#
# SPDX-License-Identifier: (GPL-2.0-or-later WITH Bootloader-exception)
# -----------------------------------------------------------------------------
import importlib
import io
import os
import re
import struct

from PyInstaller.building import splash_templates
from PyInstaller.building.datastruct import Target, TOC
from PyInstaller.building.utils import misc, _check_guts_eq, _check_guts_toc
from PyInstaller import log as logging
from PyInstaller.archive.writers import SplashWriter
from PyInstaller.compat import is_cygwin, is_win, is_darwin
from PyInstaller.depend import bindepend
from PyInstaller.utils.hooks import exec_statement
from PyInstaller.utils.hooks.tcl_tk import TK_ROOTNAME, \
    collect_tcl_tk_files, find_tcl_tk_shared_libs

try:
    from PIL import Image as PILImage
except ImportError:
    PILImage = None

logger = logging.getLogger(__name__)

# these requirement files are checked against the current splash
# screen script. If you wish to modify the splash screen and run into
# tcl errors/bad behavior, this is a good place to start and add components
# your implementation of the splash screen might use.
splash_requirements = [
    # prepended tcl/tk binaries
    os.path.join(TK_ROOTNAME, "license.terms"),
    os.path.join(TK_ROOTNAME, "text.tcl"),
    os.path.join(TK_ROOTNAME, "tk.tcl"),
    # Used for customizable font
    os.path.join(TK_ROOTNAME, "ttk", "ttk.tcl"),
    os.path.join(TK_ROOTNAME, "ttk", "fonts.tcl"),
    os.path.join(TK_ROOTNAME, "ttk", "cursors.tcl"),
    os.path.join(TK_ROOTNAME, "ttk", "utils.tcl"),
]


class Splash(Target):
    """
    Bundles the required resources for the splash screen into a file,
    which will be included in the CArchive.

    A Splash has two outputs, one is itself and one is sored in
    splash.binaries. Both need to be passed to other build targets in
    order to enable the splash screen.
    """
    typ = 'SPLASH'

    def __init__(self, image_file, binaries, datas, **kwargs):
        """
        :param str image_file:
            A path-like object to the image to be used. Only the PNG
            file format is supported.

            .. note:: If a different file format is supplied and PIL (Pillow)
                is installed, the file will be converted automatically.

            .. note:: *Windows*: Due to the implementation, the color Magenta/
                RGB(255, 0, 255) must not be used in the image or text.

            .. note:: If PIL (Pillow) is installed and the image is bigger
                than max_img_size, the image will be resized to fit
                into the specified area.
        :param TOC binaries:
            The TOC of binaries the Analysis build target found. This
            TOC includes all extensionmodules and their dependencies.
            This is required to figure out, if the users program uses
            tkinter.
        :param TOC datas:
            The TOC of data the Analysis build target found. This TOC
            includes all data-file dependencies of the modules. This is
            required to check if all splash screen requirements can be
            bundled.

        :keyword text_pos:
            An optional 2x integer tuple that represents the origin of the
            text on the splash screen image. The origin of the text is its
            lower left corner. A unit in the respective coordinate system
            is a pixel of the image, its origin lies in the top left corner
            of the image. This parameter also acts like a switch for the
            text feature. If omitted, no text will be displayed on the
            splash screen. This text will be used to show textual progress
            in onefile mode.
        :type text_pos: Tuple[int, int]
        :keyword text_size:
            The desired size of the font. If the size argument is a
            positive number, it is interpreted as a size in points. If
            size is a negative number, its absolute value is interpreted
            as a size in pixels. Default ``12``
        :type text_size: int
        :keyword text_font:
            An optional name of a font for the text. This font must be
            installed on the user system, otherwise the system default
            font is used. If this parameter is omitted, the default font
            is also used.
        :keyword text_color:
            An optional color for the text. Either RGB HTML notation
            or color names are supported. Default: black
            (Windows: Due to a implementation issue the color magenta/
            rgb(255, 0, 255) is forbidden)
        :type text_color: str
        :keyword text_default:
            The default text which will be displayed before the extraction
            starts. Default: "Initializing"
        :type text_default: str
        :keyword full_tk:
            By default Splash bundles only the necessary files for the
            splash screen (some tk components). This options enables
            adding full tk and making it a requirement, meaning all tk
            files will be unpacked before the splash screen can be started.
            This is useful during development of the splash screen script.
            Default: ``False``
        :type full_tk: bool
        :keyword minify_script:
            The splash screen is created by executing an Tcl/Tk script.
            This option enables minimizing the script, meaning removing
            all non essential parts from the script. Default: True
        :keyword rundir:
            The folder name in which tcl/tk will be extracted at runtime.
            There should be no matching folder in your application to
            avoid conflicts. Default:  ``__splash``
        :type rundir: str
        :keyword name:
            An optional alternative filename for the .res file. If
            not specified, a name is generated.
        :type name: str
        :keyword script_name:
            An optional alternative filename for the Tcl script, that
            will be generated. If not specified, a name is generated.
        :type script_name: str
        :keyword max_img_size:
            Maximum size of the splash screen image as a tuple. If the supplied
            image exceeds this limit, it will be resized to fit the maximum
            width (to keep the original aspect ratio). This option can be
            disabled by setting it to None. Default: (760, 480)
        :type max_img_size: Tuple[int, int]
        """
        from ..config import CONF
        Target.__init__(self)

        # Splash screen is not supported on macOS. It operates in a
        # secondary thread and macOS disallows UI operations in any
        # thread other than main.
        if is_darwin:
            raise SystemExit("Splash screen is not supported on macOS.")

        # Make image path relative to .spec file
        if not os.path.isabs(image_file):
            image_file = os.path.join(CONF['specpath'], image_file)
        image_file = os.path.normpath(image_file)
        if not os.path.exists(image_file):
            raise ValueError("Image file '%s' not found" % image_file)

        # Copy all arguments
        self.image_file = image_file
        self.full_tk = kwargs.get("full_tk", False)
        self.name = kwargs.get("name", None)
        self.script_name = kwargs.get("script_name", None)
        self.minify_script = kwargs.get("minify_script", True)
        self.rundir = kwargs.get("rundir", None)
        self.max_img_size = kwargs.get("max_img_size", (760, 480))

        # text options
        self.text_pos = kwargs.get("text_pos", None)
        self.text_size = kwargs.get("text_size", 12)
        self.text_font = kwargs.get("text_font", "TkDefaultFont")
        self.text_color = kwargs.get("text_color", "black")
        self.text_default = kwargs.get("text_default", "Initializing")

        # Save the generated file separately so that it is not necessary to
        # generate the data again and again
        root = os.path.splitext(self.tocfilename)[0]
        if self.name is None:
            self.name = root + '.res'
        if self.script_name is None:
            self.script_name = root + '_script.tcl'

        if self.rundir is None:
            self.rundir = self._find_rundir(binaries + datas)

        # Internal variables
        try:
            # Do not import _tkinter at the toplevel, because on some systems
            # _tkinter will fail to load, since it is not installed.
            # This would cause a runtime error in PyInstaller, since
            # this module is imported from build_main.py, instead we just
            # want to inform the user that the splash screen feature is not
            # supported on his platform
            self._tkinter_module = importlib.import_module('_tkinter')
            self._tkinter_file = self._tkinter_module.__file__
        except ModuleNotFoundError:
            raise SystemExit("You platform does not support the splash screen"
                             " feature, since tkinter is not installed. Please"
                             " install tkinter and try again.")

        # Calculated / analysed values
        self.uses_tkinter = self._uses_tkinter(binaries)
        self.script = self.generate_script()
        self.tcl_lib, self.tk_lib = find_tcl_tk_shared_libs(self._tkinter_file)
        if is_darwin:
            # Outdated Tcl/Tk 8.5 system framework is not supported.
            # Depending on macOS version, the library path will come
            # up empty (hidden system libraries on Big Sur), or will
            # be [/System]/Library/Frameworks/Tcl.framework/Tcl
            if self.tcl_lib[1] is None or 'Library/Frameworks/Tcl.framework' \
                                          in self.tcl_lib[1]:
                raise SystemExit("The splash screen feature does not support"
                                 " macOS system framework version of Tcl/Tk.")
        # Check if tcl/tk was found
        assert all(self.tcl_lib)
        assert all(self.tk_lib)
        logger.debug("Use Tcl Library from %s and Tk From %s"
                     % (self.tcl_lib, self.tk_lib))
        self.splash_requirements = set([self.tcl_lib[0], self.tk_lib[0]]
                                       + splash_requirements)  # noqa: W503

        logger.info("Collect tcl/tk binaries for the splash screen")
        tcltk_tree = collect_tcl_tk_files(self._tkinter_file)
        if self.full_tk:
            # The user wants a full copy of tk, so make all tk files
            # a requirement
            self.splash_requirements.update(toc[0] for toc in tcltk_tree)

        self.binaries = TOC()
        if not self.uses_tkinter:
            # the users script does not use tkinter, so we need to provide
            # a TOC of all necessary files
            # add the shared libraries to the binaries
            self.binaries.append((self.tcl_lib[0], self.tcl_lib[1], 'BINARY'))
            self.binaries.append((self.tk_lib[0], self.tk_lib[1], 'BINARY'))

            # Only add the intersection of the required and the collected
            # resources or add all entries if full_tk is true
            self.binaries.extend(toc for toc in tcltk_tree
                                 if toc[0] in self.splash_requirements)

        # Check if all requirements were found
        fnames = [toc[0] for toc in (binaries + datas + self.binaries)]

        def _filter(_item):
            if _item not in fnames:
                # Item is not bundled, so warn the user about it.
                # This actually may happen on some tkinter installations
                # on which there is no license.terms file
                logger.warning("The local Tcl/Tk installation is missing"
                               " the file %s. The behavior of the splash"
                               " screen is therefore undefined and may be"
                               " unsupported." % _item)
                return False
            return True

        # Remove all files which were not found
        self.splash_requirements = set(filter(_filter,
                                              self.splash_requirements))

        # Test if the tcl/tk version is supported by the bootloader.
        self.test_tk_version()

        logger.debug("Splash Requirements: %s" % self.splash_requirements)

        self.__postinit__()

    _GUTS = (
        # input parameters
        ('image_file', _check_guts_eq),
        ('name', _check_guts_eq),
        ('script_name', _check_guts_eq),
        ('text_pos', _check_guts_eq),
        ('text_size', _check_guts_eq),
        ('text_font', _check_guts_eq),
        ('text_color', _check_guts_eq),
        ('text_default', _check_guts_eq),
        ('full_tk', _check_guts_eq),
        ('minify_script', _check_guts_eq),
        ('rundir', _check_guts_eq),
        ('max_img_size', _check_guts_eq),
        # calculated/analysed values
        ('uses_tkinter', _check_guts_eq),
        ('script', _check_guts_eq),
        ('tcl_lib', _check_guts_eq),
        ('tk_lib', _check_guts_eq),
        ('splash_requirements', _check_guts_eq),
        ('binaries', _check_guts_toc),
        # internal value
        # Check if the tkinter installation changed. This is theoretically
        # possible if someone uses two different python installations of
        # the same version
        ('_tkinter_file', _check_guts_eq),
    )

    def _check_guts(self, data, last_build):
        if Target._check_guts(self, data, last_build):
            return True

        # check if image has been modified
        if misc.mtime(self.image_file) > last_build:
            logger.info("Building %s because file %s changed",
                        self.tocbasename, self.image_file)
            return True

        return False

    def assemble(self):
        logger.info("Building Splash %s" % self.name)

        # Function to resize a given image to fit into the area
        # defined by max_img_size
        def _resize_image(_image, _orig_size):
            if PILImage:
                _w, _h = _orig_size
                _ratio_w = self.max_img_size[0] / _w
                if _ratio_w < 1:
                    # Image width exceeds limit
                    _h = int(_h * _ratio_w)
                    _w = self.max_img_size[0]

                _ratio_h = self.max_img_size[1] / _h
                if _ratio_h < 1:
                    # Image height exceeds limit
                    _w = int(_w * _ratio_h)
                    _h = self.max_img_size[1]

                # If a file is given it will be open
                if isinstance(_image, PILImage.Image):
                    _img = _image
                else:
                    _img = PILImage.open(_image)
                _img_resized = _img.resize((_w, _h))

                # Save image into a stream
                _image_stream = io.BytesIO()
                _img_resized.save(_image_stream, format='PNG')
                _img.close()
                _img_resized.close()
                _image_data = _image_stream.getvalue()
                logger.info("Resized image %s from dimensions %s to"
                            " (%d, %d)" % (self.image_file,
                                           str(_orig_size),
                                           _w, _h))
                return _image_data
            else:
                raise ValueError(
                    "The splash image dimensions (w: %d, h: %d) exceed"
                    " max_img_size (w: %d, h:%d), but the image cannot"
                    " be resized due to missing PIL.Image! Either install"
                    " the Pillow package, adjust the max_img_size, or use"
                    " an image of compatible dimensions."
                    % (_orig_size[0], _orig_size[1], self.max_img_size[0],
                       self.max_img_size[1]))

        # Open image file
        image_file = open(self.image_file, 'rb')

        # Check header of the file to identify it
        if image_file.read(8) == b'\x89PNG\r\n\x1a\n':
            # self.image_file is a PNG file
            image_file.seek(16)
            img_size = (struct.unpack("!I", image_file.read(4))[0],
                        struct.unpack("!I", image_file.read(4))[0])

            if img_size > self.max_img_size:
                # The image exceeds the maximum image size, so resize it
                image = _resize_image(self.image_file, img_size)
            else:
                image = os.path.abspath(self.image_file)
        elif PILImage:
            # Pillow is installed, meaning the image can be converted
            # automatically
            img = PILImage.open(self.image_file, mode='r')

            if img.size > self.max_img_size:
                image = _resize_image(img, img.size)
            else:
                image_data = io.BytesIO()
                img.save(image_data, format='PNG')
                img.close()
                image = image_data.getvalue()
            logger.info("Converted image %s to PNG format" %
                        self.image_file)
        else:
            raise ValueError("The image %s needs to be converted to a PNG"
                             " file, but PIL.Image is not available! Either"
                             " install the Pillow package, or use a PNG image"
                             " for you splash screen." % self.image_file)

        image_file.close()

        res = SplashWriter(self.name,  # noqa: F841
                           self.splash_requirements,
                           self.tcl_lib[0],  # tcl86t.dll
                           self.tk_lib[0],  # tk86t.dll
                           TK_ROOTNAME,
                           self.rundir,
                           image,
                           self.script)

    def test_tk_version(self):
        tcl_version = float(self._tkinter_module.TCL_VERSION)
        tk_version = float(self._tkinter_module.TK_VERSION)

        # Test if tcl/tk version is supported
        if tcl_version < 8.6 or tk_version < 8.6:
            logger.warning("The installed Tcl/Tk (%s/%s) version might not"
                           " work with the splash screen feature of the"
                           " bootloader. The bootloader is tested against"
                           " Tcl/Tk 8.6" % (self._tkinter_module.TCL_VERSION,
                                            self._tkinter_module.TK_VERSION))

        # This should be impossible, since tcl/tk is released together with
        # the same version number, but just in case
        if tcl_version != tk_version:
            logger.warning("The installed version of Tcl (%s) and Tk (%s) do"
                           " not match. PyInstaller is tested against matching"
                           " versions" % (self._tkinter_module.TCL_VERSION,
                                          self._tkinter_module.TK_VERSION))

        # Test if tcl is threaded.
        # If the variable tcl_platform(threaded) exist, the tcl
        # interpreter was compiled with thread support.
        threaded = bool(exec_statement("""
        from tkinter import Tcl, TclError
        try:
            print(Tcl().getvar('tcl_platform(threaded)'))
        except TclError:
            pass"""))

        if not threaded:
            # This is a feature breaking problem, so exit
            raise SystemExit("The installed tcl version is not threaded."
                             " PyInstaller only supports the splash screen"
                             " using threaded tcl.")

    def generate_script(self):
        """Generate the script for the splash screen

        If minify_script is True, all unnecessary parts will be
        removed
        """
        d = {}
        if self.text_pos is not None:
            logger.debug("Add text support to splash screen")
            d.update({
                'pad_x': self.text_pos[0],
                'pad_y': self.text_pos[1],
                'color': self.text_color,
                'font': self.text_font,
                'font_size': self.text_size,
                'default_text': self.text_default,
            })
        script = splash_templates.build_script(text_options=d)

        if self.minify_script:
            # Remove any documentation, empty lines and unnecessary spaces
            script = '\n'.join(line for line in map(lambda l: l.strip(),
                                                    script.splitlines())
                               if not line.startswith('#')  # documentation
                               and line)  # empty lines # noqa: W503
            # Remove unnecessary spaces
            script = re.sub(' +', ' ', script)

        # Write script to disk, so that it is transparent to the use
        # what script is executed
        with open(self.script_name, "w") as script_file:
            script_file.write(script)
        return script

    @staticmethod
    def _uses_tkinter(binaries):
        # Test for _tkinter instead of tkinter, because a user
        # might use a different wrapping library for tk
        return '_tkinter' in binaries.filenames

    @staticmethod
    def _find_rundir(structure):
        # First try a name the user could understand, if one would find
        # the directory
        rundir = '__splash%s'
        candidate = rundir % ""
        counter = 0

        # Run this loop as long as a folder exist named like rundir.
        # In most cases __splash will be sufficient and this loop wont enter
        while any(e[0].startswith(candidate + os.sep) for e in structure):
            # just append to rundir a counter
            candidate = rundir % str(counter)
            counter += 1

            # The SPLASH_DATA_HEADER structure limits the name to be 16 bytes
            # at maximum. So if we exceed the limit raise an error. This will
            # never happen, since there are 10^8 different possibilities, but
            # just in case.
            assert len(candidate) <= 16

        return candidate