From 7bd9931904e2c8da57dc0ce191fa27a97c917b1c Mon Sep 17 00:00:00 2001
From: Alexander Zolotov <Alexander.Zolotov@jetbrains.com>
Date: Fri, 8 Nov 2013 15:21:15 +0400
Subject: [PATCH] Emmet: extract SurroundWithEmmetAction

---
 .../src/META-INF/XmlActions.xml               |  3 +
 .../emmet/SurroundWithEmmetAction.java        | 83 +++++++++++++++++++
 2 files changed, 86 insertions(+)
 create mode 100644 xml/impl/src/com/intellij/codeInsight/template/emmet/SurroundWithEmmetAction.java

diff --git a/platform/platform-resources/src/META-INF/XmlActions.xml b/platform/platform-resources/src/META-INF/XmlActions.xml
index 06266c625bab..ca19e8dc4dae 100644
--- a/platform/platform-resources/src/META-INF/XmlActions.xml
+++ b/platform/platform-resources/src/META-INF/XmlActions.xml
@@ -79,5 +79,8 @@
       <add-to-group group-id="RunContextPopupGroup" anchor="last"/>
       <!--<add-to-group group-id="ProjectViewPopupMenu" anchor="after" relative-to-action="EditSource"/>-->
     </action>
+    
+    <action id="SurroundWithEmmet" class="com.intellij.codeInsight.template.emmet.SurroundWithEmmetAction" 
+            text="Surround with Emmet" description="Surround selected code with Emmet abbreviation"/>
   </actions>
 </component>
diff --git a/xml/impl/src/com/intellij/codeInsight/template/emmet/SurroundWithEmmetAction.java b/xml/impl/src/com/intellij/codeInsight/template/emmet/SurroundWithEmmetAction.java
new file mode 100644
index 000000000000..4ba881666bc1
--- /dev/null
+++ b/xml/impl/src/com/intellij/codeInsight/template/emmet/SurroundWithEmmetAction.java
@@ -0,0 +1,83 @@
+/*
+ * Copyright 2000-2013 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.intellij.codeInsight.template.emmet;
+
+import com.intellij.application.options.emmet.EmmetOptions;
+import com.intellij.codeInsight.CodeInsightActionHandler;
+import com.intellij.codeInsight.actions.BaseCodeInsightAction;
+import com.intellij.codeInsight.hint.HintManager;
+import com.intellij.codeInsight.template.CustomTemplateCallback;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.SelectionModel;
+import com.intellij.openapi.fileEditor.FileDocumentManager;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.vfs.ReadonlyStatusHandler;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.PsiDocumentManager;
+import com.intellij.psi.PsiFile;
+import org.jetbrains.annotations.NotNull;
+
+public class SurroundWithEmmetAction extends BaseCodeInsightAction {
+  public SurroundWithEmmetAction() {
+    setEnabledInModalContext(true);
+  }
+
+  @Override
+  protected boolean isValidForFile(@NotNull Project project, @NotNull Editor editor, @NotNull PsiFile file) {
+    return EmmetOptions.getInstance().isEmmetEnabled() && new ZenCodingTemplate().isApplicable(file, editor.getCaretModel().getOffset(), true);
+  }
+
+  @NotNull
+  @Override
+  protected CodeInsightActionHandler getHandler() {
+    return new SurroundWithEmmetHandler();
+  }
+
+  private static class SurroundWithEmmetHandler implements CodeInsightActionHandler {
+    @Override
+    public void invoke(@NotNull Project project, @NotNull Editor editor, @NotNull PsiFile file) {
+      SelectionModel selectionModel = editor.getSelectionModel();
+      if (!selectionModel.hasSelection() && !selectionModel.hasBlockSelection()) {
+        selectionModel.selectLineAtCaret();
+      }
+
+      final Document document = editor.getDocument();
+      final VirtualFile virtualFile = FileDocumentManager.getInstance().getFile(document);
+      if (virtualFile != null) {
+        ReadonlyStatusHandler.getInstance(project).ensureFilesWritable(virtualFile);
+      }
+
+      String selection = editor.getSelectionModel().getSelectedText();
+
+      final ZenCodingTemplate template = new ZenCodingTemplate();
+      if (selection != null && template.isApplicable(file, editor.getCaretModel().getOffset(), true)) {
+        selection = selection.trim();
+        PsiDocumentManager.getInstance(project).commitAllDocuments();
+        template.wrap(selection, new CustomTemplateCallback(editor, file, true));
+      }
+      else if (!ApplicationManager.getApplication().isUnitTestMode()) {
+        HintManager.getInstance().showErrorHint(editor, "Cannot invoke Surround with Emmet in the current context");
+      }
+    }
+
+    @Override
+    public boolean startInWriteAction() {
+      return true;
+    }
+  }
+}