Merge pull request #247 from JustHumanz/userChecker

User checker

pkg/config/config.go

@@ -83,6 +83,7 @@ const (
 	CheckServerCount           = "@every 0h10m0s"
 	CheckPayload               = "@every 1h30m0s"
 	PilotGetGroups             = "@every 1h0m0s"
+	CheckUser                  = "0 1 * * *"
 
 	//Time
 	AddUserTTL      = 5 * time.Hour

pkg/database/generaldb.go

@@ -5,6 +5,7 @@ import (
 	"database/sql"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"regexp"
 	"strconv"
 	"strings"
@@ -932,3 +933,32 @@ func (i *LiveStream) RemoveCache(Key string) error {
 	}
 	return nil
 }
+
+func GetAllUser() []string {
+	var (
+		UserID []string
+	)
+	rows, err := DB.Query(`SELECT DiscordID FROM Vtuber.User where Human=1 group by DiscordID;`)
+	if err != nil {
+		log.Error(err)
+	}
+
+	defer rows.Close()
+	for rows.Next() {
+		id := ""
+		err = rows.Scan(&id)
+		if err != nil {
+			log.Error(err)
+		}
+		UserID = append(UserID, id)
+	}
+	return UserID
+}
+
+func DeleteDeletedUser(users []string) {
+	sqlq := fmt.Sprintf(`Delete FROM User where DiscordID in ('%s')`, strings.Join(users, "','"))
+	_, err := DB.Query(sqlq)
+	if err != nil {
+		log.Error(err)
+	}
+}

service/utility/runner.go

@@ -314,6 +314,46 @@ func main() {
 		log.Info("Done")
 	})
 
+	c.AddFunc(config.CheckUser, func() {
+		log.Info("Start check deleted user")
+		UsersDeleted := []string{}
+		Users := database.GetAllUser()
+		for _, v := range Users {
+			User, err := Bot.User(v)
+			if err != nil {
+				log.Error(err, v)
+			}
+			if strings.HasPrefix(User.Username, "Deleted") {
+				log.WithFields(log.Fields{
+					"userId": v,
+				}).Info("User not found or user deleted ther account")
+				UsersDeleted = append(UsersDeleted, v)
+			}
+		}
+
+		log.Info("Start deleting user ", UsersDeleted)
+		database.DeleteDeletedUser(UsersDeleted)
+	})
+
+	log.Info("Start check deleted user")
+	UsersDeleted := []string{}
+	Users := database.GetAllUser()
+	for _, v := range Users {
+		User, err := Bot.User(v)
+		if err != nil {
+			log.Error(err, v)
+		}
+		if strings.HasPrefix(User.Username, "Deleted") {
+			log.WithFields(log.Fields{
+				"userId": v,
+			}).Info("User not found or user deleted ther account")
+			UsersDeleted = append(UsersDeleted, v)
+		}
+	}
+
+	log.Info("Start deleting user ", UsersDeleted)
+	database.DeleteDeletedUser(UsersDeleted)
+
 	go pilot.RunHeartBeat(gRCPconn, "Utility")
 	runfunc.Run(Bot)
 }