AllMusic/core/sql/DataSql.java

package com.coloryr.allmusic.server.core.sql;

import com.coloryr.allmusic.server.core.AllMusic;
import com.coloryr.allmusic.server.core.objs.config.SaveObj;
import com.coloryr.allmusic.server.core.objs.enums.HudDirType;
import com.coloryr.allmusic.server.core.objs.hud.PosObj;

import java.io.File;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.Queue;
import java.util.concurrent.LinkedBlockingDeque;
import java.util.concurrent.Semaphore;

public class DataSql {
    private static final Queue<Runnable> tasks = new LinkedBlockingDeque<>();
    private static final Semaphore semaphore = new Semaphore(0);
    /**
     * 创建表用
     */
    private static final String table = "CREATE TABLE IF NOT EXISTS \"allmusic\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"name\" TEXT(60),\n" +
            "  \"info_x\" integer(6),\n" +
            "  \"info_y\" integer(6),\n" +
            "  \"info_color\" integer(20),\n" +
            "  \"info_dir\" integer(6),\n" +
            "  \"info_shadow\" integer(1),\n" +
            "  \"info_enable\" integer(1),\n" +
            "  \"lyric_x\" integer(6),\n" +
            "  \"lyric_y\" integer(6),\n" +
            "  \"lyric_color\" integer(20),\n" +
            "  \"lyric_dir\" integer(6),\n" +
            "  \"lyric_shadow\" integer(1),\n" +
            "  \"lyric_enable\" integer(1),\n" +
            "  \"list_x\" integer(6),\n" +
            "  \"list_y\" integer(6),\n" +
            "  \"list_color\" integer(20),\n" +
            "  \"list_dir\" integer(6),\n" +
            "  \"list_shadow\" integer(1),\n" +
            "  \"list_enable\" integer(1),\n" +
            "  \"pic_x\" integer(6),\n" +
            "  \"pic_y\" integer(6),\n" +
            "  \"pic_dir\" integer(6),\n" +
            "  \"pic_size\" integer(6),\n" +
            "  \"pic_enable\" integer(1),\n" +
            "  \"pic_rotate\" integer(1),\n" +
            "  \"pic_rotate_speed\" integer(6)\n" +
            ");";

    private static final String table1 = "CREATE TABLE IF NOT EXISTS \"allmusic_list\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"sid\" TEXT(40)\n" +
            ");";

    private static final String table2 = "CREATE TABLE IF NOT EXISTS \"allmusic_mute\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"sid\" TEXT(60)\n" +
            ");";

    private static final String table3 = "CREATE TABLE IF NOT EXISTS \"allmusic_mutelist\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"sid\" TEXT(60)\n" +
            ");";

    private static final String table4 = "CREATE TABLE IF NOT EXISTS \"allmusic_banlist\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"sid\" TEXT(40)\n" +
            ");";

    private static final String table5 = "CREATE TABLE IF NOT EXISTS \"allmusic_banplayer\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"sid\" TEXT(60)\n" +
            ");";
    /**
     * 数据库文件
     */
    public static File sqlFile;
    private static boolean isRun;
    /**
     * 数据库链接对象
     */
    private static Connection connection;

