Add a top-level node in the Files view to display the SDK libraries.

editor/tools/plugins/com.google.dart.tools.ui/src/com/google/dart/tools/ui/internal/filesview/ResourceContentProvider.java

 /*
- * Copyright 2011 Google Inc.
+ * Copyright 2012 Google Inc.
  * 
  * 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
  * or implied. See the License for the specific language governing permissions and limitations under
  * the License.
  */
 
 package com.google.dart.tools.ui.internal.filesview;
 
 import org.eclipse.core.filesystem.EFS;
 import org.eclipse.core.filesystem.IFileStore;
 import org.eclipse.core.resources.IContainer;
-import org.eclipse.core.resources.IProject;
 import org.eclipse.core.resources.IResource;
 import org.eclipse.core.resources.IResourceChangeEvent;
 import org.eclipse.core.resources.IResourceChangeListener;
+import org.eclipse.core.resources.IWorkspaceRoot;
 import org.eclipse.core.resources.ResourcesPlugin;
 import org.eclipse.core.runtime.CoreException;
 import org.eclipse.jface.viewers.ITreeContentProvider;
 import org.eclipse.jface.viewers.Viewer;
 import org.eclipse.swt.widgets.Display;
 
 import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * Files view content provider.
  */
 public class ResourceContentProvider implements ITreeContentProvider, IResourceChangeListener {
   private static final IResource[] NO_CHILDREN = new IResource[0];
 
   private Viewer viewer;
 
-// Synthetic libraries node support
-// private Map<IFileStore, DartLibrary> fileStoreMap;
+  private Map<IFileStore, SdkLibraryNode> sdkChildMap;
 
   public ResourceContentProvider() {
     ResourcesPlugin.getWorkspace().addResourceChangeListener(this);
   }
 
   @Override
   public void dispose() {
     ResourcesPlugin.getWorkspace().removeResourceChangeListener(this);
   }
 
   @Override
   public Object[] getChildren(Object element) {
     try {
-      if (element instanceof IProject) {
-        return getProjectChildren((IProject) element);
+      if (element instanceof IWorkspaceRoot) {
+        IWorkspaceRoot root = (IWorkspaceRoot) element;
+
+        List<Object> children = new ArrayList<Object>();
+
+        children.addAll(Arrays.asList(root.members()));
+        children.add(SdkDirectoryNode.INSTANCE);
+
+        return children.toArray();
       } else if (element instanceof IContainer) {
         IContainer container = (IContainer) element;
         return filteredMembers(container).toArray();
       } else if (element instanceof IFileStore) {
         IFileStore fileStore = (IFileStore) element;
         return fileStore.childStores(EFS.NONE, null);
+      } else if (element instanceof SdkDirectoryNode) {
+        return ((SdkDirectoryNode) element).getLibraries();
+      } else if (element instanceof SdkLibraryNode) {
+        return ((SdkLibraryNode) element).getFiles();
       }
-// Synthetic libraries node support
-//      else if (element instanceof DartLibraryImpl) {
-//        return getLibraryChildren((DartLibraryImpl) element);
-//      }
     } catch (CoreException ce) {
       //fall through
     }
+
     return NO_CHILDREN;
   }
 
   @Override
   public Object[] getElements(Object inputElement) {
     return getChildren(inputElement);
   }
 
   @Override
   public Object getParent(Object element) {
     if (element instanceof IResource) {
       return ((IResource) element).getParent();
     } else if (element instanceof IFileStore) {
       IFileStore fileStore = (IFileStore) element;
-      IFileStore parent = fileStore.getParent();
-// Synthetic libraries node support
-//      if (getFileStoreMap().containsKey(parent)) {
-//        return getFileStoreMap().get(parent);
-//      }
 
-      return parent;
+      if (getSdkParent(fileStore) != null) {
+        return getSdkParent(fileStore);
+      }
+
+      return fileStore.getParent();
+    } else if (element instanceof SdkLibraryNode) {
+      return SdkDirectoryNode.INSTANCE;
     } else {
       return null;
     }
   }
 
   @Override
   public boolean hasChildren(Object element) {
-    try {
-      if (element instanceof IProject) {
-        return filteredMembers((IProject) element).size() > 0;
-      }
-    } catch (CoreException exception) {
-      return false;
-    }
-
     return getChildren(element).length > 0;
   }
 
   @Override
   public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
     this.viewer = viewer;
   }
 
   @Override
   public void resourceChanged(IResourceChangeEvent event) {
     Display.getDefault().asyncExec(new Runnable() {
       @Override
       public void run() {
         viewer.refresh();
       }
     });
   }
 
+  private Map<IFileStore, SdkLibraryNode> createSdkChildMap() {
+    Map<IFileStore, SdkLibraryNode> map = new HashMap<IFileStore, SdkLibraryNode>();
+
+    for (SdkLibraryNode library : SdkDirectoryNode.INSTANCE.getLibraries()) {
+      for (IFileStore child : library.getFiles()) {
+        map.put(child, library);
+      }
+    }
+
+    return map;
+  }
+
   private List<IResource> filteredMembers(IContainer container) throws CoreException {
     List<IResource> children = new ArrayList<IResource>();
 
     for (IResource child : container.members()) {
       String name = child.getName();
       if (!(name.startsWith("."))) {
         children.add(child);
       }
     }
 
     return children;
   }
 
-// Synthetic libraries node support
-//  private Map<IFileStore, DartLibrary> getFileStoreMap() {
-//    if (fileStoreMap == null) {
-//      fileStoreMap = new HashMap<IFileStore, DartLibrary>();
-//
-//      try {
-//        for (DartLibrary lib : DartModelManager.getInstance().getDartModel().getBundledLibraries()) {
-//          if (lib instanceof DartLibraryImpl) {
-//            DartLibraryImpl library = (DartLibraryImpl) lib;
-//
-//            IFileStore libraryFileStore = getLibraryFileStore(library);
-//
-//            fileStoreMap.put(libraryFileStore, library);
-//          }
-//        }
-//      } catch (CoreException exception) {
-//        DartToolsPlugin.log(exception);
-//      }
-//    }
-//
-//    return fileStoreMap;
-//  }
+  private SdkLibraryNode getSdkParent(IFileStore fileStore) {
+    if (sdkChildMap == null) {
+      sdkChildMap = createSdkChildMap();
+    }
 
-//  private Object[] getLibraryChildren(DartLibraryImpl library) throws CoreException {
-//    IFileStore libraryFileStore = getLibraryFileStore(library);
-//
-//    if (libraryFileStore != null) {
-//      return libraryFileStore.childStores(EFS.NONE, null);
-//    } else {
-//      return NO_CHILDREN;
-//    }
-//  }
-
-//  private IFileStore getLibraryFileStore(DartLibraryImpl library) throws CoreException {
-//    URI uri = library.getLibrarySourceFile().getUri();
-//
-//    if (SystemLibraryManager.isDartUri(uri)) {
-//      SystemLibraryManager libraryManager = SystemLibraryManagerProvider.getSystemLibraryManager();
-//
-//      uri = libraryManager.translateDartUri(uri);
-//    }
-//
-//    if ("file".equals(uri.getScheme())) {
-//      IFileStore fileStore = EFS.getStore(uri);
-//
-//      return fileStore.getParent();
-//    } else {
-//      return null;
-//    }
-//  }
-
-  private Object[] getProjectChildren(IProject project) throws CoreException {
-    List<Object> children = new ArrayList<Object>();
-
-    List<IResource> resources = filteredMembers(project);
-    for (IResource resource : resources) {
-      children.add(resource);
-    }
-// Synthetic libraries node support
-//    if (DartProjectNature.hasDartNature(project)) {
-//      DartProject dartProject = DartCore.create(project);
-//      for (DartLibrary library : getSystemLibraries(dartProject.getDartLibraries())) {
-//        children.add(library);
-//      }
-//    }
-
-    return children.toArray();
+    return sdkChildMap.get(fileStore);
   }
 
-// Synthetic libraries node support
-//  private Set<DartLibrary> getSystemLibraries(DartLibrary[] libraries) throws CoreException {
-//    Set<DartLibrary> results = new HashSet<DartLibrary>();
-//
-//    for (DartLibrary library : libraries) {
-//      for (DartLibrary lib : library.getImportedLibraries()) {
-//        if (!lib.isLocal()) {
-//          results.add(lib);
-//        }
-//      }
-//    }
-//
-//    // TODO(pquitslund): the model should handle this.
-//    DartLibrary coreLibrary = DartModelManager.getInstance().getDartModel().getCoreLibrary();
-//    if (!coreLibrary.isLocal()) {
-//      results.add(coreLibrary);
-//    }
-//
-//    return results;
-//  }
-
 }