|
7 | 7 | import dev.felnull.pointed.teams.manager.reward.RewardAdminService; |
8 | 8 | import dev.felnull.pointed.teams.manager.reward.data.*; |
9 | 9 | import org.bukkit.Bukkit; |
| 10 | +import org.bukkit.Material; |
| 11 | +import org.bukkit.entity.Player; |
| 12 | +import org.bukkit.inventory.ItemStack; |
| 13 | +import org.bukkit.inventory.PlayerInventory; |
10 | 14 |
|
11 | 15 | import javax.sql.DataSource; |
12 | 16 | import java.sql.*; |
@@ -830,4 +834,196 @@ public boolean hasEverDistributedRewardToPlayer(long playerSubjectId, int reward |
830 | 834 | } |
831 | 835 | } |
832 | 836 |
|
| 837 | + // ========== 4-1) 一括キュー投入 ========== |
| 838 | + @Override |
| 839 | + public void enqueueToAllTeamMembers(long teamSubjectId, String scope, |
| 840 | + LocalDate fromDate, LocalDate toDate, |
| 841 | + int rewardId) throws SQLException { |
| 842 | + List<String> cmds = findEnabledCommandsByRewardId(rewardId); |
| 843 | + if (cmds.isEmpty()) return; |
| 844 | + String raw = String.join("\n", cmds); |
| 845 | + |
| 846 | + final String qMembers = |
| 847 | + "SELECT tm.player_subject_id " + |
| 848 | + "FROM " + Names.t("team_members") + " tm " + |
| 849 | + "WHERE tm.team_subject_id = ?"; |
| 850 | + |
| 851 | + final String insQ = |
| 852 | + "INSERT INTO " + Names.t("reward_pending_queue") + |
| 853 | + " (team_subject_id, scope, from_date, to_date, player_subject_id, reward_id, raw_commands) " + |
| 854 | + " VALUES (?, ?, ?, ?, ?, ?, ?)"; |
| 855 | + |
| 856 | + try (Connection c = ds.getConnection()) { |
| 857 | + c.setAutoCommit(false); |
| 858 | + try (PreparedStatement psM = c.prepareStatement(qMembers); |
| 859 | + PreparedStatement psQ = c.prepareStatement(insQ)) { |
| 860 | + psM.setLong(1, teamSubjectId); |
| 861 | + try (ResultSet rs = psM.executeQuery()) { |
| 862 | + while (rs.next()) { |
| 863 | + int i = 1; |
| 864 | + psQ.setLong(i++, teamSubjectId); |
| 865 | + psQ.setString(i++, scope); |
| 866 | + psQ.setDate(i++, java.sql.Date.valueOf(fromDate)); |
| 867 | + psQ.setDate(i++, java.sql.Date.valueOf(toDate)); |
| 868 | + psQ.setLong(i++, rs.getLong(1)); // player_subject_id |
| 869 | + psQ.setInt(i++, rewardId); |
| 870 | + psQ.setString(i++, raw); |
| 871 | + psQ.addBatch(); |
| 872 | + } |
| 873 | + } |
| 874 | + psQ.executeBatch(); |
| 875 | + c.commit(); |
| 876 | + } catch (SQLException e) { c.rollback(); throw e; } |
| 877 | + finally { c.setAutoCommit(true); } |
| 878 | + } |
| 879 | + } |
| 880 | + |
| 881 | + // ========== 4-2) 受け取り（ログイン時／コマンド） ========== |
| 882 | + @Override |
| 883 | + public int[] claimPendingRewards(UUID playerUuid) throws SQLException { |
| 884 | + Player p = Bukkit.getPlayer(playerUuid); |
| 885 | + if (p == null || !p.isOnline()) return new int[]{0, countPendingByUuid(playerUuid)}; |
| 886 | + |
| 887 | + try (Connection c = ds.getConnection()) { |
| 888 | + Long sid = findPlayerSubjectId(c, playerUuid.toString()); |
| 889 | + if (sid == null) return new int[]{0, 0}; |
| 890 | + return tryDeliverPendingForPlayerInternal(c, p, sid); |
| 891 | + } |
| 892 | + } |
| 893 | + |
| 894 | +// ===== 内部：pending を読み / 実行 / ログ / 削除 ===== |
| 895 | + |
| 896 | + private record PendingRow(long id, Long teamId, String scope, LocalDate from, LocalDate to, Integer rewardId, |
| 897 | + String rawCommands, int requiredSlots) { |
| 898 | + } |
| 899 | + |
| 900 | + private int[] tryDeliverPendingForPlayerInternal(Connection c, Player p, long playerSubjectId) throws SQLException { |
| 901 | + c.setAutoCommit(false); |
| 902 | + int delivered = 0, remaining = 0; |
| 903 | + try { |
| 904 | + List<PendingRow> rows = fetchPendingRows(c, playerSubjectId); |
| 905 | + |
| 906 | + for (PendingRow r : rows) { |
| 907 | + if (!hasFreeSlots(p.getInventory(), r.requiredSlots)) { |
| 908 | + remaining++; |
| 909 | + continue; |
| 910 | + } |
| 911 | + // 実行 → ログ → キュー削除 |
| 912 | + String executed = runCommandsNow(p, r.rawCommands); |
| 913 | + insertDispatchLog(c, r, playerSubjectId, executed); |
| 914 | + deleteQueueRow(c, r.id); |
| 915 | + delivered++; |
| 916 | + } |
| 917 | + |
| 918 | + c.commit(); |
| 919 | + } catch (SQLException e) { |
| 920 | + c.rollback(); |
| 921 | + throw e; |
| 922 | + } finally { |
| 923 | + c.setAutoCommit(true); |
| 924 | + } |
| 925 | + return new int[]{delivered, remaining}; |
| 926 | + } |
| 927 | + |
| 928 | + private List<PendingRow> fetchPendingRows(Connection c, long playerSubjectId) throws SQLException { |
| 929 | + String q = |
| 930 | + "SELECT q.id, q.team_subject_id, q.scope, q.from_date, q.to_date, " + |
| 931 | + " q.reward_id, q.raw_commands, COALESCE(r.required_slots, 0) AS required_slots " + |
| 932 | + "FROM " + Names.t("reward_pending_queue") + " q " + |
| 933 | + "LEFT JOIN " + Names.t("rewards") + " r ON r.id = q.reward_id " + |
| 934 | + "WHERE q.player_subject_id=? " + |
| 935 | + "ORDER BY q.id ASC"; |
| 936 | + List<PendingRow> out = new ArrayList<>(); |
| 937 | + try (PreparedStatement ps = c.prepareStatement(q)) { |
| 938 | + ps.setLong(1, playerSubjectId); |
| 939 | + try (ResultSet rs = ps.executeQuery()) { |
| 940 | + while (rs.next()) { |
| 941 | + out.add(new PendingRow( |
| 942 | + rs.getLong("id"), |
| 943 | + (Long)rs.getObject("team_subject_id"), |
| 944 | + rs.getString("scope"), |
| 945 | + rs.getDate("from_date").toLocalDate(), |
| 946 | + rs.getDate("to_date").toLocalDate(), |
| 947 | + (Integer)rs.getObject("reward_id"), |
| 948 | + rs.getString("raw_commands"), |
| 949 | + rs.getInt("required_slots") |
| 950 | + )); |
| 951 | + } |
| 952 | + } |
| 953 | + } |
| 954 | + return out; |
| 955 | + } |
| 956 | + |
| 957 | + private static boolean hasFreeSlots(PlayerInventory inv, int need) { |
| 958 | + if (need <= 0) return true; |
| 959 | + int empty = 0; |
| 960 | + for (ItemStack is : inv.getStorageContents()) { |
| 961 | + if (is == null || is.getType() == Material.AIR) empty++; |
| 962 | + if (empty >= need) return true; |
| 963 | + } |
| 964 | + return false; |
| 965 | + } |
| 966 | + |
| 967 | + private String runCommandsNow(Player p, String rawCommands) { |
| 968 | + String playerName = p.getName(); |
| 969 | + StringBuilder executed = new StringBuilder(); |
| 970 | + for (String raw : rawCommands.split("\\n")) { |
| 971 | + String cmd = raw |
| 972 | + .replace("{player}", playerName) |
| 973 | + .replace("{playerId}", p.getUniqueId().toString()) |
| 974 | + .replace("{rank}", "0") |
| 975 | + .replace("{points}", "0"); |
| 976 | + Bukkit.dispatchCommand(Bukkit.getConsoleSender(), cmd); |
| 977 | + if (executed.length() > 0) executed.append('\n'); |
| 978 | + executed.append(cmd); |
| 979 | + } |
| 980 | + return executed.toString(); |
| 981 | + } |
| 982 | + |
| 983 | + private void insertDispatchLog(Connection c, PendingRow r, long playerSubjectId, String executed) throws SQLException { |
| 984 | + String ins = |
| 985 | + "INSERT IGNORE INTO " + Names.t("reward_dispatch_log") + |
| 986 | + " (team_subject_id, scope, from_date, to_date, player_subject_id, rank_no, points, reward_id, executed_commands) " + |
| 987 | + " VALUES (?, ?, ?, ?, ?, 0, 0, ?, ?)"; |
| 988 | + try (PreparedStatement ps = c.prepareStatement(ins)) { |
| 989 | + int i = 1; |
| 990 | + if (r.teamId == null) ps.setNull(i++, Types.BIGINT); else ps.setLong(i++, r.teamId); |
| 991 | + ps.setString(i++, r.scope); |
| 992 | + ps.setDate(i++, java.sql.Date.valueOf(r.from)); |
| 993 | + ps.setDate(i++, java.sql.Date.valueOf(r.to)); |
| 994 | + ps.setLong(i++, playerSubjectId); |
| 995 | + if (r.rewardId == null) ps.setNull(i++, Types.INTEGER); else ps.setInt(i++, r.rewardId); |
| 996 | + ps.setString(i++, executed); |
| 997 | + ps.executeUpdate(); |
| 998 | + } |
| 999 | + } |
| 1000 | + |
| 1001 | + private void deleteQueueRow(Connection c, long id) throws SQLException { |
| 1002 | + try (PreparedStatement ps = c.prepareStatement( |
| 1003 | + "DELETE FROM " + Names.t("reward_pending_queue") + " WHERE id=?")) { |
| 1004 | + ps.setLong(1, id); |
| 1005 | + ps.executeUpdate(); |
| 1006 | + } |
| 1007 | + } |
| 1008 | + |
| 1009 | + private int countPendingByUuid(UUID uuid) throws SQLException { |
| 1010 | + try (Connection c = ds.getConnection()) { |
| 1011 | + Long sid = findPlayerSubjectId(c, uuid.toString()); |
| 1012 | + if (sid == null) return 0; |
| 1013 | + try (PreparedStatement ps = c.prepareStatement( |
| 1014 | + "SELECT COUNT(*) FROM " + Names.t("reward_pending_queue") + " WHERE player_subject_id=?")) { |
| 1015 | + ps.setLong(1, sid); |
| 1016 | + try (ResultSet rs = ps.executeQuery()) { return rs.next() ? rs.getInt(1) : 0; } |
| 1017 | + } |
| 1018 | + } |
| 1019 | + } |
| 1020 | + |
| 1021 | + private Long findPlayerSubjectId(Connection c, String uuid) throws SQLException { |
| 1022 | + try (PreparedStatement ps = c.prepareStatement( |
| 1023 | + "SELECT id FROM " + Names.t("subjects") + " WHERE type='PLAYER' AND subject_key=?")) { |
| 1024 | + ps.setString(1, uuid); |
| 1025 | + try (ResultSet rs = ps.executeQuery()) { return rs.next() ? rs.getLong(1) : null; } |
| 1026 | + } |
| 1027 | + } |
| 1028 | + |
833 | 1029 | } |
0 commit comments