@pothos/plugin-add-graphql/src/index.ts

A Pothos plugin for adding existing GraphQL types to a Pothos schema

import './global-types';
import './schema-builder';
import SchemaBuilder, { BasePlugin, type SchemaTypes } from '@pothos/core';
import { GraphQLSchema } from 'graphql';
import { addTypeToSchema } from './utils';

const pluginName = 'addGraphQL';

export default pluginName;

const builtInTypes = Object.keys(new GraphQLSchema({}).getTypeMap());
export class PothosAddGraphQLPlugin<Types extends SchemaTypes> extends BasePlugin<Types> {
  override beforeBuild(): void {
    const { schema, types } = this.builder.options.add ?? {};

    const allTypes = [
      ...(Array.isArray(types) ? types : Object.values(types ?? {})),
      ...Object.values(schema?.getTypeMap() ?? {}).filter(
        (type) => !builtInTypes.includes(type.name),
      ),
    ];

    for (const type of allTypes) {
      addTypeToSchema(this.builder, type);
    }
  }
}

SchemaBuilder.registerPlugin(pluginName, PothosAddGraphQLPlugin);