From 15f6cc70b1979882a0ccb0b68db4c0c32e534a86 Mon Sep 17 00:00:00 2001
From: Paul Schaub <vanitasvitae@fsfe.org>
Date: Wed, 3 May 2023 14:30:08 +0200
Subject: [PATCH] Add MessageMetadata.getRecipientKeyIds()

Fixes #376
---
 .../decryption_verification/MessageMetadata.java  | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/pgpainless-core/src/main/java/org/pgpainless/decryption_verification/MessageMetadata.java b/pgpainless-core/src/main/java/org/pgpainless/decryption_verification/MessageMetadata.java
index 648b99ab..cc97e81a 100644
--- a/pgpainless-core/src/main/java/org/pgpainless/decryption_verification/MessageMetadata.java
+++ b/pgpainless-core/src/main/java/org/pgpainless/decryption_verification/MessageMetadata.java
@@ -92,6 +92,21 @@ public class MessageMetadata {
         return false;
     }
 
+    /**
+     * Return a list containing all recipient keyIDs.
+     *
+     * @return list of recipients
+     */
+    public List<Long> getRecipientKeyIds() {
+        List<Long> keyIds = new ArrayList<>();
+        Iterator<EncryptedData> encLayers = getEncryptionLayers();
+        while (encLayers.hasNext()) {
+            EncryptedData layer = encLayers.next();
+            keyIds.addAll(layer.getRecipients());
+        }
+        return keyIds;
+    }
+
     public @Nonnull Iterator<EncryptedData> getEncryptionLayers() {
         return new LayerIterator<EncryptedData>(message) {
             @Override