# coding: utf-8
# Copyright 2013 The Font Bakery Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# See AUTHORS.txt for the list of Authors and LICENSE.txt for the License.

from __future__ import print_function
import sys
import defusedxml.lxml
import os as os_origin
import shutil as shutil_origin
import os.path as op
import re
import subprocess
from logger import logger

class ProcessedFile(object):

    class __ProcessedFile:
    
        def __init__(self):
            self.filepath = None
    
        def __str__(self):
            return self.filepath

        def __add__(self, other):
            return str(self) + other

        def __radd__(self, other):
            return other + str(self)

        def __len__(self):
            return len(self.filepath)

        def __getitem__(self, index):
            return self.filepath[index]

        def startswith(self, b):
            return self.filepath[:len(b) - 1] == b

    instance = None

    def __new__(cls):
        if not ProcessedFile.instance:
            ProcessedFile.instance = ProcessedFile.__ProcessedFile()
        return ProcessedFile.instance

    def __getattr__(self, name):
        return getattr(self.instance, name)

    def __setattr__(self, name):
        return setattr(self.instance, name)


class UpstreamDirectory(object):
    """ Describes structure of upstream directory
    >>> upstream = UpstreamDirectory("tests/fixtures/upstream-example")
    >>> upstream.UFO
    ['Font-Regular.ufo']
    >>> upstream.TTX
    ['Font-Light.ttx']
    >>> upstream.BIN
    ['Font-SemiBold.ttf']
    >>> upstream.METADATA
    ['METADATA.pb']
    >>> sorted(upstream.LICENSE)
    ['APACHE.txt', 'LICENSE.txt']
    >>> upstream.SFD
    ['Font-Bold.sfd']
    >>> sorted(upstream.TXT)
    ['APACHE.txt', 'LICENSE.txt']
    """

    OFL = ['open font license.markdown', 'ofl.txt', 'ofl.md']
    LICENSE = ['license.txt', 'license.md', 'copyright.txt']
    APACHE = ['apache.txt', 'apache.md']
    UFL = ['ufl.txt', 'ufl.md']

    ALL_LICENSES = OFL + LICENSE + APACHE + UFL

    def __init__(self, upstream_path):
        self.upstream_path = upstream_path

        self.UFO = []
        self.TTX = []
        self.BIN = []
        self.LICENSE = []
        self.METADATA = []
        self.SFD = []
        self.TXT = []

        self.walk()

    def get_ttx(self):
        return self.TTX

    def get_binaries(self):
        return self.BIN

    def get_fonts(self):
        return self.UFO + self.TTX + self.BIN + self.SFD
    ALL_FONTS = property(get_fonts)

    def walk(self):
        l = len(self.upstream_path)
        exclude = ['build_info', ]
        for root, dirs, files in os_origin.walk(self.upstream_path, topdown=True):
            dirs[:] = [d for d in dirs if d not in exclude]
            for f in files:
                fullpath = op.join(root, f)

                if f[-4:].lower() == '.ttx':
                    try:
                        doc = defusedxml.lxml.parse(fullpath)
                        el = doc.xpath('//ttFont[@sfntVersion]')
                        if not el:
                            continue
                    except Exception as exc:
                        msg = 'Failed to parse "{}". Error: {}'
                        logger.error(msg.format(fullpath, exc))
                        continue
                    self.TTX.append(fullpath[l:].strip('/'))

                if op.basename(f).lower() == 'metadata.pb':
                    self.METADATA.append(fullpath[l:].strip('/'))

                if f[-4:].lower() in ['.ttf', '.otf']:
                    self.BIN.append(fullpath[l:].strip('/'))

                if f[-4:].lower() == '.sfd':
                    self.SFD.append(fullpath[l:].strip('/'))

                if f[-4:].lower() in ['.txt', '.markdown', '.md', '.LICENSE']:
                    self.TXT.append(fullpath[l:].strip('/'))

                if op.basename(f).lower() in UpstreamDirectory.ALL_LICENSES:
                    self.LICENSE.append(fullpath[l:].strip('/'))

            for d in dirs:
                fullpath = op.join(root, d)
                if op.splitext(fullpath)[1].lower() == '.ufo':
                    self.UFO.append(fullpath[l:].strip('/'))


def nameTableRead(font, NameID, fallbackNameID=False):
    for record in font['name'].names:
        if record.nameID == NameID:
            return record.toUnicode()

    if fallbackNameID:
        return nameTableRead(font, fallbackNameID)

    return


def ttfautohint_installed():
    return bool([os.path.join(p, 'ttfautohint')
                 for p in os.environ.get('PATH').split(':')
                 if os.path.exists(os.path.join(p, 'ttfautohint'))])


