/* * Copyright (C)2008-2017 Haxe Foundation * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER * DEALINGS IN THE SOFTWARE. */ package hscript; import hscript.Expr; enum Token { TEof; TConst( c : Const ); TId( s : String ); TOp( s : String ); TPOpen; TPClose; TBrOpen; TBrClose; TDot; TComma; TSemicolon; TBkOpen; TBkClose; TQuestion; TDoubleDot; TMeta( s : String ); TPrepro( s : String ); } class Parser { // config / variables public var line : Int; public var opChars : String; public var identChars : String; #if haxe3 public var opPriority : Map; public var opRightAssoc : Map; #else public var opPriority : Hash; public var opRightAssoc : Hash; #end /** allows to check for #if / #else in code **/ public var preprocesorValues : Map = new Map(); /** activate JSON compatiblity **/ public var allowJSON : Bool; /** allow types declarations **/ public var allowTypes : Bool; /** allow haxe metadata declarations **/ public var allowMetadata : Bool; /** resume from parsing errors (when parsing incomplete code, during completion for example) **/ public var resumeErrors : Bool; // implementation var input : String; var readPos : Int; var char : Int; var ops : Array; var idents : Array; var uid : Int = 0; #if hscriptPos var origin : String; var tokenMin : Int; var tokenMax : Int; var oldTokenMin : Int; var oldTokenMax : Int; var tokens : List<{ min : Int, max : Int, t : Token }>; #else static inline var p1 = 0; static inline var tokenMin = 0; static inline var tokenMax = 0; #if haxe3 var tokens : haxe.ds.GenericStack; #else var tokens : haxe.FastList; #end #end public function new() { line = 1; opChars = "+*/-=!><&|^%~"; identChars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_"; var priorities = [ ["%"], ["*", "/"], ["+", "-"], ["<<", ">>", ">>>"], ["|", "&", "^"], ["==", "!=", ">", "<", ">=", "<="], ["..."], ["&&"], ["||"], ["=","+=","-=","*=","/=","%=","<<=",">>=",">>>=","|=","&=","^=","=>"], ["->"] ]; #if haxe3 opPriority = new Map(); opRightAssoc = new Map(); #else opPriority = new Hash(); opRightAssoc = new Hash(); #end for( i in 0...priorities.length ) for( x in priorities[i] ) { opPriority.set(x, i); if( i == 9 ) opRightAssoc.set(x, true); } for( x in ["!", "++", "--", "~"] ) // unary "-" handled in parser directly! opPriority.set(x, x == "++" || x == "--" ? -1 : -2); } public inline function error( err, pmin, pmax ) { if( !resumeErrors ) #if hscriptPos throw new Error(err, pmin, pmax, origin, line); #else throw err; #end } public function invalidChar(c) { error(EInvalidChar(c), readPos-1, readPos-1); } function initParser( origin ) { // line=1 - don't reset line : it might be set manualy preprocStack = []; #if hscriptPos this.origin = origin; readPos = 0; tokenMin = oldTokenMin = 0; tokenMax = oldTokenMax = 0; tokens = new List(); #elseif haxe3 tokens = new haxe.ds.GenericStack(); #else tokens = new haxe.FastList(); #end char = -1; ops = new Array(); idents = new Array(); uid = 0; for( i in 0...opChars.length ) ops[opChars.charCodeAt(i)] = true; for( i in 0...identChars.length ) idents[identChars.charCodeAt(i)] = true; } public function parseString( s : String, ?origin : String = "hscript" ) { initParser(origin); input = s; readPos = 0; var a = new Array(); while( true ) { var tk = token(); if( tk == TEof ) break; push(tk); parseFullExpr(a); } return if( a.length == 1 ) a[0] else mk(EBlock(a),0); } function unexpected( tk ) : Dynamic { error(EUnexpected(tokenString(tk)),tokenMin,tokenMax); return null; } inline function push(tk) { #if hscriptPos tokens.push( { t : tk, min : tokenMin, max : tokenMax } ); tokenMin = oldTokenMin; tokenMax = oldTokenMax; #else tokens.add(tk); #end } inline function ensure(tk) { var t = token(); if( t != tk ) unexpected(t); } inline function ensureToken(tk) { var t = token(); if( !Type.enumEq(t,tk) ) unexpected(t); } function maybe(tk) { var t = token(); if( Type.enumEq(t, tk) ) return true; push(t); return false; } function getIdent() { var tk = token(); switch( tk ) { case TId(id): return id; default: unexpected(tk); return null; } } inline function expr(e:Expr) { #if hscriptPos return e.e; #else return e; #end } inline function pmin(e:Expr) { #if hscriptPos return e == null ? 0 : e.pmin; #else return 0; #end } inline function pmax(e:Expr) { #if hscriptPos return e == null ? 0 : e.pmax; #else return 0; #end } inline function mk(e,?pmin,?pmax) : Expr { #if hscriptPos if( e == null ) return null; if( pmin == null ) pmin = tokenMin; if( pmax == null ) pmax = tokenMax; return { e : e, pmin : pmin, pmax : pmax, origin : origin, line : line }; #else return e; #end } function isBlock(e) { if( e == null ) return false; return switch( expr(e) ) { case EBlock(_), EObject(_), ESwitch(_): true; case EFunction(_,e,_,_,_,_): isBlock(e); case EVar(_, t, e, _,_): e != null ? isBlock(e) : t != null ? t.match(CTAnon(_)) : false; case EIf(_,e1,e2): if( e2 != null ) isBlock(e2) else isBlock(e1); case EBinop(_,_,e): isBlock(e); case EUnop(_,prefix,e): !prefix && isBlock(e); case EWhile(_,e): isBlock(e); case EDoWhile(_,e): isBlock(e); case EFor(_,_,e): isBlock(e); case EReturn(e): e != null && isBlock(e); case ETry(_, _, _, e): isBlock(e); case EMeta(_, _, e): isBlock(e); default: false; } } function parseFullExpr( exprs : Array ) { var e = parseExpr(); exprs.push(e); var tk = token(); // this is a hack to support var a,b,c; with a single EVar while( tk == TComma && e != null && expr(e).match(EVar(_)) ) { e = parseStructure("var"); // next variable exprs.push(e); tk = token(); } if( tk != TSemicolon && tk != TEof ) { if( isBlock(e) ) push(tk); else unexpected(tk); } } function parseObject(p1) { // parse object var fl = new Array(); while( true ) { var tk = token(); var id = null; switch( tk ) { case TId(i): id = i; case TConst(c): if( !allowJSON ) unexpected(tk); switch( c ) { case CString(s): id = s; default: unexpected(tk); } case TBrClose: break; default: unexpected(tk); break; } ensure(TDoubleDot); fl.push({ name : id, e : parseExpr() }); tk = token(); switch( tk ) { case TBrClose: break; case TComma: default: unexpected(tk); } } return parseExprNext(mk(EObject(fl),p1)); } function parseExpr() { var oldPos = readPos; var tk = token(); #if hscriptPos var p1 = tokenMin; #end switch( tk ) { case TId(id): var e = parseStructure(id, oldPos); if( e == null ) e = mk(EIdent(id)); return parseExprNext(e); case TConst(c): return parseExprNext(mk(EConst(c))); case TPOpen: tk = token(); if( tk == TPClose ) { ensureToken(TOp("->")); var eret = parseExpr(); return mk(EFunction([], mk(EReturn(eret),p1)), p1); } push(tk); var e = parseExpr(); tk = token(); switch( tk ) { case TPClose: return parseExprNext(mk(EParent(e),p1,tokenMax)); case TDoubleDot: var t = parseType(); tk = token(); switch( tk ) { case TPClose: return parseExprNext(mk(ECheckType(e,t),p1,tokenMax)); case TComma: switch( expr(e) ) { case EIdent(v): return parseLambda([{ name : v, t : t }], pmin(e)); default: } default: } case TComma: switch( expr(e) ) { case EIdent(v): return parseLambda([{name:v}], pmin(e)); default: } default: } return unexpected(tk); case TBrOpen: tk = token(); switch( tk ) { case TBrClose: return parseExprNext(mk(EObject([]),p1)); case TId(_): var tk2 = token(); push(tk2); push(tk); switch( tk2 ) { case TDoubleDot: return parseExprNext(parseObject(p1)); default: } case TConst(c): if( allowJSON ) { switch( c ) { case CString(_): var tk2 = token(); push(tk2); push(tk); switch( tk2 ) { case TDoubleDot: return parseExprNext(parseObject(p1)); default: } default: push(tk); } } else push(tk); default: push(tk); } var a = new Array(); while( true ) { parseFullExpr(a); tk = token(); if( tk == TBrClose || (resumeErrors && tk == TEof) ) break; push(tk); } return mk(EBlock(a),p1); case TOp(op): if( op == "-" ) { var start = tokenMin; var e = parseExpr(); if( e == null ) return makeUnop(op,e); switch( expr(e) ) { case EConst(CInt(i)): return mk(EConst(CInt(-i)), start, pmax(e)); case EConst(CFloat(f)): return mk(EConst(CFloat(-f)), start, pmax(e)); default: return makeUnop(op,e); } } if( opPriority.get(op) < 0 ) return makeUnop(op,parseExpr()); return unexpected(tk); case TBkOpen: var a = new Array(); tk = token(); while( tk != TBkClose && (!resumeErrors || tk != TEof) ) { push(tk); a.push(parseExpr()); tk = token(); if( tk == TComma ) tk = token(); } if( a.length == 1 && a[0] != null ) // What is this for??? switch( expr(a[0]) ) { case EFor(_), EWhile(_), EDoWhile(_): var tmp = "__a_" + (uid++); var e = mk(EBlock([ mk(EVar(tmp, null, mk(EArrayDecl([]), p1)), p1), mapCompr(tmp, a[0]), mk(EIdent(tmp),p1), ]),p1); return parseExprNext(e); default: } return parseExprNext(mk(EArrayDecl(a, nextType), p1)); case TMeta(id) if( allowMetadata ): var args = parseMetaArgs(); return mk(EMeta(id, args, parseExpr()),p1); default: return unexpected(tk); } } function parseLambda( args : Array, pmin ) { while( true ) { var id = getIdent(); var t = maybe(TDoubleDot) ? parseType() : null; args.push({ name : id, t : t }); var tk = token(); switch( tk ) { case TComma: case TPClose: break; default: unexpected(tk); break; } } ensureToken(TOp("->")); var eret = parseExpr(); return mk(EFunction(args, mk(EReturn(eret),pmin)), pmin); } function parseMetaArgs() { var tk = token(); if( tk != TPOpen ) { push(tk); return null; } var args = []; tk = token(); if( tk != TPClose ) { push(tk); while( true ) { args.push(parseExpr()); switch( token() ) { case TComma: case TPClose: break; case tk: unexpected(tk); } } } return args; } function mapCompr( tmp : String, e : Expr ) { if( e == null ) return null; var edef = switch( expr(e) ) { case EFor(v, it, e2): EFor(v, it, mapCompr(tmp, e2)); case EWhile(cond, e2): EWhile(cond, mapCompr(tmp, e2)); case EDoWhile(cond, e2): EDoWhile(cond, mapCompr(tmp, e2)); case EIf(cond, e1, e2) if( e2 == null ): EIf(cond, mapCompr(tmp, e1), null); case EBlock([e]): EBlock([mapCompr(tmp, e)]); case EParent(e2): EParent(mapCompr(tmp, e2)); default: ECall( mk(EField(mk(EIdent(tmp), pmin(e), pmax(e)), "push"), pmin(e), pmax(e)), [e]); } return mk(edef, pmin(e), pmax(e)); } function makeUnop( op, e ) { if( e == null && resumeErrors ) return null; return switch( expr(e) ) { case EBinop(bop, e1, e2): mk(EBinop(bop, makeUnop(op, e1), e2), pmin(e1), pmax(e2)); case ETernary(e1, e2, e3): mk(ETernary(makeUnop(op, e1), e2, e3), pmin(e1), pmax(e3)); default: mk(EUnop(op,true,e),pmin(e),pmax(e)); } } function makeBinop( op, e1, e ) { if( e == null && resumeErrors ) return mk(EBinop(op,e1,e),pmin(e1),pmax(e1)); return switch( expr(e) ) { case EBinop(op2,e2,e3): if( opPriority.get(op) <= opPriority.get(op2) && !opRightAssoc.exists(op) ) mk(EBinop(op2,makeBinop(op,e1,e2),e3),pmin(e1),pmax(e3)); else mk(EBinop(op, e1, e), pmin(e1), pmax(e)); case ETernary(e2,e3,e4): if( opRightAssoc.exists(op) ) mk(EBinop(op,e1,e),pmin(e1),pmax(e)); else mk(ETernary(makeBinop(op, e1, e2), e3, e4), pmin(e1), pmax(e)); default: mk(EBinop(op,e1,e),pmin(e1),pmax(e)); } } var nextIsStatic:Bool = false; var nextIsPublic:Bool = false; var nextType:CType = null; function parseStructure(id, ?oldPos:Int) { #if hscriptPos var p1 = tokenMin; #end return switch( id ) { case "if": ensure(TPOpen); var cond = parseExpr(); ensure(TPClose); var e1 = parseExpr(); var e2 = null; var semic = false; var tk = token(); if( tk == TSemicolon ) { semic = true; tk = token(); } if( Type.enumEq(tk,TId("else")) ) e2 = parseExpr(); else { push(tk); if( semic ) push(TSemicolon); } mk(EIf(cond,e1,e2),p1,(e2 == null) ? tokenMax : pmax(e2)); case "static": nextIsStatic = true; var nextToken = token(); switch(nextToken) { case TId("public"): var str = parseStructure("public"); // public attribute nextIsStatic = false; str; case TId("function"): var str = parseStructure("function"); // static function nextIsStatic = false; str; case TId("var"): var str = parseStructure("var"); // static var nextIsStatic = false; str; default: unexpected(TId("static")); nextIsStatic = false; null; } case "public": nextIsPublic = true; var nextToken = token(); switch(nextToken) { case TId("static"): var str = parseStructure("static"); // static attribute nextIsPublic = false; str; case TId("function"): var str = parseStructure("function"); // public function nextIsPublic = false; str; case TId("var"): var str = parseStructure("var"); // public var nextIsPublic = false; str; default: unexpected(TId("static")); nextIsPublic = false; null; } case "var": var ident = getIdent(); var tk = token(); var t = null; nextType = null; if( tk == TDoubleDot && allowTypes ) { t = parseType(); tk = token(); nextType = t; } var e = null; if( Type.enumEq(tk,TOp("=")) ) e = parseExpr(); else push(tk); nextType = null; mk(EVar(ident,t,e,nextIsPublic,nextIsStatic),p1,(e == null) ? tokenMax : pmax(e)); case "while": var econd = parseExpr(); var e = parseExpr(); mk(EWhile(econd,e),p1,pmax(e)); case "do": var e = parseExpr(); var tk = token(); switch(tk) { case TId("while"): // Valid default: unexpected(tk); } var econd = parseExpr(); mk(EDoWhile(econd,e),p1,pmax(econd)); case "for": ensure(TPOpen); var vname = getIdent(); ensureToken(TId("in")); var eiter = parseExpr(); ensure(TPClose); var e = parseExpr(); mk(EFor(vname,eiter,e),p1,pmax(e)); case "break": mk(EBreak); case "continue": mk(EContinue); case "else": unexpected(TId(id)); case "inline": if( !maybe(TId("function")) ) unexpected(TId("inline")); return parseStructure("function"); case "function": var tk = token(); var name = null; switch( tk ) { case TId(id): name = id; default: push(tk); } var inf = parseFunctionDecl(); mk(EFunction(inf.args, inf.body, name, inf.ret,nextIsPublic,nextIsStatic),p1,pmax(inf.body)); case "import": var oldReadPos = readPos; var tk = token(); switch( tk ) { case TPOpen: var tok = token(); switch(tok) { case TConst(c): switch(c) { case CString(s): token(); ensure(TSemicolon); push(TSemicolon); mk(EImport(s), p1); default: unexpected(tok); null; } default: unexpected(tok); null; } case TId(id): var path = [id]; var t = null; while( true ) { t = token(); if( t != TDot ) { push(t); break; } t = token(); switch( t ) { case TId(id): path.push(id); default: unexpected(t); } } ensure(TSemicolon); push(TSemicolon); var p = path.join("."); mk(EImport(p),p1); default: unexpected(tk); null; } case "return": var tk = token(); push(tk); var e = if( tk == TSemicolon ) null else parseExpr(); mk(EReturn(e),p1,if( e == null ) tokenMax else pmax(e)); case "new": var a = new Array(); a.push(getIdent()); while( true ) { var tk = token(); switch( tk ) { case TDot: a.push(getIdent()); case TPOpen: break; default: unexpected(tk); break; } } var args = parseExprList(TPClose); mk(ENew(a.join("."),args),p1); case "throw": var e = parseExpr(); mk(EThrow(e),p1,pmax(e)); case "try": var e = parseExpr(); ensureToken(TId("catch")); ensure(TPOpen); var vname = getIdent(); ensure(TDoubleDot); var t = null; if( allowTypes ) t = parseType(); else ensureToken(TId("Dynamic")); ensure(TPClose); var ec = parseExpr(); mk(ETry(e, vname, t, ec), p1, pmax(ec)); case "switch": var e = parseExpr(); var def = null, cases = []; ensure(TBrOpen); while( true ) { var tk = token(); switch( tk ) { case TId("case"): var c = { values : [], expr : null }; cases.push(c); while( true ) { var e = parseExpr(); c.values.push(e); tk = token(); switch( tk ) { case TComma: // next expr case TDoubleDot: break; default: unexpected(tk); break; } } var exprs = []; while( true ) { tk = token(); push(tk); switch( tk ) { case TId("case"), TId("default"), TBrClose: break; case TEof if( resumeErrors ): break; default: parseFullExpr(exprs); } } c.expr = if( exprs.length == 1) exprs[0]; else if( exprs.length == 0 ) mk(EBlock([]), tokenMin, tokenMin); else mk(EBlock(exprs), pmin(exprs[0]), pmax(exprs[exprs.length - 1])); case TId("default"): if( def != null ) unexpected(tk); ensure(TDoubleDot); var exprs = []; while( true ) { tk = token(); push(tk); switch( tk ) { case TId("case"), TId("default"), TBrClose: break; case TEof if( resumeErrors ): break; default: parseFullExpr(exprs); } } def = if( exprs.length == 1) exprs[0]; else if( exprs.length == 0 ) mk(EBlock([]), tokenMin, tokenMin); else mk(EBlock(exprs), pmin(exprs[0]), pmax(exprs[exprs.length - 1])); case TBrClose: break; default: unexpected(tk); break; } } mk(ESwitch(e, cases, def), p1, tokenMax); default: null; } } function parseExprNext( e1 : Expr ) { var tk = token(); switch( tk ) { case TOp(op): if( op == "->" ) { // single arg reinterpretation of `f -> e` , `(f) -> e` and `(f:T) -> e` switch( expr(e1) ) { case EIdent(i), EParent(expr(_) => EIdent(i)): var eret = parseExpr(); return mk(EFunction([{ name : i }], mk(EReturn(eret),pmin(eret))), pmin(e1)); case ECheckType(expr(_) => EIdent(i), t): var eret = parseExpr(); return mk(EFunction([{ name : i, t : t }], mk(EReturn(eret),pmin(eret))), pmin(e1)); default: } unexpected(tk); } if( opPriority.get(op) == -1 ) { if( isBlock(e1) || switch(expr(e1)) { case EParent(_): true; default: false; } ) { push(tk); return e1; } return parseExprNext(mk(EUnop(op,false,e1),pmin(e1))); } return makeBinop(op,e1,parseExpr()); case TDot: var field = getIdent(); return parseExprNext(mk(EField(e1,field),pmin(e1))); case TPOpen: return parseExprNext(mk(ECall(e1,parseExprList(TPClose)),pmin(e1))); case TBkOpen: var e2 = parseExpr(); ensure(TBkClose); return parseExprNext(mk(EArray(e1,e2),pmin(e1))); case TQuestion: var e2 = parseExpr(); ensure(TDoubleDot); var e3 = parseExpr(); return mk(ETernary(e1,e2,e3),pmin(e1),pmax(e3)); default: push(tk); return e1; } } function parseFunctionArgs() { var args = new Array(); var tk = token(); if( tk != TPClose ) { var done = false; while( !done ) { var name = null, opt = false; switch( tk ) { case TQuestion: opt = true; tk = token(); default: } switch( tk ) { case TId(id): name = id; default: unexpected(tk); break; } var arg : Argument = { name : name }; args.push(arg); if( opt ) arg.opt = true; if( allowTypes ) { if( maybe(TDoubleDot) ) arg.t = parseType(); if( maybe(TOp("=")) ) arg.value = parseExpr(); } tk = token(); switch( tk ) { case TComma: tk = token(); case TPClose: done = true; default: unexpected(tk); } } } return args; } function parseFunctionDecl() { ensure(TPOpen); var args = parseFunctionArgs(); var ret = null; if( allowTypes ) { var tk = token(); if( tk != TDoubleDot ) push(tk); else ret = parseType(); } return { args : args, ret : ret, body : parseExpr() }; } function parsePath() { var path = [getIdent()]; while( true ) { var t = token(); if( t != TDot ) { push(t); break; } path.push(getIdent()); } return path; } function parseType() : CType { var t = token(); switch( t ) { case TId(v): push(t); var path = parsePath(); var params = null; t = token(); switch( t ) { case TOp(op): if( op == "<" ) { params = []; while( true ) { params.push(parseType()); t = token(); switch( t ) { case TComma: continue; case TOp(op): if( op == ">" ) break; if( op.charCodeAt(0) == ">".code ) { #if hscriptPos tokens.add({ t : TOp(op.substr(1)), min : tokenMax - op.length - 1, max : tokenMax }); #else tokens.add(TOp(op.substr(1))); #end break; } default: } unexpected(t); break; } } else push(t); default: push(t); } return parseTypeNext(CTPath(path, params)); case TPOpen: var a = token(), b = token(); push(b); push(a); function withReturn(args) { switch token() { // I think it wouldn't hurt if ensure used enumEq case TOp('->'): case t: unexpected(t); } return CTFun(args, parseType()); } switch [a, b] { case [TPClose, _] | [TId(_), TDoubleDot]: var args = [for (arg in parseFunctionArgs()) { switch arg.value { case null: case v: error(ECustom('Default values not allowed in function types'), #if hscriptPos v.pmin, v.pmax #else 0, 0 #end); } CTNamed(arg.name, if (arg.opt) CTOpt(arg.t) else arg.t); }]; return withReturn(args); default: var t = parseType(); return switch token() { case TComma: var args = [t]; while (true) { args.push(parseType()); if (!maybe(TComma)) break; } ensure(TPClose); withReturn(args); case TPClose: parseTypeNext(CTParent(t)); case t: unexpected(t); } } case TBrOpen: var fields = []; var meta = null; while( true ) { t = token(); switch( t ) { case TBrClose: break; case TId("var"): var name = getIdent(); ensure(TDoubleDot); fields.push( { name : name, t : parseType(), meta : meta } ); meta = null; ensure(TSemicolon); case TId(name): ensure(TDoubleDot); fields.push( { name : name, t : parseType(), meta : meta } ); t = token(); switch( t ) { case TComma: case TBrClose: break; default: unexpected(t); } case TMeta(name): if( meta == null ) meta = []; meta.push({ name : name, params : parseMetaArgs() }); default: unexpected(t); break; } } return parseTypeNext(CTAnon(fields)); default: return unexpected(t); } } function parseTypeNext( t : CType ) { var tk = token(); switch( tk ) { case TOp(op): if( op != "->" ) { push(tk); return t; } default: push(tk); return t; } var t2 = parseType(); switch( t2 ) { case CTFun(args, _): args.unshift(t); return t2; default: return CTFun([t], t2); } } function parseExprList( etk ) { var args = new Array(); var tk = token(); if( tk == etk ) return args; push(tk); while( true ) { args.push(parseExpr()); tk = token(); switch( tk ) { case TComma: default: if( tk == etk ) break; unexpected(tk); break; } } return args; } // ------------------------ module ------------------------------- public function parseModule( content : String, ?origin : String = "hscript" ) { initParser(origin); input = content; readPos = 0; allowTypes = true; allowMetadata = true; var decls = []; while( true ) { var tk = token(); if( tk == TEof ) break; push(tk); decls.push(parseModuleDecl()); } return decls; } function parseMetadata() : Metadata { var meta = []; while( true ) { var tk = token(); switch( tk ) { case TMeta(name): meta.push({ name : name, params : parseMetaArgs() }); default: push(tk); break; } } return meta; } function parseParams() { if( maybe(TOp("<")) ) error(EInvalidOp("Unsupported class type parameters"), readPos, readPos); return {}; } function parseModuleDecl() : ModuleDecl { var meta = parseMetadata(); var ident = getIdent(); var isPrivate = false, isExtern = false; while( true ) { switch( ident ) { case "private": isPrivate = true; case "extern": isExtern = true; default: break; } ident = getIdent(); } switch( ident ) { case "package": var path = parsePath(); ensure(TSemicolon); return DPackage(path); case "import": var path = [getIdent()]; var star = false; while( true ) { var t = token(); if( t != TDot ) { push(t); break; } t = token(); switch( t ) { case TId(id): path.push(id); case TOp("*"): star = true; default: unexpected(t); } } ensure(TSemicolon); return DImport(path, star); case "class": var name = getIdent(); var params = parseParams(); var extend = null; var implement = []; while( true ) { var t = token(); switch( t ) { case TId("extends"): extend = parseType(); case TId("implements"): implement.push(parseType()); default: push(t); break; } } var fields = []; ensure(TBrOpen); while( !maybe(TBrClose) ) fields.push(parseField()); return DClass({ name : name, meta : meta, params : params, extend : extend, implement : implement, fields : fields, isPrivate : isPrivate, isExtern : isExtern, }); case "typedef": var name = getIdent(); var params = parseParams(); ensureToken(TOp("=")); var t = parseType(); return DTypedef({ name : name, meta : meta, params : params, isPrivate : isPrivate, t : t, }); default: unexpected(TId(ident)); } return null; } function parseField() : FieldDecl { var meta = parseMetadata(); var access = []; while( true ) { var id = getIdent(); switch( id ) { case "override": access.push(AOverride); case "public": access.push(APublic); case "private": access.push(APrivate); case "inline": access.push(AInline); case "static": access.push(AStatic); case "macro": access.push(AMacro); case "function": var name = getIdent(); var inf = parseFunctionDecl(); return { name : name, meta : meta, access : access, kind : KFunction({ args : inf.args, expr : inf.body, ret : inf.ret, }), }; case "var": var name = getIdent(); var get = null, set = null; if( maybe(TPOpen) ) { get = getIdent(); ensure(TComma); set = getIdent(); ensure(TPClose); } var type = maybe(TDoubleDot) ? parseType() : null; var expr = maybe(TOp("=")) ? parseExpr() : null; if( expr != null ) { if( isBlock(expr) ) maybe(TSemicolon); else ensure(TSemicolon); } else if( type != null && type.match(CTAnon(_)) ) { maybe(TSemicolon); } else ensure(TSemicolon); return { name : name, meta : meta, access : access, kind : KVar({ get : get, set : set, type : type, expr : expr, }), }; default: unexpected(TId(id)); break; } } return null; } // ------------------------ lexing ------------------------------- inline function readChar() { return StringTools.fastCodeAt(input, readPos++); } function readString( until ) { var c = 0; var b = new StringBuf(); var esc = false; var old = line; var s = input; #if hscriptPos var p1 = readPos - 1; #end while( true ) { var c = readChar(); if( StringTools.isEof(c) ) { line = old; error(EUnterminatedString, p1, p1); break; } if( esc ) { esc = false; switch( c ) { case 'n'.code: b.addChar('\n'.code); case 'r'.code: b.addChar('\r'.code); case 't'.code: b.addChar('\t'.code); case "'".code, '"'.code, '\\'.code: b.addChar(c); case '/'.code: if( allowJSON ) b.addChar(c) else invalidChar(c); case "u".code: if( !allowJSON ) invalidChar(c); var k = 0; for( i in 0...4 ) { k <<= 4; var char = readChar(); switch( char ) { case 48,49,50,51,52,53,54,55,56,57: // 0-9 k += char - 48; case 65,66,67,68,69,70: // A-F k += char - 55; case 97,98,99,100,101,102: // a-f k += char - 87; default: if( StringTools.isEof(char) ) { line = old; error(EUnterminatedString, p1, p1); } invalidChar(char); } } b.addChar(k); default: invalidChar(c); } } else if( c == 92 ) esc = true; else if( c == until ) break; else { if( c == 10 ) line++; b.addChar(c); } } return b.toString(); } function token() { #if hscriptPos var t = tokens.pop(); if( t != null ) { tokenMin = t.min; tokenMax = t.max; return t.t; } oldTokenMin = tokenMin; oldTokenMax = tokenMax; tokenMin = (this.char < 0) ? readPos : readPos - 1; var t = _token(); tokenMax = (this.char < 0) ? readPos - 1 : readPos - 2; return t; } function _token() { #else if( !tokens.isEmpty() ) return tokens.pop(); #end var char; if( this.char < 0 ) char = readChar(); else { char = this.char; this.char = -1; } while( true ) { if( StringTools.isEof(char) ) { this.char = char; return TEof; } switch( char ) { case 0: return TEof; case 32,9,13: // space, tab, CR #if hscriptPos tokenMin++; #end case 10: line++; // LF #if hscriptPos tokenMin++; #end case 48,49,50,51,52,53,54,55,56,57: // 0...9 var n = (char - 48) * 1.0; var exp = 0.; while( true ) { char = readChar(); exp *= 10; switch( char ) { case 48,49,50,51,52,53,54,55,56,57: n = n * 10 + (char - 48); case "e".code, "E".code: var tk = token(); var pow : Null = null; switch( tk ) { case TConst(CInt(e)): pow = e; case TOp("-"): tk = token(); switch( tk ) { case TConst(CInt(e)): pow = -e; default: push(tk); } default: push(tk); } if( pow == null ) invalidChar(char); return TConst(CFloat((Math.pow(10, pow) / exp) * n * 10)); case ".".code: if( exp > 0 ) { // in case of '0...' if( exp == 10 && readChar() == ".".code ) { push(TOp("...")); var i = Std.int(n); return TConst( (i == n) ? CInt(i) : CFloat(n) ); } invalidChar(char); } exp = 1.; case "x".code: if( n > 0 || exp > 0 ) invalidChar(char); // read hexa #if haxe3 var n = 0; while( true ) { char = readChar(); switch( char ) { case 48,49,50,51,52,53,54,55,56,57: // 0-9 n = (n << 4) + char - 48; case 65,66,67,68,69,70: // A-F n = (n << 4) + (char - 55); case 97,98,99,100,101,102: // a-f n = (n << 4) + (char - 87); default: this.char = char; return TConst(CInt(n)); } } #else var n = haxe.Int32.ofInt(0); while( true ) { char = readChar(); switch( char ) { case 48,49,50,51,52,53,54,55,56,57: // 0-9 n = haxe.Int32.add(haxe.Int32.shl(n,4), cast (char - 48)); case 65,66,67,68,69,70: // A-F n = haxe.Int32.add(haxe.Int32.shl(n,4), cast (char - 55)); case 97,98,99,100,101,102: // a-f n = haxe.Int32.add(haxe.Int32.shl(n,4), cast (char - 87)); default: this.char = char; // we allow to parse hexadecimal Int32 in Neko, but when the value will be // evaluated by Interpreter, a failure will occur if no Int32 operation is // performed var v = try CInt(haxe.Int32.toInt(n)) catch( e : Dynamic ) CInt32(n); return TConst(v); } } #end default: this.char = char; var i = Std.int(n); return TConst( (exp > 0) ? CFloat(n * 10 / exp) : ((i == n) ? CInt(i) : CFloat(n)) ); } } case ";".code: return TSemicolon; case "(".code: return TPOpen; case ")".code: return TPClose; case ",".code: return TComma; case ".".code: char = readChar(); switch( char ) { case 48,49,50,51,52,53,54,55,56,57: var n = char - 48; var exp = 1; while( true ) { char = readChar(); exp *= 10; switch( char ) { case 48,49,50,51,52,53,54,55,56,57: n = n * 10 + (char - 48); default: this.char = char; return TConst( CFloat(n/exp) ); } } case ".".code: char = readChar(); if( char != ".".code ) invalidChar(char); return TOp("..."); default: this.char = char; return TDot; } case "{".code: return TBrOpen; case "}".code: return TBrClose; case "[".code: return TBkOpen; case "]".code: return TBkClose; case "'".code, '"'.code: return TConst( CString(readString(char)) ); case "?".code: return TQuestion; case ":".code: return TDoubleDot; case '='.code: char = readChar(); if( char == '='.code ) return TOp("=="); else if ( char == '>'.code ) return TOp("=>"); this.char = char; return TOp("="); case '@'.code: char = readChar(); if( idents[char] || char == ':'.code ) { var id = String.fromCharCode(char); while( true ) { char = readChar(); if( !idents[char] ) { this.char = char; return TMeta(id); } id += String.fromCharCode(char); } } invalidChar(char); case '#'.code: char = readChar(); if( idents[char] ) { var id = String.fromCharCode(char); while( true ) { char = readChar(); if( !idents[char] ) { this.char = char; return preprocess(id); } id += String.fromCharCode(char); } } invalidChar(char); default: if( ops[char] ) { var op = String.fromCharCode(char); while( true ) { char = readChar(); if( StringTools.isEof(char) ) char = 0; if( !ops[char] ) { this.char = char; return TOp(op); } var pop = op; op += String.fromCharCode(char); if( !opPriority.exists(op) && opPriority.exists(pop) ) { if( op == "//" || op == "/*" ) return tokenComment(op,char); this.char = char; return TOp(pop); } } } if( idents[char] ) { var id = String.fromCharCode(char); while( true ) { char = readChar(); if( StringTools.isEof(char) ) char = 0; if( !idents[char] ) { this.char = char; return TId(id); } id += String.fromCharCode(char); } } invalidChar(char); } char = readChar(); } return null; } function preprocValue( id : String ) : Dynamic { return preprocesorValues.get(id); } var preprocStack : Array<{ r : Bool }>; function parsePreproCond() { var tk = token(); return switch( tk ) { case TPOpen: push(TPOpen); parseExpr(); case TId(id): mk(EIdent(id), tokenMin, tokenMax); case TOp("!"): mk(EUnop("!", true, parsePreproCond()), tokenMin, tokenMax); default: unexpected(tk); } } function evalPreproCond( e : Expr ) { switch( expr(e) ) { case EIdent(id): return preprocValue(id) != null; case EUnop("!", _, e): return !evalPreproCond(e); case EParent(e): return evalPreproCond(e); case EBinop("&&", e1, e2): return evalPreproCond(e1) && evalPreproCond(e2); case EBinop("||", e1, e2): return evalPreproCond(e1) || evalPreproCond(e2); default: error(EInvalidPreprocessor("Can't eval " + expr(e).getName()), readPos, readPos); return false; } } function preprocess( id : String ) : Token { switch( id ) { case "if": var e = parsePreproCond(); if( evalPreproCond(e) ) { preprocStack.push({ r : true }); return token(); } preprocStack.push({ r : false }); skipTokens(); return token(); case "else", "elseif" if( preprocStack.length > 0 ): if( preprocStack[preprocStack.length - 1].r ) { preprocStack[preprocStack.length - 1].r = false; skipTokens(); return token(); } else if( id == "else" ) { preprocStack.pop(); preprocStack.push({ r : true }); return token(); } else { // elseif preprocStack.pop(); return preprocess("if"); } case "end" if( preprocStack.length > 0 ): preprocStack.pop(); return token(); default: return TPrepro(id); } } function skipTokens() { var spos = preprocStack.length - 1; var obj = preprocStack[spos]; var pos = readPos; while( true ) { var tk = token(); if( tk == TEof ) error(EInvalidPreprocessor("Unclosed"), pos, pos); if( preprocStack[spos] != obj ) { push(tk); break; } } } function tokenComment( op : String, char : Int ) { var c = op.charCodeAt(1); var s = input; if( c == '/'.code ) { // comment while( char != '\r'.code && char != '\n'.code ) { char = readChar(); if( StringTools.isEof(char) ) break; } this.char = char; return token(); } if( c == '*'.code ) { /* comment */ var old = line; if( op == "/**/" ) { this.char = char; return token(); } while( true ) { while( char != '*'.code ) { if( char == '\n'.code ) line++; char = readChar(); if( StringTools.isEof(char) ) { line = old; error(EUnterminatedComment, tokenMin, tokenMin); break; } } char = readChar(); if( StringTools.isEof(char) ) { line = old; error(EUnterminatedComment, tokenMin, tokenMin); break; } if( char == '/'.code ) break; } return token(); } this.char = char; return TOp(op); } function constString( c ) { return switch(c) { case CInt(v): Std.string(v); case CFloat(f): Std.string(f); case CString(s): s; // TODO : escape + quote #if !haxe3 case CInt32(v): Std.string(v); #end } } function tokenString( t ) { return switch( t ) { case TEof: ""; case TConst(c): constString(c); case TId(s): s; case TOp(s): s; case TPOpen: "("; case TPClose: ")"; case TBrOpen: "{"; case TBrClose: "}"; case TDot: "."; case TComma: ","; case TSemicolon: ";"; case TBkOpen: "["; case TBkClose: "]"; case TQuestion: "?"; case TDoubleDot: ":"; case TMeta(id): "@" + id; case TPrepro(id): "#" + id; } } }