src/main/java/cheat/Cheat.java

package cheat;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;
import javax.persistence.PersistenceContext;
import javax.transaction.Transactional;

import org.apache.commons.lang3.tuple.Pair;
import java.io.File;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;

import com.querydsl.jpa.impl.JPAQueryFactory;

import entity.Appeal;
import entity.QAppeal;
import entity.QSeed;
import entity.User;
import entity.config;
import entity.QUser;
import entity.TTorent;

public class Cheat implements CheatInterfnterface {

    @PersistenceContext
    private EntityManagerFactory emf;

    public Cheat() {
        config cfg = new config();
        Map<String,Object> props = new HashMap<>();
        props.put("javax.persistence.jdbc.url",
                  "jdbc:mysql://" + cfg.SqlURL + "/" + cfg.Database);
        props.put("javax.persistence.jdbc.user", cfg.SqlUsername);
        props.put("javax.persistence.jdbc.password", cfg.SqlPassword);
        this.emf = Persistence.createEntityManagerFactory("myPersistenceUnit", props);
    }

    @Override
    @Transactional
    public boolean AddAppeal(Appeal appeal) {
        EntityManager entityManager = emf.createEntityManager();
        try {
            entityManager.getTransaction().begin();
            entityManager.persist(appeal);
            entityManager.getTransaction().commit();
        } catch (Exception e) {
            e.printStackTrace();
            if (entityManager.getTransaction().isActive()) {
                entityManager.getTransaction().rollback();
            }
            return false;
        }
        return true;
    }