def get_data_directory():
    FONTBAKERY_DATA_DIRNAME = 'fontbakery'
    try:
        #Windows code:
        d = op.join(os.environ["FONTBAKERY_DATA_DIRNAME"],
                    FONTBAKERY_DATA_DIRNAME)
    except KeyError:
        #GNU/Linux and Mac code:
        d = op.join(op.expanduser("~"),
                    ".local", "share", FONTBAKERY_DATA_DIRNAME)
    if not op.exists(d):
        os.makedirs(d)
    return d


def shell_cmd_repr(command, args):
    available_commands = {
        'move': 'mv {}',
        'copy': 'cp -a {}',
        'copytree': 'cp -a {}',
        'makedirs': 'mkdir -p {}'
    }
    cmdline = available_commands[command].format(' '.join(list(args)))
    return cmdline.replace(os_origin.getcwd() + os_origin.path.sep, '')


class metaclass(type):

    def __getattr__(cls, value):
        if not hasattr(cls.__originmodule__, value):
            _ = "'module' object has no attribute '%s'"
            raise AttributeError(_ % value)

        attr = getattr(cls.__originmodule__, value)
        if not callable(attr):
            return attr

        def func(*args, **kwargs):
            logger.debug('\n$ ' + shell_cmd_repr(value, args))
            try:
                result = getattr(cls.__originmodule__, value)(*args, **kwargs)
                return result
            except Exception as e:
                logger.error('Error: %s' % e.message)
                raise e

        return func


class osmetaclass(metaclass):
    __originmodule__ = os_origin


class shutilmetaclass(metaclass):
    __originmodule__ = shutil_origin


class shutil(object):
    __metaclass__ = shutilmetaclass


class os(object):
    __metaclass__ = osmetaclass


def run(command, cwd=None):
    """ Wrapper for subprocess.Popen with custom logging support.
        :param command: shell command to run, required
        :param cwd: - current working dir, required
        :param log: - logging object with .write() method, required
    """
    # Start the command
    env = os.environ.copy()

    logger.info('$ %s\n' % command.replace(os_origin.getcwd() + os.path.sep, ''))
    env.update({'PYTHONPATH': os_origin.pathsep.join(sys.path)})
    process = subprocess.Popen(command, shell=True, cwd=cwd or os_origin.getcwd(),
                               stdout=subprocess.PIPE,
                               stderr=subprocess.PIPE,
                               close_fds=True, env=env)
    out, err = process.communicate()
    if out.strip():
        logger.info(out.strip())
    if err.strip():
        logger.error(err.strip())
    return out


def formatter(start, end, step):
    return '{}-{}'.format('U+{0:04x}'.format(start), 'U+{0:04x}'.format(end))


def re_range(lst):
    n = len(lst)
    result = []
    scan = 0
    while n - scan > 2:
        step = lst[scan + 1] - lst[scan]
        if lst[scan + 2] - lst[scan + 1] != step:
            result.append('U+{0:04x}'.format(lst[scan]))
            scan += 1
            continue

        for j in range(scan+2, n-1):
            if lst[j+1] - lst[j] != step:
                result.append(formatter(lst[scan], lst[j], step))
                scan = j+1
                break
        else:
            result.append(formatter(lst[scan], lst[-1], step))
            return ','.join(result)

    if n - scan == 1:
        result.append('U+{0:04x}'.format(lst[scan]))
    elif n - scan == 2:
        result.append(','.join(map('U+{0:04x}'.format, lst[scan:])))

    return ','.join(result)


def clean_name_values(fontdata):
    styles = ['Thin', 'Extra Light', 'Light', 'Regular', 'Medium',
              'SemiBold', 'Extra Bold', 'Black', 'Thin Italic',
              'Extra Light Italic', 'Light Italic', 'Italic',
              'Medium Italic', 'SemiBold Italic', 'Extra Bold Italic',
              'Black Italic', 'Bold', 'Bold Italic']
    for style in styles:
        fontdata['names'] = [{'nameID': x['nameID'],
                             'string': x['string'].replace(style, '').strip()}
                             for x in fontdata['names'] if x['nameID'] in [1, 2, 4, 6, 16, 17, 18]]
    return fontdata


def fix_all_names(fontdata, familyname):

    isBold = isItalic = False

    if fontdata['OS/2']['fsSelection'] & 0b10000:
        isBold = True
    if fontdata['head']['macStyle'] & 0b01:
        isBold = True
    if fontdata['OS/2']['fsSelection'] & 0b00001:
        isItalic = True
    if fontdata['head']['macStyle'] & 0b10:
        isItalic = True
    if fontdata['post']['italicAngle'] != 0:
        isItalic = True

    weight = fontdata['OS/2']['usWeightClass']

    if isBold:
        weight = 700
    fontdata['OS/2']['usWeightClass'] = weight
    if 'CFF' in fontdata: fontdata['CFF']['Weight'] = fontdata['OS/2']['usWeightClass']

    rules = NameTableNamingRule({'isBold': isBold,
                                 'isItalic': isItalic,
                                 'familyName': familyname,
                                 'weight': weight})

    names = []
    passedNamesId = []

    for rec in fontdata['names']:
        string = rec['string']
        if rec['nameID'] in [1, 2, 4, 6, 16, 17, 18]:
            string = rules.apply(rec['nameID'])
            passedNamesId.append(rec['nameID'])
        names.append({'nameID': rec['nameID'], 'string': string})

    difference = set([1, 2, 4, 6, 16, 17, 18]).difference(set(passedNamesId))
    if difference:
        for nameID in difference:
            string = rules.apply(nameID)
            names.append({'nameID': nameID, 'string': string})
            
    fontdata['names'] = names

    return fontdata


