Discord: fix for new auth dialog

2 files changed, 20 insertions, 0 deletions

diff --git a/protocols/Discord/src/proto.cpp b/protocols/Discord/src/proto.cpp
index e5a62df682..d0f982fbd7 100644
--- a/protocols/Discord/src/proto.cpp
+++ b/protocols/Discord/src/proto.cpp
@@ -436,6 +436,25 @@ MCONTACT CDiscordProto::AddToList(int flags, PROTOSEARCHRESULT *psr)
 	return hContact;
 }
 
+MCONTACT CDiscordProto::AddToListByEvent(int flags, int, MEVENT hDbEvent)
+{
+	DB::EventInfo dbei;
+	dbei.cbBlob = -1;
+	if (db_event_get(hDbEvent, &dbei))
+		return 0;
+	if (mir_strcmp(dbei.szModule, m_szModuleName))
+		return 0;
+	if (dbei.eventType != EVENTTYPE_AUTHREQUEST)
+		return 0;
+
+	DB::AUTH_BLOB blob(dbei.pBlob);
+	if (flags & PALF_TEMPORARY)
+		Contact_RemoveFromList(blob.get_contact());
+	else
+		Contact_PutOnList(blob.get_contact());
+	return blob.get_contact();
+}
+
 ////////////////////////////////////////////////////////////////////////////////////////
 // SendMsg
 
diff --git a/protocols/Discord/src/proto.h b/protocols/Discord/src/proto.h
index c90aafba6a..8a4301fcad 100644
--- a/protocols/Discord/src/proto.h
+++ b/protocols/Discord/src/proto.h
@@ -356,6 +356,7 @@ public:
 
 	HANDLE   SearchBasic(const wchar_t *id) override;
 	MCONTACT AddToList(int flags, PROTOSEARCHRESULT *psr) override;
+	MCONTACT AddToListByEvent(int flags, int, MEVENT hDbEvent) override;
 	
 	int      AuthRecv(MCONTACT, PROTORECVEVENT *pre) override;
 	int      Authorize(MEVENT hDbEvent) override;