From cf00c0c57729aca8328a9216a37fea718eb4fcff Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Mon, 30 Nov 2020 10:06:56 +0100
Subject: [PATCH] added resolve prelaunch hook

---
 .../celaction/pre_celaction_registers.py      |  3 +-
 pype/hooks/resolve/pre_resolve_setup.py       | 58 +++++++++++++++++++
 2 files changed, 59 insertions(+), 2 deletions(-)
 create mode 100644 pype/hooks/resolve/pre_resolve_setup.py

diff --git a/pype/hooks/celaction/pre_celaction_registers.py b/pype/hooks/celaction/pre_celaction_registers.py
index b65deda7f5..f592026699 100644
--- a/pype/hooks/celaction/pre_celaction_registers.py
+++ b/pype/hooks/celaction/pre_celaction_registers.py
@@ -41,8 +41,7 @@ class CelactionPrelaunchHook(PreLaunchHook):
 
         # TODO: change to root path and pyblish standalone to premiere way
         pype_root_path = os.getenv("PYPE_SETUP_PATH")
-        path = os.path.join(pype_root_path,
-                            "pype.bat")
+        path = os.path.join(pype_root_path, "pype.bat")
 
         winreg.SetValueEx(hKey, "SubmitAppTitle", 0, winreg.REG_SZ, path)
 
diff --git a/pype/hooks/resolve/pre_resolve_setup.py b/pype/hooks/resolve/pre_resolve_setup.py
new file mode 100644
index 0000000000..336fabf0c4
--- /dev/null
+++ b/pype/hooks/resolve/pre_resolve_setup.py
@@ -0,0 +1,58 @@
+import os
+import importlib
+from pype.lib import PreLaunchHook
+from pype.hosts.resolve import utils
+
+
+class ResolvePrelaunch(PreLaunchHook):
+    """
+    This hook will check if current workfile path has Resolve
+    project inside. IF not, it initialize it and finally it pass
+    path to the project by environment variable to Premiere launcher
+    shell script.
+    """
+    hosts = ["resolve"]
+
+    def execute(self):
+        # making sure pyton 3.6 is installed at provided path
+        py36_dir = os.path.normpath(self.env.get("PYTHON36_RESOLVE", ""))
+        assert os.path.isdir(py36_dir), (
+            "Python 3.6 is not installed at the provided folder path. Either "
+            "make sure the `environments\resolve.json` is having correctly "
+            "set `PYTHON36_RESOLVE` or make sure Python 3.6 is installed "
+            f"in given path. \nPYTHON36_RESOLVE: `{py36_dir}`"
+        )
+        self.log.info(f"Path to Resolve Python folder: `{py36_dir}`...")
+        self.env["PYTHON36_RESOLVE"] = py36_dir
+
+        # setting utility scripts dir for scripts syncing
+        us_dir = os.path.normpath(
+            self.env.get("RESOLVE_UTILITY_SCRIPTS_DIR", "")
+        )
+        assert os.path.isdir(us_dir), (
+            "Resolve utility script dir does not exists. Either make sure "
+            "the `environments\resolve.json` is having correctly set "
+            "`RESOLVE_UTILITY_SCRIPTS_DIR` or reinstall DaVinci Resolve. \n"
+            f"RESOLVE_UTILITY_SCRIPTS_DIR: `{us_dir}`"
+        )
+        self.log.debug(f"-- us_dir: `{us_dir}`")
+
+        # correctly format path for pre python script
+        pre_py_sc = os.path.normpath(self.env.get("PRE_PYTHON_SCRIPT", ""))
+        self.env["PRE_PYTHON_SCRIPT"] = pre_py_sc
+        self.log.debug(f"-- pre_py_sc: `{pre_py_sc}`...")
+        try:
+            __import__("pype.hosts.resolve")
+            __import__("pyblish")
+
+        except ImportError:
+            self.log.warning(
+                "pyblish: Could not load Resolve integration.",
+                exc_info=True
+            )
+
+        else:
+            # Resolve Setup integration
+            importlib.reload(utils)
+            self.log.debug(f"-- utils.__file__: `{utils.__file__}`")
+            utils.setup(self.env)