Examples with RegisteredServer com.velocitypowered.api.proxy.server.RegisteredServer used on opensource projects

Example 1 with RegisteredServer

use of com.velocitypowered.api.proxy.server.RegisteredServer in project VotingPlugin by Ben12345rocks.

the class VotingPluginVelocity method addVoteParty.

public void addVoteParty() {
    if (getConfig().getVotePartyEnabled()) {
        addCurrentVotePartyVotes(1);
        if (votePartyVotes >= currentVotePartyVotesRequired) {
            debug("Current vote party total: " + votePartyVotes);
            addCurrentVotePartyVotes(-currentVotePartyVotesRequired);
            currentVotePartyVotesRequired += getConfig().getVotePartyIncreaseVotesRequired();
            voteCacheFile.setVotePartyInreaseVotesRequired(voteCacheFile.getVotePartyInreaseVotesRequired() + getConfig().getVotePartyIncreaseVotesRequired());
            if (!getConfig().getVotePartyBroadcast().isEmpty()) {
                for (RegisteredServer server : getVotePartyServers()) {
                    if (method.equals(BungeeMethod.PLUGINMESSAGING)) {
                        sendPluginMessageServer(server, "VotePartyBroadcast", config.getVotePartyBroadcast());
                    } else if (method.equals(BungeeMethod.SOCKETS)) {
                        sendServerMessageServer(server.getServerInfo().getName(), config.getVotePartyBroadcast());
                    }
                }
            }
            for (String command : getConfig().getVotePartyBungeeCommands()) {
                server.getCommandManager().executeAsync(server.getConsoleCommandSource(), command);
            }
            if (method.equals(BungeeMethod.PLUGINMESSAGING)) {
                for (RegisteredServer server : getVotePartyServers()) {
                    sendVoteParty(server);
                }
            }
        }
        voteCacheFile.save();
    }
}

Example 2 with RegisteredServer

use of com.velocitypowered.api.proxy.server.RegisteredServer in project VotingPlugin by Ben12345rocks.

the class VotingPluginVelocity method onPluginMessagingReceived.

