28569da51c342c21326af4510101db0ba28fc7a1
[idea/community.git] / plugins / groovy / src / org / jetbrains / plugins / groovy / annotator / ConfigureGroovyLibraryNotificationProvider.java
1 /*
2  * Copyright 2000-2010 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 org.jetbrains.plugins.groovy.annotator;
17
18 import com.intellij.ProjectTopics;
19 import com.intellij.openapi.compiler.CompilerManager;
20 import com.intellij.openapi.fileTypes.FileType;
21 import com.intellij.openapi.module.Module;
22 import com.intellij.openapi.module.ModuleUtil;
23 import com.intellij.openapi.progress.ProcessCanceledException;
24 import com.intellij.openapi.project.IndexNotReadyException;
25 import com.intellij.openapi.project.Project;
26 import com.intellij.openapi.roots.ModuleRootAdapter;
27 import com.intellij.openapi.roots.ModuleRootEvent;
28 import com.intellij.openapi.roots.ModuleRootListener;
29 import com.intellij.openapi.roots.ModuleRootManager;
30 import com.intellij.openapi.util.Key;
31 import com.intellij.openapi.vfs.VirtualFile;
32 import com.intellij.ui.EditorNotificationPanel;
33 import com.intellij.ui.EditorNotifications;
34 import org.jetbrains.annotations.NotNull;
35 import org.jetbrains.plugins.groovy.GroovyFileType;
36
37 import java.util.Collections;
38 import java.util.HashSet;
39 import java.util.Set;
40
41 /**
42  * @author Maxim.Medvedev
43  */
44 public class ConfigureGroovyLibraryNotificationProvider implements EditorNotifications.Provider<EditorNotificationPanel> {
45   private static final Key<EditorNotificationPanel> KEY = Key.create("configure.groovy.library");
46
47   private final Project myProject;
48
49   private final Set<FileType> supportedFileTypes;
50
51   public ConfigureGroovyLibraryNotificationProvider(Project project, final EditorNotifications notifications) {
52     myProject = project;
53     project.getMessageBus().connect(project).subscribe(ProjectTopics.PROJECT_ROOTS, new ModuleRootAdapter() {
54       @Override
55       public void rootsChanged(ModuleRootEvent event) {
56         notifications.updateAllNotifications();
57       }
58     });
59
60     supportedFileTypes = new HashSet<FileType>();
61     supportedFileTypes.add(GroovyFileType.GROOVY_FILE_TYPE);
62
63     for (GroovyFrameworkConfigNotification configNotification : GroovyFrameworkConfigNotification.EP_NAME.getExtensions()) {
64       Collections.addAll(supportedFileTypes, configNotification.getFrameworkFileTypes());
65     }
66   }
67
68   @Override
69   public Key<EditorNotificationPanel> getKey() {
70     return KEY;
71   }
72
73   @Override
74   public EditorNotificationPanel createNotificationPanel(VirtualFile file) {
75     try {
76       if (!supportedFileTypes.contains(file.getFileType())) return null;
77       if (CompilerManager.getInstance(myProject).isExcludedFromCompilation(file)) return null;
78
79       final Module module = ModuleUtil.findModuleForFile(file, myProject);
80       if (module == null) return null;
81
82       if (isMavenModule(module)) return null;
83
84       for (GroovyFrameworkConfigNotification configNotification : GroovyFrameworkConfigNotification.EP_NAME.getExtensions()) {
85         if (configNotification.hasFrameworkStructure(module)) {
86           if (!configNotification.hasFrameworkLibrary(module)) {
87             return configNotification.createConfigureNotificationPanel(module);
88           }
89           return null;
90         }
91       }
92     }
93     catch (ProcessCanceledException ignored) {
94
95     }
96     catch (IndexNotReadyException ignored) {
97
98     }
99
100     return null;
101   }
102
103   private static boolean isMavenModule(@NotNull Module module) {
104     for (VirtualFile root : ModuleRootManager.getInstance(module).getContentRoots()) {
105       if (root.findChild("pom.xml") != null) return true;
106     }
107
108     return false;
109   }
110
111 }