From 7b0f359a7f4776cdf4dcb017e06c8dcee6a3cbd4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maciej=20Krzy=C5=BCanowski?= <krzmaciek@gmail.com>
Date: Mon, 3 Jun 2024 20:40:20 +0000
Subject: [PATCH] Remove initiation after sending ChatFinishRequest

---
 server/server.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/server/server.go b/server/server.go
index 4dc6cf6..8565f7a 100644
--- a/server/server.go
+++ b/server/server.go
@@ -559,6 +559,7 @@ func testEcho(hdlCtx *HandlerContext) {
 
 func (ctx *Context) wsapiHandler(w http.ResponseWriter, r *http.Request) {
 	upgrader := websocket.Upgrader{}
+
 	conn, err := upgrader.Upgrade(w, r, nil)
 
 	if err != nil {
@@ -675,6 +676,12 @@ func (srvCtx *Context) handleUDP(data []byte, addr net.Addr) {
 		return
 	}
 
+	srvCtx.initiations = slices.DeleteFunc(srvCtx.initiations,
+		func(i *common.Initiation) bool {
+			return i.AbAPunchCode == matchedInitation.AbAPunchCode ||
+				i.AbBPunchCode == matchedInitation.AbAPunchCode
+		})
+
 	err = abA.sendRequest(common.StartChatFinishRequest{
 		OtherSideNickname: matchedInitation.AbBNick,
 		OtherSideAddress:  matchedInitation.AbBAddress,