From 3362f81c962e3fd51fb5926aa82970b79475428c Mon Sep 17 00:00:00 2001
From: Erik Osheim <erik@osheim.org>
Date: Tue, 29 Sep 2015 08:48:35 -0400
Subject: [PATCH] merged

--HG--
branch : pmacs2
---
 mode/lua.py | 28 +++++++++++++++++-----------
 1 file changed, 17 insertions(+), 11 deletions(-)

diff --git a/mode/lua.py b/mode/lua.py
index 6db8474..b00e243 100644
--- a/mode/lua.py
+++ b/mode/lua.py
@@ -9,25 +9,28 @@ from mode.pipe import Pipe
 from method.shell import Interact
 
 chr1 = '[a-zA-Z_]'
-chr2 = '[a-zA-Z_0-9]'
-word = chr1 + chr2 + '*'
+chr2 = '[a-zA-Z_0-9_]'
+word = '[a-zA-Z_]' + chr2 + '*'
+
+capword = '[a-zA-Z]' + chr2 + '*'
 
 class LuaGrammar(Grammar):
     rules = [
-        RegionRule('comment', r'--\[(?P<level>=*)\[', Grammar, r'\]%(level)s\]'),
-        #RegionRule('comment', r'--\[\[', Grammar, r'\]\]'),
+        RegionRule('comment', r'--\[\[', Grammar, r'\]\]'),
         PatternRule('comment', '--.*$'),
         PatternRule('spaces', ' +'),
         PatternRule('eol', r'\n'),
 
-        RegionRule('lua.string', "'", StringGrammar1, "'"),
-        RegionRule('lua.string', '"', StringGrammar2, '"'),
-        RegionRule('lua.string', '\[\[', Grammar, '\]\]'),
+        RegionRule('lua.string', "'", StringGrammar1, "'"), #fixme
+        RegionRule('lua.string', '"', StringGrammar2, '"'), #fixme
 
-        PatternMatchRule('x', '(function)( +)(' + word + ')',
-                         'lua.keyword', 'spaces', 'lua.function'),
-        PatternRule('lua.keyword', '(?:while|until|true|then|return|repeat|or|not|nil|local|in|if|function|for|false|end|elseif|else|done|do|break|and)(?!' + chr2 + ')'),
+        PatternMatchRule('', '(function)( +)([A-Za-z_][A-Za-z0-9_]*)(:)([A-Za-z_][A-Za-z0-9_]*)', 'lua.keyword', 'spaces', 'lua.class', 'delimiter', 'lua.function'),
+        PatternMatchRule('', '(function)( +)([A-Za-z_][A-Za-z0-9_]*)', 'lua.keyword', 'spaces', 'lua.function'),
+
+        PatternRule('lua.keyword', '(?:while|until|then|return|repeat|or|not|local|in|if|function|for|end|elseif|else|done|do|break|and)(?!' + chr2 + ')'),
+        PatternRule('lua.reserved', '(?:true|false|nil|self)(?![a-zA-Z0-9_])'),
         PatternRule('lua.internal', '_[A-Z]+'),
+        PatternRule('lua.class', '[A-Z][a-zA-Z0-9_]*'),
         PatternRule('lua.identifier', word),
 
         PatternRule('delimiter', r'(?:[=(){}\[\];:,.])'),
@@ -75,7 +78,10 @@ class Lua(Fundamental):
     opentags    = {'(': ')', '[': ']', '{': '}'}
     closetokens = ('delimiter',)
     closetags   = {')': '(', ']': '[', '}': '{'}
-    colors      = {}
+    colors      = {
+        'lua.class': ('blue225', 'default'),
+        'lua.reserved': ('magenta505', 'default'),
+    }
     actions     = [LuaCheckSyntax, LuaStart, LuaLoadFile]
     _bindings   = {
         'close-paren':      (')',),