Function PgDistinctPlugin

packages/query/src/graphql/plugins/PgDistinctPlugin.ts:17–114

(builder)

};

export const PgDistinctPlugin: Plugin = (builder) => {
  // Creates enums for each entity based on their fields
  builder.hook(
    'init',
    (args, build) => {
      const {
        graphql: {GraphQLEnumType},
        newWithHooks,
        pgIntrospectionResultsByKind,
      } = build;

      pgIntrospectionResultsByKind.class.forEach((cls: PgClass) => {
        if (!cls.isSelectable || build.pgOmit(cls, 'order')) return;
        if (!cls.namespace) return;

        const enumTypeName = getEnumName(cls.name);

        const entityEnumValues: Record<string, {value: number}> = {};
        cls?.attributes?.forEach((attr, index) => {
          if (attr.name.indexOf('_') !== 0) {
            entityEnumValues[attr.name.toUpperCase()] = {value: index};
          }
        });

        newWithHooks(
          GraphQLEnumType,
          {
            name: enumTypeName,
            values: entityEnumValues,
          },
          {
            __origin: `Adding connection "distinct" enum type for ${cls.name}.`,
            pgIntrospection: cls,
          },
          true
        );
      });

      return args;
    },
    ['AddDistinctEnumsPlugin']
  );

  // Extends schema and modifies the query
  builder.hook(
    'GraphQLObjectType:fields:field:args',
    (args, build, {addArgDataGenerator, scope: {pgFieldIntrospection}}) => {
      const {
        extend,
        graphql: {GraphQLList},
        pgSql: sql,
      } = build as Build & {extend: Extend; pgSql: typeof PgSql};

      const enumTypeName = getEnumName(pgFieldIntrospection?.name);
      const enumType = build.getTypeByName(enumTypeName) as GraphQLEnumType;

      if (!enumType) {
        return args;