source/com/intellij/facet/impl/autodetecting/facetsTree/ImplicitFacetsTreeComponent.java

   1 /*
   2  * Copyright (c) 2007 JetBrains s.r.o. All Rights Reserved.
   3  */
   4
   5 package com.intellij.facet.impl.autodetecting.facetsTree;
   6
   7 import com.intellij.facet.FacetType;
   8 import com.intellij.facet.Facet;
   9 import com.intellij.facet.impl.autodetecting.DetectedFacetManager;
  10 import com.intellij.facet.impl.autodetecting.model.DetectedFacetInfo;
  11 import com.intellij.facet.impl.autodetecting.model.FacetInfo2;
  12 import com.intellij.facet.impl.autodetecting.model.FacetInfoBackedByFacet;
  13 import com.intellij.openapi.module.Module;
  14 import com.intellij.openapi.vfs.VirtualFile;
  15 import org.jetbrains.annotations.Nullable;
  16
  17 import java.util.*;
  18
  19 /**
  20  * @author nik
  21  */
  22 public class ImplicitFacetsTreeComponent {
  23   private DetectedFacetsTree myTree;
  24   private final DetectedFacetManager myDetectedFacetManager;
  25   private Collection<DetectedFacetsTree.FacetTypeNode> myFacetTypeNodes;
  26
  27   public ImplicitFacetsTreeComponent(DetectedFacetManager detectedFacetManager, Collection<DetectedFacetInfo<Module>> detectedFacets, HashMap<DetectedFacetInfo<Module>, List<VirtualFile>> files) {
  28     myDetectedFacetManager = detectedFacetManager;
  29
  30     Map<FacetType, DetectedFacetsTree.FacetTypeNode> facetTypeNodes = new HashMap<FacetType, DetectedFacetsTree.FacetTypeNode>();
  31
  32     Collection<FacetInfo2<Module>> sortedFacets = new LinkedHashSet<FacetInfo2<Module>>();
  33     for (FacetInfo2<Module> facet : detectedFacets) {
  34       addUnderlying(facet, sortedFacets);
  35     }
  36
  37     Map<FacetInfo2<Module>, DetectedFacetsTree.FacetNode> facetNodes = new HashMap<FacetInfo2<Module>, DetectedFacetsTree.FacetNode>();
  38     for (FacetInfo2<Module> facetInfo : sortedFacets) {
  39       FacetType facetType = getRootFacetType(facetInfo);
  40       DetectedFacetsTree.FacetTypeNode facetTypeNode = facetTypeNodes.get(facetType);
  41       if (facetTypeNode == null) {
  42         facetTypeNode = new DetectedFacetsTree.FacetTypeNode(facetType);
  43         facetTypeNodes.put(facetType, facetTypeNode);
  44       }
  45
  46       Module module = facetInfo.getModule();
  47       ModuleNodeImpl moduleNode = findOrCreateModuleNode(facetTypeNode, module);
  48
  49       DetectedFacetsTree.FacetNode parentNode = null;
  50       FacetInfo2<Module> underlyingFacet = facetInfo.getUnderlyingFacetInfo();
  51       if (underlyingFacet != null) {
  52         parentNode = facetNodes.get(underlyingFacet);
  53       }
  54
  55       VirtualFile projectRoot = facetInfo.getModule().getProject().getBaseDir();
  56       DetectedFacetsTree.FacetNode facetNode;
  57       if (facetInfo instanceof DetectedFacetInfo) {
  58         facetNode = new FacetNodeImpl((DetectedFacetInfo<Module>)facetInfo, projectRoot, parentNode, files.get(facetInfo));
  59       }
  60       else {
  61         facetNode = new RealFacetNode((FacetInfoBackedByFacet)facetInfo, projectRoot, parentNode);
  62       }
  63       facetNodes.put(facetInfo, facetNode);
  64       if (parentNode == null) {
  65         moduleNode.addRootFacet(facetNode);
  66       }
  67     }
  68
  69     myFacetTypeNodes = facetTypeNodes.values();
  70     myTree = new DetectedFacetsTree(myFacetTypeNodes);
  71   }
  72
  73   public void createFacets() {
  74     for (DetectedFacetsTree.FacetTypeNode facetTypeNode : myFacetTypeNodes) {
  75       for (DetectedFacetsTree.ModuleNode moduleNode : facetTypeNode.getModuleNodes()) {
  76         boolean accept = facetTypeNode.isChecked() && moduleNode.isChecked();
  77         if (accept) {
  78           processFacetNodes(moduleNode.getRootFacets(), true, null);
  79         }
  80         else {
  81           myDetectedFacetManager.disableDetectionInModule(facetTypeNode.getFacetType(), ((ModuleNodeImpl)moduleNode).myModule);
  82         }
  83       }
  84     }
  85   }
  86
  87   private void processFacetNodes(final List<DetectedFacetsTree.FacetNode> facetNodes, final boolean accept, final Facet underlyingFacet) {
  88     for (DetectedFacetsTree.FacetNode facetNode : facetNodes) {
  89       Facet facet = null;
  90       if (facetNode instanceof FacetNodeImpl) {
  91         DetectedFacetInfo<Module> detectedFacetInfo = ((FacetNodeImpl)facetNode).getDetectedFacetInfo();
  92         if (accept && facetNode.isChecked()) {
  93           facet = myDetectedFacetManager.createFacet(detectedFacetInfo, underlyingFacet);
  94         }
  95         else {
  96           myDetectedFacetManager.disableDetectionInFile(detectedFacetInfo);
  97         }
  98       }
  99       else {
 100         facet = ((RealFacetNode)facetNode).myFacetInfo.getFacet();
 101       }
 102       processFacetNodes(facetNode.getChildren(), accept && facetNode.isChecked(), facet);
 103     }
 104   }
 105
 106   private static void addUnderlying(final FacetInfo2<Module> facet, final Collection<FacetInfo2<Module>> sortedFacets) {
 107     FacetInfo2<Module> underlying = facet.getUnderlyingFacetInfo();
 108     if (underlying != null && !sortedFacets.contains(underlying)) {
 109       addUnderlying(underlying, sortedFacets);
 110     }
 111     sortedFacets.add(facet);
 112   }
 113
 114   private static ModuleNodeImpl findOrCreateModuleNode(final DetectedFacetsTree.FacetTypeNode facetTypeNode, final Module module) {
 115     for (DetectedFacetsTree.ModuleNode node : facetTypeNode.getModuleNodes()) {
 116       ModuleNodeImpl moduleNode = (ModuleNodeImpl)node;
 117       if (moduleNode.myModule.equals(module)) {
 118         return moduleNode;
 119       }
 120     }
 121     ModuleNodeImpl moduleNode = new ModuleNodeImpl(module);
 122     facetTypeNode.addModuleNode(moduleNode);
 123     return moduleNode;
 124   }
 125
 126   private static FacetType getRootFacetType(final FacetInfo2<?> detectedFacet) {
 127     FacetInfo2<?> facet = detectedFacet;
 128     FacetInfo2<?> underlyingFacet;
 129     while ((underlyingFacet = facet.getUnderlyingFacetInfo()) != null) {
 130       facet = underlyingFacet;
 131     }
 132     return facet.getFacetType();
 133   }
 134
 135   public DetectedFacetsTree getTree() {
 136     return myTree;
 137   }
 138
 139   private static class FacetNodeImpl extends DetectedFacetsTree.FacetNode {
 140     private final DetectedFacetInfo<Module> myDetectedFacetInfo;
 141
 142     private FacetNodeImpl(DetectedFacetInfo<Module> detectedFacet, VirtualFile projectRoot, @Nullable final DetectedFacetsTree.FacetNode parent,
 143                           final List<VirtualFile> files) {
 144       super(detectedFacet, detectedFacet.getFacetType(), projectRoot, files.toArray(new VirtualFile[files.size()]), parent);
 145       myDetectedFacetInfo = detectedFacet;
 146     }
 147
 148     public DetectedFacetInfo<Module> getDetectedFacetInfo() {
 149       return myDetectedFacetInfo;
 150     }
 151
 152     public String getName() {
 153       return myDetectedFacetInfo.getFacetName();
 154     }
 155   }
 156
 157   private static class RealFacetNode extends DetectedFacetsTree.FacetNode {
 158     private final FacetInfoBackedByFacet myFacetInfo;
 159
 160     private RealFacetNode(final FacetInfoBackedByFacet facetInfo, final VirtualFile projectRoot, @Nullable final DetectedFacetsTree.FacetNode parent) {
 161       super(facetInfo, facetInfo.getFacetType(), projectRoot, VirtualFile.EMPTY_ARRAY, parent);
 162       myFacetInfo = facetInfo;
 163     }
 164
 165     public String getDescription() {
 166       return "";
 167     }
 168
 169     public String getName() {
 170       return myFacetInfo.getFacetName();
 171     }
 172   }
 173
 174   private static class ModuleNodeImpl extends DetectedFacetsTree.ModuleNode {
 175     private final Module myModule;
 176
 177     private ModuleNodeImpl(final Module module) {
 178       super(module);
 179       myModule = module;
 180     }
 181
 182     public String getModuleName() {
 183       return myModule.getName();
 184     }
 185
 186     public String getModuleDescription() {
 187       return "";
 188     }
 189   }
 190 }