    @Override
    public Appeal GetAppeal(String appealid) {
        try {
            EntityManager entityManager = emf.createEntityManager();
            JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
            QAppeal qAppeal = QAppeal.appeal;
            Appeal appeal = queryFactory
                .selectFrom(qAppeal)
                .where(qAppeal.appealid.eq(appealid))
                .fetchOne();
            return appeal;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    @Override
    public Appeal[] GetAppealList() {
        EntityManager entityManager = emf.createEntityManager();
        try {
            JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
            QAppeal qAppeal = QAppeal.appeal;
            List<Appeal> appeals = queryFactory
                .selectFrom(qAppeal)
                .fetch();
            return appeals.toArray(new Appeal[0]);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            if (entityManager != null) {
                entityManager.close();
            }
        }
    }

    @Override
    public boolean HandleAppeal(String appealid, Integer status) {
        EntityManager entityManager = emf.createEntityManager();
        try {
            Appeal appeal = GetAppeal(appealid);
            if (appeal != null) {
                appeal.status = status;
                entityManager.getTransaction().begin();
                entityManager.merge(appeal);
                User user = entityManager.find(User.class, appeal.appealuserid);
                // if (user != null && user.accountstate != false) {
                if (user != null) {
                    if (status == 1) {
                        user.accountstate = false;
                    }
                    entityManager.merge(user);
                }
                entityManager.getTransaction().commit();
                return true;
            }
            return false;
        } catch (Exception e) {
            e.printStackTrace();
            if (emf.createEntityManager().getTransaction().isActive()) {
                emf.createEntityManager().getTransaction().rollback();
            }
            return false;
        } finally {
            if (entityManager != null) {
                entityManager.close();
            }
        }
    }

    @Override
    public Pair<String, String>[] GetFakeSeed() {
        List<Pair<String, String>> fakeSeeds = new ArrayList<>();
        try {
            EntityManager entityManager = emf.createEntityManager();
            JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
            QSeed qSeed = QSeed.seed;
            List<com.querydsl.core.Tuple> results = queryFactory
                .select(qSeed.seedid, qSeed.seeduserid)
                .from(qSeed)
                .where(qSeed.faketime.gt(config.getFakeTime()))
                .fetch();
            for (com.querydsl.core.Tuple result : results) {
                String seedid = result.get(qSeed.seedid);
                String userid = result.get(qSeed.seeduserid);
                fakeSeeds.add(Pair.of(seedid, userid));
            }
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
        return fakeSeeds.toArray(new Pair[0]);
    }

    @Override
    public String[] GetPunishedUserList() {
        List<String> punishedUsers = new ArrayList<>();
        try {
            EntityManager entityManager = emf.createEntityManager();
            JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
            entity.QUser qUser = entity.QUser.user;
            List<String> results = queryFactory
                .select(qUser.userid)
                .from(qUser)
                .where(qUser.accountstate.isTrue())
                .fetch();
            for (String userid : results) {
                punishedUsers.add(userid);
            }
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
        return punishedUsers.toArray(new String[0]);
    }

    @Override
    public void DetectTrans() {
        // JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
        // QSeed qSeed = QSeed.seed;
        // List<Seed> seeds = queryFactory
        //     .selectFrom(qSeed)
        //     .fetch();
        // for (Seed seed : seeds) {
        //     QTransRecord qTransRecord = QTransRecord.transRecord;
        //     List<TransRecord> tasks = queryFactory
        //         .selectFrom(qTransRecord)
        //         .where(qTransRecord.seedid.eq(seed.seedid))
        //         .fetch();

        //     int n = tasks.size();
        //     if (n == 0) continue;

        //     double[] xArr = new double[n];
        //     for (int i = 0; i < n; i++) {
        //         TransRecord t = tasks.get(i);
        //         xArr[i] = Math.max(0, t.upload - t.download);
        //     }

        //     double sum = 0;
        //     for (double x : xArr) sum += x;
        //     double mu = sum / n;

        //     double sqSum = 0;
        //     for (double x : xArr) sqSum += (x - mu) * (x - mu);
        //     double sigma = Math.sqrt(sqSum / n);

        //     for (int i = 0; i < n; i++) {
        //         if (Math.abs(xArr[i] - mu) > 3 * sigma) {
        //             User user = entityManager.find(User.class, tasks.get(i).downloaduserid);
        //             if (user != null) {
        //                 user.detectedCount++;
        //                 user.lastDetectedTime = new java.util.Date();
        //                 entityManager.merge(user);
        //             }
        //         }
        //     }
        // }
    }

    @Override
    public void DetectFakeSeed(){
    }

    @Override
    public boolean DetectFakeSeed(String seedid){
        return false;
    }

    @Override
    public void PunishUser(){
        EntityManager entityManager = emf.createEntityManager();
        JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
        entity.QUser qUser = entity.QUser.user;
        List<User> users = queryFactory
            .selectFrom(qUser)
            .where(qUser.detectedCount.gt(config.getCheatTime()) 
                .or(qUser.fakeDetectedCount.gt(config.getFakeTime())))
            .fetch();

        for (User user : users) {
            user.accountstate = true;
            entityManager.merge(user);
        }
    }

    @Override
    public User[] GetCheatUsers() {
        EntityManager entityManager = emf.createEntityManager();
        List<User> cheatUsers = new ArrayList<>();
        try {
            JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
            QUser qUser = QUser.user;
            cheatUsers = queryFactory
                .selectFrom(qUser)
                .where(qUser.accountstate.eq(true))
                .fetch();
            return cheatUsers.toArray(new User[0]);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            if (entityManager != null){
                entityManager.close();
            }
        }
    }

    @Override
    public User[] GetSuspiciousUsers() {
        EntityManager entityManager = emf.createEntityManager();
        List<User> suspiciousUsers = new ArrayList<>();
        try {
            JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
            QUser qUser = QUser.user;
            suspiciousUsers = queryFactory
                .selectFrom(qUser)
                .where((qUser.detectedCount.gt(0)
                    .or(qUser.fakeDetectedCount.gt(0)))
                    .and(qUser.accountstate.eq(false)))
                .fetch();
            return suspiciousUsers.toArray(new User[0]);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            if (entityManager != null){
                entityManager.close();
            }
        }
    }
    
    @Override
    public int UnbanUser(String userid) {
        EntityManager entityManager = emf.createEntityManager();
        try {
            User user = entityManager.find(User.class, userid);
            if (user == null) {
                return 1; // 用户不存在
            }
            if (user.accountstate) {
                user.accountstate = false; // 解封用户
                entityManager.getTransaction().begin();
                entityManager.merge(user);
                entityManager.getTransaction().commit();
                return 0; // 成功解封
            }
            return 2; // 用户未被封禁
        } catch (Exception e) {
            e.printStackTrace();
            if (entityManager.getTransaction().isActive()) {
                entityManager.getTransaction().rollback();
            }
            return -1; // 出现异常
        } finally {
            if (entityManager != null){
                entityManager.close();
            }
        }
    }

    @Override
    public int BanUser(String userid) {
        EntityManager entityManager = emf.createEntityManager();
        try {
            User user = entityManager.find(User.class, userid);
            if (user == null) {
                return 1; // 用户不存在
            }
            if (!user.accountstate) {
                user.accountstate = true; // 封禁用户
                entityManager.getTransaction().begin();
                entityManager.merge(user);
                entityManager.getTransaction().commit();
                return 0; // 成功封禁
            }
            return 2; // 用户已被封禁
        } catch (Exception e) {
            e.printStackTrace();
            if (entityManager.getTransaction().isActive()) {
                entityManager.getTransaction().rollback();
            }
            return -1; // 出现异常
        } finally {
            if (entityManager != null){
                entityManager.close();
            }
        }
    }

    @Override
    public int SubmitAppeal(String userid, String content, File file) {
        EntityManager entityManager = emf.createEntityManager();
        try {
            User user = entityManager.find(User.class, userid);
            if (user == null) {
                return 1; // 用户不存在
            }
            Appeal appeal = new Appeal();
            appeal.appealid = java.util.UUID.randomUUID().toString();
            appeal.appealuserid = userid;
            appeal.content = content;
            appeal.user = user; // 设置关联的用户
            Path storageDir = Paths.get(config.APPEAL_STORAGE_DIR);
            if (!Files.exists(storageDir)) {
                Files.createDirectories(storageDir);
            }
            String filename = file.getName();
            Path target = storageDir.resolve(appeal.appealid + "_" + filename);
            Files.copy(file.toPath(), target, StandardCopyOption.REPLACE_EXISTING);
            appeal.status = 0; // 初始状态为未处理
            appeal.fileURL = target.toString(); // 设置文件存储路径
            entityManager.getTransaction().begin();
            entityManager.persist(appeal);
            entityManager.getTransaction().commit();
            return 0; // 成功提交申诉
        } catch (Exception e) {
            e.printStackTrace();
            if (entityManager.getTransaction().isActive()) {
                entityManager.getTransaction().rollback();
            }
            return -1; // 出现异常
        } finally {
            if (entityManager != null){
                entityManager.close();
            }
        }
    }
}