Copy file to correct location in selected theme

Author: pmclain

src/com/magento/idea/magento2plugin/actions/generation/dialog/OverrideInThemeDialog.java

@@ -5,9 +5,8 @@
 package com.magento.idea.magento2plugin.actions.generation.dialog;
 
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.popup.JBPopupFactory;
 import com.intellij.psi.PsiFile;
-import com.jetbrains.php.lang.psi.elements.Method;
-import com.jetbrains.php.lang.psi.elements.PhpClass;
 import com.magento.idea.magento2plugin.actions.generation.dialog.validator.OverrideInThemeDialogValidator;
 import com.magento.idea.magento2plugin.actions.generation.generator.OverrideInThemeGenerator;
 import com.magento.idea.magento2plugin.indexes.ModuleIndex;
@@ -68,6 +67,7 @@ public void actionPerformed(ActionEvent e) {
 
     private void onOK() {
         if (!validator.validate(project)) {
+            JBPopupFactory.getInstance().createMessage("Invalid theme selection.").showCenteredInCurrentWindow(project);
             return;
         }
 

src/com/magento/idea/magento2plugin/actions/generation/generator/OverrideInThemeGenerator.java

@@ -1,9 +1,18 @@
 package com.magento.idea.magento2plugin.actions.generation.generator;
 
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.popup.JBPopupFactory;
 import com.intellij.psi.PsiDirectory;
 import com.intellij.psi.PsiFile;
 import com.magento.idea.magento2plugin.indexes.ModuleIndex;
+import com.magento.idea.magento2plugin.util.RegExUtil;
+import com.magento.idea.magento2plugin.util.magento.GetComponentNameByDirectory;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
 public class OverrideInThemeGenerator {
 
@@ -25,12 +34,85 @@ public OverrideInThemeGenerator(Project project) {
     public void execute(PsiFile baseFile, String themeName) {
         ModuleIndex moduleIndex = ModuleIndex.getInstance(project);
         PsiDirectory directory = moduleIndex.getModuleDirectoryByModuleName(themeName);
+        GetComponentNameByDirectory getComponentNameByDirectory = GetComponentNameByDirectory.getInstance(project);
+        String componentName = getComponentNameByDirectory.execute(baseFile.getContainingDirectory());
+
+        List<String> pathComponents;
+        if (isModule(componentName)) {
+            pathComponents = getModulePathComponents(baseFile, componentName);
+        } else if (isTheme(componentName)) {
+            pathComponents = getThemePathComponents(baseFile);
+        } else {
+            return;
+        }
+
+        directory = getTargetDirectory(directory, pathComponents);
+
+        if (directory.findFile(baseFile.getName()) != null) {
+            JBPopupFactory.getInstance()
+                    .createMessage("Override already exists in selected theme.")
+                    .showCenteredInCurrentWindow(project);
+            directory.findFile(baseFile.getName()).navigate(true);
+            return;
+        }
 
-        //TODO: Copy file to correct path when source is module
-        //TODO: Copy file to correct path when source is theme
-        //TODO: Throw error if theme file already exists
         directory.copyFileFrom(baseFile.getName(), baseFile);
         PsiFile newFile = directory.findFile(baseFile.getName());
         newFile.navigate(true);
     }
+
+    private boolean isModule(String componentName) {
+        Pattern modulePattern = Pattern.compile(RegExUtil.Magento.MODULE_NAME);
+        Matcher moduleMatcher = modulePattern.matcher(componentName);
+
+        return moduleMatcher.find();
+    }
+
+    private boolean isTheme(String componentName) {
+        Pattern themePattern = Pattern.compile(RegExUtil.Magento.THEME_NAME);
+        Matcher themeMatcher = themePattern.matcher(componentName);
+
+        return themeMatcher.find();
+    }
+
+    private List<String> getModulePathComponents(PsiFile file, String componentName) {
+        List<String> pathComponents = new ArrayList<>();
+        PsiDirectory parent = file.getParent();
+        while (!parent.getName().equals("frontend") && !parent.getName().equals("adminhtml")) {
+            pathComponents.add(parent.getName());
+            parent = parent.getParent();
+        }
+        pathComponents.add(componentName);
+        Collections.reverse(pathComponents);
+
+        return pathComponents;
+    }
+
+    private List<String> getThemePathComponents(PsiFile file) {
+        List<String> pathComponents = new ArrayList<>();
+        Pattern pattern = Pattern.compile(RegExUtil.Magento.MODULE_NAME);
+
+        PsiDirectory parent = file.getParent();
+        do {
+            pathComponents.add(parent.getName());
+            parent = parent.getParent();
+        } while (!pattern.matcher(parent.getName()).find());
+        pathComponents.add(parent.getName());
+        Collections.reverse(pathComponents);
+
+        return pathComponents;
+    }
+
+    private PsiDirectory getTargetDirectory(PsiDirectory directory, List<String> pathComponents) {
+        for (int i = 0; i < pathComponents.size(); i++) {
+            String directoryName = pathComponents.get(i);
+            if (directory.findSubdirectory(directoryName) != null) {
+                directory = directory.findSubdirectory(directoryName);
+            } else {
+                directory = directory.createSubdirectory(directoryName);
+            }
+        }
+
+        return directory;
+    }
 }