• Home
• com.twitter
• scrooge-generator_2.11
• 4.7.0
• source code
• Document.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.twitter.scrooge.AST.Document.scala Maven / Gradle / Ivy

package com.twitter.scrooge.ast

case class Document(headers: Seq[Header], defs: Seq[Definition]) extends DocumentNode {
  def namespace(language: String): Option[Identifier] = {
    (headers collect {
      // first try to find language specific namespace scope
      case Namespace(l, x) if l == language => x
    }).headOption orElse(headers collect {
      // then see if universal namespace scope is defined
      case Namespace(l, x) if l == "*" => x
    }).headOption
  }

  def mapNamespaces(namespaceMap: Map[String,String]): Document = {
    copy(
      headers = headers map {
        case header @ Namespace(_, ns) => {
          namespaceMap.get(ns.fullName) map {
            newNs => header.copy(id = Identifier(newNs))
          } getOrElse(header)
        }
        case include @ Include(_, doc) => {
          include.copy(document = doc.mapNamespaces(namespaceMap))
        }
        case header => header
      }
    )
  }

  def consts: Seq[ConstDefinition] = defs.collect { case c: ConstDefinition => c }
  def enums: Seq[Enum] = defs.collect { case e: Enum => e }
  def structs: Seq[StructLike] = defs.collect { case s: StructLike => s }
  def services: Seq[Service] = defs.collect { case s: Service => s }
}