    /**
     * 初始化数据库
     */
    public static void init() {
        try {
            AllMusic.log.info("正在初始化数据库");
            if (connection != null)
                connection.close();
            Class.forName("org.sqlite.JDBC");
            connection = DriverManager.getConnection("jdbc:sqlite:" + sqlFile.getPath());
            Statement stat = connection.createStatement();
            stat.execute(table);
            stat.execute(table1);
            stat.execute(table2);
            stat.execute(table3);
            stat.execute(table4);
            stat.execute(table5);
            stat.close();

            Cache.updateData();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 检查玩家是否在数据库
     *
     * @param name 用户名
     * @return 结果
     */
    public static boolean check(String name) {
        try {
            name = name.toLowerCase(Locale.ROOT);
            boolean have = false;
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT id FROM allmusic WHERE name ='" + name + "'");
            if (set.next()) {
                have = true;
            }
            set.close();
            stat.close();
            return have;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 更新玩家Hud数据
     *
     * @param name 用户名
     * @param hud  Hud数据
     */
    private static void update(String name, SaveObj hud) {
        name = name.toLowerCase(Locale.ROOT);
        String sql = "";
        try {
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            sql = "UPDATE allmusic SET " +
                    "info_x=" + hud.info.x + "," +
                    "info_y=" + hud.info.y + "," +
                    "info_color=" + hud.info.color + "," +
                    "info_dir=" + hud.info.dir.ordinal() + "," +
                    "info_shadow=" + (hud.info.shadow ? 1 : 0) + "," +
                    "info_enable=" + (hud.info.enable ? 1 : 0) + "," +
                    "lyric_x=" + hud.lyric.x + "," +
                    "lyric_y=" + hud.lyric.y + "," +
                    "lyric_color=" + hud.lyric.color + "," +
                    "lyric_dir=" + hud.lyric.dir.ordinal() + "," +
                    "lyric_shadow=" + (hud.lyric.shadow ? 1 : 0) + "," +
                    "lyric_enable=" + (hud.lyric.enable ? 1 : 0) + "," +
                    "list_x=" + hud.list.x + "," +
                    "list_y=" + hud.list.y + "," +
                    "list_color=" + hud.list.color + "," +
                    "list_dir=" + hud.list.dir.ordinal() + "," +
                    "list_shadow=" + (hud.list.shadow ? 1 : 0) + "," +
                    "list_enable=" + (hud.list.enable ? 1 : 0) + "," +
                    "pic_x=" + hud.pic.x + "," +
                    "pic_y=" + hud.pic.y + "," +
                    "pic_dir=" + hud.pic.dir.ordinal() + "," +
                    "pic_enable=" + (hud.pic.enable ? 1 : 0) + "," +
                    "pic_size=" + hud.pic.color + "," +
                    "pic_rotate=" + (hud.pic.shadow ? 1 : 0) + "," +
                    "pic_rotate_speed=" + hud.picRotateSpeed + " WHERE name='" + name + "'";
            stat.execute(sql);
            stat.close();
        } catch (Exception e) {
            AllMusic.log.info(sql);
            e.printStackTrace();
        }
    }

    /**
     * 添加用户Hud数据
     *
     * @param name 用户名
     * @param hud  数据
     */
    public static void addUser(String name, SaveObj hud) {
        name = name.toLowerCase(Locale.ROOT);
        String sql = "";
        try {
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            if (check(name)) {
                update(name, hud);
            } else {
                Statement stat = connection.createStatement();
                sql = "INSERT INTO allmusic (name,info_x,info_y," +
                        "info_enable,lyric_x,lyric_y,lyric_enable,list_x,list_y,list_enable," +
                        "pic_x,pic_y,pic_enable,pic_size,pic_rotate,pic_rotate_speed," +
                        "info_color,info_dir,info_shadow,lyric_color,lyric_dir," +
                        "lyric_shadow,list_color,list_dir,list_shadow,pic_dir)" +
                        "VALUES ('" + name + "'," +
                        hud.info.x + "," +
                        hud.info.y + "," +
                        (hud.info.enable ? 1 : 0) + "," +
                        hud.lyric.x + "," +
                        hud.lyric.y + "," +
                        (hud.lyric.enable ? 1 : 0) + "," +
                        hud.list.x + "," +
                        hud.list.y + "," +
                        (hud.list.enable ? 1 : 0) + "," +
                        hud.pic.x + "," +
                        hud.pic.y + "," +
                        (hud.pic.enable ? 1 : 0) + "," +
                        hud.pic.color + "," +
                        (hud.pic.shadow ? 1 : 0) + "," +
                        hud.picRotateSpeed + "," +
                        hud.info.color + "," +
                        hud.info.dir.ordinal() + "," +
                        (hud.info.shadow ? 1 : 0) + "," +
                        hud.lyric.color + "," +
                        hud.lyric.dir.ordinal() + "," +
                        (hud.lyric.shadow ? 1 : 0) + "," +
                        hud.list.color + "," +
                        hud.list.dir.ordinal() + "," +
                        (hud.list.shadow ? 1 : 0) + "," +
                        hud.pic.dir.ordinal() +
                        ")";
                stat.execute(sql);
                stat.close();
            }
        } catch (Exception e) {
            AllMusic.log.info(sql);
            e.printStackTrace();
        }
    }

    /**
     * 读取所有数据
     */
    public static SaveObj readHud(String name) {
        try {
            AllMusic.log.info("正在读取玩家数据:" + name);
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT info_x,info_y,info_enable,lyric_x," +
                    "lyric_y,lyric_enable,list_x,list_y,list_enable,pic_x,pic_y,pic_enable,pic_size," +
                    "pic_rotate,pic_rotate_speed,info_color,info_dir,info_shadow,lyric_color," +
                    "lyric_dir,lyric_shadow,list_color,list_dir,list_shadow,pic_dir FROM allmusic WHERE name='" + name + "'");
            HudDirType[] vas = HudDirType.values();
            SaveObj obj = null;
            if (set.next()) {
                obj = new SaveObj();
                PosObj pos1 = new PosObj();
                pos1.x = set.getInt(1);
                pos1.y = set.getInt(2);
                obj.info = pos1;
                obj.info.enable = set.getInt(3) == 1;
                PosObj pos2 = new PosObj();
                pos2.x = set.getInt(4);
                pos2.y = set.getInt(5);
                obj.lyric = pos2;
                obj.lyric.enable = set.getInt(6) == 1;
                PosObj pos3 = new PosObj();
                pos3.x = set.getInt(7);
                pos3.y = set.getInt(8);
                obj.list = pos3;
                obj.list.enable = set.getInt(9) == 1;
                PosObj pos4 = new PosObj();
                pos4.x = set.getInt(10);
                pos4.y = set.getInt(11);
                obj.pic = pos4;
                obj.pic.enable = set.getInt(12) == 1;
                obj.pic.color = set.getInt(13);
                obj.pic.shadow = set.getInt(14) == 1;
                obj.picRotateSpeed = set.getInt(15);
                obj.info.color = set.getInt(16);
                obj.info.dir = vas[set.getInt(17)];
                obj.info.shadow = set.getInt(18) == 1;
                obj.lyric.color = set.getInt(19);
                obj.lyric.dir = vas[set.getInt(20)];
                obj.lyric.shadow = set.getInt(21) == 1;
                obj.list.color = set.getInt(22);
                obj.list.dir = vas[set.getInt(23)];
                obj.list.shadow = set.getInt(24) == 1;
                obj.pic.dir = vas[set.getInt(25)];
            }
            stat.close();

            return obj;
        } catch (Exception e) {
            AllMusic.log.warning("数据库读取错误，请删除关闭服务器删除数据库，在启动服务器");
            e.printStackTrace();
        }

        return null;
    }

    /**
     * 随机获取空闲歌单歌曲
     */
    public static String readListItem() {
        try {
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT sid FROM allmusic_list ORDER BY random() limit 1");
            String name = null;
            if (set.next()) {
                name = set.getString(1);
            }
            stat.close();
            return name;
        } catch (Exception e) {
            AllMusic.log.warning("数据库读取错误，请删除关闭服务器删除数据库，在启动服务器");
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 读取空闲歌单列表
     */
    public static String readListItem(int index) {
        try {
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT sid FROM allmusic_list limit 1 offset " + index);
            String name = null;
            if (set.next()) {
                name = set.getString(1);
            }
            stat.close();
            return name;
        } catch (Exception e) {
            AllMusic.log.warning("数据库读取错误，请删除关闭服务器删除数据库，在启动服务器");
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 获取空闲歌单歌曲数量
     *
     * @return 歌曲数量
     */
    public static int getListSize() {
        try {
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT count(sid) FROM allmusic_list");
            int count = 0;
            if (set.next()) {
                count = set.getInt(1);
            }
            stat.close();
            return count;
        } catch (Exception e) {
            AllMusic.log.warning("数据库读取错误，请删除关闭服务器删除数据库，在启动服务器");
            e.printStackTrace();
        }
        return 0;
    }

    public static void addIdleList(List<String> list) {
        task(() -> {
            try {
                AllMusic.log.info("添加" + list.size() + "首歌到空闲歌单");
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("INSERT INTO allmusic_list (sid) VALUES (?)");
                // 遍历 List<String> 并插入每个字符串
                for (String str : list) {
                    pstmt.setString(1, str); // 设置参数
                    pstmt.addBatch(); // 添加到批处理
                }
                pstmt.executeBatch();
                pstmt.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void clearIdleList() {
        task(() -> {
            try {
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                Statement stat = connection.createStatement();
                stat.execute("DELETE FROM allmusic_list");
                stat.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void addBanMusic(String music) {
        task(() -> {
            try {
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("INSERT INTO allmusic_banlist (sid) VALUES (?)");
                pstmt.setString(1, music); // 设置参数
                pstmt.execute();
                pstmt.close();
                Cache.banMusic.add(music);
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void removeBanMusic(String music) {
        task(() -> {
            try {
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("DELETE FROM allmusic_banlist WHERE sid=?");
                pstmt.setString(1, music); // 设置参数
                pstmt.execute();
                pstmt.close();
                Cache.banMusic.remove(music);
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    /**
     * check a music in ban list?
     *
     * @param name 用户名
     * @return 结果
     */
    public static boolean checkBanMusic(String name) {
        return Cache.banMusic.contains(name);
    }


    /***
     * a way to get ban music list
     * @return ban music list
     */
    public static List<String> getBanMusicList() {
        try {
            List<String> banList = new ArrayList<>();
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT sid FROM allmusic_banlist");
            while (set.next()) {
                String musicName = set.getString("sid");
                if (musicName != null) {
                    banList.add(musicName);
                }
            }
            set.close();
            stat.close();
            return banList;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return new ArrayList<>();
    }

    public static void clearBan() {
        task(() -> {
            try {
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                Statement stat = connection.createStatement();
                stat.execute("DELETE FROM allmusic_banlist");
                stat.close();
                Cache.banMusic.clear();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void addBanPlayer(String player) {
        task(() -> {
            try {
                String player1 = player.toLowerCase(Locale.ROOT);
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("INSERT INTO allmusic_banplayer (sid) VALUES (?)");
                pstmt.setString(1, player1); // 设置参数
                pstmt.execute();
                pstmt.close();
                Cache.banPlayers.add(player);
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void removeBanPlayer(String player) {
        task(() -> {
            try {
                String player1 = player.toLowerCase(Locale.ROOT);
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("DELETE FROM allmusic_banplayer WHERE sid=?");
                pstmt.setString(1, player1); // 设置参数
                pstmt.execute();
                pstmt.close();
                Cache.banPlayers.remove(player);
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    /**
     * 检查玩家是否在数据库
     *
     * @param name 用户名
     * @return 结果
     */
    public static boolean checkBanPlayer(String name) {
        return Cache.banPlayers.contains(name);
    }

    /***
     * @return a list of ban players
     */
    public static List<String> getBanPlayerList() {
        try {
            List<String> banList = new ArrayList<>();
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT sid FROM allmusic_banplayer");
            while (set.next()) {
                String playerName = set.getString("sid");
                if (playerName != null) {
                    banList.add(playerName);
                }
            }
            set.close();
            stat.close();
            return banList;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return new ArrayList<>();
    }

    public static void clearBanPlayer() {
        task(() -> {
            try {
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                Statement stat = connection.createStatement();
                stat.execute("DELETE FROM allmusic_banlist");
                stat.close();
                Cache.banPlayers.clear();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void addMutePlayer(String player) {
        task(() -> {
            try {
                String player1 = player.toLowerCase(Locale.ROOT);
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("INSERT INTO allmusic_mute (sid) VALUES (?)");
                pstmt.setString(1, player1); // 设置参数
                pstmt.execute();
                pstmt.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void removeMutePlayer(String player) {
        task(() -> {
            try {
                String player1 = player.toLowerCase(Locale.ROOT);
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("DELETE FROM allmusic_mute WHERE sid=?");
                pstmt.setString(1, player1); // 设置参数
                pstmt.execute();
                pstmt.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    /**
     * 检查玩家是否在数据库
     *
     * @param name 用户名
     * @return 结果
     */
    public static boolean checkMutePlayer(String name) {
        try {
            name = name.toLowerCase(Locale.ROOT);
            boolean have = false;
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT id FROM allmusic_mute WHERE sid ='" + name + "'");
            if (set.next()) {
                have = true;
            }
            set.close();
            stat.close();
            return have;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    public static void addMuteListPlayer(String player) {
        task(() -> {
            try {
                String player1 = player.toLowerCase(Locale.ROOT);
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("INSERT INTO allmusic_mutelist (sid) VALUES (?)");
                pstmt.setString(1, player1); // 设置参数
                pstmt.execute();
                pstmt.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void removeMuteListPlayer(String player) {
        task(() -> {
            try {
                String player1 = player.toLowerCase(Locale.ROOT);
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                PreparedStatement pstmt = connection.prepareStatement("DELETE FROM allmusic_mutelist WHERE sid=?");
                pstmt.setString(1, player1); // 设置参数
                pstmt.execute();
                pstmt.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    /**
     * 检查玩家是否在数据库
     *
     * @param name 用户名
     * @return 结果
     */
    public static boolean checkMuteListPlayer(String name) {
        try {
            name = name.toLowerCase(Locale.ROOT);
            boolean have = false;
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT id FROM allmusic_mutelist WHERE sid ='" + name + "'");
            if (set.next()) {
                have = true;
            }
            set.close();
            stat.close();
            return have;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    public static void task(Runnable runnable) {
        tasks.add(runnable);
        semaphore.release();
    }

    /**
     * 启用Hud数据库
     */
    public static void start() {
        init();
        Thread thread = new Thread(DataSql::run);
        isRun = true;
        thread.start();
    }

    /**
     * 停止数据库
     */
    public static void stop() {
        isRun = false;
        semaphore.release();
    }

    private static void run() {
        Runnable runnable;
        while (isRun) {
            try {
                semaphore.acquire();
                if (!isRun)
                    return;
                do {
                    runnable = tasks.poll();
                    if (runnable != null) {
                        runnable.run();
                    }
                }
                while (runnable != null);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}