com.apollographql.apollo.compiler.ir.Operation.kt Maven / Gradle / Ivy

/**
 * Copyright 2018-2019 Amazon.com,
 * Inc. or its affiliates. All Rights Reserved.
 *
 * SPDX-License-Identifier: Apache-2.0
 */

package com.apollographql.apollo.compiler.ir

import com.apollographql.apollo.api.Operation
import com.apollographql.apollo.compiler.SchemaTypeSpecBuilder
import com.apollographql.apollo.compiler.withBuilder
import com.squareup.javapoet.TypeSpec
import javax.lang.model.element.Modifier

data class Operation(
    val operationName: String,
    val operationType: String,
    val variables: List,
    val source: String,
    val fields: List,
    val filePath: String,
    val fragmentsReferenced: List,
    val operationId: String

) : CodeGenerator {
  override fun toTypeSpec(context: CodeGenerationContext): TypeSpec =
      SchemaTypeSpecBuilder(
          typeName = DATA_TYPE_NAME,
          fields = fields,
          fragmentSpreads = emptyList(),
          inlineFragments = emptyList(),
          context = context
      )
          .build(Modifier.PUBLIC, Modifier.STATIC)
          .toBuilder()
          .addSuperinterface(Operation.Data::class.java)
          .build()
          .let {
            if (context.generateModelBuilder) {
              it.withBuilder()
            } else {
              it
            }
          }

    fun normalizedOperationName(useSemanticNaming: Boolean): String = when (operationType) {
        TYPE_MUTATION -> normalizedOperationName(useSemanticNaming, "Mutation")
        TYPE_QUERY -> normalizedOperationName(useSemanticNaming, "Query")
        TYPE_SUBSCRIPTION -> normalizedOperationName(useSemanticNaming, "Subscription")
        else -> throw IllegalArgumentException("Unknown operation type $operationType")
    }

    private fun normalizedOperationName(useSemanticNaming: Boolean, operationNameSuffix: String): String {
        return if (useSemanticNaming && !operationName.endsWith(operationNameSuffix)) {
            operationName.capitalize() + operationNameSuffix
        } else {
            operationName.capitalize()
        }
    }

    fun isMutation() = operationType == TYPE_MUTATION

    fun isQuery() = operationType == TYPE_QUERY

    fun isSubscription() = operationType == TYPE_SUBSCRIPTION

  companion object {
    val DATA_TYPE_NAME = "Data"
    val TYPE_MUTATION = "mutation"
    val TYPE_QUERY = "query"
    val TYPE_SUBSCRIPTION = "subscription"
  }
}