xml/dom-openapi/src/com/intellij/util/xml/highlighting/AddDomElementQuickFix.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
  17 package com.intellij.util.xml.highlighting;
  18
  19 import com.intellij.codeInspection.LocalQuickFix;
  20 import com.intellij.codeInspection.ProblemDescriptor;
  21 import com.intellij.openapi.project.Project;
  22 import com.intellij.psi.xml.XmlTag;
  23 import com.intellij.util.xml.DomBundle;
  24 import com.intellij.util.xml.DomElement;
  25 import org.jetbrains.annotations.NotNull;
  26
  27 /**
  28  * @author Dmitry Avdeev
  29  */
  30 public class AddDomElementQuickFix<T extends DomElement> implements LocalQuickFix {
  31
  32   protected final T myElement;
  33   protected final String myName;
  34
  35   public AddDomElementQuickFix(@NotNull T element) {
  36     myElement = (T)element.createStableCopy();
  37     myName = computeName();
  38   }
  39
  40   @NotNull
  41   public String getName() {
  42     return myName;
  43   }
  44
  45   private String computeName() {
  46     final String name = myElement.getXmlElementName();
  47     return isTag() ?
  48            DomBundle.message("add.element.fix.name", name) :
  49            DomBundle.message("add.attribute.fix.name", name);
  50   }
  51
  52   private boolean isTag() {
  53     return myElement.getXmlElement() instanceof XmlTag;
  54   }
  55
  56   @NotNull
  57   public String getFamilyName() {
  58     return DomBundle.message("quick.fixes.family");
  59   }
  60
  61   public void applyFix(@NotNull Project project, @NotNull ProblemDescriptor descriptor) {
  62     myElement.ensureXmlElementExists();
  63   }
  64 }