From c1268ec253a1c9b4ee7e3d8dc5acb76712a8a035 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Thu, 25 Aug 2022 15:55:09 +0200
Subject: [PATCH] implemented hamrony addon

---
 openpype/hosts/harmony/__init__.py | 17 ++++++++---------
 openpype/hosts/harmony/addon.py    | 24 ++++++++++++++++++++++++
 2 files changed, 32 insertions(+), 9 deletions(-)
 create mode 100644 openpype/hosts/harmony/addon.py

diff --git a/openpype/hosts/harmony/__init__.py b/openpype/hosts/harmony/__init__.py
index d2f710d83d..9177eaa285 100644
--- a/openpype/hosts/harmony/__init__.py
+++ b/openpype/hosts/harmony/__init__.py
@@ -1,11 +1,10 @@
-import os
+from .addon import (
+    HARMONY_HOST_DIR,
+    HarmonyAddon,
+)
 
 
-def add_implementation_envs(env, _app):
-    """Modify environments to contain all required for implementation."""
-    openharmony_path = os.path.join(
-        os.environ["OPENPYPE_REPOS_ROOT"], "openpype", "hosts",
-        "harmony", "vendor", "OpenHarmony"
-    )
-    # TODO check if is already set? What to do if is already set?
-    env["LIB_OPENHARMONY_PATH"] = openharmony_path
+__all__ = (
+    "HARMONY_HOST_DIR",
+    "HarmonyAddon",
+)
diff --git a/openpype/hosts/harmony/addon.py b/openpype/hosts/harmony/addon.py
new file mode 100644
index 0000000000..b051d68abb
--- /dev/null
+++ b/openpype/hosts/harmony/addon.py
@@ -0,0 +1,24 @@
+import os
+from openpype.modules import OpenPypeModule
+from openpype.modules.interfaces import IHostModule
+
+HARMONY_HOST_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+class HarmonyAddon(OpenPypeModule, IHostModule):
+    name = "harmony"
+    host_name = "harmony"
+
+    def initialize(self, module_settings):
+        self.enabled = True
+
+    def add_implementation_envs(self, env, _app):
+        """Modify environments to contain all required for implementation."""
+        openharmony_path = os.path.join(
+            HARMONY_HOST_DIR, "vendor", "OpenHarmony"
+        )
+        # TODO check if is already set? What to do if is already set?
+        env["LIB_OPENHARMONY_PATH"] = openharmony_path
+
+    def get_workfile_extensions(self):
+        return [".zip"]