1 files changed, 20 insertions, 69 deletions
diff --git a/src/main/scala/cc/spray/boilerplate/Generator.scala b/src/main/scala/cc/spray/boilerplate/Generator.scala
index d171d7b..9a9b5ab 100644
--- a/src/main/scala/cc/spray/boilerplate/Generator.scala
+++ b/src/main/scala/cc/spray/boilerplate/Generator.scala
@@ -1,75 +1,26 @@
 package cc.spray.boilerplate
 
-import sbt._
-import sbt.Keys._
+import util.parsing.combinator.RegexParsers
 
 object Generator {
-  def generateFromTemplates(streams: TaskStreams, sourceDir: File, targetDir: File): Seq[File] = {
-    val files = sourceDir ** "*.template"
-
-    def changeExtension(f: File): File = {
-      val (ext, name) = f.getName.reverse.span(_ != '.')
-      new File(f.getParent, name.reverse.toString+"scala")
-    }
-
-    val mapping = (files x rebase(sourceDir, targetDir)).map {
-      case (orig, target) => (orig, changeExtension(target))
-    }
-
-    mapping foreach {
-      case (templateFile, target) =>
-        if (templateFile.lastModified > target.lastModified) {
-          streams.log.info("Generating '%s'" format target.getName)
-          val template = IO.read(templateFile)
-          IO.write(target, generateFromTemplate(template))
-        } else
-          streams.log.debug("Template '%s' older than target. Ignoring." format templateFile.getName)
-    }
-
-    mapping.map(_._2)
-  }
-
-  val ExpandReplacement = """(?s:\*\{(.*)\}\*)""".r
-  def generateFromTemplate(template: String): String =
-    ExpandReplacement.replaceAllIn(template, { m =>
-      val format = m.group(1)
-      (1 to 22).map(generate(format)).mkString
-    })
-
-  val SimpleReplacement = """\[\{(.*)\}\]""".r
-  val EnumerateReplacement = """\{\{([^}]*)\}([^}]+)?\}""".r
-
-  def generate(format: String)(idx: Int): String = {
-    val numChars = idx.toString.length
-    def formatNum0(num: Int) = num.formatted("%0"+numChars+"d")
-    def formatNumSpace(num: Int) = num.formatted("%"+numChars+"d")
-
-    def replaceSimple(pattern: String) =
-      SimpleReplacement.replaceAllIn(pattern, { m =>
-        val pattern = m.group(1)
-        replaceInPattern(pattern)(idx)
-      })
-
-    def replaceExpand(pattern: String) =
-      EnumerateReplacement.replaceAllIn(pattern, { m =>
-        val pattern = m.group(1)
-        val separator = m.group(2) match {
-          case null =>
-            if (pattern.endsWith(", ") || pattern.contains("\n")) "" else ", "
-          case sep => sep
-        }
-
-        (1 to idx).map(replaceInPattern(pattern)).mkString(separator)
-      })
-    def replaceInPattern(pattern: String)(idx: Int): String =
-             // in likely identifiers replace by '04' etc.
-      pattern.replaceAll("(?<!\\d)1(?!\\d)", formatNum0(idx))
-             .replaceAll("(?<=\\w)(?<!\\d)0(?!\\d)", formatNum0(idx - 1))
-
-             // in other places replace by ' 4' etc.
-             .replaceAll("(?<!\\w)(?<!\\d)1(?!\\d)", formatNumSpace(idx))
-             .replaceAll("(?<!\\w)(?<!\\d)0(?!\\d)", formatNumSpace(idx - 1))
-
-    replaceExpand(replaceSimple(format))
+  def generateFromTemplate(template: String, expandTo: Int): String =
+    generate(TemplateParser.parse(template))(expandTo)
+
+  def formatNum0(num: Int) = num.formatted("%d")
+  def formatNumSpace(num: Int) = num.formatted("%d")
+  def replaceInPattern(pattern: String)(idx: Int): String =
+           // in likely identifiers replace by '04' etc.
+    pattern.replaceAll("(?<=\\w)(?<!\\d)(?<!##)1(?!\\d)", formatNum0(idx))
+           .replaceAll("(?<=\\w)(?<!\\d)(?<!##)0(?!\\d)", formatNum0(idx - 1))
+
+           // in other places replace by ' 4' etc.
+           .replaceAll("(?<!\\w)(?<!\\d)(?<!##)1(?!\\d)", formatNumSpace(idx))
+           .replaceAll("(?<!\\w)(?<!\\d)(?<!##)0(?!\\d)", formatNumSpace(idx - 1))
+
+  def generate(format: TemplateElement)(idx: Int): String = format match {
+    case Sequence(els) => els.map(e => generate(e)(idx)).mkString
+    case Expand(inner, sep) => (1 to idx).map(generate(inner)).mkString(sep)
+    case LiteralString(lit) => replaceInPattern(lit)(idx)
+    case FixedString(lit) => lit
   }
 }