scala.class_companion.ssp Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spindle-codegen-templates_2.9.1 Show documentation
spindle-codegen-templates
The newest version!
<%
  // Copyright 2013 Foursquare Labs Inc. All Rights Reserved.

  import com.foursquare.spindle.codegen.runtime.{PrimitiveRenderType, StructLike}
%>
<%@ val cls: StructLike %>
<%@ val clsContainer: String %>
<%
def escapeQuotes(str: String): String = str.replace("\"", "\\\"")
%>\

object ${cls.name} extends ${cls.name}Meta {
<% render("class_companion_builder.ssp", Map("cls" -> cls)) %>

  implicit val companionProvider: ${cls.name}CompanionProvider = new ${cls.name}CompanionProvider
}

class ${cls.name}Meta
    extends #if(clsContainer.nonEmpty)Java${clsContainer}.#endJava${cls.name}Meta[${cls.name}, Raw${cls.name}, ${cls.name}Meta]
#for (pkField <- cls.primaryKeyField)
    with com.foursquare.spindle.HasMetaPrimaryKey[${pkField.renderType.text}, ${cls.name}]
#end
    with com.foursquare.spindle.RecordProvider[${cls.name}] \
{
  override def recordName: String = "${cls.name}"

<% render("class_companion_tfields.ssp", Map("cls" -> cls)) %>

  override def createUntypedRawRecord: com.foursquare.spindle.UntypedRecord = createRawRecord
  override def createRecord: ${cls.name} = createRawRecord
  override def createRawRecord: Raw${cls.name} = new Raw${cls.name}

  override def untypedIfInstanceFrom(x: AnyRef): Option[com.foursquare.spindle.UntypedRecord] = ifInstanceFrom(x)
  override def ifInstanceFrom(x: AnyRef): Option[${cls.name}] = {
    if (x.isInstanceOf[${cls.name}]) Some(x.asInstanceOf[${cls.name}]) else None
  }

  override val annotations: com.foursquare.spindle.Annotations =
#if (cls.annotations.nonEmpty)
    new com.foursquare.spindle.Annotations(scala.collection.immutable.Vector(
      ${cls.annotations.toSeq.map(annot => "(\"%s\", \"%s\")".format(annot._1, escapeQuotes(annot._2))).mkString(",\n      ")}
    ))
#else
    com.foursquare.spindle.Annotations.empty
#end

<% render("class_companion_rfields.ssp", Map("cls" -> cls)) %>

#for (pkField <- cls.primaryKeyField)
  override def primaryKey: com.foursquare.spindle.FieldDescriptor[${pkField.renderType.text}, ${cls.name}, ${cls.name}Meta] = this.${pkField.name}
#end
\

  def apply(
      ${cls.fields.map(field => "%s: %s".format(field.escapedName, field.renderType.text)).mkString(",\n      ")}
  ): ${cls.name} = {
    val ret = this.createRawRecord
#for (field <- cls.fields)
    ret.${field.escapedName}_=(${field.escapedName})
#end
    ret
  }
}

class ${cls.name}CompanionProvider extends com.foursquare.spindle.CompanionProvider[${cls.name}] {
  type CompanionT = ${cls.name}Meta
  override def provide: ${cls.name}Meta = ${cls.name}
}