Improve VirtualDocument

packages/lsp/src/virtual/document.ts

@@ -461,9 +461,6 @@ export class VirtualDocument implements IDisposable {
    * Clear the virtual document and all related stuffs
    */
   clear(): void {
-    this.unusedStandaloneDocuments.forEach(item =>
-      item.forEach(doc => doc.dispose())
-    );
     this.unusedStandaloneDocuments.clear();
 
     for (let document of this.foreignDocuments.values()) {
@@ -745,7 +742,7 @@ export class VirtualDocument implements IDisposable {
             );
             continue;
           }
-          let foreignDocument = this.chooseForeignDocument(extractor);
+          let foreignDocument = this._chooseForeignDocument(extractor);
           foreignDocumentsMap.set(result.range, {
             virtualLine: foreignDocument.lastVirtualLine,
             virtualDocument: foreignDocument,
@@ -800,8 +797,6 @@ export class VirtualDocument implements IDisposable {
    * Close all foreign documents.
    */
   closeAllForeignDocuments(): void {
-    console.log('closing');
-
     for (let document of this.foreignDocuments.values()) {
       this.closeForeign(document);
     }
@@ -896,6 +891,16 @@ export class VirtualDocument implements IDisposable {
     } as ISourcePosition;
   }
 
+  /**
+   * Compute the position in root document from the position of
+   * a virtual document.
+   */
+  transformVirtualToRoot(position: IVirtualPosition): IRootPosition | null {
+    let editor = this.virtualLines.get(position.line)!.editor;
+    let editorPosition = this.transformVirtualToEditor(position);
+    return this.root.transformFromEditorToRoot(editor, editorPosition!);
+  }
+
   /**
    * Get the corresponding editor of the virtual line.
    */
@@ -973,7 +978,7 @@ export class VirtualDocument implements IDisposable {
   /**
    * Get the foreign document that can be opened with the input extractor.
    */
-  private chooseForeignDocument(
+  private _chooseForeignDocument(
     extractor: IForeignCodeExtractor
   ): VirtualDocument {
     let foreignDocument: VirtualDocument;
@@ -1012,13 +1017,16 @@ export class VirtualDocument implements IDisposable {
     standalone: boolean,
     fileExtension: string
   ): VirtualDocument {
-    let document = new VirtualDocument({
+    let document = new (this.constructor as new (
+      ...args: ConstructorParameters<typeof VirtualDocument>
+    ) => VirtualDocument)({
       ...this.options,
       parent: this,
       standalone: standalone,
       fileExtension: fileExtension,
       language: language
     });
+
     const context: Document.IForeignContext = {
       foreignDocument: document,
       parentHost: this