Adding foreign code extractors for spark sql and exporting other requ…

…ired classes
jupyter-lsp · Sep 15, 2023 · b80b4d1 · b80b4d1
1 parent dc46506
commit b80b4d1
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/packages/jupyterlab-lsp/src/index.ts b/packages/jupyterlab-lsp/src/index.ts
@@ -3,6 +3,12 @@
 /** General public tokens, including lumino Tokens and namespaces */
 export * from './tokens';
 
+/** Export the required classes */
+export {FileEditorAdapter} from './adapters/file_editor/file_editor';
+export {FileEditorContextMenuEntryPoint} from './adapters/file_editor';
+export {PositionConverter} from './converter';
+export {VirtualDocument} from './virtual/document';
+
 /** Generated JSON Schema types for server responses and settings */
 export * as SCHEMA from './_schema';
 

diff --git a/packages/jupyterlab-lsp/src/transclusions/ipython-sql/extractors.ts b/packages/jupyterlab-lsp/src/transclusions/ipython-sql/extractors.ts
@@ -37,6 +37,20 @@ export let foreign_code_extractors: IForeignCodeExtractorsRegistry = {
       foreign_capture_groups: [1],
       is_standalone: false,
       file_extension: 'sql'
+    }),
+    new RegExpForeignCodeExtractor({
+      language: 'sql',
+      pattern: `^%%spark -c sql(?: (?:${SQL_URL_PATTERN}|${COMMAND_PATTERN}|(?:\\w+ << )|(?:\\w+@\\w+)))?\n?((?:.+\n)?(?:[^]*))`,
+      foreign_capture_groups: [1],
+      is_standalone: true,
+      file_extension: 'sql'
+    }),
+    new RegExpForeignCodeExtractor({
+      language: 'sql',
+      pattern: `(?:^|\n)%spark -c sql (?:${SQL_URL_PATTERN}|${COMMAND_PATTERN}|(.*))\n?`,
+      foreign_capture_groups: [1],
+      is_standalone: false,
+      file_extension: 'sql'
     })
   ]
 };