com.google.eclipse.protobuf.ui/src/com/google/eclipse/protobuf/ui/outline/ProtobufOutlineTreeProvider.java

/*
 * Copyright (c) 2011 Google Inc.
 *
 * All rights reserved. This program and the accompanying materials are made available under the terms of the Eclipse
 * Public License v1.0 which accompanies this distribution, and is available at
 *
 * http://www.eclipse.org/legal/epl-v10.html
 */
package com.google.eclipse.protobuf.ui.outline;

import static com.google.common.collect.Lists.newArrayList;
import static com.google.eclipse.protobuf.ui.outline.Messages.importDeclarations;

import java.util.List;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.xtext.ui.editor.outline.IOutlineNode;
import org.eclipse.xtext.ui.editor.outline.impl.*;

import com.google.eclipse.protobuf.protobuf.*;
import com.google.eclipse.protobuf.protobuf.Package;

/**
 * Customization of the default outline structure.
 *
 * @author alruiz@google.com (Alex Ruiz)
 */
public class ProtobufOutlineTreeProvider extends DefaultOutlineTreeProvider {
  private static final List<Class<? extends EObject>> IGNORED_ELEMENT_TYPES = newArrayList();

  static {
    IGNORED_ELEMENT_TYPES.add(BooleanLink.class);
    IGNORED_ELEMENT_TYPES.add(FieldOption.class);
    IGNORED_ELEMENT_TYPES.add(MessageLink.class);
  }

  boolean _isLeaf(Extensions extensions) {
    return true;
  }

  boolean _isLeaf(MessageField field) {
    return true;
  }

  boolean _isLeaf(Option option) {
    return true;
  }

  boolean _isLeaf(Package aPackage) {
    return true;
  }

  protected void _createChildren(DocumentRootNode parent, Protobuf protobuf) {
    OutlineViewModel model = new OutlineViewModel(protobuf);
    for (EObject aPackage : model.packages()) {
      createNode(parent, aPackage);
    }
    addGroup(parent, protobuf, model.imports(), "imports", importDeclarations);
    for (EObject e : model.remainingElements()) {
      createNode(parent, e);
    }
  }

  private void addGroup(DocumentRootNode parent, Protobuf protobuf, List<? extends EObject> group, String imageKey,
      String text) {
    if (group.isEmpty()) {
      return;
    }
    SimpleOutlineNode groupNode = new SimpleOutlineNode(parent, protobuf, labelProvider.getImage(imageKey), text, false);
    for (EObject o : group) {
      createNode(groupNode, o);
    }
  }

  @Override protected void createNode(IOutlineNode parent, EObject modelElement) {
    if (isIgnored(modelElement)) {
      return;
    }
    super.createNode(parent, modelElement);
  }

  private boolean isIgnored(EObject modelElement) {
    for (Class<? extends EObject> ignoredType : IGNORED_ELEMENT_TYPES) {
      if (ignoredType.isInstance(modelElement)) {
        return true;
      }
    }
    return false;
  }
}