summaryrefslogtreecommitdiff
path: root/scalatexApi/src/main/scala/scalatex/stages/IndentHandler.scala
diff options
context:
space:
mode:
Diffstat (limited to 'scalatexApi/src/main/scala/scalatex/stages/IndentHandler.scala')
-rw-r--r--scalatexApi/src/main/scala/scalatex/stages/IndentHandler.scala16
1 files changed, 11 insertions, 5 deletions
diff --git a/scalatexApi/src/main/scala/scalatex/stages/IndentHandler.scala b/scalatexApi/src/main/scala/scalatex/stages/IndentHandler.scala
index 180d134..697ccc3 100644
--- a/scalatexApi/src/main/scala/scalatex/stages/IndentHandler.scala
+++ b/scalatexApi/src/main/scala/scalatex/stages/IndentHandler.scala
@@ -49,8 +49,15 @@ object IndentHandler extends (String => String){
// println(spaces, current)
val declRemainder = successRemainder(Parser.parse(current.trim, _.templateDeclaration()))
-
- val exprRemainder = successRemainder(Parser.parse(current.trim, _.expression())).filter(_ == current.trim)
+// println("::::::" + current.trim)
+// println(successRemainder(Parser.parse(current.trim, _.expression())))
+ val exprRemainder = successRemainder(Parser.parse(current.trim, _.expression())).filter {
+ // Either it takes up the entire line or the line ends with a =>, in which case
+ // we assume the guy wanted to pass a lambda
+ x =>
+// println(x)
+ x == current.trim || current.trim.endsWith("=>")
+ }
/**
@@ -81,14 +88,13 @@ object IndentHandler extends (String => String){
val newFirst =
if (!before.startsWith("@else")) before
else before.dropRight("@else".length)+ "} else"
-
+ println(before, after, delta)
if (delta > 0 && noBraceLine(after)) newFirst + "{" + after
else if (delta <= 0 && noBraceLine(after) && after.trim != "") newFirst + "{" + after + "}" * (1 - elseCorrection)
else current
-
-
}
+
val closing = "}" * (-delta - elseCorrection)
val newStack =