Mercurial Hosting > luan
annotate core/src/luan/impl/ThemeParser.java @ 600:b926e53910dd
change theming indentation
author | Franklin Schmidt <fschmidt@gmail.com> |
---|---|
date | Mon, 28 Sep 2015 00:32:27 -0600 |
parents | 8370c4009cce |
children | c5ad80f869da |
rev | line source |
---|---|
584 | 1 package luan.impl; |
2 | |
3 import java.util.Map; | |
4 import java.util.HashMap; | |
5 import java.util.List; | |
6 import java.util.ArrayList; | |
7 import luan.LuanSource; | |
8 import luan.LuanTable; | |
9 import luan.LuanElement; | |
10 import luan.LuanState; | |
11 import luan.LuanFunction; | |
12 import luan.LuanException; | |
13 import luan.modules.PackageLuan; | |
14 | |
15 | |
16 public final class ThemeParser { | |
17 | |
18 public static LuanFunction compile(LuanState luan,LuanSource source) throws LuanException { | |
19 try { | |
20 FnDef fnDef = new ThemeParser(source).parse(); | |
21 final LuanStateImpl luanImpl = (LuanStateImpl)luan; | |
22 return new Closure(luanImpl,fnDef); | |
23 } catch(ParseException e) { | |
24 //e.printStackTrace(); | |
25 throw new LuanException(luan, e.getFancyMessage() ); | |
26 } | |
27 } | |
28 | |
29 private static final class Frame { | |
30 final Frame parent; | |
31 final List<String> symbols = new ArrayList<String>(); | |
32 int stackSize = 0; | |
33 final boolean isVarArg; | |
34 final List<String> upValueSymbols = new ArrayList<String>(); | |
35 final List<UpValue.Getter> upValueGetters = new ArrayList<UpValue.Getter>(); | |
36 | |
37 Frame() { | |
38 this.parent = null; | |
39 isVarArg = true; | |
40 } | |
41 | |
42 Frame(Frame parent) { | |
43 this.parent = parent; | |
44 isVarArg = false; | |
45 if( upValueIndex(MOD) != 0 ) | |
46 throw new RuntimeException(); | |
47 } | |
48 | |
49 int stackIndex(String name) { | |
50 int i = symbols.size(); | |
51 while( --i >= 0 ) { | |
52 if( symbols.get(i).equals(name) ) | |
53 return i; | |
54 } | |
55 return -1; | |
56 } | |
57 | |
58 int upValueIndex(String name) { | |
59 int i = upValueSymbols.size(); | |
60 while( --i >= 0 ) { | |
61 if( upValueSymbols.get(i).equals(name) ) | |
62 return i; | |
63 } | |
64 if( parent==null ) | |
65 return -1; | |
66 i = parent.stackIndex(name); | |
67 if( i != -1 ) { | |
68 upValueGetters.add(new UpValue.StackGetter(i)); | |
69 } else { | |
70 i = parent.upValueIndex(name); | |
71 if( i == -1 ) | |
72 return -1; | |
73 upValueGetters.add(new UpValue.NestedGetter(i)); | |
74 } | |
75 upValueSymbols.add(name); | |
76 return upValueSymbols.size() - 1; | |
77 } | |
78 | |
79 void addUpValueGetter(String name,UpValue.Getter upValueGetter) { | |
80 upValueSymbols.add(name); | |
81 upValueGetters.add(upValueGetter); | |
82 } | |
83 } | |
84 | |
85 private static final String IO = "-IO-"; | |
86 private static final String MOD = "-MOD-"; | |
87 private static final String ENV = "-ENV-"; | |
594 | 88 private static final String INDENT = "-INDENT-"; |
584 | 89 private static final UpValue.Getter[] NO_UP_VALUE_GETTERS = new UpValue.Getter[0]; |
90 | |
91 private final LuanSource source; | |
92 private final Parser parser; | |
93 private Frame frame = new Frame(); | |
94 | |
95 private ThemeParser(LuanSource source) { | |
96 this.source = source; | |
97 this.parser = new Parser(this.source); | |
98 } | |
99 | |
100 private LuanElement se(int start) { | |
101 return se(start,null); | |
102 } | |
103 | |
104 private LuanElement se(int start,String text) { | |
105 return new LuanElement(source,start,parser.currentIndex(),text); | |
106 } | |
107 | |
108 private int symbolsSize() { | |
109 return frame.symbols.size(); | |
110 } | |
111 | |
112 private void addSymbol(String name) { | |
113 frame.symbols.add(name); | |
114 if( frame.stackSize < symbolsSize() ) | |
115 frame.stackSize = symbolsSize(); | |
116 } | |
117 | |
118 private FnDef newFnDef(int start,Stmt stmt) { | |
119 return new FnDef( se(start), stmt, frame.stackSize, symbolsSize(), frame.isVarArg, frame.upValueGetters.toArray(NO_UP_VALUE_GETTERS) ); | |
120 } | |
121 | |
122 private int stackIndex(String name) { | |
123 return frame.stackIndex(name); | |
124 } | |
125 | |
594 | 126 private void popSymbols(int n) { |
127 List<String> symbols = frame.symbols; | |
128 while( n-- > 0 ) { | |
129 symbols.remove(symbols.size()-1); | |
130 } | |
131 } | |
132 | |
584 | 133 private int upValueIndex(String name) { |
134 return frame.upValueIndex(name); | |
135 } | |
136 | |
137 private ParseException exception(String msg) { | |
138 parser.failure(); | |
139 return parser.exception(msg); | |
140 } | |
141 | |
142 private Expr env() { | |
143 return new GetLocalVar(null,stackIndex(ENV)); | |
144 } | |
145 | |
146 private FnDef parse() throws ParseException { | |
147 List<Stmt> stmts = new ArrayList<Stmt>(); | |
148 int stackStart = symbolsSize(); | |
149 { | |
150 addSymbol(IO); | |
151 LuanElement se = se(0,"require 'luan:Io'"); | |
152 FnCall requireCall = new FnCall( se, new ConstExpr(se,PackageLuan.requireFn), new ConstExpr(se,"luan:Io") ); | |
153 SetStmt setStmt = new SetStmt( new SetLocalVar(stackIndex(IO)), new ExpressionsExpr(requireCall) ); | |
154 stmts.add(setStmt); | |
155 } | |
156 { | |
157 addSymbol(MOD); | |
158 LuanElement se = se(0,"local M = {}"); | |
594 | 159 TableExpr.Field indent = new TableExpr.Field(new ConstExpr(null,INDENT),new ConstExpr(null,"")); |
160 TableExpr tableExpr = new TableExpr( se, new TableExpr.Field[]{indent}, ExpList.emptyExpList ); | |
584 | 161 SetStmt setStmt = new SetStmt( new SetLocalVar(stackIndex(MOD)), tableExpr ); |
162 stmts.add(setStmt); | |
163 } | |
164 while( !parser.endOfInput() ) { | |
165 Stmt def = parseDef(); | |
166 if( def != null ) { | |
167 stmts.add(def); | |
168 } else { | |
169 parser.anyChar(); | |
170 } | |
171 } | |
172 stmts.add( new ReturnStmt(null,new GetLocalVar(null,stackIndex(MOD))) ); | |
173 Stmt block = new Block( stmts.toArray(new Stmt[0]), stackStart, symbolsSize() ); | |
174 FnDef fnDef = newFnDef(0,block); | |
175 return fnDef; | |
176 } | |
177 | |
178 private Stmt parseDef() throws ParseException { | |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
179 int start = parser.currentIndex(); |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
180 if( !parser.match("{define:") ) |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
181 return null; |
594 | 182 String name = parseName(); |
183 if( name==null ) | |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
184 throw parser.exception("invalid block name"); |
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
185 if( !parser.match("}") ) |
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
186 throw parser.exception("unclosed define tag"); |
594 | 187 String spaces = ""; |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
188 boolean indent = BlankLine(); |
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
189 if( indent ) { |
594 | 190 int startSpaces = parser.currentIndex(); |
191 InlineSpaces(); | |
192 spaces = parser.textFrom(startSpaces); | |
193 } | |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
194 Expr table = new GetLocalVar(null,stackIndex(MOD)); |
594 | 195 Settable fnName = new SetTableEntry(se(start),table,new ConstExpr(null,name)); |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
196 frame = new Frame(frame); |
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
197 addSymbol(ENV); |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
198 Stmt block = parseBody("define:"+name,spaces,indent); |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
199 FnDef fnDef = newFnDef(start,block); |
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
200 frame = frame.parent; |
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
201 Stmt rtn = new SetStmt(fnName,fnDef); |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
202 return rtn; |
584 | 203 } |
204 | |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
205 private Stmt parseBody(String tagName,String spaces,boolean afterIndent) throws ParseException { |
584 | 206 List<Stmt> stmts = new ArrayList<Stmt>(); |
207 int stackStart = symbolsSize(); | |
594 | 208 int start = parser.currentIndex(); |
209 { | |
210 addSymbol(INDENT); | |
211 final Expr env = env(); | |
212 Expr exp = new ExprImpl(se(start,"indent")) { | |
213 @Override public Object eval(LuanStateImpl luan) throws LuanException { | |
214 LuanTable tbl = (LuanTable)env.eval(luan); | |
215 String indent = (String)tbl.get(luan,INDENT); | |
216 if( indent==null ) throw new NullPointerException(); | |
217 return indent; | |
218 } | |
219 }; | |
220 // Expr exp = new IndexExpr( se(start,"indent"), env(), new ConstExpr(null,INDENT) ); | |
221 SetStmt setStmt = new SetStmt( new SetLocalVar(stackIndex(INDENT)), exp ); | |
222 stmts.add(setStmt); | |
223 } | |
224 int end = start; | |
225 while( !matchEndTag(tagName) ) { | |
584 | 226 if( parser.endOfInput() ) |
227 throw exception("unclosed block"); | |
594 | 228 Stmt block = parseBlock(spaces); |
584 | 229 if( block != null ) { |
594 | 230 addText(start,end,stmts,false); |
231 start = parser.currentIndex(); | |
584 | 232 stmts.add(block); |
594 | 233 afterIndent = false; |
584 | 234 continue; |
235 } | |
594 | 236 { |
237 String extraSpaces = null; | |
238 if( afterIndent ) { | |
239 int startSpaces = parser.currentIndex(); | |
240 InlineSpaces(); | |
241 extraSpaces = parser.textFrom(startSpaces); | |
242 end = parser.currentIndex(); | |
243 } | |
244 Stmt simpleTag = parseSimpleTag(extraSpaces); | |
245 if( simpleTag != null ) { | |
246 addText(start,end,stmts,false); | |
247 start = parser.currentIndex(); | |
248 stmts.add(simpleTag); | |
249 afterIndent = false; | |
250 continue; | |
251 } | |
252 if( extraSpaces!=null && extraSpaces.length() > 0 ) | |
253 continue; | |
254 } | |
255 if( EndOfLine() ) { | |
256 end = parser.currentIndex(); | |
257 afterIndent = false; | |
258 if( parser.match(spaces) ) { | |
259 addText(start,end,stmts,true); | |
260 start = parser.currentIndex(); | |
261 afterIndent = true; | |
262 } | |
584 | 263 continue; |
264 } | |
265 parser.anyChar(); | |
594 | 266 end = parser.currentIndex(); |
267 afterIndent = false; | |
584 | 268 } |
594 | 269 addText(start,end,stmts,false); |
270 Stmt block = new Block( stmts.toArray(new Stmt[0]), stackStart, symbolsSize() ); | |
271 popSymbols(1); | |
584 | 272 return block; |
273 } | |
274 | |
594 | 275 private boolean matchEndTag(String tagName) { |
276 parser.begin(); | |
600
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
277 if( tagName.startsWith("define:") ) |
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
278 EndOfLine(); |
b926e53910dd
change theming indentation
Franklin Schmidt <fschmidt@gmail.com>
parents:
595
diff
changeset
|
279 return parser.match("{/") && parser.match(tagName) && parser.match('}') ? parser.success() : parser.failure(); |
584 | 280 } |
281 | |
594 | 282 private void addText(int start,int end,List<Stmt> stmts,boolean indent) { |
283 List<Expressions> args = new ArrayList<Expressions>(); | |
284 if( start < end ) { | |
285 String text = parser.text.substring(start,end); | |
286 args.add( new ConstExpr(null,text) ); | |
287 } | |
288 if( indent ) { | |
289 args.add( new GetLocalVar(null,stackIndex(INDENT)) ); | |
290 } | |
291 if( !args.isEmpty() ) { | |
292 Expr io = new GetUpVar(null,upValueIndex(IO)); | |
293 Expr stdoutExp = new IndexExpr( se(start,"stdout"), io, new ConstExpr(null,"stdout") ); | |
294 Expr writeExp = new IndexExpr( se(start,"write"), stdoutExp, new ConstExpr(null,"write") ); | |
295 FnCall writeCall = new FnCall( se(start), writeExp, ExpList.build(args) ); | |
296 stmts.add( new ExpressionsStmt(writeCall) ); | |
297 } | |
298 } | |
299 | |
300 private Stmt parseBlock(String spaces) throws ParseException { | |
584 | 301 int start = parser.begin(); |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
302 if( !parser.match("{block:") ) |
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
303 return parser.failure(null); |
594 | 304 String name = parseName(); |
305 if( name==null ) | |
306 throw exception("invalid block name"); | |
307 if( !parser.match("}") ) | |
308 return null; | |
584 | 309 frame = new Frame(frame); |
310 addSymbol(ENV); | |
594 | 311 Stmt block = parseBody("block:"+name,spaces,false); |
584 | 312 FnDef fnDef = newFnDef(start,block); |
313 frame = frame.parent; | |
314 // String rtn = "<% env." + tag.name + "(" + (tag.attrs.isEmpty() ? "nil" : table(tag.attrs)) + ",env,function(env) %>" + block + "<% end) %>"; | |
315 Expr env = env(); | |
594 | 316 Expr fn = new IndexExpr( se(start,"block:"+name), env, new ConstExpr(null,name) ); |
584 | 317 List<Expressions> args = new ArrayList<Expressions>(); |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
318 args.add( env ); |
584 | 319 args.add( fnDef ); |
320 FnCall fnCall = new FnCall( se(start), fn, ExpList.build(args) ); | |
321 Stmt rtn = new ExpressionsStmt(fnCall); | |
322 return parser.success(rtn); | |
323 } | |
324 | |
594 | 325 private Stmt parseSimpleTag(String spaces) throws ParseException { |
584 | 326 int start = parser.begin(); |
327 if( !parser.match("{") ) | |
328 return parser.failure(null); | |
329 String name = parseName(); | |
330 if( name==null ) | |
331 return parser.failure(null); | |
332 if( !parser.match("}") ) | |
333 return parser.failure(null); | |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
334 // rtn = "<% env." + name + (attrs.isEmpty() ? "()" : table(attrs)) + " %>"; |
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
335 Expr env = env(); |
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
336 Expr fn = new IndexExpr( se(start,name), env, new ConstExpr(null,name) ); |
594 | 337 if( spaces!=null && spaces.length() > 0 ) { |
338 final Expr oldEnv = env; | |
339 final Expr oldIndentExpr = new GetLocalVar(null,stackIndex(INDENT)); | |
340 final String addSpaces = spaces; | |
341 env = new ExprImpl(se(start,"indent_env")) { | |
342 @Override public Object eval(LuanStateImpl luan) throws LuanException { | |
343 LuanTable mt = new LuanTable(); | |
344 mt.rawPut("__index",oldEnv.eval(luan)); | |
345 LuanTable tbl = new LuanTable(); | |
346 tbl.setMetatable(mt); | |
347 String oldIndent = (String)oldIndentExpr.eval(luan); | |
348 String newIndent = oldIndent + addSpaces; | |
349 tbl.rawPut(INDENT,newIndent); | |
350 return tbl; | |
351 } | |
352 }; | |
353 } | |
595
8370c4009cce
remove theme attributes
Franklin Schmidt <fschmidt@gmail.com>
parents:
594
diff
changeset
|
354 FnCall fnCall = new FnCall( se(start), fn, env ); |
584 | 355 Stmt rtn = new ExpressionsStmt(fnCall); |
356 return parser.success(rtn); | |
357 } | |
358 | |
594 | 359 private void InlineSpaces() { |
360 while( parser.anyOf(" \t") ); | |
361 } | |
362 | |
363 private boolean BlankLine() { | |
364 parser.begin(); | |
365 while( parser.anyOf(" \t") ); | |
366 return EndOfLine() ? parser.success() : parser.failure(); | |
367 } | |
368 | |
369 private boolean EndOfLine() { | |
370 return parser.match( "\r\n" ) || parser.match( '\r' ) || parser.match( '\n' ); | |
371 } | |
372 | |
584 | 373 private String parseName() { |
374 int start = parser.begin(); | |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
375 if( !NameChar() ) |
584 | 376 return parser.failure(null); |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
377 while( NameChar() ); |
584 | 378 String match = parser.textFrom(start); |
379 return parser.success(match); | |
380 } | |
381 | |
593
92c9fa5e39e6
remove theme "Get" and "Set", and add "define"
Franklin Schmidt <fschmidt@gmail.com>
parents:
587
diff
changeset
|
382 private boolean NameChar() { |
587 | 383 return parser.inCharRange('a', 'z') || parser.inCharRange('A', 'Z') |
384 || parser.inCharRange('0', '9') || parser.anyOf("-_"); | |
584 | 385 } |
386 | |
387 } |