add AddUserInChat

This commit is contained in:
Владимир Фёдоров 2022-04-25 02:54:43 +07:00
parent 6ccd5b478e
commit 3005afda25
2 changed files with 25 additions and 0 deletions

View File

@ -39,3 +39,25 @@ func AddChat(chatID int64) error {
}
return result.Err()
}
func AddUserInChat(chatID int64, username string) error {
if err := AddChat(chatID); err != nil {
return err
}
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
client, err := mongo.Connect(ctx, options.Client().ApplyURI(cfg.MongoURL))
if err != nil {
return err
}
collection := client.Database(cfg.DBName).Collection(cfg.ChatsCollectionName)
_, err = collection.UpdateOne(
ctx,
bson.E{Key: "chat_id", Value: chatID},
bson.M{"addToSet": bson.M{"users": username}},
)
return err
}

View File

@ -80,6 +80,9 @@ func run() {
text := update.Message.Text
chatID := update.Message.Chat.ID
username := update.Message.From.UserName
_ = db.AddUserInChat(chatID, username)
if strings.Contains(strings.ToLower(text), tag) {
_, _ = bot.Send(tgbot.NewMessage(chatID, allUsers))