plugins/maven/src/main/java/org/jetbrains/idea/maven/dom/intentions/AddMavenDependencyQuickFix.java

   1 /*
   2  * Copyright 2000-2009 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.idea.maven.dom.intentions;
  17
  18 import com.intellij.codeInsight.intention.IntentionAction;
  19 import com.intellij.codeInsight.intention.LowPriorityAction;
  20 import com.intellij.openapi.application.Result;
  21 import com.intellij.openapi.command.WriteCommandAction;
  22 import com.intellij.openapi.editor.Editor;
  23 import com.intellij.openapi.project.Project;
  24 import com.intellij.psi.PsiElement;
  25 import com.intellij.psi.PsiFile;
  26 import com.intellij.psi.PsiJavaCodeReferenceElement;
  27 import com.intellij.util.IncorrectOperationException;
  28 import com.intellij.util.xml.DomUtil;
  29 import org.jetbrains.annotations.NotNull;
  30 import org.jetbrains.annotations.Nullable;
  31 import org.jetbrains.idea.maven.dom.MavenDomBundle;
  32 import org.jetbrains.idea.maven.dom.MavenDomUtil;
  33 import org.jetbrains.idea.maven.dom.model.MavenDomProjectModel;
  34 import org.jetbrains.idea.maven.indices.MavenArtifactSearchDialog;
  35 import org.jetbrains.idea.maven.model.MavenId;
  36 import org.jetbrains.idea.maven.project.MavenProject;
  37
  38 import java.util.List;
  39 import java.util.regex.Pattern;
  40
  41 public class AddMavenDependencyQuickFix implements IntentionAction, LowPriorityAction {
  42
  43   private static final Pattern CLASSNAME_PATTERN = Pattern.compile("(\\p{javaJavaIdentifierStart}\\p{javaJavaIdentifierPart}*\\.)*\\p{Lu}\\p{javaJavaIdentifierPart}+");
  44
  45   private final PsiJavaCodeReferenceElement myRef;
  46
  47   public AddMavenDependencyQuickFix(PsiJavaCodeReferenceElement ref) {
  48     myRef = ref;
  49   }
  50
  51   @NotNull
  52   public String getText() {
  53     return "Add Maven Dependency...";
  54   }
  55
  56   @NotNull
  57   public String getFamilyName() {
  58     return MavenDomBundle.message("inspection.group");
  59   }
  60
  61   public boolean isAvailable(@NotNull Project project, Editor editor, PsiFile file) {
  62     return myRef.isValid() && MavenDomUtil.findContainingProject(file) != null && looksLikeClassName(getReferenceText());
  63   }
  64
  65   private static boolean looksLikeClassName(@Nullable String text) {
  66     if (text == null) return false;
  67     //if (true) return true;
  68     return CLASSNAME_PATTERN.matcher(text).matches();
  69   }
  70
  71   public void invoke(@NotNull final Project project, Editor editor, final PsiFile file) throws IncorrectOperationException {
  72     if (!myRef.isValid()) return;
  73
  74     MavenProject mavenProject = MavenDomUtil.findContainingProject(file);
  75     if (mavenProject == null) return;
  76
  77     final List<MavenId> ids = MavenArtifactSearchDialog.searchForClass(project, getReferenceText());
  78     if (ids.isEmpty()) return;
  79
  80     final MavenDomProjectModel model = MavenDomUtil.getMavenDomProjectModel(project, mavenProject.getFile());
  81     if (model == null) return;
  82
  83     new WriteCommandAction(project, "Add Maven Dependency", DomUtil.getFile(model)) {
  84       @Override
  85       protected void run(Result result) throws Throwable {
  86         for (MavenId each : ids) {
  87           MavenDomUtil.createDomDependency(model, null, each);
  88         }
  89       }
  90     }.execute();
  91   }
  92
  93   public String getReferenceText() {
  94     PsiJavaCodeReferenceElement result = myRef;
  95     while (true) {
  96       PsiElement parent = result.getParent();
  97       if (!(parent instanceof PsiJavaCodeReferenceElement)) {
  98         break;
  99       }
 100
 101       result = (PsiJavaCodeReferenceElement)parent;
 102     }
 103
 104     return result.getQualifiedName();
 105   }
 106
 107   public boolean startInWriteAction() {
 108     return false;
 109   }
 110 }