ResourceBundleGrouper.java

/*
 * Copyright 2000-2014 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
 *
 * 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.intellij.lang.properties.projectView;

import com.intellij.ide.projectView.TreeStructureProvider;
import com.intellij.ide.projectView.ViewSettings;
import com.intellij.ide.util.treeView.AbstractTreeNode;
import com.intellij.lang.properties.CustomResourceBundle;
import com.intellij.lang.properties.PropertiesImplUtil;
import com.intellij.lang.properties.ResourceBundle;
import com.intellij.lang.properties.psi.PropertiesFile;
import com.intellij.openapi.actionSystem.PlatformDataKeys;
import com.intellij.openapi.application.ReadAction;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.project.DumbAware;
import com.intellij.openapi.project.Project;
import com.intellij.psi.PsiFile;
import gnu.trove.THashMap;
import org.jetbrains.annotations.NotNull;

import java.util.*;

public class ResourceBundleGrouper implements TreeStructureProvider, DumbAware {
  private final static Logger LOG = Logger.getInstance(ResourceBundleGrouper.class);

  private final Project myProject;

  public ResourceBundleGrouper(Project project) {
    myProject = project;
  }

  @Override
  @NotNull
  public Collection<AbstractTreeNode> modify(@NotNull AbstractTreeNode parent, @NotNull final Collection<AbstractTreeNode> children, final ViewSettings settings) {
    if (parent instanceof ResourceBundleNode) return children;

    return ReadAction.compute(() -> {
      Map<ResourceBundle, Collection<PropertiesFile>> childBundles = new THashMap<>();
      for (AbstractTreeNode child : children) {
        Object f = child.getValue();
        if (f instanceof PsiFile) {
          PropertiesFile propertiesFile = PropertiesImplUtil.getPropertiesFile((PsiFile)f);
          if (propertiesFile != null) {
            boolean isProcessed = false;
            for (Collection<PropertiesFile> files : childBundles.values()) {
              if (files.contains(propertiesFile)) {
                isProcessed = true;
                break;
              }
            }
            if (isProcessed) continue;
            PropertiesImplUtil.ResourceBundleWithCachedFiles resourceBundleWithCachedFiles =
              PropertiesImplUtil.getResourceBundleWithCachedFiles(propertiesFile);
            final ResourceBundle bundle = resourceBundleWithCachedFiles.getBundle();
            Collection<PropertiesFile> files = childBundles.get(bundle);
            if (files == null) {
              files = new LinkedHashSet<>();
              childBundles.put(bundle, files);
            }
            files.add(propertiesFile);
            files.addAll(resourceBundleWithCachedFiles.getFiles());
          }
        }
      }

      List<AbstractTreeNode> result = new ArrayList<>();
      for (Map.Entry<ResourceBundle, Collection<PropertiesFile>> entry : childBundles.entrySet()) {
        ResourceBundle resourceBundle = entry.getKey();
        Collection<PropertiesFile> files = entry.getValue();
        if (files.size() != 1) {
          result.add(new ResourceBundleNode(myProject, resourceBundle, settings));
        }
      }

      for (AbstractTreeNode child : children) {
        Object f = child.getValue();
        if (f instanceof PsiFile) {
          PropertiesFile propertiesFile = PropertiesImplUtil.getPropertiesFile((PsiFile)f);
          if (propertiesFile != null) {
            ResourceBundle bundle = null;
            for (Map.Entry<ResourceBundle, Collection<PropertiesFile>> e : childBundles.entrySet()) {
              if (e.getValue().contains(propertiesFile)) {
                bundle = e.getKey();
                break;
              }
            }
            LOG.assertTrue(bundle != null);
            if (childBundles.get(bundle).size() != 1) {
              continue;
            }
            else if (bundle instanceof CustomResourceBundle) {
              final CustomResourceBundlePropertiesFileNode node =
                new CustomResourceBundlePropertiesFileNode(myProject, (PsiFile)f, settings);
              result.add(node);
            }
          }
        }
        result.add(child);
      }

      return result;
    });
  }

  @Override
  public Object getData(Collection<AbstractTreeNode> selected, String dataName) {
    if (selected == null) return null;
    if (PlatformDataKeys.DELETE_ELEMENT_PROVIDER.is(dataName)) {
      for (AbstractTreeNode selectedElement : selected) {
        Object element = selectedElement.getValue();
        if (element instanceof ResourceBundle) {
          return new ResourceBundleDeleteProvider();
        }
      }
    } else if (ResourceBundle.ARRAY_DATA_KEY.is(dataName)) {
      final List<ResourceBundle> selectedElements = new ArrayList<>();
      for (AbstractTreeNode node : selected) {
        final Object value = node.getValue();
        if (value instanceof ResourceBundle) {
          selectedElements.add((ResourceBundle)value);
        }
      }
      return selectedElements.isEmpty() ? null : selectedElements.toArray(new ResourceBundle[selectedElements.size()]);
    }
    return null;
  }
}