Spaces:
Build error
Build error
File size: 3,166 Bytes
d9494a5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | import * as Sentry from '@sentry/node';
import { type Request } from 'express';
import { DocumentNode, parse } from 'graphql';
import { type Plugin } from 'graphql-yoga';
import { isNull } from '@sniptt/guards';
import { type DirectExecutionService } from 'src/engine/api/graphql/direct-execution/direct-execution.service';
import { classifyTopLevelFields } from 'src/engine/api/graphql/direct-execution/utils/classify-top-level-fields.util';
import { findOperationDefinition } from 'src/engine/api/graphql/direct-execution/utils/find-operation-definition.util';
import { isSubscriptionOperation } from 'src/engine/api/graphql/direct-execution/utils/is-subscription-operation.util';
import { type FeatureFlagService } from 'src/engine/core-modules/feature-flag/services/feature-flag.service';
import { UserInputError } from 'src/engine/core-modules/graphql/utils/graphql-errors.util';
export type DirectExecutionPluginConfig = {
directExecutionService: DirectExecutionService;
featureFlagService: FeatureFlagService;
};
export function useDirectExecution(
config: DirectExecutionPluginConfig,
): Plugin {
return {
onRequest: async ({ endResponse, serverContext }) => {
const req = (serverContext as unknown as { req: Request }).req;
if (!req.workspace?.id || !req.body?.query) {
return;
}
const queryString = req.body.query as string;
const operationName = req.body.operationName as string | undefined;
let document: DocumentNode;
try {
document = parse(queryString);
} catch {
return;
}
const operationDefinition = findOperationDefinition(
document,
operationName,
);
if (
!operationDefinition ||
isSubscriptionOperation(document, operationName)
) {
return;
}
const workspaceResolverNames =
await config.directExecutionService.getWorkspaceResolverNames(
req.workspace.id,
);
if (!workspaceResolverNames) {
return;
}
const { hasIntrospectionFields, hasWorkspaceFields, hasCoreFields } =
classifyTopLevelFields(document, operationName, workspaceResolverNames);
if (hasCoreFields && hasWorkspaceFields) {
const error = new UserInputError(
'This query cannot be executed as a single request. Please split it into separate queries.',
);
return endResponse(Response.json({ errors: [error.toJSON()] }));
}
if (hasCoreFields) {
return;
}
if (Sentry.isInitialized()) {
const transactionName =
operationName || operationDefinition.name?.value || '';
Sentry.setTags({
operationName: transactionName,
operation: operationDefinition.operation,
});
Sentry.getCurrentScope().setTransactionName(transactionName);
}
const result = await config.directExecutionService.execute(
req,
document,
hasIntrospectionFields,
hasWorkspaceFields,
);
if (isNull(result)) {
return;
}
return endResponse(Response.json(result));
},
};
}
|