class NameTableNamingRule(object):

    def __init__(self, fontconfig):
        self.fontconfig = fontconfig

    def apply(self, nameID):
        """ Return func """
        return getattr(self, 'ruleNameID_{}'.format(nameID))()

    @property
    def isRegular(self):
        return self.fontconfig['weight'] == 400 \
            and not (self.fontconfig['isItalic'] or self.fontconfig['isBold'])

    @property
    def isItalic(self):
        return self.fontconfig['weight'] == 400 and self.fontconfig['isItalic']

    @property
    def isBlack(self):
        return self.fontconfig['weight'] == 900

    @property
    def isThin(self):
        return self.fontconfig['weight'] == 250

    @property
    def isLight(self):
        return self.fontconfig['weight'] == 300

    @property
    def isMedium(self):
        return self.fontconfig['weight'] == 500

    def ruleNameID_1(self):
        if self.isBlack:
            return self.fontconfig['familyName'] + ' Black'
        if self.isThin:
            return self.fontconfig['familyName'] + ' Thin'
        if self.isLight:
            return self.fontconfig['familyName'] + ' Light'
        if self.isMedium:
            return self.fontconfig['familyName'] + ' Medium'

        if not self.fontconfig['isBold'] and not self.fontconfig['isItalic']:
            return self.fontconfig['familyName'] + ' Regular'
        return self.fontconfig['familyName']

    def ruleNameID_2(self):
        if self.fontconfig['isBold'] and not self.fontconfig['isItalic']:
            return 'Bold'
        if self.fontconfig['isItalic'] and not self.fontconfig['isBold']:
            return 'Italic'
        if self.fontconfig['isItalic'] and self.fontconfig['isBold']:
            return 'Bold Italic'
        return 'Regular'

    def ruleNameID_4(self):
        if self.fontconfig['isBold'] and not self.fontconfig['isItalic']:
            return self.ruleNameID_1() + ' Bold'
        elif self.fontconfig['isItalic'] and not self.fontconfig['isBold']:
            return self.ruleNameID_1() + ' Italic'
        elif self.fontconfig['isItalic'] and self.fontconfig['isBold']:
            return self.ruleNameID_1() + ' Bold Italic'
        return self.ruleNameID_1()

    def ruleNameID_6(self):
        psn = self.fontconfig['familyName'].replace(' ', '') + '-'
        # import pdb; pdb.set_trace()
        if self.isBlack:
            psn += 'Black'
            if self.fontconfig['isItalic']:
                psn += 'Italic'
        if self.fontconfig['isBold']:
            psn += 'Bold'
            if self.fontconfig['isItalic']:
                psn += 'Italic'
        elif self.isThin:
            psn += 'Thin'
            if self.fontconfig['isItalic']:
                psn += 'Italic'
        elif self.isLight:
            psn += 'Light'
            if self.fontconfig['isItalic']:
                psn += 'Italic'
        elif self.isMedium:
            psn += 'Medium'
            if self.fontconfig['isItalic']:
                psn += 'Italic'
        elif self.isItalic:
            return psn + 'Italic'
        elif self.isRegular:
            return psn + 'Regular'
        psn = psn.strip('-')
        return psn

    def ruleNameID_16(self):
        return self.fontconfig['familyName']

    def ruleNameID_17(self):
        tsn = 'Regular'
        if self.isBlack:
            tsn = 'Black'
            if self.fontconfig['isItalic']:
                tsn += ' Italic'
        if self.fontconfig['isBold']:
            tsn = 'Bold'
            if self.fontconfig['isItalic']:
                tsn += ' Italic'
        elif self.isThin:
            tsn = 'Thin'
            if self.fontconfig['isItalic']:
                tsn += ' Italic'
        elif self.isLight:
            tsn = 'Light'
            if self.fontconfig['isItalic']:
                tsn += ' Italic'
        elif self.isMedium:
            tsn = 'Medium'
            if self.fontconfig['isItalic']:
                tsn += ' Italic'
        elif self.isRegular:
            tsn = 'Regular'
        elif self.fontconfig['isItalic']:
            tsn = 'Italic'
        return tsn

    def ruleNameID_18(self):
        return self.ruleNameID_4()