@gridsome/source-graphql/transforms.js

GraphQL source for Gridsome

const {
  GraphQLObjectType,
  GraphQLSchema
} = require('gridsome/graphql')

const {
  visitSchema,
  VisitSchemaKind
} = require(`graphql-tools/dist/transforms/visitSchema`)

const {
  createResolveType,
  fieldMapToFieldConfigMap
} = require(`graphql-tools/dist/stitching/schemaRecreation`)

class NamespaceUnderFieldTransform {
  constructor (typeName, fieldName) {
    this.typeName = typeName
    this.fieldName = fieldName
  }

  transformSchema (schema) {
    const query = schema.getQueryType()

    let newQuery // eslint-disable-line
    const nestedType = new GraphQLObjectType({
      name: this.typeName,
      fields: () =>
        fieldMapToFieldConfigMap(
          query.getFields(),
          createResolveType(typeName => {
            if (typeName === query.name) {
              return newQuery
            } else {
              return schema.getType(typeName)
            }
          }),
          true
        )
    })
    newQuery = new GraphQLObjectType({
      name: query.name,
      fields: {
        [this.fieldName]: {
          type: nestedType,
          resolve: (parent, args, context, info) => {
            return {}
          }
        }
      }
    })
    const typeMap = schema.getTypeMap()
    const allTypes = Object.keys(typeMap)
      .filter(name => name !== query.name)
      .map(key => typeMap[key])

    return new GraphQLSchema({
      query: newQuery,
      types: allTypes
    })
  }
}

class StripNonQueryTransform {
  transformSchema (schema) {
    return visitSchema(schema, {
      [VisitSchemaKind.MUTATION] () {
        return null
      },
      [VisitSchemaKind.SUBSCRIPTION] () {
        return null
      }
    })
  }
}

module.exports = {
  NamespaceUnderFieldTransform,
  StripNonQueryTransform
}