format.specific.avrohuggers.SpecificSchemahugger.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of avrohugger-core_2.13 Show documentation

avrohugger-core

There is a newer version: 2.9.0

package avrohugger
package format
package specific
package avrohuggers

import format.abstractions.avrohuggers.Schemahugger
import trees.{ SpecificCaseClassTree, SpecificObjectTree }
import matchers.TypeMatcher
import stores.{ClassStore, SchemaStore}

import org.apache.avro.Schema
import org.apache.avro.Schema.Type.{ FIXED, RECORD }

import treehugger.forest.Tree

object SpecificSchemahugger extends Schemahugger {

  def toTrees(
    schemaStore: SchemaStore,
    classStore: ClassStore,
    namespace: Option[String],
    schema: Schema,
    typeMatcher: TypeMatcher,
    maybeBaseTrait: Option[String],
    maybeFlags: Option[List[Long]],
    restrictedFields: Boolean,
    targetScalaPartialVersion: String): List[Tree] = {

    schema.getType match {
      case RECORD =>
        val caseClassDef = SpecificCaseClassTree.toCaseClassDef(
          classStore,
          namespace,
          schema,
          typeMatcher,
          maybeBaseTrait,
          maybeFlags,
          restrictedFields,
          targetScalaPartialVersion)
        val companionDef = SpecificObjectTree.toCaseCompanionDef(
          schema,
          maybeFlags,
          schemaStore,
          typeMatcher)
        List(caseClassDef, companionDef)
      case FIXED =>
        val caseClassDef = SpecificCaseClassTree.toFixedDef(
          schema,
          namespace,
          maybeFlags,
          schemaStore,
          typeMatcher,
          classStore,
          targetScalaPartialVersion)
        val companionDef = SpecificObjectTree.toCaseCompanionDef(
          schema,
          maybeFlags,
          schemaStore,
          typeMatcher)
        List(caseClassDef, companionDef)
      case _ => sys.error("Only RECORD or FIXED can be toplevel definitions")

    }
  }

}