From b833ed799237fee9b6ba0b74675b5d340025279b Mon Sep 17 00:00:00 2001
From: MHSanaei <ho3ein.sanaei@gmail.com>
Date: Sat, 17 Jun 2023 19:11:16 +0330
Subject: new - tg Login Notification #584

---
 web/service/setting.go | 5 +++++
 web/service/tgbot.go   | 5 +++++
 2 files changed, 10 insertions(+)

(limited to 'web/service')

diff --git a/web/service/setting.go b/web/service/setting.go
index 677ccbb2..9bf74641 100644
--- a/web/service/setting.go
+++ b/web/service/setting.go
@@ -39,6 +39,7 @@ var defaultValueMap = map[string]string{
 	"tgBotChatId":        "",
 	"tgRunTime":          "@daily",
 	"tgBotBackup":        "false",
+	"tgBotLoginNotify":   "false",
 	"tgCpu":              "0",
 	"tgLang":             "en-US",
 	"secretEnable":       "false",
@@ -266,6 +267,10 @@ func (s *SettingService) GetTgBotBackup() (bool, error) {
 	return s.getBool("tgBotBackup")
 }
 
+func (s *SettingService) GetTgBotLoginNotify() (bool, error) {
+	return s.getBool("tgBotLoginNotify")
+}
+
 func (s *SettingService) GetTgCpu() (int, error) {
 	return s.getInt("tgCpu")
 }
diff --git a/web/service/tgbot.go b/web/service/tgbot.go
index 751de41b..39e7771a 100644
--- a/web/service/tgbot.go
+++ b/web/service/tgbot.go
@@ -643,6 +643,11 @@ func (t *Tgbot) UserLoginNotify(username string, ip string, time string, status
 		return
 	}
 
+	loginNotifyEnabled, err := t.settingService.GetTgBotLoginNotify()
+	if err != nil || !loginNotifyEnabled {
+		return
+	}
+
 	msg := ""
 	if status == LoginSuccess {
 		msg += t.I18nBot("tgbot.messages.loginSuccess")
-- 
cgit v1.2.3