Add string interpolation

HaxeFoundation · May 4, 2017 · b92e8a8 · b92e8a8
1 parent 51506e0
commit b92e8a8
Show file tree

Hide file tree

Showing 2 changed files with 72 additions and 2 deletions.
diff --git a/hscript/Expr.hx b/hscript/Expr.hx
@@ -27,7 +27,7 @@ package hscript;
 enum Const {
 	CInt( v : Int );
 	CFloat( f : Float );
-	CString( s : String );
+	CString( s : String, ?interpolated:Bool );
 	#if !haxe3
 	CInt32( v : haxe.Int32 );
 	#end

diff --git a/hscript/Parser.hx b/hscript/Parser.hx
@@ -310,6 +310,12 @@ class Parser {
 				e = mk(EIdent(id));
 			return parseExprNext(e);
 		case TConst(c):
+			switch (c) {
+				case CString(s, interpolated):
+					if (interpolated)
+						return parseExprNext(interpolate(s));
+				default:
+			}
 			return parseExprNext(mk(EConst(c)));
 		case TPOpen:
 			var e = parseExpr();
@@ -389,6 +395,70 @@ class Parser {
 			return unexpected(tk);
 		}
 	}
+
+	function interpolate(s:String) {
+		var exprs:Array<Expr> = [];
+		var dollarIndex = s.indexOf('$');
+
+		while (dollarIndex > -1) {
+			var i = dollarIndex;
+			var char = s.charAt(++i);
+			switch (char) {
+				case '$':
+					s = s.substring(0, i) + s.substr(i + 1);
+				case '{':
+					var expr = "";
+					var depth = 0;
+
+					var precedingSub = s.substring(0, i - 1);
+					if (precedingSub != "")
+						exprs.push(mk(EConst(CString(precedingSub))));
+
+					while (depth >= 0) {
+						char = s.charAt(++i);
+						if (char == '{')
+							depth++;
+						else if (char == '}') {
+							depth--;
+							if (depth < 0)
+								break;
+						}
+						expr += char;
+					}
+
+					exprs.push(parseString(expr #if hscriptPos, origin #end));
+					s = s.substr(i + 1);
+				case c if (c == '_' || c >= 'a' && c <= 'z' || c >= 'A' && c <= 'Z'): // [a-zA-Z_]
+					var ident = c;
+					var precedingSub = s.substring(0, i - 1);
+					if (precedingSub != "")
+						exprs.push(mk(EConst(CString(precedingSub))));
+					while (true) {
+						char = s.charAt(++i);
+						if (char == '_' 
+						|| char >= 'a' && char <= 'z' 
+						|| char >= 'A' && char <= 'Z' 
+						|| char >= '0' && char <= '9') // [a-zA-Z0-9_]
+							ident += char;
+						else break;
+					}
+					exprs.push(mk(EIdent(ident)));
+					s = s.substr(i);
+			}
+			dollarIndex = s.indexOf('$');
+		}
+
+		if (exprs.length == 0)
+			exprs.push(mk(EConst(CString(s))));
+
+		var expr = exprs[0];
+		for (i in 1...exprs.length) {
+			var nextExpr = exprs[i];
+			expr = mk(EBinop('+', expr, nextExpr));
+		}
+
+		return expr;
+	}
 
 	function mapCompr( tmp : String, e : Expr ) {
 		var edef = switch( expr(e) ) {
@@ -1070,7 +1140,7 @@ class Parser {
 			case 125: return TBrClose;
 			case 91: return TBkOpen;
 			case 93: return TBkClose;
-			case 39: return TConst( CString(readString(39)) );
+			case 39: return TConst( CString(readString(39), true) );
 			case 34: return TConst( CString(readString(34)) );
 			case 63: return TQuestion;
 			case 58: return TDoubleDot;