don't include the shell api twice

Author: lerouxb

packages/mongodb-ts-autocomplete/scripts/extract-types.ts

@@ -1,13 +1,7 @@
 /* eslint-disable no-console */
 import { promises as fs } from 'fs';
 import path from 'path';
-import { api as ShellApiText } from '@mongosh/shell-api/api';
-
-function replaceImports(code: string) {
-  // This just makes it possible to work on mql.ts because then the
-  // IDE finds the import.
-  return code.replace(/'bson'/g, "'/bson.ts'");
-}
+import { replaceImports } from '../src/autocompleter';
 
 async function loadSources(sources: Record<string, string>) {
   const result: Record<string, string> = {};
@@ -25,7 +19,6 @@ async function run() {
     '/mql.ts': path.join(__dirname, '..', 'src', 'fixtures', 'mql.ts'),
   };
   const files = await loadSources(input);
-  files['/shell-api.ts'] = replaceImports(ShellApiText);
   const code = `
 const files = ${JSON.stringify(files)};
 export default files;

packages/mongodb-ts-autocomplete/src/autocompleter.ts

@@ -1,6 +1,7 @@
 import Autocompleter from '@mongodb-js/ts-autocomplete';
 import type { AutoCompletion } from '@mongodb-js/ts-autocomplete';
 import autocompleteTypes from './fixtures/autocomplete-types';
+import { api as ShellApiText } from '@mongosh/shell-api/api';
 
 import type { JSONSchema } from 'mongodb-schema';
 import { toTypescriptTypeDefinition } from 'mongodb-schema';
@@ -13,6 +14,12 @@ import {
 import { CachingAutocompletionContext } from './autocompletion-context';
 import type { AutocompletionContext } from './autocompletion-context';
 
+export function replaceImports(code: string) {
+  // This just makes it possible to work on mql.ts because then the
+  // IDE finds the import.
+  return code.replace(/'bson'/g, "'/bson.ts'");
+}
+
 type MongoDBAutocompleterOptions = {
   context: AutocompletionContext;
 };
@@ -159,7 +166,10 @@ export class MongoDBAutocompleter {
 
     this.connectionSchemas = Object.create(null);
 
-    this.autocompleter.updateCode(autocompleteTypes);
+    this.autocompleter.updateCode({
+      ...autocompleteTypes,
+      '/shell-api.ts': replaceImports(ShellApiText),
+    });
   }
 
   addConnection(connectionId: string): ConnectionSchema {