src/main/scala/spray/boilerplate/Generator.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

/*
 * sbt-boilerplate is distributed under the 2-Clause BSD license. See the LICENSE file in the root
 * of the repository.
 *
 * Copyright (c) 2012 Johannes Rudolph
 */
package spray.boilerplate

import util.parsing.combinator.RegexParsers

object Generator {
  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)(?<!##)2(?!\\d)", formatNum0(idx + 1))
           .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)(?<!##)2(?!\\d)", formatNumSpace(idx + 1))
           .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
  }
}