Outline view for information from analysis server.

editor/tools/plugins/com.google.dart.tools.ui/src/com/google/dart/tools/ui/internal/text/editor/DartEditor.java

 /*
  * Copyright (c) 2012, the Dart project authors.
  * 
  * Licensed under the Eclipse Public License v1.0 (the "License"); you may not use this file except
  * in compliance with the License. You may obtain a copy of the License at
  * 
  * http://www.eclipse.org/legal/epl-v10.html
  * 
  * Unless required by applicable law or agreed to in writing, software distributed under the License
  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
@@ skipping 14 matching lines @@
 import com.google.dart.engine.ast.visitor.NodeLocator;
 import com.google.dart.engine.context.AnalysisContext;
 import com.google.dart.engine.element.CompilationUnitElement;
 import com.google.dart.engine.index.Index;
 import com.google.dart.engine.search.SearchEngine;
 import com.google.dart.engine.search.SearchEngineFactory;
 import com.google.dart.engine.services.assist.AssistContext;
 import com.google.dart.engine.source.FileBasedSource;
 import com.google.dart.engine.source.Source;
 import com.google.dart.engine.utilities.source.SourceRange;
+import com.google.dart.server.Outline;
 import com.google.dart.tools.core.DartCore;
 import com.google.dart.tools.core.DartCoreDebug;
 import com.google.dart.tools.core.MessageConsole;
+import com.google.dart.tools.core.analysis.model.AnalysisServerData;
+import com.google.dart.tools.core.analysis.model.AnalysisServerOutlineListener;
 import com.google.dart.tools.core.analysis.model.Project;
 import com.google.dart.tools.core.analysis.model.ProjectManager;
 import com.google.dart.tools.core.formatter.DefaultCodeFormatterConstants;
 import com.google.dart.tools.core.model.CompilationUnit;
 import com.google.dart.tools.core.model.DartElement;
 import com.google.dart.tools.core.model.DartModelException;
 import com.google.dart.tools.core.model.SourceReference;
 import com.google.dart.tools.core.utilities.general.SourceRangeFactory;
 import com.google.dart.tools.internal.corext.refactoring.util.ExecutionUtils;
 import com.google.dart.tools.internal.corext.refactoring.util.RunnableEx;
@@ skipping 1659 matching lines @@
 
   /** Preference key for matching brackets */
   protected final static String MATCHING_BRACKETS = PreferenceConstants.EDITOR_MATCHING_BRACKETS;
 
   /** Preference key for matching brackets color */
   protected final static String MATCHING_BRACKETS_COLOR = PreferenceConstants.EDITOR_MATCHING_BRACKETS_COLOR;
 
   protected final static char[] BRACKETS = {'{', '}', '(', ')', '[', ']', '<', '>'};
 
   /** The outline page */
-  protected DartOutlinePage fOutlinePage;
+  protected DartOutlinePage fOutlinePage_OLD;
+
+  /** The outline page */
+  protected DartOutlinePage_NEW fOutlinePage_NEW;
+
+  private AnalysisServerOutlineListener analysisServerOutlineListener = new AnalysisServerOutlineListener() {
+    @Override
+    public void computedOutline(String contextId, Source source, final Outline outline) {
+      if (fOutlinePage_NEW != null) {
+        Display.getDefault().asyncExec(new Runnable() {
+          @Override
+          public void run() {
+            fOutlinePage_NEW.setInput(outline);
+          }
+        });
+      }
+    }
+  };
 
   /** Outliner context menu Id */
   protected String fOutlinerContextMenuId;
   /**
    * The editor selection changed listener.
    */
   private EditorSelectionChangedListener fEditorSelectionChangedListener;
   /** The editor's bracket matcher */
   protected DartPairMatcher fBracketMatcher = new DartPairMatcher(BRACKETS);
 
