168 lines · 6.2 KB · TypeScript
RawDownload
import { Effect, Layer, Option } from "effect";import { HttpClient, HttpClientRequest } from "effect/unstable/http"; import { GraphqlInvocationError } from "./errors";import { type OperationBinding, InvocationResult } from "./types"; const endpointWithQueryParams = (endpoint: string, queryParams: Record<string, string>): string => {  if (Object.keys(queryParams).length === 0) return endpoint;  const url = new URL(endpoint);  for (const [name, value] of Object.entries(queryParams)) {    url.searchParams.set(name, value);  }  return url.toString();}; export const endpointForTelemetry = (endpoint: string): string => {  if (!URL.canParse(endpoint)) return endpoint;  const url = new URL(endpoint);  url.search = "";  url.hash = "";  return url.toString();}; /** The operation string to send for a call. A caller-supplied `select` overrides *  the default scalar-leaf selection: it is spliced into the field's selection *  set (`field { <select> }`) so nested/list data can be requested per call. Falls *  back to the stored default operation when `select` is absent or the binding *  predates the prefix/suffix split. `select` is a control input, never a GraphQL *  variable. Shared with plugin.invokeTool so the validated string matches the *  sent string exactly. */export const effectiveOperationString = (  operation: OperationBinding,  args: Record<string, unknown>,): string => {  const customSelect = typeof args.select === "string" ? args.select.trim() : "";  return customSelect.length > 0 &&    operation.operationPrefix != null &&    operation.operationSuffix != null    ? `${operation.operationPrefix} { ${customSelect} }${operation.operationSuffix}`    : operation.operationString;}; // ---------------------------------------------------------------------------// Response helpers// --------------------------------------------------------------------------- const isJsonContentType = (ct: string | null | undefined): boolean => {  if (!ct) return false;  const normalized = ct.split(";")[0]?.trim().toLowerCase() ?? "";  return (    normalized === "application/json" || normalized.includes("+json") || normalized.includes("json")  );}; // ---------------------------------------------------------------------------// Public API — execute a GraphQL operation// --------------------------------------------------------------------------- export const invoke = Effect.fn("GraphQL.invoke")(function* (  operation: OperationBinding,  args: Record<string, unknown>,  endpoint: string,  resolvedHeaders: Record<string, string>,  resolvedQueryParams: Record<string, string> = {},) {  const client = yield* HttpClient.HttpClient;  const requestEndpoint = endpointWithQueryParams(endpoint, resolvedQueryParams);  const telemetryEndpoint = endpointForTelemetry(endpoint);   yield* Effect.annotateCurrentSpan({    "http.method": "POST",    "http.url": telemetryEndpoint,    "plugin.graphql.endpoint": telemetryEndpoint,    "plugin.graphql.operation_kind": operation.kind,    "plugin.graphql.field_name": operation.fieldName,    "plugin.graphql.headers.resolved_count": Object.keys(resolvedHeaders).length,    "plugin.graphql.query_params.resolved_count": Object.keys(resolvedQueryParams).length,  });   // Build the GraphQL request body  const variables: Record<string, unknown> = {};  for (const varName of operation.variableNames) {    if (args[varName] !== undefined) {      variables[varName] = args[varName];    }  }   // Also pick up any variables from a "variables" container  if (typeof args.variables === "object" && args.variables !== null) {    Object.assign(variables, args.variables);  }   // `select` (a control input, not a GraphQL variable) is applied here and never  // enters `variables`. plugin.invokeTool validates this same string before we  // reach the network.  const operationString = effectiveOperationString(operation, args);   let request = HttpClientRequest.post(requestEndpoint).pipe(    HttpClientRequest.setHeader("Content-Type", "application/json"),    HttpClientRequest.bodyJsonUnsafe({      query: operationString,      variables: Object.keys(variables).length > 0 ? variables : undefined,    }),  );   for (const [name, value] of Object.entries(resolvedHeaders)) {    request = HttpClientRequest.setHeader(request, name, value);  }   const response = yield* client.execute(request).pipe(    Effect.mapError(      (err) =>        new GraphqlInvocationError({          message: "GraphQL request failed",          statusCode: Option.none(),          cause: err,        }),    ),  );   const status = response.status;  const contentType = response.headers["content-type"] ?? null;   const body: unknown = isJsonContentType(contentType)    ? yield* response.json.pipe(Effect.catch(() => response.text))    : yield* response.text;   // GraphQL responses are always 200 with { data, errors }  const gqlBody = body as { data?: unknown; errors?: unknown[] } | null;  const hasErrors = Array.isArray(gqlBody?.errors) && gqlBody.errors.length > 0;   yield* Effect.annotateCurrentSpan({    "http.status_code": status,    "plugin.graphql.has_errors": hasErrors,    "plugin.graphql.error_count": hasErrors ? gqlBody!.errors!.length : 0,  });   return InvocationResult.make({    status,    data: gqlBody?.data ?? null,    errors: hasErrors ? gqlBody!.errors : null,    body,    headers: { ...response.headers },  });}); // ---------------------------------------------------------------------------// Invoke a GraphQL operation with a provided HttpClient layer// --------------------------------------------------------------------------- export const invokeWithLayer = (  operation: OperationBinding,  args: Record<string, unknown>,  endpoint: string,  resolvedHeaders: Record<string, string>,  resolvedQueryParams: Record<string, string>,  httpClientLayer: Layer.Layer<HttpClient.HttpClient>,) =>  invoke(operation, args, endpoint, resolvedHeaders, resolvedQueryParams).pipe(    Effect.provide(httpClientLayer),    Effect.withSpan("plugin.graphql.invoke", {      attributes: {        "plugin.graphql.endpoint": endpointForTelemetry(endpoint),        "plugin.graphql.operation_kind": operation.kind,        "plugin.graphql.field_name": operation.fieldName,      },    }),  );