import color, mode2, tab2
from lex2 import Grammar, PatternRule, NocasePatternRule, RegionRule
from mode_python import StringGrammar

class SqlGrammar(Grammar):
    rules = [
        PatternRule(r'comment', r'--.*\n$'),
        RegionRule(r'comment', '/\*', Grammar, '\*/'),
        PatternRule(r'delimiter', r'[();,\.:\$\[\]]'),
        NocasePatternRule(r'attribute', r'(?:check|exists|unique|not null|default|primary key|minvalue|foreign key|references)(?![A-Za-z0-9_])'),
        NocasePatternRule(r'operator', r'(?:case|when|then|else|end|not|and|or|is not|is|in|not in)(?![A-Za-z0-9_])'),
        NocasePatternRule(r'keyword', r'(?:create database|create index|create sequence|create table|create trigger|create view|select|insert|update|delete|drop database|drop index|drop sequence|drop table|drop trigger|drop view|create user|alter user|drop user|drop function|grant|revoke|create function|create or replace function|create or replace view|create language|create operator|create type)(?![A-Za-z0-9_])'),
        NocasePatternRule(r'pseudokeyword', r'(?:returns|language|right join|left join|inner join|outer join|join|where|null|true|false|into|values|as|from|order by|asc|desc|limit|distinct|cascade|using|on)(?![A-Za-z0-9_])'),
        NocasePatternRule(r'type', r'(?:void|row|serial|varchar|float|integer|int|text|timestamptz|timestamp|datetz|date|timetz|time|boolean|bool)(?![A-Za-z0-9_])'),
        PatternRule(r'function', r'(?:nextval|current_timestamp|current_time|current_date)(?![A-Za-z0-9_])'),
        RegionRule(r'string', "'", StringGrammar, "'"),
        RegionRule(r'quoted', '"', StringGrammar, '"'),
        PatternRule(r'bareword', r'[A-Za-z0-9_]+'),
        PatternRule(r'empty', r'^ *\n$'),
        PatternRule(r'eol', r'; *\n'),
        PatternRule(r'continuation', r'\n'),
    ]

class SqlTabber(tab2.StackTabber):
    def is_base(self, y):
        if y == 0:
            return True
        highlighter = self.mode.window.buffer.highlights[self.mode.name()]
        if not highlighter.tokens[y]:
            return False
        t = highlighter.tokens[y][0]
        return t.name == 'function'
    def _handle_other_token(self, currlvl, y, i):
        token  = self.get_token(y, i)
        if self.is_rightmost_token(y, i):
            if not self._empty() and token.name == 'continuation':
                self._opt_append('cont', currlvl + 4)
            elif token.name == 'eol':
                self._opt_pop("cont")
        return currlvl

class Sql(mode2.Fundamental):
    grammar     = SqlGrammar
    tabbercls   = SqlTabber
    opentokens  = ('delimiter',)
    opentags    = {'(': ')', '[': ']', '{': '}'}
    closetokens = ('delimiter',)
    closetags   = {')': '(', ']': '[', '}': '{'}
    def __init__(self, w):
        mode2.Fundamental.__init__(self, w)
        self.add_bindings('close-paren', (')',))
        self.add_bindings('close-brace', ('}',))
        self.add_bindings('close-bracket', (']',))
        self.colors = {
            'comment':        color.build('red', 'default'),
            'operator':       color.build('yellow', 'default'),
            'attribute':      color.build('magenta', 'default'),
            'keyword':        color.build('cyan', 'default'),
            'pseudokeyword':  color.build('cyan', 'default'),
            'type':           color.build('green', 'default'),
            'function':       color.build('yellow', 'default'),
            'quoted':         color.build('yellow', 'default'),
            'string.start':   color.build('green', 'default'),
            'string.null':    color.build('green', 'default'),
            'string.escaped': color.build('magenta', 'default'),
            'string.octal':   color.build('magenta', 'default'),
            'string.end':     color.build('green', 'default'),
            'bareword':       color.build('default', 'default'),
        }
    def name(self):
        return "Sql"