link entities

rruckley · Oct 12, 2024 · 3dc95e9 · 3dc95e9
1 parent a4c49eb
commit 3dc95e9
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/tmf667/document.rs b/src/tmf667/document.rs
@@ -5,6 +5,8 @@ use crate::{
 };
 use tmflib_derive::{HasId,HasName,HasLastUpdate,HasRelatedParty,HasDescription};
 use crate::common::related_party::RelatedParty;
+use crate::common::related_entity::RelatedEntity;
+use crate::vec_insert;
 use serde::{Deserialize,Serialize};
 
 const CLASS_PATH : &str = "document";
@@ -56,7 +58,10 @@ pub struct Document {
     #[serde(skip_serializing_if = "Option::is_none")]
     creation_date: Option<DateTime>,
     // Referenced objects
+    /// Parties
     related_party: Option<Vec<RelatedParty>>,
+    /// Related Entities
+    related_entity : Option<Vec<RelatedEntity>>,
     /// Attachement
     attachment : AttachmentRefOrValue,
 }
@@ -88,6 +93,17 @@ impl Document {
         self.document_type = Some(r#type.into());
         self
     }
+
+    /// Link another TMF entity during creation
+    pub fn link<T : HasName>(mut self, entity : T) -> Document {
+        self.link_entity(entity);
+        self
+    }
+
+    /// Link another TMF entity into this document
+    pub fn link_entity<T : HasName>(&mut self, entity : T) {
+        vec_insert(&mut self.related_entity,RelatedEntity::from(entity));
+    }
 }
 
 impl From<AttachmentRefOrValue> for Document {