Add support for GraphQL in Notebooks (#236)

* Add support for GraphQL in Notebooks

* Add changeset

.changeset/wise-gorillas-greet.md

@@ -0,0 +1,5 @@
+---
+"vscode-apollo": minor
+---
+
+Add support for GraphQL in Notebook cells

src/language-server/project/internal.ts

@@ -48,6 +48,7 @@ const fileAssociations: { [extension: string]: string } = {
   ".re": "reason",
   ".ex": "elixir",
   ".exs": "elixir",
+  ".ipynb": "python",
 };
 
 export interface GraphQLInternalProjectConfig extends GraphQLProjectConfig {

src/language-server/server.ts

@@ -157,13 +157,14 @@ connection.onInitialized(async () => {
 });
 
 const documents = new TextDocuments(TextDocument);
+const schemes = ["file", "vscode-notebook-cell"];
 
 // Make the text document manager listen on the connection
 // for open, change and close text document events
 documents.listen(connection);
 
-function isFile(uri: string) {
-  return URI.parse(uri).scheme === "file";
+function isEnabledDocument(uri: string) {
+  return schemes.includes(URI.parse(uri).scheme);
 }
 
 documents.onDidChangeContent((params) => {
@@ -173,8 +174,7 @@ documents.onDidChangeContent((params) => {
   );
   if (!project) return;
 
-  // Only watch changes to files
-  if (!isFile(params.document.uri)) {
+  if (!isEnabledDocument(params.document.uri)) {
     return;
   }
 
@@ -213,8 +213,7 @@ connection.onDidChangeWatchedFiles((params) => {
       continue;
     }
 
-    // Only watch changes to files
-    if (!isFile(uri)) {
+    if (!isEnabledDocument(uri)) {
       continue;
     }
 

src/tools/utilities/languageInformation.ts

@@ -30,7 +30,7 @@ export const minimumKnownExtensions: Record<
   typescriptreact: [".tsx"],
   vue: [".vue"],
   svelte: [".svelte"],
-  python: [".py"],
+  python: [".py", ".ipynb"],
   ruby: [".rb"],
   dart: [".dart"],
   reason: [".re"],