10572324896

Committed 27 Aug 2024 05:35AM UTC coverage: 96.742% (+0.1%) from 96.633%

Build Type

push

Committed by

Commit Message

minor code cleanup

Run Details

1 of 2 new or added lines in 2 files covered. (50.0%)

1 existing line in 1 file now uncovered.

861 of 890 relevant lines covered (96.74%)

1.83 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

99.05

/src/main/scala/tscfg/generators/java/JavaGen.scala

package tscfg.generators.java

import tscfg.DefineCase.ExtendsDefineCase
import tscfg.ModelBuilder
import tscfg.codeDefs.javaDef
import tscfg.exceptions.ObjectDefinitionException
import tscfg.generators._
import tscfg.generators.java.javaUtil._
import tscfg.model._
import tscfg.ns.NamespaceMan
import tscfg.util.escapeString

class JavaGen(
    genOpts: GenOpts,
    rootNamespace: NamespaceMan = new NamespaceMan
) extends Generator(genOpts) {

  import defs._

  implicit val methodNames: MethodNames = MethodNames()

  def generate(objectType: ObjectType): GenResult = {
    genResults = GenResult()

    methodNames.checkUserSymbol(className)
    val res = generateForObj(objectType, className = className, isRoot = true)

    val packageStr = s"package ${genOpts.packageName};\n\n"

    val definition = (packageStr + res.definition).trim
    genResults.copy(code = definition)
  }

  private def generate(
      typ: Type,
      classNamePrefixOpt: Option[String],
      className: String,
      annTypeForAbstractClassName: Option[AnnType] = None
  ): Res = typ match {

    case et: EnumObjectType =>
      generateForEnum(et, classNamePrefixOpt, className)

    case ot: ObjectType =>
      generateForObj(
        ot,
        classNamePrefixOpt,
        className,
        annTypeForAbstractClassName = annTypeForAbstractClassName
      )

    case aot: AbstractObjectType =>
      generateForObj(
        aot,
        classNamePrefixOpt,
        className,
        annTypeForAbstractClassName = annTypeForAbstractClassName
      );

    case ort: ObjectRefType => generateForObjRef(ort)

    case lt: ListType => generateForList(lt, classNamePrefixOpt, className)

    case bt: BasicType => generateForBasic(bt)
  }

  private def generateForObj(
      ot: ObjectRealType,
      classNamePrefixOpt: Option[String] = None,
      className: String,
      isRoot: Boolean = false,
      annTypeForAbstractClassName: Option[AnnType] = None
  ): Res = {

    val classNameAdjusted = adjustClassName(className)
    genResults =
      genResults.copy(classNames = genResults.classNames + classNameAdjusted)

    val symbols = ot.members.keys.toList.sorted
    symbols.foreach(methodNames.checkUserSymbol)

    val results = symbols.map { symbol =>
      val a = ot.members(symbol)
      val res = generate(
        a.t,
        classNamePrefixOpt = Some(classNameAdjusted + "."),
        className = javaUtil.getClassName(symbol),
        annTypeForAbstractClassName = Some(a)
      )
      (symbol, res, a)
    }

    val classDeclMembers = results.map { case (symbol, res, a) =>
      val memberType = res.javaType
      val typeFull = if (a.optional && a.default.isEmpty) {
        val typeActual = a.t match {
          case ObjectRefType(_, simpleName) => simpleName
          case _                            => toObjectType(memberType).toString
        }

        if (genOpts.useOptionals)
          s"java.util.Optional<$typeActual>"
        else typeActual
      }
      else {
        a.t match {
          case ObjectRefType(_, simpleName) => simpleName
          case _                            => memberType.toString
        }
      }
      val javaId = javaIdentifier(symbol)
      (typeFull, javaId, a)
    }

    val classDeclMembersStr = classDeclMembers
      .flatMap { case (type_, javaId, a) =>
        if (a.isDefine) None
        else
          Some {
            genResults = genResults
              .copy(fields = genResults.fields + (javaId -> type_))
            if (genOpts.genRecords)
              s"$type_ $javaId" + dbg("<decl>")
            else
              s"public final $type_ $javaId;" + dbg("<decl>")
          }
      }
      .mkString(if (genOpts.genRecords) ",\n  " else "\n  ")

    val classMemberGettersStr = if (genOpts.genGetters) {
      classDeclMembers
        .filterNot(_._3.isDefine)
        .map { case (typ, javaId, _) =>
          val getter = s"get${javaId.capitalize}"
          genResults = genResults
            .copy(getters = genResults.getters + (getter -> typ))
          s"public final $typ $getter() { return $javaId; }"
        }
        .mkString("\n  ", "\n  ", "")
    }
    else ""

    val membersStr = {
      val defined = results.map(_._2.definition).filter(_.nonEmpty)
      if (defined.isEmpty) ""
      else {
        defined.map(_.replaceAll("\n", "\n  ")).mkString("\n  ")
      }
    }

    implicit val listAccessors =
      collection.mutable.LinkedHashMap[String, String]()

    val ctorMembersStr = if (genOpts.genRecords) {
      results
        .flatMap { case (symbol, res, a) =>
          a.defineCase match {
            case Some(dc) if dc.isAbstract =>
              throw ObjectDefinitionException(
                s"record cannot be abstract: $symbol"
              )
            case Some(_: ExtendsDefineCase) =>
              throw ObjectDefinitionException(
                s"record cannot extend classes: $symbol"
              )
            case Some(_) =>
              None
            case None =>
              Some(
                instance(a, res, path = escapeString(symbol), isRoot = isRoot)
              )
          }
        }
        .mkString("this(\n      ", ",\n      ", "\n    );")
    }
    else {
      results
        .flatMap { case (symbol, res, a) =>
          if (a.isDefine) None
          else
            Some {
              val javaId = javaIdentifier(symbol)
              "this." + javaId + " = " + instance(
                a,
                res,
                path = escapeString(symbol)
              ) + ";"
            }
        }
        .mkString("\n    ")
    }

    val elemAccessorsStr = {
      val objOnes =
        if (listAccessors.isEmpty) ""
        else {
          "\n" + listAccessors.keys.toList.sorted
            .map { methodName =>
              listAccessors(methodName)
            }
            .mkString("\n")
        }
      val rootOnes =
        if (!isRoot) ""
        else {
          if (rootListAccessors.isEmpty) ""
          else {
            "\n\n" + rootListAccessors.keys.toList.sorted
              .map { methodName =>
                rootListAccessors(methodName)
              }
              .mkString("\n")
          }
        }
      objOnes + rootOnes
    }

    val rootAuxClasses = if (isRoot) {
      javaDef("$TsCfgValidator")
    }
    else ""

    val (ctorParams, errHandlingDecl, errHandlingDispatch) = if (isRoot) {
      (
        "com.typesafe.config.Config c",
        if (genOpts.genRecords)
          ""
        else
          """final $TsCfgValidator $tsCfgValidator = new $TsCfgValidator();
            |    final java.lang.String parentPath = "";
            |    """.stripMargin,
        s"""
             |    $$tsCfgValidator.validate();""".stripMargin
      )
    }
    else
      (
        "com.typesafe.config.Config c, java.lang.String parentPath, $TsCfgValidator $tsCfgValidator",
        "",
        ""
      )

    val (extendsString, superString) = {
      annTypeForAbstractClassName match {
        case None =>
          ("", "")

        case Some(annType) =>
          annType.nameIsImplementsIsExternal match {
            case Some((cn, isImplements, isExternal)) =>
              if (isImplements)
                (s"implements $cn ", "")
              else if (isExternal)
                (s"extends $cn ", "")
              else
                (
                  s"extends $cn ",
                  "\n    super(c, parentPath, $tsCfgValidator);"
                )

            case None =>
              ("", "")
          }
      }
    }

    val classStr =
      ot match {
        case _: ObjectType if genOpts.genRecords =>
          val beginningBody =
            if (isRoot)
              "\n  static final $TsCfgValidator $tsCfgValidator = new $TsCfgValidator();\n"
            else ""

          s"""public ${if (isRoot) "" else "static "}record $classNameAdjusted(
             |  $classDeclMembersStr$classMemberGettersStr
             |) $extendsString{$beginningBody
             |  $membersStr
             |  public $classNameAdjusted($ctorParams) {$superString
             |    $errHandlingDecl$ctorMembersStr$errHandlingDispatch
             |  }$elemAccessorsStr
             |$rootAuxClasses}
             |""".stripMargin

        case _: ObjectType =>
          val staticStr = if (isRoot) "" else "static "
          s"""public ${staticStr}class $classNameAdjusted $extendsString{
             |  $classDeclMembersStr$classMemberGettersStr
             |  $membersStr
             |  public $classNameAdjusted($ctorParams) {$superString
             |    $errHandlingDecl$ctorMembersStr$errHandlingDispatch
             |  }$elemAccessorsStr
             |$rootAuxClasses}
             |""".stripMargin

        case _: AbstractObjectType =>
          s"""public abstract static class $classNameAdjusted $extendsString{
             |  $classDeclMembersStr$classMemberGettersStr
             |  $membersStr
             |  public $classNameAdjusted($ctorParams) {$superString
             |    $errHandlingDecl$ctorMembersStr$errHandlingDispatch
             |  }$elemAccessorsStr
             |}
             |""".stripMargin
      }

    val baseType =
      classNamePrefixOpt.getOrElse("") + classNameAdjusted + dbg("<Y>")
    Res(ot, javaType = BaseJavaType(baseType), definition = classStr)
  }

  private def generateForObjRef(ort: ObjectRefType): Res = {
    val className = ort.simpleName
    genResults = genResults.copy(classNames = genResults.classNames + className)

    Res(ort, javaType = BaseJavaType(className + dbg("<X>")))
  }

  private def generateForList(
      lt: ListType,
      classNamePrefixOpt: Option[String],
      className: String
  ): Res = {
    val className2 =
      className + (if (className.endsWith("$Elm")) "" else "$Elm")
    val elem        = generate(lt.t, classNamePrefixOpt, className2)
    val elemRefType = toObjectType(elem.javaType)
    Res(lt, javaType = ListJavaType(elemRefType), definition = elem.definition)
  }

  private def generateForBasic(b: BasicType): Res = {
    Res(
      b,
      javaType = BaseJavaType(name = b match {
        case STRING  => "java.lang.String"
        case INTEGER => "int"
        case LONG    => "long"
        case DOUBLE  => "double"
        case BOOLEAN => "boolean"
        case SIZE    => "long"
        case DURATION(_) =>
          if (genOpts.useDurations) "java.time.Duration" else "long"
      })
    )
  }

  private def generateForEnum(
      et: EnumObjectType,
      classNamePrefixOpt: Option[String] = None,
      className: String,
  ): Res = {

    val classNameAdjusted = adjustClassName(className)
    val members           = et.members.mkString("", ",\n  ", ";")
    val str =
      s"""|public enum $classNameAdjusted {
         |  $members
         |}""".stripMargin

    val baseType = classNamePrefixOpt.getOrElse("") + classNameAdjusted
    Res(et, javaType = BaseJavaType(baseType), definition = str)
  }

  private val rootListAccessors =
    collection.mutable.LinkedHashMap[String, String]()

  /** Avoids duplicate class names as they are not supported by Java (the
    * namespaces implied by nesting are not enough to distinguish inner classes
    * with the same name)
    */
  private def adjustClassName(className: String): String = {
    classNameCounter.get(className) match {
      case None =>
        classNameCounter.put(className, 1)
        className
      case Some(counter) =>
        classNameCounter.put(className, counter + 1)
        className + (counter + 1)
    }
  }

  private val classNameCounter = collection.mutable.HashMap[String, Int]()

  private def toObjectType(javaType: JavaType): JavaType = javaType match {
    case BaseJavaType(name) =>
      name match {
        case "int"     => BaseJavaType("java.lang.Integer")
        case "long"    => BaseJavaType("java.lang.Long")
        case "double"  => BaseJavaType("java.lang.Double")
        case "boolean" => BaseJavaType("java.lang.Boolean")
        case _         => javaType
      }
    case other => other
  }

  private def instance(
      a: AnnType,
      res: Res,
      path: String,
      isRoot: Boolean = false
  )(implicit
      listAccessors: collection.mutable.LinkedHashMap[String, String]
  ): String = {

    val objRefResolution: Option[String] = a.t match {
      case ort: ObjectRefType => objectRefInstance(ort, res, path)
      case _                  => None
    }

    objRefResolution.getOrElse {
      a.t match {
        case bt: BasicType    => basicInstance(a, bt, path)
        case _: ObjectAbsType => objectInstance(a, res, path, isRoot)
        case lt: ListType     => listInstance(a, lt, res, path)
      }
    }
  }

  private def objectRefInstance(
      ort: ObjectRefType,
      res: Res,
      path: String
  ): Option[String] = {
    val namespace = rootNamespace.resolve(ort.namespace)
    namespace.getDefine(ort.simpleName) flatMap { t =>
      t match {
        case _: EnumObjectType => Some(enumInstance(res, path))
        case _                 => None
      }
    }
  }

  private def enumInstance(res: Res, path: String): String = {
    val className = res.javaType.toString
    // for now, for java, just let `.valueOf` complain if the value is invalid:
    s"""$className.valueOf(c.getString("$path"))"""
  }

  private def objectInstance(
      a: AnnType,
      res: Res,
      path: String,
      isRoot: Boolean
  )(implicit
      listAccessors: collection.mutable.Map[String, String]
  ): String = {
    val className = res.javaType.toString

    val ppArg =
      if (isRoot)
        s""", "$path.", $$tsCfgValidator"""
      else
        s""", parentPath + "$path.", $$tsCfgValidator"""

    def reqConfigCall = {
      val methodName = "$_reqConfig"
      listAccessors += methodName -> javaDef(methodName)
      s"""$methodName(parentPath, c, "$path", $$tsCfgValidator)"""
    }

    if (genOpts.assumeAllRequired)
      s"""new $className($reqConfigCall$ppArg)"""
    else if (a.optional) {
      if (genOpts.useOptionals) {
        s"""c.$hasPath("$path") ? java.util.Optional.of(new $className(c.getConfig("$path")$ppArg)) : java.util.Optional.empty()"""
      }
      else {
        s"""c.$hasPath("$path") ? new $className(c.getConfig("$path")$ppArg) : null"""
      }
    }
    else {
      // TODO revisit #33 handling of object as always optional
      s"""c.$hasPath("$path") ? new $className(c.getConfig("$path")$ppArg) : new $className(com.typesafe.config.ConfigFactory.parseString("$path{}")$ppArg)"""
    }
  }

  private def listInstance(a: AnnType, lt: ListType, res: Res, path: String)(
      implicit listAccessors: collection.mutable.Map[String, String]
  ): String = {
    val javaType = res.javaType.asInstanceOf[ListJavaType]
    val base     = listMethodName(javaType, lt, path)
    if (a.optional) {
      if (genOpts.useOptionals) {
        s"""c.$hasPath("$path") ? java.util.Optional.of($base) : java.util.Optional.empty()"""
      }
      else {
        s"""c.$hasPath("$path") ? $base : null"""
      }
    }
    else base
  }

  private def basicInstance(a: AnnType, bt: BasicType, path: String)(implicit
      listAccessors: collection.mutable.Map[String, String]
  ): String = {
    val getter = tsConfigUtil.basicGetter(bt, path, genOpts.useDurations)

    a.default match {
      case Some(v) =>
        val value = tsConfigUtil.basicValue(a.t, v, genOpts.useDurations)
        (bt, value) match {
          case (BOOLEAN, "true")  => s"""!c.$hasPath("$path") || c.$getter"""
          case (BOOLEAN, "false") => s"""c.$hasPath("$path") && c.$getter"""
          case (DURATION(_), duration) if genOpts.useDurations =>
            s"""c.$hasPath("$path") ? c.$getter : java.time.Duration.parse("$duration")"""
          case _ => s"""c.$hasPath("$path") ? c.$getter : $value"""
        }

      case None if a.optional && genOpts.useOptionals =>
        s"""c.$hasPath("$path") ? java.util.Optional.of(c.$getter) : java.util.Optional.empty()"""
      case None if a.optional =>
        s"""c.$hasPath("$path") ? c.$getter : null"""

      case _ =>
        bt match {
          case DURATION(_) => s"""c.$getter"""
          case _ =>
            val (methodName, methodCall) =
              tsConfigUtil.basicRequiredGetter(bt, path)
            listAccessors += methodName -> javaDef(methodName)
            methodCall
        }
    }
  }

  private def listMethodName(
      javaType: ListJavaType,
      lt: ListType,
      path: String
  )(implicit
      listAccessors: collection.mutable.Map[String, String],
      methodNames: MethodNames
  ): String = {

    val (_, methodName) = rec(javaType, lt, "")
    methodName + s"""(c.getList("$path"), parentPath, $$tsCfgValidator)"""
  }

  private def rec(ljt: ListJavaType, lt: ListType, prefix: String)(implicit
      listAccessors: collection.mutable.Map[String, String],
      methodNames: MethodNames
  ): (Boolean, String) = {

    val (isBasic, elemMethodName) = ljt.jt match {
      case bst: BaseJavaType =>
        val basic      = lt.t.isInstanceOf[BasicType]
        val methodName = baseName(lt.t, bst.toString)
        if (basic) {
          rootListAccessors += methodName -> methodNames
            .basicElemAccessDefinition(methodName)
          rootListAccessors += methodNames.expE -> methodNames.expEDef
        }
        (basic, methodName)

      case lst: ListJavaType =>
        rec(lst, lt.t.asInstanceOf[ListType], prefix + methodNames.listPrefix)
    }

    val (methodName, methodBody) =
      listMethodDefinition(elemMethodName, ljt.jt, lt)

    if (isBasic)
      rootListAccessors += methodName -> methodBody
    else
      listAccessors += methodName -> methodBody

    (isBasic, methodName)
  }

  private def baseName(t: Type, name: String)(implicit
      methodNames: MethodNames
  ): String = t match {
    case STRING      => methodNames.strA
    case INTEGER     => methodNames.intA
    case LONG        => methodNames.lngA
    case DOUBLE      => methodNames.dblA
    case BOOLEAN     => methodNames.blnA
    case SIZE        => methodNames.sizA
    case DURATION(_) => methodNames.durA

    case _: ObjectAbsType => name.replace('.', '_')

    case _: ListType => throw new AssertionError()
  }

  private def listMethodDefinition(
      elemMethodName: String,
      javaType: JavaType,
      lt: ListType
  )(implicit methodNames: MethodNames): (String, String) = {

    val elem = if (elemMethodName.startsWith(methodNames.listPrefix)) {
      s"$elemMethodName((com.typesafe.config.ConfigList)cv, parentPath, $$tsCfgValidator)"
    }
    else if (elemMethodName.startsWith("$")) {
      s"$elemMethodName(cv)"
    }
    else {
      val adjusted = elemMethodName.replace("_", ".")
      val objRefResolution = lt.t match {
        case ort: ObjectRefType =>
          val namespace = rootNamespace.resolve(ort.namespace)
          namespace.getDefine(ort.simpleName) flatMap { t =>
            t match {
              case _: EnumObjectType =>
                Some(s"""$adjusted.valueOf(cv.unwrapped().toString())""")
              case _ => None
            }
          }

        case _ => None
      }
      objRefResolution.getOrElse {
        s"new $adjusted(((com.typesafe.config.ConfigObject)cv).toConfig(), parentPath, $$tsCfgValidator)"
      }
    }

    val methodName = methodNames.listPrefix + elemMethodName
    val methodDef =
      s"""  private static java.util.List<$javaType> $methodName(com.typesafe.config.ConfigList cl, java.lang.String parentPath, $$TsCfgValidator $$tsCfgValidator) {
         |    java.util.ArrayList<$javaType> al = new java.util.ArrayList<>();
         |    for (com.typesafe.config.ConfigValue cv: cl) {
         |      al.add($elem);
         |    }
         |    return java.util.Collections.unmodifiableList(al);
         |  }""".stripMargin
    (methodName, methodDef)
  }
}

