crypto/helper: allow overriding post-decrypt function

mautrix · Sep 12, 2024 · e16b681 · e16b681
1 parent d472be3
commit e16b681
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/crypto/cryptohelper/cryptohelper.go b/crypto/cryptohelper/cryptohelper.go
@@ -38,7 +38,8 @@ type CryptoHelper struct {
 
 	LoginAs *mautrix.ReqLogin
 
-	ASEventProcessor crypto.ASEventProcessor
+	ASEventProcessor  crypto.ASEventProcessor
+	CustomPostDecrypt func(context.Context, *event.Event)
 
 	DBAccountID string
 }
@@ -320,7 +321,9 @@ func (helper *CryptoHelper) HandleEncrypted(ctx context.Context, evt *event.Even
 
 func (helper *CryptoHelper) postDecrypt(ctx context.Context, decrypted *event.Event) {
 	decrypted.Mautrix.EventSource |= event.SourceDecrypted
-	if helper.ASEventProcessor != nil {
+	if helper.CustomPostDecrypt != nil {
+		helper.CustomPostDecrypt(ctx, decrypted)
+	} else if helper.ASEventProcessor != nil {
 		helper.ASEventProcessor.Dispatch(ctx, decrypted)
 	} else {
 		helper.client.Syncer.(mautrix.DispatchableSyncer).Dispatch(ctx, decrypted)