packages/plugins/graphql/server/src/services/builders/resolvers/query.ts - develop - githubmirror/strapi

packages/plugins/graphql/server/src/services/builders/resolvers/query.ts

80 строк2 KB

Ben Irvin

feat: publicationFilter param in REST and document service (#25793)

21 май 2026, 12:31

21 май 2026, 12:3105ff01d

import { omit } from 'lodash/fp';
import type { Schema } from '@strapi/types';
import type { Context } from '../../types';

import { mergePublicationFilterFromGraphQLArgs } from './merge-publication-args';

/** Merge sanitized query with resolver args so GraphQL-coerced publication args are not dropped. */
const mergeDocumentListParams = (
  sanitizedQuery: Record<string, unknown>,
  args: Record<string, unknown>
) => {
  const { status, ...rest } = sanitizedQuery;

  const merged: Record<string, unknown> = {
    ...rest,
    status: (status as 'draft' | 'published' | undefined) ?? 'published',
  };

  const { publicationFilter } = mergePublicationFilterFromGraphQLArgs(args);
  if (publicationFilter !== undefined) {
    merged.publicationFilter = publicationFilter;
  }

  return merged;
};

export default ({ strapi }: Context) => ({
  buildQueriesResolvers({ contentType }: { contentType: Schema.ContentType }) {
    const { uid } = contentType;

    return {
      async findMany(parent: any, args: any, ctx: any) {
        await strapi.contentAPI.validate.query(args, contentType, {
          auth: ctx?.state?.auth,
        });

        const sanitizedQuery = (await strapi.contentAPI.sanitize.query(args, contentType, {
          auth: ctx?.state?.auth,
        })) as Record<string, unknown>;

        return strapi.documents!(uid).findMany(
          mergeDocumentListParams(sanitizedQuery, args) as any
        );
      },

      async findFirst(parent: any, args: any, ctx: any) {
        await strapi.contentAPI.validate.query(args, contentType, {
          auth: ctx?.state?.auth,
        });

        const sanitizedQuery = (await strapi.contentAPI.sanitize.query(args, contentType, {
          auth: ctx?.state?.auth,
        })) as Record<string, unknown>;

        return strapi.documents!(uid).findFirst(
          mergeDocumentListParams(sanitizedQuery, args) as any
        );
      },

      async findOne(parent: any, args: any, ctx: any) {
        const { documentId } = args;

        await strapi.contentAPI.validate.query(args, contentType, {
          auth: ctx?.state?.auth,
        });

        const sanitizedQuery = (await strapi.contentAPI.sanitize.query(args, contentType, {
          auth: ctx?.state?.auth,
        })) as Record<string, unknown>;

        const merged = mergeDocumentListParams(
          omit(['id', 'documentId'], sanitizedQuery) as Record<string, unknown>,
          args
        );

        return strapi.documents!(uid).findOne({ ...merged, documentId } as any);
      },
    };
  },
});