# -*- coding: utf-8 -*-
# Copyright (C) by the Spot authors, see the AUTHORS file for details.
#
# This file is part of Spot, a model checking library.
#
# Spot 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.
#
# Spot 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 <http://www.gnu.org/licenses/>.

# This file is named "aux_.py" for compatibility with Windows'
# historical limitations, but should really be imported as "spot.aux".

"""
Auxiliary functions for Spot's Python bindings
"""

from functools import lru_cache
import subprocess
import sys
import os
import errno
import contextlib
import re


def extend(*classes):
    """
    Decorator that extends all the given classes with the contents
    of the class currently being defined.
    """
    def wrap(this):
        for cls in classes:
            for (name, val) in this.__dict__.items():
                if name not in ('__dict__', '__weakref__') \
                   and not (name == '__doc__' and val is None):
                    setattr(cls, name, val)
        return classes[0]
    return wrap


# Work around a bug introduced in GraphViz 2.42.x, where the scale
# parameter is inverted.  https://gitlab.com/graphviz/graphviz/issues/1605
# In our case, the scale parameters should both be <= 1, so we can
# detect when that is not the case.
svgscale_regex = re.compile(r'transform="scale\(([\d.]+) ([\d.]+)\) rotate')

def _gvfix(matchobj):
        xs = float(matchobj.group(1))
        ys = float(matchobj.group(2))
        if xs >= 1 and ys >= 1:
            xs = 1/xs
            ys = 1/ys
        return 'transform="scale({} {}) rotate'.format(xs, ys)


# Add a small LRU cache so that when we display automata into a
# interactive widget, we avoid some repeated calls to dot for
# identical inputs.
@lru_cache(maxsize=64)
def str_to_svg(s):
    """
    Send some text to dot for conversion to SVG.
    """
    try:
        p = subprocess.run(['dot', '-Tsvg'],
                           capture_output=True, text=True,
                           input=s.decode('utf-8') if type(s) is bytes else s)
    except FileNotFoundError:
        print("The command 'dot' seems to be missing on your system.\n"
              "Please install the GraphViz package "
              "and make sure 'dot' is in your PATH.", file=sys.stderr)
        raise
    if p.stderr: print(p.stderr, file=sys.stderr, flush=True)
    p.check_returncode()
    return svgscale_regex.sub(_gvfix, p.stdout)


def ostream_to_svg(ostr):
    """
    Encode an ostringstream to SVG.
    """
    return str_to_svg(ostr.str())


def rm_f(filename):
    """
    Remove filename if it exists.
    """
    try:
        os.remove(filename)
    except OSError as e:
        if e.errno != errno.ENOENT:
            raise


@contextlib.contextmanager
def tmpdir():
    cwd = os.getcwd()
    tmpdir = os.environ.get('SPOT_TMPDIR') or os.environ.get('TMPDIR') or '.'
    try:
        os.chdir(tmpdir)
        yield
    finally:
        os.chdir(cwd)
