Feature: Preserving Folder Structure

dub.sdl

@@ -4,8 +4,8 @@ authors "Inochi2D Project"
 copyright "Copyright © 2020, Inochi2D Project"
 license "BSD 2-clause"
 dependency "psd-d" version="~>0.6.1"
-dependency "inmath" version="~>1.0.5"
-dependency "kra-d" version="~>0.5.5"
+dependency "inmath" version="~>1.0.6"
+dependency "kra-d" version="~>0.5.6"
 dependency "tinyfiledialogs" version="~>0.10.1"
 dependency "facetrack-d" version="~>0.7.6"
 dependency "bindbc-sdl" version="~>1.1.2"

source/creator/core/package.d

@@ -22,7 +22,6 @@ import std.exception;
 import bindbc.sdl;
 import bindbc.opengl;
 import inochi2d;
-import tinyfiledialogs;
 import std.string;
 import std.stdio;
 import std.conv;
@@ -629,6 +628,8 @@ void incBeginLoopNoEv() {
         incRenderDialogs();
     }
     incStatusUpdate();
+
+    incHandleDialogHandlers();
 }
 
 void incSetDefaultLayout() {

source/creator/core/settings.d

@@ -24,6 +24,11 @@ string incSettingsPath() {
 void incSettingsLoad() {
     if (exists(incSettingsPath())) {
         settings = parseJSON(readText(incSettingsPath()));
+
+        // File Handling
+        // Always ask the user whether to preserve the folder structure during import
+        // also see incGetKeepLayerFolder()
+        settings["KeepLayerFolder"] = "Ask";
     }
 }
 
@@ -91,4 +96,4 @@ T incSettingsGet(T)(string name, T default_) {
 */
 bool incSettingsCanGet(string name) {
     return (name in settings) !is null;
-}
+}

source/creator/io/kra.d

@@ -24,12 +24,7 @@ private {
 bool incImportShowKRADialog() {
     TFD_Filter[] filters = [{ ["*.kra"], "Krita Document (*.kra)" }];
     string file = incShowImportDialog(filters, _("Import..."));
-
-    if (file) {
-        incImportKRA(file, IncKRAImportSettings(false));
-        return true;
-    }
-    return false;
+    return incAskImportKRA(file);
 }
 
 class IncKRALayer {
@@ -142,8 +137,45 @@ struct IncKRAImportSettings {
     bool keepStructure = false;
 }
 
+/**
+    Imports a KRA file with user prompt.
+    also see incAskImportPSD()
+*/
+bool incAskImportKRA(string file) {
+    if (!file) return false;
+
+    KRALoadHandler handler = new KRALoadHandler(file);
+    return incKeepStructDialog(handler);
+}
+
+class KRALoadHandler : ImportKeepHandler {
+    private string file;
+
+    this(string file) {
+        super();
+        this.file = file;
+    }
+
+    override
+    bool load(AskKeepLayerFolder select) {
+        switch (select) {
+            case AskKeepLayerFolder.NotPreserve:
+                incImportKRA(file, IncKRAImportSettings(false));
+                return true;
+            case AskKeepLayerFolder.Preserve:
+                incImportKRA(file, IncKRAImportSettings(true));
+                return true;
+            case AskKeepLayerFolder.Cancel:
+                return false;
+            default:
+                throw new Exception("Invalid selection");
+        }
+    }
+}
+
 /**
     Imports a KRA file.
+    Note: You should invoke incAskImportKRA for UI interaction.
 */
 void incImportKRA(string file, IncKRAImportSettings settings = IncKRAImportSettings.init) {
     incNewProject();

source/creator/io/package.d

@@ -10,6 +10,8 @@ public import creator.io.kra;
 public import creator.io.inpexport;
 public import creator.io.videoexport;
 public import creator.io.imageexport;
+import creator.widgets: DialogButtons;
+import creator.widgets.dialog;
 
 import tinyfiledialogs;
 public import tinyfiledialogs : TFD_Filter;
@@ -73,6 +75,13 @@ private {
     }
 }
 
+string incToDString(c_str cstr1) {
+    if (cstr1 !is null) {
+        return cast(string) cstr1.fromStringz;
+    }
+    return null;
+}
+
 string incShowImportDialog(const(TFD_Filter)[] filters, string title, bool multiple = false) {
     version (linux) {
         try {
@@ -86,19 +95,11 @@ string incShowImportDialog(const(TFD_Filter)[] filters, string title, bool multi
 
             // FALLBACK: If xdg-desktop-portal is not available then try tinyfiledialogs.
             c_str filename = tinyfd_openFileDialog(title.toStringz, "", filters, multiple);
-            if (filename !is null) {
-                string file = cast(string) filename.fromStringz;
-                return file.dup;
-            }
-            return null;
+            return incToDString(filename).dup;
         }
     } else {
         c_str filename = tinyfd_openFileDialog(title.toStringz, "", filters, multiple);
-        if (filename !is null) {
-            string file = cast(string) filename.fromStringz;
-            return file.dup;
-        }
-        return null;
+        return incToDString(filename).dup;
     }
 }
 
@@ -114,15 +115,11 @@ string incShowOpenFolderDialog(string title = "Open...") {
 
             // FALLBACK: If xdg-desktop-portal is not available then try tinyfiledialogs.
             c_str filename = tinyfd_selectFolderDialog(title.toStringz, null);
-            if (filename !is null)
-                return cast(string) filename.fromStringz.dup;
-            return null;
+            return incToDString(filename).dup;
         }
     } else {
         c_str filename = tinyfd_selectFolderDialog(title.toStringz, null);
-        if (filename !is null)
-            return cast(string) filename.fromStringz.dup;
-        return null;
+        return incToDString(filename).dup;
     }
 }
 
@@ -138,19 +135,11 @@ string incShowOpenDialog(const(TFD_Filter)[] filters, string title = "Open...")
 
             // FALLBACK: If xdg-desktop-portal is not available then try tinyfiledialogs.
             c_str filename = tinyfd_openFileDialog(title.toStringz, "", filters, false);
-            if (filename !is null) {
-                string file = cast(string) filename.fromStringz;
-                return file.dup;
-            }
-            return null;
+            return incToDString(filename).dup;
         }
     } else {
         c_str filename = tinyfd_openFileDialog(title.toStringz, "", filters, false);
-        if (filename !is null) {
-            string file = cast(string) filename.fromStringz;
-            return file.dup;
-        }
-        return null;
+        return incToDString(filename).dup;
     }
 }
 
