From 8c67b96a0f3672edaea4e283b62f1caa777015e8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=A0imon=20Brandner?= <simon.bra.ag@gmail.com>
Date: Tue, 1 Jun 2021 10:42:21 +0200
Subject: [PATCH] Save all events
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Šimon Brandner <simon.bra.ag@gmail.com>
---
 src/components/structures/CallEventGrouper.ts | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/components/structures/CallEventGrouper.ts b/src/components/structures/CallEventGrouper.ts
index c654c08636..84f178b75f 100644
--- a/src/components/structures/CallEventGrouper.ts
+++ b/src/components/structures/CallEventGrouper.ts
@@ -33,10 +33,14 @@ const SUPPORTED_STATES = [
 ];
 
 export default class CallEventGrouper extends EventEmitter {
-    invite: MatrixEvent;
+    events: Array<MatrixEvent> = [];
     call: MatrixCall;
     state: CallState;
 
+    private get invite(): MatrixEvent {
+        return this.events.find((event) => event.getType() === EventType.CallInvite);
+    }
+
     public answerCall = () => {
         this.call?.answer();
     }
@@ -80,10 +84,11 @@ export default class CallEventGrouper extends EventEmitter {
     }
 
     public add(event: MatrixEvent) {
-        if (event.getType() === EventType.CallInvite) this.invite = event;
-        if (event.getType() === EventType.CallHangup) this.state = CallState.Ended;
+        this.events.push(event);
+        const type = event.getType();
 
-        if (this.call) return;
+        if (type === EventType.CallHangup) this.state = CallState.Ended;
+        else if (type === EventType.CallReject) this.state = CallState.Ended;
         const callId = event.getContent().call_id;
         this.call = CallHandler.sharedInstance().getCallById(callId);
         if (!this.call) return;