diff --git a/internal/bot/bot.go b/internal/bot/bot.go index 75a5df5..178d695 100644 --- a/internal/bot/bot.go +++ b/internal/bot/bot.go @@ -75,8 +75,8 @@ func (b *Bot) handleMessage(message *tgbotapi.Message) { // Handle commands if message.IsCommand() { switch message.Command() { - case "submit": - b.handleSubmitCommand(message) + // case "submit": + // b.handleSubmitCommand(message) case "leaderboards": b.handleLeaderboardCommand(message) case "help": @@ -493,6 +493,16 @@ func (b *Bot) CheckSubmissions() error { } else { mention = user.FirstName } + submission := &models.Submission{ + UserID: user.ID, + ProblemID: todaysChallenge.ID, + Date: today, + } + err = b.db.AddSubmission(submission) + if err != nil { + log.Printf("Error adding submission for user %d: %v", user.ID, err) + continue + } messageText := fmt.Sprintf("🎉 %s has just submitted today's challenge (Day %d):\n\n", mention, dayNumber) diff --git a/internal/database/database.go b/internal/database/database.go index 24367b3..7bcc489 100644 --- a/internal/database/database.go +++ b/internal/database/database.go @@ -342,7 +342,7 @@ func (db *DB) GetLeetcodeProfile(id int64) (*models.User, error) { if rows.Next() { var user models.User - err := rows.Scan(&user.ID, &user.Username, &user.FirstName, &user.LastName) + err := rows.Scan(&user.ID, &user.Username) if err != nil { return nil, err }