From 1a04025606bef046ad914f2188fd8df0fa9400f5 Mon Sep 17 00:00:00 2001
From: andrii <a.zinkevych@atwix.com>
Date: Fri, 21 Oct 2022 11:36:08 +0300
Subject: [PATCH] 1205: Created a new action "Override email template in a
 project theme"

---
 resources/META-INF/plugin.xml                 |  3 +
 .../OverrideEmailTemplateInThemeAction.java   | 74 +++++++++++++++++++
 2 files changed, 77 insertions(+)
 create mode 100644 src/com/magento/idea/magento2plugin/actions/generation/OverrideEmailTemplateInThemeAction.java

diff --git a/resources/META-INF/plugin.xml b/resources/META-INF/plugin.xml
index 04e5261d4..232172073 100644
--- a/resources/META-INF/plugin.xml
+++ b/resources/META-INF/plugin.xml
@@ -132,6 +132,9 @@
         <action id="OverrideTemplateInTheme.Menu" class="com.magento.idea.magento2plugin.actions.generation.OverrideTemplateInThemeAction">
             <add-to-group group-id="ProjectViewPopupMenu"/>
         </action>
+        <action id="OverrideEmailTemplateInTheme.Menu" class="com.magento.idea.magento2plugin.actions.generation.OverrideEmailTemplateInThemeAction">
+            <add-to-group group-id="ProjectViewPopupMenu"/>
+        </action>
         <action id="OverrideLayoutInTheme.Menu" class="com.magento.idea.magento2plugin.actions.generation.OverrideLayoutInThemeAction">
             <add-to-group group-id="ProjectViewPopupMenu"/>
         </action>
diff --git a/src/com/magento/idea/magento2plugin/actions/generation/OverrideEmailTemplateInThemeAction.java b/src/com/magento/idea/magento2plugin/actions/generation/OverrideEmailTemplateInThemeAction.java
new file mode 100644
index 000000000..4ecf91015
--- /dev/null
+++ b/src/com/magento/idea/magento2plugin/actions/generation/OverrideEmailTemplateInThemeAction.java
@@ -0,0 +1,74 @@
+/*
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+package com.magento.idea.magento2plugin.actions.generation;
+
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.PsiFile;
+import com.magento.idea.magento2plugin.MagentoIcons;
+import com.magento.idea.magento2plugin.actions.generation.dialog.OverrideTemplateInThemeDialog;
+import com.magento.idea.magento2plugin.magento.packages.Areas;
+import org.jetbrains.annotations.NotNull;
+
+public class OverrideEmailTemplateInThemeAction extends OverrideFileInThemeAction {
+
+    public static final String ACTION_NAME = "Override email template in a project theme";
+    public static final String ACTION_DESCRIPTION = "Override email template file in project theme";
+    public static final String EMAIL_DIRECTORY = "email";
+    public static final String HTML = "html";
+
+    public OverrideEmailTemplateInThemeAction() {
+        super(ACTION_NAME, ACTION_DESCRIPTION, MagentoIcons.MODULE);
+    }
+
+    @Override
+    public void actionPerformed(final @NotNull AnActionEvent event) {
+        final Project project = event.getProject();
+
+        if (project == null || psiFile == null) {
+            return;
+        }
+        OverrideTemplateInThemeDialog.open(project, psiFile);
+    }
+
+    @Override
+    protected boolean isOverrideAllowed(
+            final @NotNull PsiFile file,
+            final @NotNull Project project
+    ) {
+        final VirtualFile virtualFile = file.getVirtualFile();
+
+        if (virtualFile == null) {
+            return false;
+        }
+        final String fileExtension = virtualFile.getExtension();
+
+        if (fileExtension == null) {
+            return false;
+        }
+
+        if (!HTML.equals(fileExtension)) {
+            return false;
+        }
+
+        return isEmailTemplateFile(virtualFile);
+    }
+
+    private boolean isEmailTemplateFile(final @NotNull VirtualFile virtualFile) {
+        final VirtualFile directory = virtualFile.getParent();
+
+        if (directory == null) {
+            return false;
+        }
+
+        if (EMAIL_DIRECTORY.equals(directory.getName())) {
+            return Areas.getAreaByString(directory.getParent().getName()) != null;
+        } else {
+            return isEmailTemplateFile(directory);
+        }
+    }
+}