java/idea-ui/src/com/intellij/openapi/roots/ui/configuration/artifacts/sourceItems/SourceItemNodeBase.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.openapi.roots.ui.configuration.artifacts.sourceItems;
  17
  18 import com.intellij.ide.projectView.PresentationData;
  19 import com.intellij.ide.util.treeView.NodeDescriptor;
  20 import com.intellij.openapi.extensions.Extensions;
  21 import com.intellij.openapi.roots.ui.configuration.artifacts.ArtifactEditorEx;
  22 import com.intellij.openapi.roots.ui.configuration.artifacts.nodes.ArtifactsTreeNode;
  23 import com.intellij.packaging.artifacts.Artifact;
  24 import com.intellij.packaging.ui.ArtifactEditorContext;
  25 import com.intellij.packaging.ui.PackagingSourceItem;
  26 import com.intellij.packaging.ui.PackagingSourceItemsProvider;
  27 import com.intellij.packaging.ui.TreeNodePresentation;
  28 import com.intellij.ui.treeStructure.SimpleNode;
  29 import org.jetbrains.annotations.Nullable;
  30
  31 import java.util.ArrayList;
  32 import java.util.Collection;
  33 import java.util.List;
  34
  35 /**
  36  * @author nik
  37  */
  38 public abstract class SourceItemNodeBase extends ArtifactsTreeNode {
  39   private Artifact myArtifact;
  40   private final ArtifactEditorEx myArtifactEditor;
  41
  42   public SourceItemNodeBase(ArtifactEditorContext context, NodeDescriptor parentDescriptor, final TreeNodePresentation presentation,
  43                             ArtifactEditorEx artifactEditor) {
  44     super(context, parentDescriptor, presentation);
  45     myArtifact = artifactEditor.getArtifact();
  46     myArtifactEditor = artifactEditor;
  47   }
  48
  49   protected ArtifactEditorEx getArtifactEditor() {
  50     return myArtifactEditor;
  51   }
  52
  53   @Override
  54   protected void update(PresentationData presentation) {
  55     final Artifact artifact = myArtifactEditor.getArtifact();
  56     if (!myArtifact.equals(artifact)) {
  57       myArtifact = artifact;
  58     }
  59     super.update(presentation);
  60   }
  61
  62   @Override
  63   protected SimpleNode[] buildChildren() {
  64     final PackagingSourceItemsProvider[] providers = Extensions.getExtensions(PackagingSourceItemsProvider.EP_NAME);
  65     List<SimpleNode> children = new ArrayList<SimpleNode>();
  66     for (PackagingSourceItemsProvider provider : providers) {
  67       final Collection<? extends PackagingSourceItem> items = provider.getSourceItems(myContext, myArtifact, getSourceItem());
  68       for (PackagingSourceItem item : items) {
  69         if (myArtifact.getArtifactType().isSuitableItem(item)) {
  70           children.add(new SourceItemNode(myContext, this, item, myArtifactEditor));
  71         }
  72       }
  73     }
  74     return children.toArray(new SimpleNode[children.size()]);
  75   }
  76
  77   @Nullable
  78   protected abstract PackagingSourceItem getSourceItem();
  79 }