import re
from PyQt5 import QtGui, Qsci

class _LexerSelector(object):
    _lexer = None
    def match(self, filename, filedata):
        return False

    def lexer(self, cfg=None):
        """
        Return a configured instance of the lexer
        """
        return self.cfg_lexer(self._lexer(), cfg) #pylint: disable=E1102

    def cfg_lexer(self, lexer, cfg=None):
        if cfg:
            font = QtGui.QFont()
            fontstr = cfg.getFont()
            font.fromString(fontstr)
            size = cfg.getFontSize()
        else:
            font = QtGui.QFont('Monospace')
            size = 9

        font.setPointSize(size)
        lexer.setFont(font, -1)
        return lexer

class _FilenameLexerSelector(_LexerSelector):
    """
    Base class for lexer selector based on file name matching
    """
    extensions = ()
    def match(self, filename, filedata):
        filename = filename.lower()
        for ext in self.extensions:
            if filename.endswith(ext):
                return True
        return False

class _ScriptLexerSelector(_FilenameLexerSelector):
    """
    Base class for lexer selector based on content pattern matching
    """
    regex = None
    headersize = 3
    def match(self, filename, filedata):
        if super(_ScriptLexerSelector, self).match(filename, filedata):
            return True
        if self.regex:
            for line in filedata.splitlines()[:self.headersize]:
                if len(line)<1000 and self.regex.match(line):
                    return True
        return False

class PythonLexerSelector(_ScriptLexerSelector):
    extensions = ('.py', '.pyw')
    _lexer = Qsci.QsciLexerPython
    regex = re.compile(r'^#[!].*python')

class BashLexerSelector(_ScriptLexerSelector):
    extensions = ('.sh', '.bash')
    _lexer = Qsci.QsciLexerBash
    regex = re.compile(r'^#[!].*sh')

class PerlLexerSelector(_ScriptLexerSelector):
    extensions = ('.pl', '.perl')
    _lexer = Qsci.QsciLexerPerl
    regex = re.compile(r'^#[!].*perl')

class RubyLexerSelector(_ScriptLexerSelector):
    extensions = ('.rb', '.ruby')
    _lexer = Qsci.QsciLexerRuby
    regex = re.compile(r'^#[!].*ruby')

class LuaLexerSelector(_ScriptLexerSelector):
    extensions = ('.lua', )
    _lexer = Qsci.QsciLexerLua
    regex = None

class CppLexerSelector(_FilenameLexerSelector):
    extensions = ('.c', '.cpp', '.cxx', '.h', '.hpp', '.hxx')
    _lexer = Qsci.QsciLexerCPP

class CSSLexerSelector(_FilenameLexerSelector):
    extensions = ('.css',)
    _lexer = Qsci.QsciLexerCSS

class HTMLLexerSelector(_FilenameLexerSelector):
    extensions = ('.htm', '.html', '.xhtml', '.xml')
    _lexer = Qsci.QsciLexerHTML

class MakeLexerSelector(_FilenameLexerSelector):
    extensions = ('.mk', 'makefile')
    _lexer = Qsci.QsciLexerMakefile

class SQLLexerSelector(_FilenameLexerSelector):
    extensions = ('.sql',)
    _lexer = Qsci.QsciLexerSQL

class JSLexerSelector(_FilenameLexerSelector):
    extensions = ('.js',)
    _lexer = Qsci.QsciLexerJavaScript

class JavaLexerSelector(_FilenameLexerSelector):
    extensions = ('.java',)
    _lexer = Qsci.QsciLexerJava

class TeXLexerSelector(_FilenameLexerSelector):
    extensions = ('.tex', '.latex',)
    _lexer = Qsci.QsciLexerTeX

class DiffLexerSelector(_ScriptLexerSelector):
    extensions = ()
    _lexer = Qsci.QsciLexerDiff
    regex = re.compile(r'^@@ [-]\d+,\d+ [+]\d+,\d+ @@$')
    def cfg_lexer(self, lexer, cfg=None):
        """
        Return a configured instance of the lexer
        """
        if cfg:
            lexer.setDefaultPaper(QtGui.QColor(cfg.getDiffBGColor()))
            lexer.setColor(QtGui.QColor(cfg.getDiffFGColor()), -1)
            lexer.setColor(QtGui.QColor(cfg.getDiffPlusColor()), 6)
            lexer.setColor(QtGui.QColor(cfg.getDiffMinusColor()), 5)
            lexer.setColor(QtGui.QColor(cfg.getDiffSectionColor()), 4)
            font = QtGui.QFont()
            fontstr = cfg.getFont()
            font.fromString(fontstr)
            size = cfg.getFontSize()
        else:
            font = QtGui.QFont('Monospace')
            size = 9

        font.setPointSize(size)
        lexer.setFont(font, -1)

        bfont = QtGui.QFont(font)
        bfont.setBold(True)
        lexer.setFont(bfont, 5)
        lexer.setFont(bfont, 6)

        return lexer


lexers = [cls() for clsname, cls in globals().items() if not clsname.startswith('_') and isinstance(cls, type) and \
          issubclass(cls, (_LexerSelector, _FilenameLexerSelector, _ScriptLexerSelector))]

def get_lexer(filename, filedata, fileflag=None, cfg=None):
    if fileflag == "=":
        return DiffLexerSelector().lexer(cfg)
    for lselector in lexers:
        if lselector.match(filename, filedata):
            return lselector.lexer(cfg)
    return None