@@ -166,19 +155,11 @@ string incShowSaveDialog(const(TFD_Filter)[] filters, string fname, string title
 
             // FALLBACK: If xdg-desktop-portal is not available then try tinyfiledialogs.
             c_str filename = tinyfd_saveFileDialog(title.toStringz, fname.toStringz, filters);
-            if (filename !is null) {
-                string file = cast(string) filename.fromStringz;
-                return file.dup;
-            }
-            return null;
+            return incToDString(filename).dup;
         }
     } else {
         c_str filename = tinyfd_saveFileDialog(title.toStringz, fname.toStringz, filters);
-        if (filename !is null) {
-            string file = cast(string) filename.fromStringz;
-            return file.dup;
-        }
-        return null;
+        return incToDString(filename).dup;
     }
 }
 
@@ -221,3 +202,72 @@ void incCreatePartsFromFiles(string[] files) {
         }
     }
 }
+
+string incGetKeepLayerFolder() {
+    if (incSettingsCanGet("KeepLayerFolder"))
+        return incSettingsGet!string("KeepLayerFolder");
+    else
+        // also see incSettingsLoad()
+        // Preserve the original behavior for existing users
+        return "NotPreserve";
+}
+
+bool incSetKeepLayerFolder(string select) {
+    incSettingsSet("KeepLayerFolder", select);
+    return true;
+}
+
+enum AskKeepLayerFolder {
+    Preserve, NotPreserve, Cancel
+}
+
+const(char)* INC_KEEP_STRUCT_DIALOG_NAME = "ImportKeepFolderStructPopup";
+
+/**
+    Function for importing pop-up dialog
+*/
+bool incKeepStructDialog(ImportKeepHandler handler) {
+    if (incGetKeepLayerFolder() == "Preserve") {
+        handler.load(AskKeepLayerFolder.Preserve);
+    } else if (incGetKeepLayerFolder() == "NotPreserve") {
+        handler.load(AskKeepLayerFolder.NotPreserve);
+    } else {
+        incRegisterDialogHandler(handler);
+
+        // Show dialog
+        incDialog(
+            INC_KEEP_STRUCT_DIALOG_NAME,
+            __("Import File"),
+            _("Do you want to preserve the folder structure of the imported file? You can change this in the settings."),
+            DialogLevel.Warning,
+            DialogButtons.Yes | DialogButtons.No | DialogButtons.Cancel
+        );
+    }
+
+    return true;
+}
+
+class ImportKeepHandler : DialogHandler {
+    this () {
+        super(INC_KEEP_STRUCT_DIALOG_NAME);
+    }
+
+    override
+    bool onClick(DialogButtons button) {
+        switch (button) {
+            case DialogButtons.Cancel:
+                return this.load(AskKeepLayerFolder.Cancel);
+            case DialogButtons.Yes:
+                return this.load(AskKeepLayerFolder.Preserve);
+            case DialogButtons.No:
+                return this.load(AskKeepLayerFolder.NotPreserve);
+            default:
+                throw new Exception("Invalid button");
+        }
+    }
+
+    bool load(AskKeepLayerFolder select) {
+        // override this
+        return false;
+    }
+}