@@ skipping 197 matching lines @@
     }
     // OK, schedule selection update
     resolvedUnit = unit;
     ExecutionUtils.runLogAsync(new RunnableEx() {
       @Override
       public void run() {
         if (isDisposed()) {
           return;
         }
         // update Outline
-        if (resolvedUnit != null && fOutlinePage != null) {
-          if (fOutlinePage != null) {
-            fOutlinePage.setInput(resolvedUnit);
+        if (resolvedUnit != null && fOutlinePage_OLD != null) {
+          if (fOutlinePage_OLD != null) {
+            fOutlinePage_OLD.setInput(resolvedUnit);
           }
           applySelectionToOutline();
         }
         // update selection listeners
         fireDartSelectionListeners();
         // update occurrences
         fForcedMarkOccurrencesSelection = getSelectionProvider().getSelection();
         updateOccurrenceAnnotations((ITextSelection) fForcedMarkOccurrencesSelection, resolvedUnit);
       }
     });
@@ skipping 46 matching lines @@
     if (isSemanticHighlightingEnabled()) {
       installSemanticHighlighting();
     }
 
     PlatformUI.getWorkbench().addWindowListener(fActivationListener);
   }
 
   @Override
   public void dispose() {
     if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
-      DartCore.getAnalysisServerData().unsubscribeNavigation(
-          getInputAnalysisContextId(),
-          getInputSource());
+      AnalysisServerData analysisServerData = DartCore.getAnalysisServerData();
+      String contextId = getInputAnalysisContextId();
+      Source source = getInputSource();
+      if (contextId != null && source != null) {
+        analysisServerData.unsubscribeNavigation(contextId, source);
+        analysisServerData.unsubscribeOutline(contextId, source, analysisServerOutlineListener);
+      }
     }
 
     DartX.todo("folding");
     if (fProjectionModelUpdater != null) {
       fProjectionModelUpdater.uninstall();
       fProjectionModelUpdater = null;
     }
 
     if (fProjectionSupport != null) {
       fProjectionSupport.dispose();
@@ skipping 118 matching lines @@
       }
     }
   }
 
   @SuppressWarnings("rawtypes")
   @Override
   public Object getAdapter(Class required) {
 
     DartX.todo("outline");
     if (IContentOutlinePage.class.equals(required)) {
-      if (fOutlinePage == null) {
-        fOutlinePage = createOutlinePage();
+      if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
+        if (fOutlinePage_NEW == null) {
+          fOutlinePage_NEW = createOutlinePage_NEW();
+        }
+        return fOutlinePage_NEW;
+      } else {
+        if (fOutlinePage_OLD == null) {
+          fOutlinePage_OLD = createOutlinePage_OLD();
+        }
+        return fOutlinePage_OLD;
       }
-      return fOutlinePage;
     }
 
     if (IEncodingSupport.class.equals(required)) {
       return fEncodingSupport;
     }
 
     if (required == IShowInTargetList.class) {
       return new IShowInTargetList() {
         @Override
         public String[] getShowInTargetIds() {
@@ skipping 338 matching lines @@
         return widget.isVisible();
       }
     }
     return false;
   }
 
   /**
    * Informs the editor that its outliner has been closed.
    */
   public void outlinePageClosed() {
-    if (fOutlinePage != null) {
-      fOutlinePage = null;
-      resetHighlightRange();
+    if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
+      if (fOutlinePage_NEW != null) {
+        fOutlinePage_NEW = null;
+        resetHighlightRange();
+      }
+    } else {
+      if (fOutlinePage_OLD != null) {
+        fOutlinePage_OLD = null;
+        resetHighlightRange();
+      }
     }
   }
 
   public void removeDartSelectionListener(ISelectionChangedListener listener) {
     dartSelectionListeners.remove(listener);
   }
 
   /**
    * Resets the foldings structure according to the folding preferences.
    */
@@ skipping 15 matching lines @@
     int length = element.getDisplayName().length();
     selectAndReveal(offset, length);
   }
 
   public void setPreferences(IPreferenceStore store) {
     uninstallSemanticHighlighting();
     super.setPreferenceStore(store);
     installSemanticHighlighting();
   }
 
