From 15c6370153d5f283783d03a4af178ddcff5fa97a Mon Sep 17 00:00:00 2001
From: Alex Kaminskii <alex@openreplay.com>
Date: Mon, 29 Aug 2022 10:19:00 +0200
Subject: [PATCH] fix(backend): fix rewrite-message filter

---
 backend/cmd/sink/main.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/backend/cmd/sink/main.go b/backend/cmd/sink/main.go
index e5a0afa78..40a295c12 100644
--- a/backend/cmd/sink/main.go
+++ b/backend/cmd/sink/main.go
@@ -71,8 +71,12 @@ func main() {
 
 				msg := iter.Message()
 				// Process assets
-				if iter.Type() == MsgSetNodeAttributeURLBased || iter.Type() == MsgSetCSSDataURLBased || iter.Type() == MsgCSSInsertRuleURLBased {
-					msg = assetMessageHandler.ParseAssets(sessionID, msg.Decode())
+				if iter.Type() == MsgSetNodeAttributeURLBased ||
+					iter.Type() == MsgSetCSSDataURLBased ||
+					iter.Type() == MsgCSSInsertRuleURLBased ||
+					iter.Type() == AdoptedSSReplaceURLBased ||
+					iter.Type() == AdoptedSSInsertRuleURLBased {
+					msg = assetMessageHandler.ParseAssets(sessionID, msg.Decode()) // TODO: filter type only once (use iterator inide or bring ParseAssets out here).
 				}
 
 				// Filter message