From 104e03b13d2185b97ac926992abb4f13ac9756b8 Mon Sep 17 00:00:00 2001
From: Roy Nieterau <roy_nieterau@hotmail.com>
Date: Fri, 5 Jul 2024 15:59:55 +0200
Subject: [PATCH] Add toggle to settings

---
 server/settings/publish_plugins.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/server/settings/publish_plugins.py b/server/settings/publish_plugins.py
index 36bb3f7340..930fa9d0a3 100644
--- a/server/settings/publish_plugins.py
+++ b/server/settings/publish_plugins.py
@@ -84,6 +84,17 @@ class CollectUSDLayerContributionsModel(BaseSettingsModel):
         return value
 
 
+class AyonEntityURIModel(BaseSettingsModel):
+    use_ayon_entity_uri: bool = SettingsField(
+        title="Use AYON Entity URI",
+        description=(
+            "When enabled the USD paths written using the contribution "
+            "workflow will use ayon entity URIs instead of resolved published "
+            "paths. You can only load these if you use the AYON USD Resolver."
+        )
+    )
+
+
 class PluginStateByHostModelProfile(BaseSettingsModel):
     _layout = "expanded"
     # Filtering
@@ -857,6 +868,14 @@ class PublishPuginsModel(BaseSettingsModel):
         default_factory=ExtractBurninModel,
         title="Extract Burnin"
     )
+    ExtractUSDAssetContribution: AyonEntityURIModel = SettingsField(
+        default_factory=AyonEntityURIModel,
+        title="Extract USD Asset Contribution",
+    )
+    ExtractUSDLayerContribution: AyonEntityURIModel = SettingsField(
+        default_factory=AyonEntityURIModel,
+        title="Extract USD Layer Contribution",
+    )
     PreIntegrateThumbnails: PreIntegrateThumbnailsModel = SettingsField(
         default_factory=PreIntegrateThumbnailsModel,
         title="Override Integrate Thumbnail Representations"
@@ -1161,6 +1180,12 @@ DEFAULT_PUBLISH_VALUES = {
             }
         ]
     },
+    "ExtractUSDAssetContribution": {
+        "use_ayon_entity_uri": True,
+    },
+    "ExtractUSDLayerContribution": {
+        "use_ayon_entity_uri": True,
+    },
     "PreIntegrateThumbnails": {
         "enabled": True,
         "integrate_profiles": []