commonMain.com.apollographql.apollo3.api.internal.ResponseParser.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of apollo-api-jvm Show documentation
Apollo GraphQL API classes
There is a newer version: 4.0.0-beta.7
package com.apollographql.apollo3.api.internal

import com.apollographql.apollo3.api.ApolloResponse
import com.apollographql.apollo3.api.CustomScalarAdapters
import com.apollographql.apollo3.api.Error
import com.apollographql.apollo3.api.Operation
import com.apollographql.apollo3.api.json.JsonReader
import com.apollographql.apollo3.api.json.MapJsonReader
import com.apollographql.apollo3.api.json.readAny
import com.apollographql.apollo3.api.nullable
import com.benasher44.uuid.uuid4
import okio.use

/**
 * [ResponseParser] parses network responses, including data, errors and extensions from a [JsonReader]
 */
internal object ResponseParser {
  fun  parse(
      jsonReader: JsonReader,
      operation: Operation,
      customScalarAdapters: CustomScalarAdapters,
  ): ApolloResponse  {
    @Suppress("NAME_SHADOWING")
    return jsonReader.use { jsonReader ->
      jsonReader.beginObject()

      var data: D? = null
      var errors: List? = null
      var extensions: Map? = null
      while (jsonReader.hasNext()) {
        @Suppress("UNCHECKED_CAST")
        when (jsonReader.nextName()) {
          "data" -> data = operation.adapter().nullable().fromJson(jsonReader, customScalarAdapters)
          "errors" -> errors = jsonReader.readErrors()
          "extensions" -> extensions = jsonReader.readAny() as? Map
          else -> jsonReader.skipValue()
        }
      }

      jsonReader.endObject()

      ApolloResponse.Builder(requestUuid = uuid4(), operation = operation, data = data).errors(errors)
          .extensions(extensions)
          .build()
    }
  }

  fun parseError(
      payload: Map,
  ) = MapJsonReader(payload).readError()

  @Suppress("UNCHECKED_CAST")
  private fun JsonReader.readErrors(): List {
    if (peek() == JsonReader.Token.NULL) {
      nextNull()
      return emptyList()
    }

    beginArray()
    val list = mutableListOf()
    while (hasNext()) {
      list.add(readError())
    }
    endArray()
    return list
  }

  @Suppress("UNCHECKED_CAST")
  private fun JsonReader.readError(): Error {
    var message = ""
    var locations: List? = null
    var path: List? = null
    var extensions: Map? = null
    var nonStandardFields: MutableMap? = null
    beginObject()
    while (hasNext()) {
      when (val name = nextName()) {
        "message" -> message = nextString() ?: ""
        "locations" -> {
          locations = readErrorLocations()
        }
        "path" -> {
          path = readPath()
        }
        "extensions" -> {
          extensions = readAny() as? Map?
        }
        else -> {
          if (nonStandardFields == null) nonStandardFields = mutableMapOf()
          nonStandardFields[name] = readAny()
        }
      }
    }
    endObject()

    return Error(message, locations, path, extensions, nonStandardFields)
  }

  private fun JsonReader.readPath(): List? {
    if (peek() == JsonReader.Token.NULL) {
      return nextNull()
    }

    val list = mutableListOf()
    beginArray()
    while (hasNext()) {
      when (peek()) {
        JsonReader.Token.NUMBER, JsonReader.Token.LONG -> list.add(nextInt())
        else -> list.add(nextString()!!)
      }
    }
    endArray()

    return list
  }

  private fun JsonReader.readErrorLocations(): List? {
    if (peek() == JsonReader.Token.NULL) {
      return nextNull()
    }
    val list = mutableListOf()
    beginArray()
    while (hasNext()) {
      list.add(readErrorLocation())
    }
    endArray()

    return list
  }

  private fun JsonReader.readErrorLocation(): Error.Location {
    var line: Int = -1
    var column: Int = -1
    beginObject()
    while (hasNext()) {
      when (nextName()) {
        "line" -> line = nextInt()
        "column" -> column = nextInt()
        else -> skipValue()
      }
    }
    endObject()
    return Error.Location(line, column)
  }
}