added extended attribute getter for notation-go (#46)

* added extended attribute getter for notation-go Signed-off-by: Patrick Zheng <[email protected]> * update Signed-off-by: Patrick Zheng <[email protected]> Signed-off-by: Patrick Zheng <[email protected]>
notaryproject · Aug 22, 2022 · 9757c1e · 9757c1e
1 parent 8b03c75
commit 9757c1e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/signature/types.go b/signature/types.go
@@ -2,6 +2,7 @@ package signature
 
 import (
 	"crypto/x509"
+	"errors"
 	"time"
 )
 
@@ -119,3 +120,14 @@ type Payload struct {
 	// Content contains the raw bytes of the payload.
 	Content []byte
 }
+
+// ExtendedAttribute fetches the specified Attribute with provided key from
+// signerInfo.SignedAttributes.ExtendedAttributes
+func (signerInfo *SignerInfo) ExtendedAttribute(key string) (Attribute, error) {
+	for _, attr := range signerInfo.SignedAttributes.ExtendedAttributes {
+		if attr.Key == key {
+			return attr, nil
+		}
+	}
+	return Attribute{}, errors.New("key not in ExtendedAttributes")
+}