chore: introduce query planner to router-bridge

.prettierrc.js

@@ -15,7 +15,7 @@ module.exports = {
   requirePragma: true,
   overrides: [
     {
-      files: '{docs/{,source/**},.,{gateway-js,federation-js,harmonizer,federation-integration-testsuite-js,query-planner-js,subgraph-js}/**,test}/{*.js,*.ts}',
+      files: '{docs/{,source/**},.,{gateway-js,federation-js,harmonizer,federation-integration-testsuite-js,query-planner-js,router-bridge,subgraph-js}/**,test}/{*.js,*.ts}',
       options: {
         requirePragma: false,
         trailingComma: 'all',

router-bridge/build.rs

@@ -25,8 +25,8 @@ fn main() {
     };
 
     match (&bridge_last_update, &js_dist_last_update) {
-        // the federation folder has evolved since the last time we built harmonizer.
-        (Ok(bridge), Some(federation)) if federation > bridge => update_bridge(),
+        // the federation folder has evolved since the last time we built the rust-bridge.
+        (Ok(bridge), Some(js)) if js > bridge => update_bridge(),
         // Os didn't allow to query for metadata, we can't know for sure the bridge is up to date.
         (Err(_), _) => update_bridge(),
         _ => {

router-bridge/js-src/do_introspect.ts

@@ -1,14 +1,13 @@
-import type { batchIntrospect } from './introspection';
+import type { batchIntrospect } from '.';
+import type { OperationResult } from './types';
+
 /**
  * There are several global properties that we make available in our V8 runtime
  * and these are the types for those that we expect to use within this script.
  * They'll be stripped in the emitting of this file as JS, of course.
  */
 declare var bridge: { batchIntrospect: typeof batchIntrospect };
 
-// TODO: Maybe tighten the type, and put it somewhere else
-type OperationResult = { Ok: any, Err?: undefined } | { Ok?: undefined, Err: any };
-
 declare var done: (operationResult: OperationResult) => void;
 declare var sdl: string;
 declare var queries: string[];

router-bridge/js-src/do_plan.ts

@@ -0,0 +1,35 @@
+import type { plan } from '.';
+import type { OperationResult } from './types';
+import { BuildQueryPlanOptions } from '@apollo/query-planner';
+
+/**
+ * There are several global properties that we make available in our V8 runtime
+ * and these are the types for those that we expect to use within this script.
+ * They'll be stripped in the emitting of this file as JS, of course.
+ */
+declare var bridge: { plan: typeof plan };
+
+declare var done: (operationResult: OperationResult) => void;
+declare var schemaString: string;
+declare var queryString: string;
+declare var options: BuildQueryPlanOptions;
+declare var operationName: string | undefined;
+
+if (!options) {
+  done({
+    Err: [{ message: 'Error in JS-Rust-land: options is missing.' }],
+  });
+}
+
+const planResult = bridge.plan(
+  schemaString,
+  queryString,
+  options,
+  operationName,
+);
+
+if (planResult.errors?.length > 0) {
+  done({ Err: planResult.errors });
+} else {
+  done({ Ok: planResult.data });
+}

router-bridge/js-src/index.ts

@@ -1,2 +1,2 @@
-export { introspect, batchIntrospect } from "./introspection";
-export { parse as parseGraphqlDocument } from "graphql";
+export { introspect, batchIntrospect } from './introspection';
+export { plan } from './plan';

router-bridge/js-src/plan.ts

@@ -0,0 +1,36 @@
+import { ExecutionResult, parse, validate } from 'graphql';
+import {
+  QueryPlanner,
+  buildOperationContext,
+  BuildQueryPlanOptions,
+  buildComposedSchema,
+} from '@apollo/query-planner';
+
+export function plan(
+  schemaString: string,
+  queryString: string,
+  options: BuildQueryPlanOptions,
+  operationName?: string,
+): ExecutionResult {
+  try {
+    const schema = parse(schemaString);
+    const query = parse(queryString);
+    const composedSchema = buildComposedSchema(schema);
+
+    const validationErrors = validate(composedSchema, query);
+    if (validationErrors.length > 0) {
+      return { errors: validationErrors };
+    }
+
+    const operationContext = buildOperationContext(
+      composedSchema,
+      query,
+      operationName,
+    );
+
+    const planner = new QueryPlanner(composedSchema);
+    return { data: planner.buildQueryPlan(operationContext, options) };
+  } catch (e) {
+    return { errors: [e] };
+  }
+}

router-bridge/js-src/runtime.js

@@ -7,11 +7,11 @@ Deno.core.ops();
 const _newline = new Uint8Array([10]);
 
 function print(value) {
-    Deno.core.dispatchByName('op_print', 0, value.toString(), _newline);
+  Deno.core.dispatchByName('op_print', 0, value.toString(), _newline);
 }
 
 function done(result) {
-    Deno.core.opSync('op_result', result);
+  Deno.core.opSync('op_result', result);
 }
 
 // We build some of the preliminary objects that our Rollup-built package is
@@ -24,7 +24,7 @@ node_fetch_1 = {};
 // particular, to determine whether or not we are running in a debug
 // mode.  For the purposes of harmonizer, we don't gain anything from
 // running in such a mode.
-process = {argv: [], env: {"NODE_ENV": "production"}};
+process = { argv: [], env: { NODE_ENV: 'production' } };
 // Some JS runtime implementation specific bits that we rely on that
 // need to be initialized as empty objects.
 global = {};

router-bridge/js-src/types.ts

@@ -0,0 +1,3 @@
+export type OperationResult =
+  | { Ok: any; Err?: undefined }
+  | { Ok?: undefined; Err: any };

router-bridge/js-src/url_polyfill.ts

@@ -1,3 +1,3 @@
-import "fast-text-encoding";
-import * as url from "whatwg-url";
+import 'fast-text-encoding';
+import * as url from 'whatwg-url';
 export default url;

router-bridge/package.json

@@ -14,10 +14,9 @@
   "author": "Apollo <opensource@apollographql.com>",
   "license": "MIT",
   "dependencies": {
+    "@apollo/query-planner": "file:../query-planner-js",
     "fast-text-encoding": "1.0.3",
+    "graphql": "15.6.1",
     "whatwg-url": "9.1.0"
-  },
-  "peerDependencies": {
-    "graphql": "^14.5.0 || ^15.0.0"
   }
 }

router-bridge/src/introspect.rs

@@ -87,7 +87,7 @@ pub type IntrospectionResult = Result<Vec<IntrospectionResponse>, IntrospectionE
 /// The `batch_introspect` function receives a [`string`] representing the SDL and invokes JavaScript
 /// introspection on it, with the `queries` to run against the SDL.
 ///
-pub fn batch_introspect(sdl: String, queries: Vec<String>) -> IntrospectionResult {
+pub fn batch_introspect(sdl: &str, queries: Vec<String>) -> IntrospectionResult {
     Js::new()
         .with_parameter("sdl", sdl)
         .with_parameter("queries", queries)
@@ -112,11 +112,8 @@ mod tests {
         }
         "#;
 
-        let introspected = batch_introspect(
-            raw_sdl.to_string(),
-            vec![DEFAULT_INTROSPECTION_QUERY.to_string()],
-        )
-        .unwrap();
+        let introspected =
+            batch_introspect(raw_sdl, vec![DEFAULT_INTROSPECTION_QUERY.to_string()]).unwrap();
         insta::assert_snapshot!(serde_json::to_string(&introspected).unwrap());
     }
 
@@ -129,8 +126,7 @@ mod tests {
         let response = batch_introspect(
             "schema {
                 query: Query
-            }"
-            .to_string(),
+            }",
             vec![DEFAULT_INTROSPECTION_QUERY.to_string()],
         )
         .unwrap();
@@ -147,8 +143,7 @@ mod tests {
         let response = batch_introspect(
             "schema {
                 query: Query
-            }"
-            .to_string(),
+            }",
             vec![DEFAULT_INTROSPECTION_QUERY.to_string()],
         )
         .unwrap();

router-bridge/src/lib.rs

@@ -7,3 +7,4 @@
 #![warn(missing_docs, future_incompatible, unreachable_pub, rust_2018_idioms)]
 pub mod introspect;
 mod js;
+pub mod plan;