annotate return values of IndexableSetContributor.getAdditionalRootsToIndex as @NotNull
[idea/community.git] / xml / impl / src / com / intellij / javaee / ExternalResourcesRootsProvider.java
1 /*
2  * Copyright 2000-2015 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.javaee;
17
18 import com.intellij.codeInsight.daemon.impl.quickfix.FetchExtResourceAction;
19 import com.intellij.openapi.util.NotNullLazyValue;
20 import com.intellij.openapi.vfs.LocalFileSystem;
21 import com.intellij.openapi.vfs.VfsUtilCore;
22 import com.intellij.openapi.vfs.VirtualFile;
23 import com.intellij.util.containers.ContainerUtil;
24 import com.intellij.util.indexing.IndexableSetContributor;
25 import gnu.trove.THashSet;
26 import org.jetbrains.annotations.NotNull;
27
28 import java.util.Map;
29 import java.util.Set;
30
31 /**
32  * @author Dmitry Avdeev
33  */
34 public class ExternalResourcesRootsProvider extends IndexableSetContributor {
35   private final NotNullLazyValue<Set<String>> myStandardResources = new NotNullLazyValue<Set<String>>() {
36     @NotNull
37     @Override
38     protected Set<String> compute() {
39       ExternalResourceManagerExImpl manager = (ExternalResourceManagerExImpl)ExternalResourceManager.getInstance();
40       Set<ExternalResourceManagerExImpl.Resource> dirs = new THashSet<ExternalResourceManagerExImpl.Resource>();
41       Set<String> set = new THashSet<String>();
42       for (Map<String, ExternalResourceManagerExImpl.Resource> map : manager.getStandardResources()) {
43         for (ExternalResourceManagerExImpl.Resource resource : map.values()) {
44           ExternalResourceManagerExImpl.Resource dir = new ExternalResourceManagerExImpl.Resource(
45             resource.directoryName(), resource);
46
47           if (dirs.add(dir)) {
48             String url = resource.getResourceUrl();
49             if (url != null) {
50               set.add(url.substring(0, url.lastIndexOf('/') + 1));
51             }
52           }
53         }
54       }
55       return set;
56     }
57   };
58
59   @NotNull
60   @Override
61   public Set<VirtualFile> getAdditionalRootsToIndex() {
62     Set<VirtualFile> roots = new THashSet<VirtualFile>();
63     for (String url : myStandardResources.getValue()) {
64       VirtualFile file = VfsUtilCore.findRelativeFile(url, null);
65       if (file != null) {
66         roots.add(file);
67       }
68     }
69
70     String path = FetchExtResourceAction.getExternalResourcesPath();
71     VirtualFile extResources = LocalFileSystem.getInstance().findFileByPath(path);
72     ContainerUtil.addIfNotNull(extResources, roots);
73
74     return roots;
75   }
76 }