UNPKG

@theguild/federation-composition

Version:
152 lines (150 loc) 5.72 kB
import { Kind, parse } from "graphql"; import { print } from "./graphql/printer.js"; import { transformSupergraphToPublicSchema } from "./graphql/transform-supergraph-to-public-schema.js"; import { sdl as authenticatedSDL } from "./specifications/authenticated.js"; import { sdl as costSDL } from "./specifications/cost.js"; import { sdl as inaccessibleSDL } from "./specifications/inaccessible.js"; import { sdl as joinSDL } from "./specifications/join.js"; import { sdl as linkSDL, printLink } from "./specifications/link.js"; import { sdl as policySDL } from "./specifications/policy.js"; import { sdl as requiresScopesSDL } from "./specifications/requires-scopes.js"; import { sdl as tagSDL } from "./specifications/tag.js"; import { validate } from "./validate.js"; export function composeServices(services, __internal) { const validationResult = validate(services, __internal); if (!validationResult.success) { return { errors: validationResult.errors, }; } const rootTypes = { query: false, mutation: false, subscription: false, }; for (const def of validationResult.supergraph) { if (def.name.value === "Query") { rootTypes.query = true; } else if (def.name.value === "Mutation") { rootTypes.mutation = true; } else if (def.name.value === "Subscription") { rootTypes.subscription = true; } if (rootTypes.query === true && rootTypes.mutation === true && rootTypes.subscription === true) { break; } } const usedTagSpec = validationResult.specs.tag; const usedCostSpec = validationResult.specs.cost; const usedInaccessibleSpec = validationResult.specs.inaccessible; const usedPolicySpec = validationResult.specs.policy; const usedRequiresScopesSpec = validationResult.specs.requiresScopes; const usedAuthenticatedSpec = validationResult.specs.authenticated; let _publicSdl; let costLinkImports = ""; if (usedCostSpec.used) { const imports = []; if (usedCostSpec.names.cost && usedCostSpec.names.cost !== "cost") { imports.push(createLinkImportValue("@cost", `@${usedCostSpec.names.cost}`)); } if (usedCostSpec.names.listSize) { imports.push(createLinkImportValue("@listSize", usedCostSpec.names.listSize !== "listSize" ? `@${usedCostSpec.names.listSize}` : null)); } if (imports.length) { costLinkImports = `, import: [${imports.join(", ")}]`; } } const federationVersionToJoinVersion = { "v1.0": "v0.3", "v2.0": "v0.3", "v2.1": "v0.3", "v2.2": "v0.3", "v2.3": "v0.3", "v2.4": "v0.3", "v2.5": "v0.3", "v2.6": "v0.3", "v2.7": "v0.4", "v2.8": "v0.5", "v2.9": "v0.5", }; const core = ` schema @link(url: "https://specs.apollo.dev/link/v1.0") @link(url: "https://specs.apollo.dev/join/${federationVersionToJoinVersion[validationResult.federationVersion]}", for: EXECUTION) ${usedTagSpec ? '@link(url: "https://specs.apollo.dev/tag/v0.3")' : ""} ${usedCostSpec.used ? `@link(url: "https://specs.apollo.dev/cost/v0.1"${costLinkImports})` : ""} ${usedInaccessibleSpec ? '@link(url: "https://specs.apollo.dev/inaccessible/v0.2", for: SECURITY)' : ""} ${usedPolicySpec ? '@link(url: "https://specs.apollo.dev/policy/v0.1", for: SECURITY)' : ""} ${usedRequiresScopesSpec ? '@link(url: "https://specs.apollo.dev/requiresScopes/v0.1", for: SECURITY)' : ""} ${usedAuthenticatedSpec ? '@link(url: "https://specs.apollo.dev/authenticated/v0.1", for: SECURITY)' : ""} ${validationResult.links.map(printLink).join("\n ")} { ${rootTypes.query ? "query: Query" : ""} ${rootTypes.mutation ? "mutation: Mutation" : ""} ${rootTypes.subscription ? "subscription: Subscription" : ""} } ${joinSDL(validationResult.federationVersion)} ${linkSDL} ${usedTagSpec ? tagSDL : ""} ${usedCostSpec.used ? costSDL({ cost: usedCostSpec.names.cost ?? "cost", listSize: usedCostSpec.names.listSize ?? "listSize", }) : ""} ${usedInaccessibleSpec ? inaccessibleSDL : ""} ${usedPolicySpec ? policySDL : ""} ${usedRequiresScopesSpec ? requiresScopesSDL : ""} ${usedAuthenticatedSpec ? authenticatedSDL : ""} `; return { supergraphSdl: ` ${core} ${print({ kind: Kind.DOCUMENT, definitions: validationResult.supergraph, })} `, get publicSdl() { if (_publicSdl) { return _publicSdl; } _publicSdl = print(transformSupergraphToPublicSchema({ kind: Kind.DOCUMENT, definitions: parse(core).definitions.concat(validationResult.supergraph), })); return _publicSdl; }, }; } function createLinkImportValue(name, alias) { if (alias) { return `{ name: "${name}", as: "${alias}" }`; } return `"${name}"`; } export function assertCompositionSuccess(compositionResult, message) { if (compositionHasErrors(compositionResult)) { throw new Error(message || "Unexpected test failure"); } } export function assertCompositionFailure(compositionResult, message) { if (!compositionHasErrors(compositionResult)) { throw new Error(message || "Unexpected test failure"); } } export function compositionHasErrors(compositionResult) { return "errors" in compositionResult && !!compositionResult.errors; }