From c31cdf94d18da2946d2c620460a7c5980527667b Mon Sep 17 00:00:00 2001
From: Ondrej Samohel <ondrej.samohel@orbi.tools>
Date: Tue, 27 Jul 2021 10:30:26 +0200
Subject: [PATCH] initial support for configurable dirmap

---
 openpype/hosts/maya/api/__init__.py           | 18 ++++++++++
 .../defaults/project_settings/maya.json       | 13 ++++++++
 .../defaults/project_settings/unreal.json     |  3 +-
 .../projects_schema/schema_project_maya.json  | 33 +++++++++++++++++++
 4 files changed, 65 insertions(+), 2 deletions(-)

diff --git a/openpype/hosts/maya/api/__init__.py b/openpype/hosts/maya/api/__init__.py
index 4697d212de..027fa871e8 100644
--- a/openpype/hosts/maya/api/__init__.py
+++ b/openpype/hosts/maya/api/__init__.py
@@ -26,6 +26,24 @@ INVENTORY_PATH = os.path.join(PLUGINS_DIR, "inventory")
 
 
 def install():
+    from openpype.settings import get_project_settings
+
+    project_settings = get_project_settings(os.getenv("AVALON_PROJECT"))
+    mapping = project_settings["maya"]["maya-dirmap"]["paths"] or {}
+    if mapping.get("source-path") and project_settings["maya"]["maya-dirmap"]["enabled"] is True:
+        log.info("Processing directory mapping ...")
+        cmds.dirmap(en=True)
+    for k, sp in enumerate(mapping["source-path"]):
+        try:
+            print("{} -> {}".format(sp, mapping["destination-path"][k]))
+            cmds.dirmap(m=[sp, mapping["destination-path"][k]])
+            cmds.dirmap(m=[mapping["destination-path"][k], sp])
+        except IndexError:
+            # missing corresponding destination path
+            log.error(("invalid dirmap mapping, missing corresponding"
+                       " destination directory."))
+            break
+
     pyblish.register_plugin_path(PUBLISH_PATH)
     avalon.register_plugin_path(avalon.Loader, LOAD_PATH)
     avalon.register_plugin_path(avalon.Creator, CREATE_PATH)
diff --git a/openpype/settings/defaults/project_settings/maya.json b/openpype/settings/defaults/project_settings/maya.json
index 284a1a0040..b92dc52b92 100644
--- a/openpype/settings/defaults/project_settings/maya.json
+++ b/openpype/settings/defaults/project_settings/maya.json
@@ -7,6 +7,19 @@
         "workfile": "ma",
         "yetiRig": "ma"
     },
+    "maya-dirmap": {
+        "enabled": true,
+        "paths": {
+            "source-path": [
+                "foo1",
+                "foo2"
+            ],
+            "destination-path": [
+                "bar1",
+                "bar2"
+            ]
+        }
+    },
     "create": {
         "CreateLook": {
             "enabled": true,
diff --git a/openpype/settings/defaults/project_settings/unreal.json b/openpype/settings/defaults/project_settings/unreal.json
index 46b9ca2a18..dad61cd1f0 100644
--- a/openpype/settings/defaults/project_settings/unreal.json
+++ b/openpype/settings/defaults/project_settings/unreal.json
@@ -1,6 +1,5 @@
 {
     "project_setup": {
-        "dev_mode": true,
-        "install_unreal_python_engine": false
+        "dev_mode": true
     }
 }
\ No newline at end of file
diff --git a/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json b/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json
index 0a59cab510..e70c0da708 100644
--- a/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json
+++ b/openpype/settings/entities/schemas/projects_schema/schema_project_maya.json
@@ -14,6 +14,39 @@
                                 "type": "text"
                             }
         },
+        {
+           "type": "dict",
+            "collapsible": true,
+            "checkbox_key": "enabled",
+            "key": "maya-dirmap",
+            "label": "Maya Directory Mapping",
+            "is_group": true,
+            "children": [
+                {
+                    "type": "boolean",
+                    "key": "enabled",
+                    "label": "Enabled"
+                },
+                {
+                    "type": "dict",
+                    "key": "paths",
+                    "children": [
+                        {
+                            "type": "list",
+                            "object_type": "text",
+                            "key": "source-path",
+                            "label": "Source Path"
+                        },
+                        {
+                            "type": "list",
+                            "object_type": "text",
+                            "key": "destination-path",
+                            "label": "Destination Path"
+                        }
+                    ]
+                }
+            ]
+        },
         {
             "type": "schema",
             "name": "schema_maya_create"