object JavaGen {

  import tscfg.{model, util}

  import _root_.java.io.{File, FileWriter, PrintWriter}

  // $COVERAGE-OFF$
  def generate(
      filename: String,
      showOut: Boolean = false,
      assumeAllRequired: Boolean = false,
      genGetters: Boolean = false,
      genRecords: Boolean = false,
      useOptionals: Boolean = false,
      useDurations: Boolean = false
  ): GenResult = {
    val file   = new File("src/main/tscfg/" + filename)
    val source = io.Source.fromFile(file).mkString.trim

    if (showOut)
      println("source:\n  |" + source.replaceAll("\n", "\n  |"))

    val className = "Java" + {
      val noPath = filename.substring(filename.lastIndexOf('/') + 1)
      val noDef  = noPath.replaceAll("""^def\.""", "")
      val symbol = noDef.substring(0, noDef.indexOf('.'))
      util.upperFirst(symbol) + "Cfg"
    }

    val rootNamespace = new NamespaceMan

    val buildResult =
      ModelBuilder(rootNamespace, source, assumeAllRequired = assumeAllRequired)
    val objectType = buildResult.objectType
    if (showOut) {
      println(
        "\nobjectType:\n  |" + model.util
          .format(objectType)
          .replaceAll("\n", "\n  |")
      )
      if (buildResult.warnings.nonEmpty) {
        println("warnings:")
        buildResult.warnings.foreach(w =>
          println(s"   line ${w.line}: ${w.source}: ${w.message}")
        )
      }
    }

    val genOpts = GenOpts(
      "tscfg.example",
      className,
      genGetters = genGetters,
      genRecords = genRecords,
      useOptionals = useOptionals,
      assumeAllRequired = assumeAllRequired,
      useDurations = useDurations
    )

    val generator = new JavaGen(genOpts, rootNamespace)

    val results = generator.generate(objectType)

    // println("\n" + results.code)

    val destFilename = s"src/test/java/tscfg/example/$className.java"
    val destFile     = new File(destFilename)
    val out          = new PrintWriter(new FileWriter(destFile), true)
    out.println(results.code)
    out.close()
    results
  }