-  public void setSelection(LightNodeElement element, boolean moveCursor) {
+  public void setSelection_NEW(Outline element, boolean moveCursor) {
+    // validate Outline
+    if (element == null) {
+      return;
+    }
+    // prepare range
+    int offset = element.getOffset();
+    int length = element.getLength();
+    // prepare ISourceViewer
+    ISourceViewer sourceViewer = getSourceViewer();
+    if (sourceViewer == null) {
+      return;
+    }
+    // highlight range (not selection - just highlighting on left editor band)
+    if (offset < 0) {
+      return;
+    }
+    setHighlightRange(offset, length, moveCursor);
+    // do we want to change selection?
+    if (!moveCursor) {
+      return;
+    }
+    // prepare StyledText
+    StyledText textWidget = sourceViewer.getTextWidget();
+    if (textWidget == null) {
+      return;
+    }
+    // set selection in StyledText
+    try {
+      textWidget.setRedraw(false);
+      sourceViewer.revealRange(offset, length);
+      sourceViewer.setSelectedRange(offset, length);
+      fForcedMarkOccurrencesSelection = getSelectionProvider().getSelection();
+      updateOccurrenceAnnotations((ITextSelection) fForcedMarkOccurrencesSelection, getInputUnit());
+    } finally {
+      textWidget.setRedraw(true);
+    }
+  }
+
+  public void setSelection_OLD(LightNodeElement element, boolean moveCursor) {
     // validate LightNodeElement
     if (element == null) {
       return;
     }
     // prepare range
     int offset = element.getNameOffset();
     int length = element.getNameLength();
     // prepare ISourceViewer
     ISourceViewer sourceViewer = getSourceViewer();
     if (sourceViewer == null) {
@@ skipping 33 matching lines @@
     // TODO(scheglov)
 //      synchronizeOutlinePage(computeHighlightRangeSourceElement());
   }
 
   public void updatedTitleImage(Image image) {
     setTitleImage(image);
   }
 
   @Override
   protected void adjustHighlightRange(int offset, int length) {
-    Object element = getElementAt(offset, false);
+    if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
+      if (fOutlinePage_NEW != null) {
+        fOutlinePage_NEW.select(offset);
+      }
+    } else {
+      Object element = getElementAt(offset, false);
 
-    if (element instanceof LightNodeElement) {
-      LightNodeElement sourceElement = (LightNodeElement) element;
-      int elementOffset = sourceElement.getNameOffset();
-      int elementLength = sourceElement.getNameLength();
+      if (element instanceof LightNodeElement) {
+        LightNodeElement sourceElement = (LightNodeElement) element;
+        int elementOffset = sourceElement.getNameOffset();
+        int elementLength = sourceElement.getNameLength();
 
-      ISourceViewer viewer = getSourceViewer();
-      if (viewer instanceof ITextViewerExtension5) {
-        ITextViewerExtension5 extension = (ITextViewerExtension5) viewer;
-        extension.exposeModelRange(new Region(elementOffset, elementLength));
-      }
+        ISourceViewer viewer = getSourceViewer();
+        if (viewer instanceof ITextViewerExtension5) {
+          ITextViewerExtension5 extension = (ITextViewerExtension5) viewer;
+          extension.exposeModelRange(new Region(elementOffset, elementLength));
+        }
 
-      setHighlightRange(elementOffset, elementLength, true);
-      if (fOutlinePage != null) {
-        fOutlinePage.select((LightNodeElement) element);
+        setHighlightRange(elementOffset, elementLength, true);
+        if (fOutlinePage_OLD != null) {
+          fOutlinePage_OLD.select((LightNodeElement) element);
+        }
       }
     }
 
     ISourceViewer viewer = getSourceViewer();
     if (viewer instanceof ITextViewerExtension5) {
       ITextViewerExtension5 extension = (ITextViewerExtension5) viewer;
       extension.exposeModelRange(new Region(offset, length));
     } else {
       resetHighlightRange();
     }
@@ skipping 381 matching lines @@
     action.setActionDefinitionId(ITextEditorActionDefinitionIds.SELECT_WORD_NEXT);
     setAction(ITextEditorActionDefinitionIds.SELECT_WORD_NEXT, action);
     textWidget.setKeyBinding(SWT.CTRL | SWT.SHIFT | SWT.ARROW_RIGHT, SWT.NULL);
   }
 
   /**
    * Creates the outline page used with this editor.
    * 
    * @return the created Dart outline page
    */
-  protected DartOutlinePage createOutlinePage() {
-    DartOutlinePage page = new DartOutlinePage(fOutlinerContextMenuId, this);
-    setOutlinePageInput(page, getEditorInput());
+  protected DartOutlinePage_NEW createOutlinePage_NEW() {
+    DartOutlinePage_NEW page = new DartOutlinePage_NEW(fOutlinerContextMenuId, this);
+    setOutlinePageInput_NEW(page, getEditorInput());
     return page;
   }
 
+  /**
+   * Creates the outline page used with this editor.
+   * 
+   * @return the created Dart outline page
+   */
+  protected DartOutlinePage createOutlinePage_OLD() {
+    DartOutlinePage page = new DartOutlinePage(fOutlinerContextMenuId, this);
+    setOutlinePageInput_OLD(page, getEditorInput());
+    return page;
+  }
+
   @Override
   protected final ISourceViewer createSourceViewer(Composite parent, IVerticalRuler verticalRuler,
       int styles) {
 
     IPreferenceStore store = getPreferenceStore();
     final ISourceViewer viewer = createDartSourceViewer(
         parent,
         verticalRuler,
         getOverviewRuler(),
         isOverviewRulerVisible(),
@@ skipping 69 matching lines @@
 
     EditorUtility.addGTKPasteHack(viewer);
 
     return viewer;
   }
 
   protected void doSelectionChanged(ISelection selection) {
     if (selection instanceof IStructuredSelection) {
       IStructuredSelection structuredSelection = (IStructuredSelection) selection;
       for (Object selectionObject : structuredSelection.toList()) {
-        if (selectionObject instanceof LightNodeElement) {
-          setSelection((LightNodeElement) selectionObject, !isActivePart());
-          break;
+        if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
+          if (selectionObject instanceof Outline) {
+            setSelection_NEW((Outline) selectionObject, !isActivePart());
+            break;
+          }
+        } else {
+          if (selectionObject instanceof LightNodeElement) {
+            setSelection_OLD((LightNodeElement) selectionObject, !isActivePart());
+            break;
+          }
         }
       }
     }
   }
 
   @Override
   protected void doSetInput(IEditorInput input) throws CoreException {
     AnalysisContext sdkContext = DartCore.getProjectManager().getSdkContext();
     if (input instanceof IFileEditorInput) {
       IFileEditorInput fileInput = (IFileEditorInput) input;
       inputResourceFile = fileInput.getFile();
     } else if (input instanceof IURIEditorInput) {
       IURIEditorInput uriInput = (IURIEditorInput) input;
       URI uri = uriInput.getURI();
       if (uri != null && uri.getScheme().equals("file") && uri.isAbsolute()) {
         try {
           inputJavaFile = new File(uri);
         } catch (Throwable e) {
         }
       }
     }
 
     if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
-      DartCore.getAnalysisServerData().subscribeNavigation(
-          getInputAnalysisContextId(),
-          getInputSource());
+      AnalysisServerData analysisServerData = DartCore.getAnalysisServerData();
+      String contextId = getInputAnalysisContextId();
+      Source source = getInputSource();
+      if (contextId != null && source != null) {
+        analysisServerData.subscribeNavigation(contextId, source);
+      }
     }
 
     ISourceViewer sourceViewer = getSourceViewer();
     if (!(sourceViewer instanceof ISourceViewerExtension2)) {
       setPreferenceStore(createCombinedPreferenceStore(input));
       internalDoSetInput(input);
       return;
     }
 
     // uninstall & unregister preference store listener
@@ skipping 681 matching lines @@
     fOpenEditorActionGroup.fillContextMenu(menu);
     fOpenEditorActionGroup.setContext(null);
   }
 
   /**
    * Sets the input of the editor's outline page.
    * 
    * @param page the Dart outline page
    * @param input the editor input
    */
-  protected void setOutlinePageInput(DartOutlinePage page, IEditorInput input) {
+  protected void setOutlinePageInput_NEW(DartOutlinePage_NEW page, IEditorInput input) {
     if (page == null) {
       return;
     }
+    AnalysisServerData analysisServerData = DartCore.getAnalysisServerData();
+    String contextId = getInputAnalysisContextId();
+    Source source = getInputSource();
+    if (contextId != null && source != null) {
+      analysisServerData.subscribeOutline(contextId, source, analysisServerOutlineListener);
+    }
+  }
+
+  /**
+   * Sets the input of the editor's outline page.
+   * 
+   * @param page the Dart outline page
+   * @param input the editor input
+   */
+  protected void setOutlinePageInput_OLD(DartOutlinePage page, IEditorInput input) {
+    if (page == null) {
+      return;
+    }
     page.setInput(resolvedUnit);
   }
 
   /**
    * Sets the outliner's context menu ID.
    * 
    * @param menuId the menu ID
    */
   protected void setOutlinerContextMenuId(String menuId) {
     fOutlinerContextMenuId = menuId;
@@ skipping 166 matching lines @@
 
   /**
    * Synchronizes the outliner selection with the given element position in the editor.
    * 
    * @param element the Dart element to select
    * @param checkIfOutlinePageActive <code>true</code> if check for active outline page needs to be
    *          done
    */
   protected void synchronizeOutlinePage(Object /* ASTNode */element,
       boolean checkIfOutlinePageActive) {
-    if (fOutlinePage != null && element != null
+    if (fOutlinePage_OLD != null && element != null
         && !(checkIfOutlinePageActive && isOutlinePageActive())) {
       // TODO(scheglov)
 //        ((DartOutlinePage) fOutlinePage).select((LightNodeElement) element);
 //        ((DartOutlinePage) fOutlinePage).setInput(
 //            (com.google.dart.engine.ast.CompilationUnit) ((LightNodeElement) element).getNode().getRoot(),
 //            (LightNodeElement) element);
     }
   }
 
   protected void uninstallOccurrencesFinder() {
@@ skipping 292 matching lines @@
     if (unit == null) {
       return;
     }
     // prepare selection
     ISelection selection = getSelectionProvider().getSelection();
     if (!(selection instanceof TextSelection)) {
       return;
     }
     int offset = ((TextSelection) selection).getOffset();
     // apply selected element
-    LightNodeElement element = computeHighlightRangeSourceElement(resolvedUnit, offset);
-    if (element != null && fOutlinePage != null) {
-      fOutlinePage.select(element);
+    if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
+      if (fOutlinePage_NEW != null) {
+        fOutlinePage_NEW.select(offset);
+      }
+    } else {
+      LightNodeElement element = computeHighlightRangeSourceElement(resolvedUnit, offset);
+      if (element != null && fOutlinePage_OLD != null) {
+        fOutlinePage_OLD.select(element);
+      }
     }
   }
 
   /**
    * Creates and returns the preference store for this Dart editor with the given input.
    * 
    * @param input The editor input for which to create the preference store
    * @return the preference store for this editor
    */
   private IPreferenceStore createCombinedPreferenceStore(IEditorInput input) {
@@ skipping 174 matching lines @@
       if (reconciler != null) {
         reconciler.install(dartSourceViewer);
         dartSourceViewer.setReconciler(reconciler);
       }
     }
 
     if (fEncodingSupport != null) {
       fEncodingSupport.reset();
     }
 
-    setOutlinePageInput(fOutlinePage, input);
+    if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
+      setOutlinePageInput_NEW(fOutlinePage_NEW, input);
+    } else {
+      setOutlinePageInput_OLD(fOutlinePage_OLD, input);
+    }
 
     if (isShowingOverrideIndicators()) {
       installOverrideIndicator(false);
     }
   }
 
   /*
    * Tells whether the content is editable.
    */
   private boolean isContentEditable() {
     return inputResourceFile != null && !DartCore.isContainedInPackages(inputResourceFile)
         && !inputResourceFile.isReadOnly();
   }
 
   /**
    * @return {@code true} if this {@link DartEditor} was already disposed.
    */
   private boolean isDisposed() {
     return getSourceViewer() == null;
   }
 
   private boolean isEditorHoverProperty(String property) {
     return PreferenceConstants.EDITOR_TEXT_HOVER_MODIFIERS.equals(property);
   }
 
   private boolean isOutlinePageActive() {
     IWorkbenchPart part = getActivePart();
-    return part instanceof ContentOutline
-        && ((ContentOutline) part).getCurrentPage() == fOutlinePage;
+    if (DartCoreDebug.ENABLE_ANALYSIS_SERVER) {
+      return part instanceof ContentOutline
+          && ((ContentOutline) part).getCurrentPage() == fOutlinePage_NEW;
+    } else {
+      return part instanceof ContentOutline
+          && ((ContentOutline) part).getCurrentPage() == fOutlinePage_OLD;
+    }
   }
 
   private boolean isRemoveTrailingWhitespaceEnabled() {
     return PreferenceConstants.getPreferenceStore().getBoolean(
         PreferenceConstants.EDITOR_REMOVE_TRAILING_WS);
   }
 
   /**
    * @return <code>true</code> if Semantic Highlighting is enabled.
    */
@@ skipping 59 matching lines @@
               textHover,
               t,
               ITextViewerExtension2.DEFAULT_HOVER_STATE_MASK);
         }
       } else {
         sourceViewer.setTextHover(configuration.getTextHover(sourceViewer, t), t);
       }
     }
   }
 }