commonMain.com.apollographql.apollo.api.internal.SimpleResponseWriter.kt Maven / Gradle / Ivy

Go to download
package com.apollographql.apollo.api.internal

import com.apollographql.apollo.api.ResponseField
import com.apollographql.apollo.api.ScalarType
import com.apollographql.apollo.api.ScalarTypeAdapters
import com.apollographql.apollo.api.internal.json.JsonWriter
import com.apollographql.apollo.api.internal.json.Utils.writeToJson
import com.apollographql.apollo.api.internal.json.use
import okio.Buffer
import okio.IOException

class SimpleResponseWriter(private val scalarTypeAdapters: ScalarTypeAdapters) : ResponseWriter {
  private val data: MutableMap = LinkedHashMap()

  @Throws(IOException::class)
  fun toJson(indent: String?): String {
    return Buffer().apply {
      JsonWriter.of(this).use { jsonWriter ->
        jsonWriter.indent = indent
        jsonWriter.beginObject()
        jsonWriter.name("data")
        writeToJson(data, jsonWriter)
        jsonWriter.endObject()
      }
    }.readUtf8()
  }

  override fun writeString(field: ResponseField, value: String?) {
    data[field.responseName] = value
  }

  override fun writeInt(field: ResponseField, value: Int?) {
    data[field.responseName] = value
  }

  override fun writeLong(field: ResponseField, value: Long?) {
    data[field.responseName] = value
  }

  override fun writeDouble(field: ResponseField, value: Double?) {
    data[field.responseName] = value
  }

  override fun writeBoolean(field: ResponseField, value: Boolean?) {
    data[field.responseName] = value
  }

  override fun writeCustom(field: ResponseField.CustomTypeField, value: Any?) {
    if (value == null) {
      data[field.responseName] = null
    } else {
      val typeAdapter = scalarTypeAdapters.adapterFor(field.scalarType)
      val customTypeValue = typeAdapter.encode(value)
      data[field.responseName] = customTypeValue.value
    }
  }

  override fun writeObject(field: ResponseField, marshaller: ResponseFieldMarshaller?) {
    if (marshaller == null) {
      data[field.responseName] = null
    } else {
      val objectResponseWriter = SimpleResponseWriter(scalarTypeAdapters)
      marshaller.marshal(objectResponseWriter)
      data[field.responseName] = objectResponseWriter.data
    }
  }

  override fun writeFragment(marshaller: ResponseFieldMarshaller?) {
    marshaller?.marshal(this)
  }

  override fun  writeList(field: ResponseField, values: List?, listWriter: ResponseWriter.ListWriter) {
    if (values == null) {
      data[field.responseName] = null
    } else {
      val listItemWriter = CustomListItemWriter(scalarTypeAdapters)
      listWriter.write(values, listItemWriter)
      data[field.responseName] = listItemWriter.data
    }
  }

  private class CustomListItemWriter internal constructor(private val scalarTypeAdapters: ScalarTypeAdapters) : ResponseWriter.ListItemWriter {
    internal val data = ArrayList()

    override fun writeString(value: String?) {
      data.add(value)
    }

    override fun writeInt(value: Int?) {
      data.add(value)
    }

    override fun writeLong(value: Long?) {
      data.add(value)
    }

    override fun writeDouble(value: Double?) {
      data.add(value)
    }

    override fun writeBoolean(value: Boolean?) {
      data.add(value)
    }

    override fun writeCustom(scalarType: ScalarType, value: Any?) {
      if (value == null) {
        data.add(null)
      } else {
        val typeAdapter = scalarTypeAdapters.adapterFor(scalarType)
        val customTypeValue = typeAdapter.encode(value)
        data.add(customTypeValue.value)
      }
    }

    override fun writeObject(marshaller: ResponseFieldMarshaller?) {
      if (marshaller == null) {
        data.add(null)
      } else {
        val objectResponseWriter = SimpleResponseWriter(scalarTypeAdapters)
        marshaller.marshal(objectResponseWriter)
        data.add(objectResponseWriter.data)
      }
    }

    override fun  writeList(items: List?, listWriter: ResponseWriter.ListWriter) {
      if (items == null) {
        data.add(null)
      } else {
        val listItemWriter = CustomListItemWriter(scalarTypeAdapters)
        listWriter.write(items, listItemWriter)
        data.add(listItemWriter.data)
      }
    }
  }
}