| /* |
| * Copyright (C) 2014 The Android Open Source Project |
| * |
| * 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.android.tools.idea.gradle.projectView; |
| |
| import com.android.tools.idea.gradle.util.GradleUtil; |
| import com.intellij.facet.ProjectFacetManager; |
| import com.intellij.ide.projectView.PresentationData; |
| import com.intellij.ide.projectView.ProjectViewNode; |
| import com.intellij.ide.projectView.ProjectViewNodeDecorator; |
| import com.intellij.ide.projectView.impl.nodes.PsiDirectoryNode; |
| import com.intellij.openapi.module.Module; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.roots.ProjectRootManager; |
| import com.intellij.openapi.vfs.VirtualFile; |
| import com.intellij.packageDependencies.ui.PackageDependenciesNode; |
| import com.intellij.psi.PsiDirectory; |
| import com.intellij.ui.ColoredTreeCellRenderer; |
| import org.jetbrains.android.facet.AndroidFacet; |
| import org.jetbrains.annotations.NotNull; |
| |
| /** Provides custom icons for modules based on the module type. */ |
| public class ModuleNodeIconDecorator implements ProjectViewNodeDecorator { |
| @Override |
| public void decorate(ProjectViewNode node, PresentationData data) { |
| if (!(node instanceof PsiDirectoryNode)) { |
| return; |
| } |
| |
| final PsiDirectoryNode psiDirectoryNode = (PsiDirectoryNode)node; |
| PsiDirectory psiDirectory = psiDirectoryNode.getValue(); |
| |
| Project project = psiDirectory.getProject(); |
| if (!ProjectFacetManager.getInstance(project).hasFacets(AndroidFacet.ID)) { |
| return; |
| } |
| |
| VirtualFile folder = psiDirectory.getVirtualFile(); |
| Module module = ProjectRootManager.getInstance(project).getFileIndex().getModuleForFile(folder); |
| if (module != null && isModuleFolder(folder, module)) { |
| data.setIcon(GradleUtil.getModuleIcon(module)); |
| } |
| } |
| |
| private static boolean isModuleFolder(@NotNull VirtualFile folder, @NotNull Module module) { |
| VirtualFile moduleFile = module.getModuleFile(); |
| return moduleFile != null && folder.equals(moduleFile.getParent()); |
| } |
| |
| @Override |
| public void decorate(PackageDependenciesNode node, ColoredTreeCellRenderer cellRenderer) { |
| } |
| } |