source/creator/io/psd.d

@@ -24,12 +24,7 @@ private {
 bool incImportShowPSDDialog() {
     TFD_Filter[] filters = [{ ["*.psd"], "Photoshop Document (*.psd)" }];
     string file = incShowImportDialog(filters, _("Import..."));
-
-    if (file) {
-        incImportPSD(file, IncPSDImportSettings(false));
-        return true;
-    }
-    return false;
+    return incAskImportPSD(file);
 }
 
 class IncPSDLayer {
@@ -142,8 +137,45 @@ struct IncPSDImportSettings {
     bool keepStructure = false;
 }
 
+/**
+    Imports a PSD file with user prompt.
+    also see incAskImportKRA()
+*/
+bool incAskImportPSD(string file) {
+    if (!file) return false;
+
+    PSDLoadHandler handler = new PSDLoadHandler(file);
+    return incKeepStructDialog(handler);
+}
+
+class PSDLoadHandler : ImportKeepHandler {
+    private string file;
+
+    this(string file) {
+        super();
+        this.file = file;
+    }
+
+    override
+    bool load(AskKeepLayerFolder select) {
+        switch (select) {
+            case AskKeepLayerFolder.NotPreserve:
+                incImportPSD(file, IncPSDImportSettings(false));
+                return true;
+            case AskKeepLayerFolder.Preserve:
+                incImportPSD(file, IncPSDImportSettings(true));
+                return true;
+            case AskKeepLayerFolder.Cancel:
+                return false;
+            default:
+                throw new Exception("Invalid selection");
+        }
+    }
+}
+
 /**
     Imports a PSD file.
+    Note: You should invoke incAskImportPSD for UI interaction.
 */
 void incImportPSD(string file, IncPSDImportSettings settings = IncPSDImportSettings.init) {
     incNewProject();

source/creator/panels/viewport.d

@@ -241,7 +241,12 @@ protected:
 
                     // Allow dragging PSD in to main window
                     case ".psd":
-                        incImportPSD(file);
+                        incAskImportPSD(file);
+                        break mainLoop;
+
+                    // Allow dragging KRA in to main window
+                    case ".kra":
+                        incAskImportKRA(file);
                         break mainLoop;
 
                     default:

source/creator/viewport/vertex/package.d

@@ -179,6 +179,7 @@ void incViewportVertexConfirmBar() {
         }
 
         // In case of a warning popup preventing application.
+        // TODO: if incDialogButtonSelected does not work, we may implement a DialogHandler for this.
         if (incDialogButtonSelected("CONFIRM_VERTEX_APPLY") == DialogButtons.Yes) {
             incMeshEditApply();
         }