Fixed directory for source model

Author: ProkopovVitaliy

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

@@ -49,7 +49,11 @@ public SourceModelGenerator(
 
     @Override
     protected AbstractPhpFile initFile() {
-        return new SourceModelFile(data.getModuleName(), data.getClassName());
+        return new SourceModelFile(
+                data.getModuleName(),
+                data.getClassName(),
+                data.getDirectory()
+        );
     }
 
     @Override

src/com/magento/idea/magento2plugin/magento/files/SourceModelFile.java

@@ -11,6 +11,7 @@ public class SourceModelFile extends AbstractPhpFile {
     public static final String HUMAN_READABLE_NAME = "Source model class";
     public static final String TEMPLATE = "Magento Source Model Class";
     public static final String DEFAULT_DIR = "Model/Source";
+    private String directory;
 
     /**
      * Constructor.
@@ -24,9 +25,27 @@ public SourceModelFile(
         super(moduleName, className);
     }
 
+    /**
+     * Constructor.
+     *
+     * @param className String
+     */
+    public SourceModelFile(
+            @NotNull final String moduleName,
+            @NotNull final String className,
+            final String directory
+    ) {
+        this(moduleName, className);
+        this.directory = directory;
+    }
+
     @Override
     public String getDirectory() {
-        return DEFAULT_DIR;
+        if (directory == null) {
+            return DEFAULT_DIR;
+        }
+
+        return directory;
     }
 
     @Override