include webext-bridge modified source

1 files changed, 28 insertions, 0 deletions

diff --git a/extension/src/webext-bridge/internal/delivery-logger.ts b/extension/src/webext-bridge/internal/delivery-logger.ts
new file mode 100644
index 0000000..395f035
--- /dev/null
+++ b/extension/src/webext-bridge/internal/delivery-logger.ts
@@ -0,0 +1,28 @@
+import type { InternalMessage } from '../types'
+import type { EndpointFingerprint } from './endpoint-fingerprint'
+
+export interface DeliveryReceipt {
+  message: InternalMessage
+  to: EndpointFingerprint
+  from: {
+    endpointId: string
+    fingerprint: EndpointFingerprint
+  }
+}
+
+export const createDeliveryLogger = () => {
+  let logs: ReadonlyArray<DeliveryReceipt> = []
+
+  return {
+    add: (...receipts: DeliveryReceipt[]) => {
+      logs = [...logs, ...receipts]
+    },
+    remove: (message: string | DeliveryReceipt[]) => {
+      logs
+        = typeof message === 'string'
+          ? logs.filter(receipt => receipt.message.transactionId !== message)
+          : logs.filter(receipt => !message.includes(receipt))
+    },
+    entries: () => logs,
+  }
+}