From aadc6e8a03cffbc52414d08aeb780633a66dffb9 Mon Sep 17 00:00:00 2001
From: Nefrace <nefrace@gmail.com>
Date: Mon, 8 Jul 2024 20:18:07 +0300
Subject: [PATCH] Fixed registration handler

---
 src/db.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/db.go b/src/db.go
index a1b8176..6767ebf 100644
--- a/src/db.go
+++ b/src/db.go
@@ -126,7 +126,12 @@ func NewActivation(code string) error {
 }
 
 func UseActivation(code string) bool {
-	_, err := db.Exec(`delete from activations where code = $1`, code)
+	exists := false
+	err := db.Get(&exists, `SELECT exists(SELECT 1 FROM chats WHERE code = $1);`, code)
+	if err != nil || !exists {
+		return false
+	}
+	_, err = db.Exec(`delete from activations where code = $1`, code)
 	return err == nil
 }