com.lahzouz.java.graphql.client.maven.plugin.Introspection.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of apollo-client-maven-plugin Show documentation
Maven plugin for generating graphql clients
There is a newer version: 3.3.1
Show newest version
package com.lahzouz.java.graphql.client.maven.plugin

import org.apache.http.HttpHeaders
import org.apache.http.client.entity.GzipDecompressingEntity
import org.apache.http.client.methods.HttpPost
import org.apache.http.config.RegistryBuilder
import org.apache.http.conn.socket.ConnectionSocketFactory
import org.apache.http.conn.socket.PlainConnectionSocketFactory
import org.apache.http.conn.ssl.NoopHostnameVerifier
import org.apache.http.conn.ssl.SSLConnectionSocketFactory
import org.apache.http.entity.StringEntity
import org.apache.http.impl.client.HttpClientBuilder
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager
import org.apache.http.ssl.SSLContextBuilder
import org.apache.http.util.EntityUtils
import java.nio.charset.Charset
import java.security.cert.X509Certificate
import javax.net.ssl.SSLContext

object Introspection {

    fun getIntrospectionSchema(
        host: String,
        useSelfSignedCertificat: Boolean,
        customHeaders: Map
    ): String {

        var clientBuilder = HttpClientBuilder.create().disableContentCompression()

        if (useSelfSignedCertificat) {
            val sslContext: SSLContext = SSLContextBuilder()
                .loadTrustMaterial(null) { _: Array?, _: String? -> true }
                .build()

            clientBuilder = HttpClientBuilder.create()
                .setSSLContext(sslContext)
                .setConnectionManager(
                    PoolingHttpClientConnectionManager(
                        RegistryBuilder.create()
                            .register("http", PlainConnectionSocketFactory.INSTANCE)
                            .register(
                                "https",
                                SSLConnectionSocketFactory(
                                    sslContext,
                                    NoopHostnameVerifier.INSTANCE
                                )
                            )
                            .build()
                    )
                )
        }

        val request = HttpPost(host)
        val body = StringEntity(introspectionQuery)

        request.apply {
            entity = body
            setHeader(HttpHeaders.ACCEPT, "application/json")
            setHeader(HttpHeaders.CONTENT_TYPE, "application/json")
            setHeader(HttpHeaders.ACCEPT_ENCODING, "gzip, identity")
            setHeader(HttpHeaders.ACCEPT_CHARSET, "utf-8")
            setHeader(HttpHeaders.USER_AGENT, "Apollo Client Maven Plugin")
            customHeaders.forEach { (name, value) -> setHeader(name, value) }
        }

        val client = clientBuilder.build()
        val response = client.execute(request)
        var schema = ""
        if (200 == response.statusLine.statusCode) {
            var entity = response.entity
            val contentEncodingHeader = entity.contentEncoding
            if (contentEncodingHeader != null) {
                val encodings = contentEncodingHeader.elements
                for (i in encodings.indices) {
                    if (encodings[i].name.equals("gzip", ignoreCase = true)) {
                        entity = GzipDecompressingEntity(entity)
                        break
                    }
                }
            }
            schema = EntityUtils.toString(entity, Charset.forName("utf-8").name())
        }
        client.close()
        return schema
    }

    private const val rawIntrospectionQuery = "\\n" +
        "    query IntrospectionQuery {\\n" +
        "      __schema {\\n" +
        "        queryType { name }\\n" +
        "        mutationType { name }\\n" +
        "        subscriptionType { name }\\n" +
        "        types {\\n" +
        "          ...FullType\\n" +
        "        }\\n" +
        "        directives {\\n" +
        "          name\\n" +
        "          description\\n" +
        "          locations\\n" +
        "          args {\\n" +
        "            ...InputValue\\n" +
        "          }\\n" +
        "        }\\n" +
        "      }\\n" +
        "    }\\n" +
        "\\n" +
        "    fragment FullType on __Type {\\n" +
        "      kind\\n" +
        "      name\\n" +
        "      description\\n" +
        "      fields(includeDeprecated: true) {\\n" +
        "        name\\n" +
        "        description\\n" +
        "        args {\\n" +
        "          ...InputValue\\n" +
        "        }\\n" +
        "        type {\\n" +
        "          ...TypeRef\\n" +
        "        }\\n" +
        "        isDeprecated\\n" +
        "        deprecationReason\\n" +
        "      }\\n" +
        "      inputFields {\\n" +
        "        ...InputValue\\n" +
        "      }\\n" +
        "      interfaces {\\n" +
        "        ...TypeRef\\n" +
        "      }\\n" +
        "      enumValues(includeDeprecated: true) {\\n" +
        "        name\\n" +
        "        description\\n" +
        "        isDeprecated\\n" +
        "        deprecationReason\\n" +
        "      }\\n" +
        "      possibleTypes {\\n" +
        "        ...TypeRef\\n" +
        "      }\\n" +
        "    }\\n" +
        "\\n" +
        "    fragment InputValue on __InputValue {\\n" +
        "      name\\n" +
        "      description\\n" +
        "      type { ...TypeRef }\\n" +
        "      defaultValue\\n" +
        "    }\\n" +
        "\\n" +
        "    fragment TypeRef on __Type {\\n" +
        "      kind\\n" +
        "      name\\n" +
        "      ofType {\\n" +
        "        kind\\n" +
        "        name\\n" +
        "        ofType {\\n" +
        "          kind\\n" +
        "          name\\n" +
        "          ofType {\\n" +
        "            kind\\n" +
        "            name\\n" +
        "            ofType {\\n" +
        "              kind\\n" +
        "              name\\n" +
        "              ofType {\\n" +
        "                kind\\n" +
        "                name\\n" +
        "                ofType {\\n" +
        "                  kind\\n" +
        "                  name\\n" +
        "                  ofType {\\n" +
        "                    kind\\n" +
        "                    name\\n" +
        "                  }\\n" +
        "                }\\n" +
        "              }\\n" +
        "            }\\n" +
        "          }\\n" +
        "        }\\n" +
        "      }\\n" +
        "    }\\n" +
        "  "

    private const val introspectionQuery = "{\"query\": \"$rawIntrospectionQuery\",\"variables\":{},\"operationName\":\"IntrospectionQuery\"}"
}