java/java-impl/src/com/intellij/codeInsight/daemon/impl/quickfix/JetBrainsAnnotationsExternalLibraryResolver.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.codeInsight.daemon.impl.quickfix;
  17
  18 import com.intellij.codeInsight.AnnotationUtil;
  19 import com.intellij.codeInsight.daemon.quickFix.ExternalLibraryDescriptor;
  20 import com.intellij.codeInsight.daemon.quickFix.ExternalLibraryResolver;
  21 import com.intellij.openapi.application.PathManager;
  22 import com.intellij.openapi.module.EffectiveLanguageLevelUtil;
  23 import com.intellij.openapi.module.Module;
  24 import com.intellij.openapi.util.io.FileUtil;
  25 import com.intellij.pom.java.LanguageLevel;
  26 import com.intellij.util.ThreeState;
  27 import org.jetbrains.annotations.NotNull;
  28 import org.jetbrains.annotations.Nullable;
  29
  30 import java.io.File;
  31 import java.util.Collections;
  32 import java.util.List;
  33
  34 /**
  35  * @author nik
  36  */
  37 public class JetBrainsAnnotationsExternalLibraryResolver extends ExternalLibraryResolver {
  38   private static final ExternalLibraryDescriptor JAVA5 = new JetBrainsAnnotationsLibraryDescriptor(false);
  39   private static final ExternalLibraryDescriptor JAVA8 = new JetBrainsAnnotationsLibraryDescriptor(true);
  40
  41   @Nullable
  42   @Override
  43   public ExternalClassResolveResult resolveClass(@NotNull String shortClassName, @NotNull ThreeState isAnnotation, @NotNull Module contextModule) {
  44     if (AnnotationUtil.isJetbrainsAnnotation(shortClassName)) {
  45       ExternalLibraryDescriptor libraryDescriptor = getAnnotationsLibraryDescriptor(contextModule);
  46       return new ExternalClassResolveResult("org.jetbrains.annotations." + shortClassName, libraryDescriptor);
  47     }
  48     return null;
  49   }
  50
  51   @NotNull
  52   public static ExternalLibraryDescriptor getAnnotationsLibraryDescriptor(@NotNull Module contextModule) {
  53     boolean java8 = EffectiveLanguageLevelUtil.getEffectiveLanguageLevel(contextModule).isAtLeast(LanguageLevel.JDK_1_8);
  54     return java8 ? JAVA8 : JAVA5;
  55   }
  56
  57   private static class JetBrainsAnnotationsLibraryDescriptor extends ExternalLibraryDescriptor {
  58     private boolean myJava8;
  59
  60     public JetBrainsAnnotationsLibraryDescriptor(boolean java8) {
  61       super("com.intellij", "annotations", null);
  62       myJava8 = java8;
  63     }
  64
  65     @NotNull
  66     @Override
  67     public List<String> getLibraryClassesRoots() {
  68       List<String> paths;
  69       if (myJava8) {
  70         paths = Collections.singletonList(FileUtil.toSystemIndependentName(new File(PathManager.getHomePath(), "redist/annotations-java8.jar").getAbsolutePath()));
  71       }
  72       else {
  73         paths = Collections.singletonList(FileUtil.toSystemIndependentName(new File(PathManager.getLibPath(), "annotations.jar").getAbsolutePath()));
  74       }
  75       return paths;
  76     }
  77   }
  78 }