import mode, tab
from lex import Grammar, PatternRule, NocasePatternRule, RegionRule, NocaseRegionRule, OverridePatternRule
from mode.python import StringGrammar1, StringGrammar2

class BitStringGrammar(Grammar):
    rules = [PatternRule(r'data', r'[01]+')]

class HexStringGrammar(Grammar):
    rules = [NocasePatternRule(r'data', r'[0-9a-f]+')]

base_rules = [
    PatternRule(r'spaces', r' +'),
    OverridePatternRule(r'comment', r'--@@:(?P<token>[.a-zA-Z0-9_]+):(?P<mode>[.a-zA-Z0-9_]+) *$'),
    PatternRule(r'comment', r'--.*\n$'),
    RegionRule(r'comment', '/\*', Grammar, '\*/'),
    PatternRule(r'delimiter', r':=|[():;,\.\$\[\]~]'),
]

end_rules = [
    NocasePatternRule(r'sql_bareword', r'[a-z_][a-z0-9_]*'),
    PatternRule(r'empty', r'^ *\n$'),
    PatternRule(r'eol', r'\n'),
]

class PlPgSqlGrammar(Grammar): pass

sql_type_rule = NocasePatternRule(r'sql_type', r'(?:void|row|serial|varchar|float|integer|int|text|timestamptz|timestamp|datetz|date|timetz|time|boolean|bool)(?![A-Za-z0-9_])')

class FunctionGrammar(Grammar):
    rules = base_rules + [
        PatternRule(r'name', r'[a-zA-Z_][a-zA-Z0-9_]*(?=\()'),
        NocasePatternRule(r'sql_keyword', r'(?:as|returns|language)'),
        sql_type_rule,
        NocasePatternRule(r'language', r'(?<=language ) *[a-zA-Z_][a-zA-Z0-9_]+'),
        RegionRule(r'definition', "'", PlPgSqlGrammar, "'(?!')"),
    ] + end_rules

function_rules = [
    NocaseRegionRule(r'function', r'create function', FunctionGrammar, r';'),
    NocaseRegionRule(r'function', r'create or replace function', FunctionGrammar, r';'),
]

sql_rules = [
    NocasePatternRule(r'attribute', r'(?:check|exists|unique|not null|default|primary key|minvalue|foreign key|references)(?![A-Za-z0-9_])'),
    NocasePatternRule(r'sql_operator', r'(?:case|when|then|else|end|not in|not|and|or|is not|is|in|between|overlaps|like|ilike|similar)(?![A-Za-z0-9_])'),
    NocasePatternRule(r'sql_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|unknown|null|true|false|into|values|as|from|order by|asc|desc|limit|distinct|cascade|using|on)(?![A-Za-z0-9_])'),
    sql_type_rule,
    PatternRule(r'sql_builtin', r'(?:nextval|current_timestamp|current_time|current_date)(?![A-Za-z0-9_])'),
    PatternRule(r'number', r'[0-9]+(?:e[+-]?[0-9]+)?'),
    PatternRule(r'number', r'[0-9]+\.(?:[0-9]+)?(?:e[+-]?[0-9]+)?'),
    PatternRule(r'number', r'\.[0-9]+(?:e[+-]?[0-9]+)'),
    RegionRule(r'sql_quoted', '"', StringGrammar2, '"'),
]

string_rules = [
    RegionRule(r'string', "B'", BitStringGrammar, "'"),
    RegionRule(r'string', "X'", HexStringGrammar, "'"),
    RegionRule(r'string', "'", StringGrammar1, "'"),
]

PlPgSqlGrammar.rules = base_rules + sql_rules + [
    RegionRule(r'string', "B''", BitStringGrammar, "''"),
    RegionRule(r'string', "X''", HexStringGrammar, "''"),
    RegionRule(r'string', "''", StringGrammar1, "''"),
] + end_rules

class SqlGrammar(Grammar):
    rules = base_rules + function_rules + sql_rules + string_rules + end_rules

class SqlTabber(tab.StackTabber):
    wst  = ('null', 'eol',)
    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):
        w     = self.mode.tabwidth
        token = self.get_token(y, i)
        s     = token.string.lower()
        if token.name == 'delimiter' and s == ';':
            self._opt_pop('cont')
        elif token.name == 'sql_keyword':
            if s == 'declare':
                self._append('declare', currlvl + w)
            elif s == 'begin':
                currlvl -= w
            elif s == 'end':
                self._opt_pop('declare')
                currlvl = self.get_curr_level()
            elif s == 'end if':
                self._opt_pop('if')
                currlvl = self.get_curr_level()
            elif s == 'end loop':
                self._opt_pop('loop')
                currlvl = self.get_curr_level()
            elif s == 'else':
                currlvl -= w
            elif s == 'if':
                self._append('if', currlvl + w)
            elif s in ('while', 'for'):
                self._append('loop', currlvl + w)

        if self.is_rightmost_token(y, i):
            if not self._empty() and token.name == 'continuation':
                self._append('cont', currlvl + w)
            elif token.name == 'eol' and not self.markers:
                self._opt_pop("cont")
        return currlvl

class Sql(mode.Fundamental):
    modename    = 'Sql'
    extensions  = ['.sql']
    grammar     = SqlGrammar
    tabbercls   = SqlTabber
    commentc    = '--'
    opentokens  = ('delimiter',)
    opentags    = {'(': ')', '[': ']', '{': '}'}
    closetokens = ('delimiter',)
    closetags   = {')': '(', ']': '[', '}': '{'}
    colors      = {
        'sql_operator':      ('yellow', 'default', 'bold'),
        'attribute':         ('magenta', 'default', 'bold'),
        'sql_keyword':       ('cyan', 'default', 'bold'),
        'pseudokeyword':     ('cyan', 'default', 'bold'),
        'sql_type':          ('green', 'default', 'bold'),
        'sql_builtin':       ('yellow', 'default', 'bold'),
        'sql_quoted.start':  ('yellow', 'default', 'bold'),
        'sql_quoted.data':   ('yellow', 'default', 'bold'),
        'sql_quoted.null':   ('yellow', 'default', 'bold'),
        'sql_quoted.end':    ('yellow', 'default', 'bold'),
        'sql_variable':      ('yellow', 'default', 'bold'),
        'sql_bareword':      ('default', 'default', 'bold'),
        'function.start':    ('cyan', 'default', 'bold'),
        'function.data':     ('default', 'default', 'bold'),
        'function.null':     ('default', 'default', 'bold'),
        'function.name':     ('magenta', 'default', 'bold'),
        'function.language': ('magenta', 'default', 'bold'),
        'function.end':      ('default', 'default', 'bold'),

        'function.definition.start':        ('magenta', 'default', 'bold'),
        'function.definition.sql_bareword': ('magenta', 'default', 'bold'),
        'function.definition.data':         ('magenta', 'default', 'bold'),
        'function.definition.null':         ('magenta', 'default', 'bold'),
        'function.definition.end':          ('magenta', 'default', 'bold'),
    }
    _bindings = {
        'close-paren': (')',),
        'close-brace': ('}',),
        'close-bracket': (']',),
    }

install = Sql.install