import mode, tab from lex import Grammar, PatternRule, NocasePatternRule, RegionRule, NocaseRegionRule from mode.python import StringGrammar1, StringGrammar2 from method import CommentRegion, UncommentRegion class PlPgSqlGrammar(Grammar): rules = [ PatternRule(r'spaces', r' +'), 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'sql_keyword', r'(?:declare|begin|end if|end loop|end|raise notice|return|if|while|for|else)(?![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|next|asc|desc|limit|distinct|cascade|alias for|using|on)(?![A-Za-z0-9_])'), NocasePatternRule(r'sql_type', r'(?:void|row|serial|varchar|float|integer|int|text|timestamptz|timestamp|datetz|date|timetz|time|boolean|bool|record|row)(?![A-Za-z0-9_])'), PatternRule(r'sql_builtin', r'(?:nextval|current_timestamp|current_time|current_date)(?![A-Za-z0-9_])'), RegionRule(r'string', "''", StringGrammar1, "''"), RegionRule(r'sql_quoted', '"', StringGrammar2, '"'), PatternRule(r'sql_variable', r'\$[1-9][0-9]*'), PatternRule(r'sql_bareword', r'[A-Za-z0-9_]+'), PatternRule(r'empty', r'^ *\n$'), PatternRule(r'eol', r'\n'), ] class FunctionGrammar(Grammar): rules = [ PatternRule(r'spaces', r' +'), PatternRule(r'comment', r'--.*\n$'), RegionRule(r'comment', '/\*', Grammar, '\*/'), PatternRule(r'delimiter', r':=|[():;,\.\$\[\]~]'), PatternRule(r'name', r'[a-zA-Z_][a-zA-Z0-9_]*(?=\()'), NocasePatternRule(r'sql_keyword', r'(?:as|returns|language)'), 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_])'), NocasePatternRule(r'language', r'(?<=language ) *[a-zA-Z_][a-zA-Z0-9_]+'), RegionRule(r'definition', "'", PlPgSqlGrammar, "'(?!')"), PatternRule(r'sql_bareword', r'[A-Za-z0-9_]+'), PatternRule(r'empty', r'^ *\n$'), PatternRule(r'eol', r'\n'), ] class BitStringGrammar(Grammar): rules = [PatternRule(r'data', r'[01]+')] class HexStringGrammar(Grammar): rules = [NocasePatternRule(r'data', r'[0-9a-f]+')] class SqlGrammar(Grammar): rules = [ PatternRule(r'spaces', r' +'), PatternRule(r'comment', r'--.*\n$'), RegionRule(r'comment', '/\*', Grammar, '\*/'), PatternRule(r'delimiter', r':=|[():;,\.\$\[\]~]'), NocaseRegionRule(r'function', r'create function', FunctionGrammar, r';'), NocaseRegionRule(r'function', r'create or replace function', FunctionGrammar, r';'), 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_])'), 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_])'), 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'string', "B'", BitStringGrammar, "'"), RegionRule(r'string', "X'", HexStringGrammar, "'"), RegionRule(r'string', "'", StringGrammar1, "'"), RegionRule(r'sql_quoted', '"', StringGrammar2, '"'), NocasePatternRule(r'sql_bareword', r'[a-z_][a-z0-9_]*'), PatternRule(r'empty', r'^ *\n$'), PatternRule(r'eol', r'\n'), ] class SqlCommentRegion(CommentRegion): commentc = '--' class SqlUncommentRegion(UncommentRegion): commentc = '--' 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 opentokens = ('delimiter',) opentags = {'(': ')', '[': ']', '{': '}'} closetokens = ('delimiter',) closetags = {')': '(', ']': '[', '}': '{'} actions = [SqlCommentRegion, SqlUncommentRegion] 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'), } def __init__(self, w): mode.Fundamental.__init__(self, w) self.add_bindings('close-paren', (')',)) self.add_bindings('close-brace', ('}',)) self.add_bindings('close-bracket', (']',)) self.add_bindings('sql-comment-region', ('C-c #',)) self.add_bindings('sql-uncomment-region', ('C-u C-c #',)) install = Sql.install