@Subscribe
public void onPluginMessagingReceived(PluginMessageEvent event) {
    if (event.getIdentifier().getId().equals(CHANNEL.getId())) {
        ByteArrayInputStream instream = new ByteArrayInputStream(event.getData());
        DataInputStream in = new DataInputStream(instream);
        try {
            ByteArrayOutputStream outstream = new ByteArrayOutputStream();
            DataOutputStream out = new DataOutputStream(outstream);
            String subchannel = in.readUTF();
            int size = in.readInt();
            // check for status message returns
            if (subchannel.equalsIgnoreCase("statusokay")) {
                String server = in.readUTF();
                logger.info("Status okay for " + server);
                return;
            } else if (subchannel.equalsIgnoreCase("login")) {
                String player = in.readUTF();
                debug("Login: " + player);
                if (server.getPlayer(player).isPresent()) {
                    Player p = server.getPlayer(player).get();
                    if (p.getCurrentServer().isPresent()) {
                        final RegisteredServer server = p.getCurrentServer().get().getServer();
                        final Player p1 = p;
                        timer.schedule(new TimerTask() {

                            @Override
                            public void run() {
                                checkCachedVotes(server);
                                checkOnlineVotes(p1, p.getUniqueId().toString(), server);
                            }
                        }, 0);
                    }
                }
                return;
            } else if (subchannel.equalsIgnoreCase("timeupdate")) {
                String str = in.readUTF();
                String[] data = str.split(Pattern.quote("//"));
                if (data.length == 3) {
                    timeHandle.setMonth(data[0]);
                    timeHandle.setDay(Integer.parseInt(data[1]));
                    timeHandle.setWeek(Integer.parseInt(data[2]));
                }
            } else {
                // reforward message
                out.writeUTF(subchannel);
                out.writeInt(size);
                for (int i = 0; i < size; i++) {
                    out.writeUTF(in.readUTF());
                }
                for (RegisteredServer send : server.getAllServers()) {
                    if (send.getPlayersConnected().size() > 0) {
                        send.sendPluginMessage(CHANNEL, outstream.toByteArray());
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

Example 3 with RegisteredServer

use of com.velocitypowered.api.proxy.server.RegisteredServer in project VotingPlugin by Ben12345rocks.

the class VotingPluginVelocity method onProxyDisable.

@Subscribe
public void onProxyDisable(ProxyShutdownEvent event) {
    if (method.equals(BungeeMethod.PLUGINMESSAGING)) {
        logger.info("VotingPlugin saving vote cache: " + cachedVotes.size() + "/" + cachedOnlineVotes.size());
        for (Entry<RegisteredServer, ArrayList<OfflineBungeeVote>> entry : cachedVotes.entrySet()) {
            RegisteredServer server = entry.getKey();
            int num = 0;
            for (OfflineBungeeVote voteData : entry.getValue()) {
                voteCacheFile.addVote(server.getServerInfo().getName(), num, voteData);
                num++;
            }
        }
        for (Entry<String, ArrayList<OfflineBungeeVote>> entry : cachedOnlineVotes.entrySet()) {
            String name = entry.getKey();
            int num = 0;
            for (OfflineBungeeVote voteData : entry.getValue()) {
                voteCacheFile.addVoteOnline(name, num, voteData);
                num++;
            }
        }
    }
    if (timeHandle != null) {
        timeHandle.save();
    }
    voteCacheFile.save();
    nonVotedPlayersCache.save();
    if (mysql != null) {
        mysql.shutDown();
    }
    logger.info("VotingPlugin disabled");
}

Example 4 with RegisteredServer

use of com.velocitypowered.api.proxy.server.RegisteredServer in project VotingPlugin by Ben12345rocks.

the class VotingPluginVelocity method onProxyInitialization.

@Subscribe
public void onProxyInitialization(ProxyInitializeEvent event) {
    File configFile = new File(dataDirectory.toFile(), "bungeeconfig.yml");
    configFile.getParentFile().mkdirs();
    if (!configFile.exists()) {
        try {
            configFile.createNewFile();
        } catch (IOException e) {
            e.printStackTrace();
        }
        InputStream toCopyStream = VotingPluginVelocity.class.getClassLoader().getResourceAsStream("bungeeconfig.yml");
        try (FileOutputStream fos = new FileOutputStream(configFile)) {
            byte[] buf = new byte[2048];
            int r;
            while (-1 != (r = toCopyStream.read(buf))) {
                fos.write(buf, 0, r);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    config = new Config(configFile);
    server.getChannelRegistrar().register(CHANNEL);
    method = BungeeMethod.getByName(config.getBungeeMethod());
    boolean mysqlLoaded = true;
    try {
        if (!config.getString(config.getNode("Host"), "").isEmpty()) {
            loadMysql();
        } else {
            mysqlLoaded = false;
            logger.error("MySQL settings not set in bungeeconfig.yml");
        }
    } catch (Exception e) {
        mysqlLoaded = false;
        e.printStackTrace();
    }
    CommandMeta meta = server.getCommandManager().metaBuilder("votingpluginbungee").aliases("vpb").build();
    server.getCommandManager().register(meta, new VotingPluginVelocityCommand(this));
    if (mysqlLoaded) {
        uuidPlayerNameCache = mysql.getRowsUUIDNameQuery();
        voteCacheFile = new VoteCache(new File(dataDirectory.toFile(), "votecache.yml"));
        nonVotedPlayersCache = new NonVotedPlayersCache(new File(dataDirectory.toFile(), "nonvotedplayerscache.yml"), this);
        timeHandle = new TimeHandle(voteCacheFile.getNode("Time", "Month").getString(""), voteCacheFile.getNode("Time", "Day").getInt(), voteCacheFile.getNode("Time", "Week").getInt()) {

            @Override
            public void save() {
                voteCacheFile.getNode("Time", "Month").setValue(timeHandle.getMonth());
                voteCacheFile.getNode("Time", "Day").setValue(timeHandle.getDay());
                voteCacheFile.getNode("Time", "Week").setValue(timeHandle.getWeek());
            }
        };
        if (method.equals(BungeeMethod.PLUGINMESSAGING)) {
            try {
                for (String serverToCheck : voteCacheFile.getServers()) {
                    ArrayList<OfflineBungeeVote> vote = new ArrayList<OfflineBungeeVote>();
                    for (String num : voteCacheFile.getServerVotes(serverToCheck)) {
                        ConfigurationNode data = voteCacheFile.getServerVotes(serverToCheck, num);
                        vote.add(new OfflineBungeeVote(data.getNode("Name").getString(), data.getNode("UUID").getString(), data.getNode("Service").getString(), data.getNode("Time").getLong(), data.getNode("Real").getBoolean(), data.getNode("TEXT").getString()));
                    }
                    cachedVotes.put(server.getServer(serverToCheck).get(), vote);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
            try {
                for (String player : voteCacheFile.getPlayers()) {
                    ArrayList<OfflineBungeeVote> vote = new ArrayList<OfflineBungeeVote>();
                    for (String num : voteCacheFile.getOnlineVotes(player)) {
                        ConfigurationNode data = voteCacheFile.getOnlineVotes(player, num);
                        vote.add(new OfflineBungeeVote(data.getNode("Name").getString(), data.getNode("UUID").getString(), data.getNode("Service").getString(), data.getNode("Time").getLong(), data.getNode("Real").getBoolean(), data.getNode("TEXT").getString()));
                    }
                    cachedOnlineVotes.put(player, vote);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
            voteCacheFile.clearData();
            server.getScheduler().buildTask(this, () -> {
                for (RegisteredServer server : cachedVotes.keySet()) {
                    checkCachedVotes(server);
                }
                for (String player : cachedOnlineVotes.keySet()) {
                    if (server.getPlayer(UUID.fromString(player)).isPresent()) {
                        checkOnlineVotes(server.getPlayer(UUID.fromString(player)).get(), player, null);
                    }
                }
            }).delay(15L, TimeUnit.SECONDS).repeat(30l, TimeUnit.SECONDS).schedule();
            server.getScheduler().buildTask(this, () -> {
                if (nonVotedPlayersCache != null) {
                    debug("Checking nonvotedplayerscache.yml...");
                    nonVotedPlayersCache.check();
                }
            }).delay(1L, TimeUnit.MINUTES).repeat(60l, TimeUnit.MINUTES).schedule();
        } else if (method.equals(BungeeMethod.SOCKETS)) {
            encryptionHandler = new EncryptionHandler(new File(dataDirectory.toFile(), "secretkey.key"));
            socketHandler = new SocketHandler(server.getPluginManager().getPlugin("votingplugin").get().getDescription().getVersion().get(), config.getBungeeHost(), config.getBungeePort(), encryptionHandler, config.getDebug());
            socketHandler.add(new SocketReceiver() {

                @Override
                public void onReceive(String[] data) {
                    if (data.length > 1) {
                        if (data.length > 2) {
                            if (data[0].equalsIgnoreCase("Broadcast")) {
                                sendServerMessage(data);
                            }
                        }
                    }
                }
            });
            socketHandler.add(new SocketReceiver() {

                @Override
                public void onReceive(String[] data) {
                    if (data.length > 1) {
                        if (data[0].equalsIgnoreCase("StatusOkay")) {
                            String server = data[1];
                            logger.info("Voting communicaton okay with " + server);
                        }
                    }
                }
            });
            clientHandles = new HashMap<String, ClientHandler>();
            List<String> l = config.getBlockedServers();
            for (ConfigurationNode d : config.getSpigotServers()) {
                String s = d.getKey().toString();
                if (!l.contains(s)) {
                    clientHandles.put(s, new ClientHandler(d.getNode("Host").getString("0.0.0.0"), d.getNode("Port").getInt(1298), encryptionHandler, config.getDebug()));
                }
            }
        }
    }
    currentVotePartyVotesRequired = getConfig().getVotePartyVotesRequired() + voteCacheFile.getVotePartyInreaseVotesRequired();
    votePartyVotes = voteCacheFile.getVotePartyCurrentVotes();
    try {
        getVersionFile();
        if (versionFile != null) {
            versionFile.delete();
            versionFile.getParentFile().delete();
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
    Metrics metrics = metricsFactory.make(this, 11547);
    metrics.addCustomChart(new SimplePie("bungee_method", () -> getConfig().getBungeeMethod().toString()));
    metrics.addCustomChart(new SimplePie("sendtoallservers", () -> "" + getConfig().getSendVotesToAllServers()));
    metrics.addCustomChart(new SimplePie("allowunjoined", () -> "" + getConfig().getAllowUnJoined()));
    metrics.addCustomChart(new SimplePie("pointsonvote", () -> "" + getConfig().getPointsOnVote()));
    metrics.addCustomChart(new SimplePie("bungeemanagetotals", () -> "" + getConfig().getBungeeManageTotals()));
    metrics.addCustomChart(new SimplePie("waitforuseronline", () -> "" + getConfig().getWaitForUserOnline()));
    metrics.addCustomChart(new SimplePie("plugin_version", () -> "" + version));
    if (!buildNumber.equals("NOTSET")) {
        metrics.addCustomChart(new SimplePie("dev_build_number", () -> "" + buildNumber));
    }
    logger.info("VotingPlugin velocity loaded, method: " + method.toString() + ", PluginMessagingVersion: " + BungeeVersion.getPluginMessageVersion() + ", Internal Jar Version: " + version);
    if (!buildNumber.equals("NOTSET")) {
        logger.info("Detected using dev build number: " + buildNumber);
    }
}

Example 5 with RegisteredServer

use of com.velocitypowered.api.proxy.server.RegisteredServer in project VotingPlugin by Ben12345rocks.

the class VotingPluginVelocity method vote.

public synchronized void vote(String player, String service, boolean realVote) {
    try {
        if (player == null || player.isEmpty()) {
            logger.info("No name from vote on " + service);
            return;
        }
        String uuid = getUUID(player);
        if (uuid.isEmpty()) {
            if (config.getAllowUnJoined()) {
                debug("Fetching UUID online, since allowunjoined is enabled");
                UUID u = null;
                try {
                    u = fetchUUID(player);
                } catch (Exception e) {
                    if (getConfig().getDebug()) {
                        e.printStackTrace();
                    }
                }
                if (u == null) {
                    debug("Failed to get uuid for " + player);
                    return;
                }
                uuid = u.toString();
            } else {
                logger.info("Ignoring vote from " + player + " since player hasn't joined before");
                return;
            }
        }
        player = getProperName(uuid, player);
        BungeeMessageData text = null;
        addVoteParty();
        if (getConfig().getBungeeManageTotals()) {
            if (mysql == null) {
                logger.error("Mysql is not loaded correctly, stopping vote processing");
                return;
            }
            if (!mysql.getUuids().contains(uuid)) {
                mysql.update(uuid, "PlayerName", new DataValueString(player));
            }
            ArrayList<Column> data = mysql.getExactQuery(new Column("uuid", new DataValueString(uuid)));
            int allTimeTotal = getValue(data, "AllTimeTotal", 1);
            int monthTotal = getValue(data, "MonthTotal", 1);
            int weeklyTotal = getValue(data, "WeeklyTotal", 1);
            int dailyTotal = getValue(data, "DailyTotal", 1);
            int points = getValue(data, "Points", getConfig().getPointsOnVote());
            int milestoneCount = getValue(data, "MilestoneCount", 1);
            text = new BungeeMessageData(allTimeTotal, monthTotal, weeklyTotal, dailyTotal, points, milestoneCount, votePartyVotes, currentVotePartyVotesRequired);
            ArrayList<Column> update = new ArrayList<Column>();
            update.add(new Column("AllTimeTotal", new DataValueInt(allTimeTotal)));
            update.add(new Column("MonthTotal", new DataValueInt(monthTotal)));
            update.add(new Column("WeeklyTotal", new DataValueInt(weeklyTotal)));
            update.add(new Column("DailyTotal", new DataValueInt(dailyTotal)));
            update.add(new Column("Points", new DataValueInt(points)));
            update.add(new Column("MilestoneCount", new DataValueInt(milestoneCount)));
            debug("Setting totals " + text.toString());
            mysql.update(uuid, update);
        } else {
            text = new BungeeMessageData(0, 0, 0, 0, 0, 0, votePartyVotes, currentVotePartyVotesRequired);
        }
        /*
			 * String text = mysqlUpdate(data, uuid, "AllTimeTotal", 1) + "//" +
			 * mysqlUpdate(data, uuid, "MonthTotal", 1) + "//" + mysqlUpdate(data, uuid,
			 * "WeeklyTotal", 1) + "//" + mysqlUpdate(data, uuid, "DailyTotal", 1) + "//" +
			 * mysqlUpdate(data, uuid, "Points", 1) + "//" + mysqlUpdate(data, uuid,
			 * "MilestoneCount", 1);
			 */
        long time = LocalDateTime.now().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli();
        if (method.equals(BungeeMethod.PLUGINMESSAGING)) {
            if (config.getSendVotesToAllServers()) {
                for (RegisteredServer s : server.getAllServers()) {
                    if (!config.getBlockedServers().contains(s.getServerInfo().getName())) {
                        boolean forceCache = false;
                        Player p = null;
                        if (server.getPlayer(UUID.fromString(uuid)).isPresent()) {
                            p = server.getPlayer(UUID.fromString(uuid)).get();
                        }
                        if ((p == null || !p.isActive()) && getConfig().getWaitForUserOnline()) {
                            forceCache = true;
                            debug("Forcing vote to cache");
                        }
                        if (s.getPlayersConnected().isEmpty() || forceCache) {
                            // cache
                            if (!cachedVotes.containsKey(s)) {
                                cachedVotes.put(s, new ArrayList<OfflineBungeeVote>());
                            }
                            ArrayList<OfflineBungeeVote> list = cachedVotes.get(s);
                            list.add(new OfflineBungeeVote(player, uuid, service, time, realVote, text.toString()));
                            cachedVotes.put(s, list);
                            debug("Caching vote for " + player + " on " + service + " for " + s);
                        } else {
                            // send
                            sendPluginMessageServer(s, "Vote", player, uuid, service, "" + time, Boolean.TRUE.toString(), "" + realVote, text.toString(), "" + getConfig().getBungeeManageTotals(), "" + BungeeVersion.getPluginMessageVersion(), "" + config.getBroadcast());
                        }
                        if (config.getBroadcast()) {
                            sendPluginMessageServer(s, "VoteBroadcast", uuid, player, service);
                        }
                    }
                }
            } else {
                Player p = null;
                if (server.getPlayer(UUID.fromString(uuid)).isPresent()) {
                    p = server.getPlayer(UUID.fromString(uuid)).get();
                }
                if (p != null && p.isActive() && !config.getBlockedServers().contains(p.getCurrentServer().get().getServerInfo().getName())) {
                    sendPluginMessageServer(p.getCurrentServer().get().getServer(), "VoteOnline", player, uuid, service, "" + time, Boolean.TRUE.toString(), "" + realVote, text.toString(), "" + getConfig().getBungeeManageTotals(), "" + BungeeVersion.getPluginMessageVersion(), "" + config.getBroadcast());
                } else {
                    if (!cachedOnlineVotes.containsKey(uuid)) {
                        cachedOnlineVotes.put(uuid, new ArrayList<OfflineBungeeVote>());
                    }
                    ArrayList<OfflineBungeeVote> list = cachedOnlineVotes.get(uuid);
                    if (list == null) {
                        list = new ArrayList<OfflineBungeeVote>();
                    }
                    list.add(new OfflineBungeeVote(player, uuid, service, time, realVote, text.toString()));
                    cachedOnlineVotes.put(uuid, list);
                    debug("Caching online vote for " + player + " on " + service);
                }
                for (RegisteredServer s : server.getAllServers()) {
                    sendPluginMessageServer(s, "VoteUpdate", uuid, "" + votePartyVotes, "" + currentVotePartyVotesRequired);
                    if (config.getBroadcast()) {
                        sendPluginMessageServer(s, "VoteBroadcast", uuid, player, service);
                    }
                }
            }
        } else if (method.equals(BungeeMethod.SOCKETS)) {
            sendSocketVote(player, service, text);
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}