@graphql-hive/federation-gateway-audit/src/test-suites/requires-with-argument/c.subgraph.ts

Audit tool for Apollo Federation Gateway

import { createSubgraph } from "../../subgraph.js";
import { comments, posts } from "./data.js";

export default createSubgraph("c", {
  typeDefs: /* GraphQL */ `
    extend schema
      @link(
        url: "https://specs.apollo.dev/federation/v2.3"
        import: ["@key", "@external", "@requires"]
      )

    type Query {
      feed: [Post]
    }

    type Post @key(fields: "id") {
      id: ID!
    }

    type Comment @key(fields: "id") {
      id: ID!
      authorId: ID
      body: String!
    }
  `,
  resolvers: {
    Query: {
      feed() {
        return posts.map((post) => ({ id: post.id }));
      },
    },
    Post: {
      __resolveReference(key: { id: string }) {
        const post = posts.find((post) => post.id === key.id);

        if (!post) {
          return null;
        }

        return {
          id: post.id,
        };
      },
      comments(post: { id: string }, { limit }: { limit: number }) {
        return comments
          .filter((comment) => comment.postId === post.id)
          .slice(0, limit)
          .map((c) => ({
            id: c.id,
            authorId: c.authorId,
            body: c.body,
          }));
      },
    },
    Comment: {
      __resolveReference(key: { id: string }) {
        const comment = comments.find((c) => c.id === key.id);

        if (!comment) {
          return null;
        }

        return {
          id: comment.id,
          authorId: comment.authorId,
          body: comment.body,
        };
      },
    },
  },
});