###############################################################################
# Name: mako.py                                                               #
# Purpose: Define Mako syntax for highlighting and other features             #
# Author: Cody Precord <cprecord@editra.org>                                  #
# Copyright: (c) 2007 Cody Precord <staff@editra.org>                         #
# License: wxWindows License                                                  #
###############################################################################

"""
FILE: mako.py
AUTHOR: Cody Precord
@summary: Lexer configuration module for Mako Templates.

"""

__author__ = "Cody Precord <cprecord@editra.org>"
__svnid__ = "$Id: _mako.py 68798 2011-08-20 17:17:05Z CJP $"
__revision__ = "$Revision: 68798 $"

#-----------------------------------------------------------------------------#
# Imports
import wx.stc as stc
from pygments.token import Token
from pygments.lexers import get_lexer_by_name

# Local Imports
import synglob
import syndata

#-----------------------------------------------------------------------------#
# Style Id's

STC_MAKO_DEFAULT, \
STC_MAKO_COMMENT, \
STC_MAKO_NUMBER, \
STC_MAKO_STRING, \
STC_MAKO_STRINGEOL, \
STC_MAKO_SCALAR, \
STC_MAKO_OPERATOR, \
STC_MAKO_PREPROCESSOR, \
STC_MAKO_ATTRIBUTE, \
STC_MAKO_TAG, \
STC_MAKO_BUILTIN, \
STC_MAKO_KEYWORD = range(12)

#-----------------------------------------------------------------------------#

#---- Keyword Specifications ----#

# Python Keywords
KEYWORDS = "include inherit namespace page"

#---- Syntax Style Specs ----#
SYNTAX_ITEMS = [ (STC_MAKO_DEFAULT, 'default_style'),
                 (STC_MAKO_COMMENT, 'comment_style'),
                 (STC_MAKO_NUMBER, 'number_style'),
                 (STC_MAKO_STRING, 'string_style'),
                 (STC_MAKO_STRINGEOL, 'stringeol_style'),
                 (STC_MAKO_SCALAR, 'scalar_style'),
                 (STC_MAKO_OPERATOR, 'operator_style'),
                 (STC_MAKO_PREPROCESSOR, 'pre_style'),
                 (STC_MAKO_ATTRIBUTE, 'keyword2_style'),
                 (STC_MAKO_TAG, 'keyword_style'),       # Need new tag
                 (STC_MAKO_BUILTIN, 'keyword4_style'),
                 (STC_MAKO_KEYWORD, 'keyword_style'), ]

#-----------------------------------------------------------------------------#

class SyntaxData(syndata.SyntaxDataBase):
    """SyntaxData object for Mako""" 
    def __init__(self, langid):
        super(SyntaxData, self).__init__(langid)

        # Setup
        self.SetLexer(stc.STC_LEX_CONTAINER)
        self.RegisterFeature(synglob.FEATURE_STYLETEXT, StyleText)

    def GetKeywords(self):
        """Returns Specified Keywords List """
        return [(1, KEYWORDS)]

    def GetSyntaxSpec(self):
        """Syntax Specifications """
        return SYNTAX_ITEMS

    def GetCommentPattern(self):
        """Returns a list of characters used to comment a block of code """
        return [u"#",]

#-----------------------------------------------------------------------------#

def StyleText(stc, start, end):
    """Style the text
    @param stc: Styled text control instance
    @param start: Start position
    @param end: end position

    """
    cpos = 0
    stc.StartStyling(cpos, 0x1f)
    lexer = get_lexer_by_name("html+mako")
    doctxt = stc.GetTextRange(0, end)
    wineol = stc.GetEOLChar() == "\r\n"
    for token, txt in lexer.get_tokens(doctxt):
#        print token, txt
        style = TOKEN_MAP.get(token, STC_MAKO_DEFAULT)
        if style == STC_MAKO_PREPROCESSOR and txt.startswith(u'#'):
            style = STC_MAKO_COMMENT
#        elif style == STC_MAKO_STRING and txt[-1] not in '"\'':
#            style = STC_MAKO_STRINGEOL

        tlen = len(txt)
        if wineol and "\n" in txt:
            tlen += txt.count("\n")

        if tlen:
            stc.SetStyling(tlen, style)
        cpos += tlen
        stc.StartStyling(cpos, 0x1f)

#-----------------------------------------------------------------------------#

TOKEN_MAP = { Token.Literal.String : STC_MAKO_STRING,
              Token.Comment.Preproc : STC_MAKO_PREPROCESSOR,
              Token.Comment : STC_MAKO_COMMENT,
              Token.Name.Builtin : STC_MAKO_BUILTIN,
              Token.Operator : STC_MAKO_OPERATOR,
              Token.Punctuation : STC_MAKO_OPERATOR,
              Token.Number : STC_MAKO_NUMBER,
              Token.Keyword : STC_MAKO_KEYWORD,
              Token.Name.Attribute : STC_MAKO_ATTRIBUTE,
              Token.String.Interpol : STC_MAKO_SCALAR,
              Token.Name.Tag : STC_MAKO_TAG }