From 8fc5bfb7990c8b49dff6a256c03b94e12441e3fe Mon Sep 17 00:00:00 2001
From: wijnand <wikoreman@gmail.com>
Date: Fri, 23 Mar 2018 15:12:44 +0100
Subject: [PATCH] Introduce ValidateNodeIdsInDatabase

---
 .../publish/validate_node_ids_in_database.py  | 55 +++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 colorbleed/plugins/maya/publish/validate_node_ids_in_database.py

diff --git a/colorbleed/plugins/maya/publish/validate_node_ids_in_database.py b/colorbleed/plugins/maya/publish/validate_node_ids_in_database.py
new file mode 100644
index 0000000000..855e6aa4a1
--- /dev/null
+++ b/colorbleed/plugins/maya/publish/validate_node_ids_in_database.py
@@ -0,0 +1,55 @@
+import pyblish.api
+
+import avalon.io as io
+
+import colorbleed.api
+from colorbleed.maya import lib
+
+
+class ValidateNodeIdsInDatabase(pyblish.api.InstancePlugin):
+    """Validate if the CB Id is related to an asset in the database
+
+    All nodes with the `cbId` attribute will be validated to ensure that
+    the loaded asset in the scene is related to the current project.
+
+    Tip: If there is an asset which is being reused from a different project
+    please ensure the asset is republished in the new project
+
+    """
+
+    order = colorbleed.api.ValidatePipelineOrder
+    label = 'Node Ids in Database'
+    hosts = ['maya']
+    families = ["*"]
+
+    actions = [colorbleed.api.SelectInvalidAction]
+
+    def process(self, instance):
+        invalid = self.get_invalid(instance)
+        if invalid:
+            raise RuntimeError("Found asset IDs which are not related to"
+                               "current project in instance: "
+                               "`%s`" % instance.name)
+
+    @classmethod
+    def get_invalid(cls, instance):
+
+        invalid = []
+
+        # Get all id required nodes
+        id_required_nodes = lib.get_id_required_nodes(referenced_nodes=True,
+                                                      nodes=instance[:])
+
+        # check ids against database ids
+        db_asset_ids = io.find({"type": "asset"}).distinct("_id")
+        db_asset_ids = set(str(i) for i in db_asset_ids)
+
+        # Get all asset IDs
+        for node in id_required_nodes:
+            cb_id = lib.get_id(node)
+            asset_id = cb_id.split(":", 1)[0]
+            if asset_id not in db_asset_ids:
+                cls.log.error("`%s` has unassociated assetfd ID" % node)
+                invalid.append(node)
+
+        return invalid