  def main(args: Array[String]): Unit = {
    val filename = args.headOption.getOrElse("example/example.conf")
    val results  = generate(filename, showOut = true)
    println(s"""classNames: ${results.classNames}
         |fields    : ${results.fields}
      """.stripMargin)
  }

  // $COVERAGE-ON$

}

private[java] object defs {

  abstract sealed class JavaType

  case class BaseJavaType(name: String) extends JavaType {
    override def toString: String = name
  }

  case class ListJavaType(jt: JavaType) extends JavaType {
    override def toString: String = s"java.util.List<$jt>"
  }

  case class Res(typ: Type, javaType: JavaType, definition: String = "")

}

private[java] case class MethodNames() {
  val strA       = "$_str"
  val intA       = "$_int"
  val lngA       = "$_lng"
  val dblA       = "$_dbl"
  val blnA       = "$_bln"
  val durA       = "$_dur" // TODO review this one as it's not actually used
  val sizA       = "$_siz"
  val expE       = "$_expE"
  val listPrefix = "$_L"

  def checkUserSymbol(symbol: String): Unit = {
    if (symbol.startsWith("$_"))
      println(
        s"""
           |WARNING: Symbol $symbol may cause conflict with generated code.
           |         Avoid the $$_ prefix in your spec's identifiers.
         """.stripMargin
      )
  }

  // definition of methods used to access list's elements of basic type
  val basicElemAccessDefinition: Map[String, String] = {
    List(strA, intA, lngA, dblA, blnA, sizA)
      .map(k => k -> javaDef(k))
      .toMap
  }

  val expEDef: String = javaDef(expE)
}

carueda / tscfg / 10572324896

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous