From 52fd675bbd7d9ce70df80ef38da4ee3fa05dd84d Mon Sep 17 00:00:00 2001
From: Jack Grigg <thestr4d@gmail.com>
Date: Sat, 24 Aug 2024 02:12:31 +0000
Subject: [PATCH] age: Add `IdentityFile::to_recipients`

---
 age/CHANGELOG.md    |  1 +
 age/src/identity.rs | 27 +++++++++++++++++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/age/CHANGELOG.md b/age/CHANGELOG.md
index 9186e0f..42ea7c7 100644
--- a/age/CHANGELOG.md
+++ b/age/CHANGELOG.md
@@ -11,6 +11,7 @@ to 1.0.0 are beta releases.
 ## [Unreleased]
 ### Added
 - `age::Decryptor::{decrypt, decrypt_async, is_scrypt}`
+- `age::IdentityFile::to_recipients`
 - `age::scrypt`, providing recipient and identity types for passphrase-based
   encryption.
 - Partial French translation!
diff --git a/age/src/identity.rs b/age/src/identity.rs
index 300a29f..ce166d2 100644
--- a/age/src/identity.rs
+++ b/age/src/identity.rs
@@ -134,6 +134,23 @@ impl IdentityFile {
         Ok(IdentityFile { identities })
     }
 
+    /// Returns recipients for the identities in this file.
+    ///
+    /// Plugin identities will be merged into one [`Recipient`] per unique plugin.
+    ///
+    /// [`Recipient`]: crate::Recipient
+    pub fn to_recipients(
+        &self,
+        callbacks: impl Callbacks,
+    ) -> Result<Vec<Box<dyn crate::Recipient + Send>>, EncryptError> {
+        let mut recipients = RecipientsAccumulator::new();
+        recipients.with_identities_ref(self);
+        recipients.build(
+            #[cfg(feature = "plugin")]
+            callbacks,
+        )
+    }
+
     /// Returns the identities in this file.
     pub fn into_identities(self) -> Vec<IdentityFileEntry> {
         self.identities
@@ -188,6 +205,16 @@ impl RecipientsAccumulator {
         }
     }
 
+    pub(crate) fn with_identities_ref(&mut self, identity_file: &IdentityFile) {
+        for entry in &identity_file.identities {
+            match entry {
+                IdentityFileEntry::Native(i) => self.recipients.push(Box::new(i.to_public())),
+                #[cfg(feature = "plugin")]
+                IdentityFileEntry::Plugin(i) => self.plugin_identities.push(i.clone()),
+            }
+        }
+    }
+
     #[cfg_attr(not(feature = "plugin"), allow(unused_mut))]
     pub(crate) fn build(
         mut self,