IDEA-114837 Should add library in test scope if adding a maven dependency from test...
[idea/community.git] / 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.openapi.roots.ProjectRootManager;
25 import com.intellij.openapi.vfs.VirtualFile;
26 import com.intellij.psi.PsiElement;
27 import com.intellij.psi.PsiFile;
28 import com.intellij.psi.PsiJavaCodeReferenceElement;
29 import com.intellij.util.IncorrectOperationException;
30 import com.intellij.util.xml.DomUtil;
31 import org.jetbrains.annotations.NotNull;
32 import org.jetbrains.annotations.Nullable;
33 import org.jetbrains.idea.maven.dom.MavenDomBundle;
34 import org.jetbrains.idea.maven.dom.MavenDomUtil;
35 import org.jetbrains.idea.maven.dom.model.MavenDomDependency;
36 import org.jetbrains.idea.maven.dom.model.MavenDomProjectModel;
37 import org.jetbrains.idea.maven.indices.MavenArtifactSearchDialog;
38 import org.jetbrains.idea.maven.model.MavenId;
39 import org.jetbrains.idea.maven.project.MavenProject;
40
41 import java.util.List;
42 import java.util.regex.Pattern;
43
44 public class AddMavenDependencyQuickFix implements IntentionAction, LowPriorityAction {
45
46   private static final Pattern CLASSNAME_PATTERN = Pattern.compile("(\\p{javaJavaIdentifierStart}\\p{javaJavaIdentifierPart}*\\.)*\\p{Lu}\\p{javaJavaIdentifierPart}+");
47
48   private final PsiJavaCodeReferenceElement myRef;
49
50   public AddMavenDependencyQuickFix(PsiJavaCodeReferenceElement ref) {
51     myRef = ref;
52   }
53
54   @NotNull
55   public String getText() {
56     return "Add Maven Dependency...";
57   }
58
59   @NotNull
60   public String getFamilyName() {
61     return MavenDomBundle.message("inspection.group");
62   }
63
64   public boolean isAvailable(@NotNull Project project, Editor editor, PsiFile file) {
65     return myRef.isValid() && MavenDomUtil.findContainingProject(file) != null && looksLikeClassName(getReferenceText());
66   }
67
68   private static boolean looksLikeClassName(@Nullable String text) {
69     if (text == null) return false;
70     //if (true) return true;
71     return CLASSNAME_PATTERN.matcher(text).matches();
72   }
73
74   public void invoke(@NotNull final Project project, Editor editor, final PsiFile file) throws IncorrectOperationException {
75     if (!myRef.isValid()) return;
76
77     MavenProject mavenProject = MavenDomUtil.findContainingProject(file);
78     if (mavenProject == null) return;
79
80     final List<MavenId> ids = MavenArtifactSearchDialog.searchForClass(project, getReferenceText());
81     if (ids.isEmpty()) return;
82
83     final MavenDomProjectModel model = MavenDomUtil.getMavenDomProjectModel(project, mavenProject.getFile());
84     if (model == null) return;
85
86     new WriteCommandAction(project, "Add Maven Dependency", DomUtil.getFile(model)) {
87       @Override
88       protected void run(Result result) throws Throwable {
89         boolean isTestSource = false;
90
91         VirtualFile virtualFile = file.getOriginalFile().getVirtualFile();
92         if (virtualFile != null) {
93           isTestSource = ProjectRootManager.getInstance(project).getFileIndex().isInSourceContent(virtualFile);
94         }
95
96         for (MavenId each : ids) {
97           MavenDomDependency dependency = MavenDomUtil.createDomDependency(model, null, each);
98           if (isTestSource) {
99             dependency.getScope().setStringValue("test");
100           }
101         }
102       }
103     }.execute();
104   }
105
106   public String getReferenceText() {
107     PsiJavaCodeReferenceElement result = myRef;
108     while (true) {
109       PsiElement parent = result.getParent();
110       if (!(parent instanceof PsiJavaCodeReferenceElement)) {
111         break;
112       }
113
114       result = (PsiJavaCodeReferenceElement)parent;
115     }
116
117     return result.getQualifiedName();
118   }
119
120   public boolean startInWriteAction() {
121     return false;
122   }
123 }