BauSystem2.0/BauSystem_Main/src/de/steamwar/bausystem/BauSystem.java

/*
 * This file is a part of the SteamWar software.
 *
 * Copyright (C) 2021  SteamWar.de-Serverteam
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

package de.steamwar.bausystem;

import com.comphenix.tinyprotocol.TinyProtocol;
import de.steamwar.bausystem.configplayer.Config;
import de.steamwar.bausystem.features.tpslimit.TPSFreezeUtils;
import de.steamwar.bausystem.features.world.RamUsage;
import de.steamwar.bausystem.linkage.LinkageUtils;
import de.steamwar.bausystem.region.loader.PrototypeLoader;
import de.steamwar.bausystem.region.loader.RegionLoader;
import de.steamwar.bausystem.region.loader.Updater;
import de.steamwar.bausystem.worlddata.WorldData;
import de.steamwar.message.Message;
import lombok.Getter;
import org.bukkit.Bukkit;
import org.bukkit.World;
import org.bukkit.event.Listener;
import org.bukkit.plugin.Plugin;
import org.bukkit.plugin.java.JavaPlugin;
import org.bukkit.scheduler.BukkitRunnable;
import org.bukkit.scheduler.BukkitTask;

import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.Consumer;
import java.util.logging.Level;

public class BauSystem extends JavaPlugin implements Listener {

    // This should be treated as final!
    public static Message MESSAGE;

    @Getter
    private static BauSystem instance;

    private World world;

    @Override
    public void onEnable() {
        world = Bukkit.getWorlds().get(0);

        // LOGGER
        fixLogging();

        MESSAGE = new Message("BauSystem", getClassLoader());

        instance = this;
        SWUtils.setBausystem(instance);

        try {
            PrototypeLoader.load();
            RegionLoader.load();
        } catch (SecurityException e) {
            Bukkit.getLogger().log(Level.SEVERE, e.getMessage(), e);
            Bukkit.shutdown();
            return;
        }

        new Updater(PrototypeLoader.file, PrototypeLoader::load);
        new Updater(RegionLoader.file, RegionLoader::load);

        LinkageUtils.link();
        RamUsage.init();

        // This could disable any watchdog stuff. We need to investigate if this is a problem.
        /*
        Thread thread = new Thread(() -> {
            while (true) {
                WatchdogThread.tick();
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            }
        });
        thread.setName("WatchdogThread ticker");
        thread.setDaemon(true);
        thread.start();
         */
    }

    @Override
    public void onDisable() {
        LinkageUtils.unlink();

        WorldData.write();
        Config.getInstance().saveAll();
        TinyProtocol.instance.close();
    }

    private void fixLogging() {
        System.setErr(new PrintStream(new OutputStream() {
            private StringBuilder current = new StringBuilder();

            @Override
            public void write(int b) throws IOException {
                if (b == '\n') {
                    String logging = current.toString();
                    if (logging.contains("SLF4J")) {
                        Bukkit.getLogger().info(logging);
                    } else {
                        Bukkit.getLogger().warning(logging);
                    }
                    current = new StringBuilder();
                } else {
                    current.append((char) b);
                }
            }
        }));
    }

    public static BukkitTask runTaskLater(Plugin plugin, Runnable runnable, long delay) {
        return new BukkitRunnable() {
            private int counter = 1;

            @Override
            public void run() {
                if (TPSFreezeUtils.isFrozen()) return;
                if (counter >= delay) {
                    runnable.run();
                    cancel();
                    return;
                }
                counter++;
            }
        }.runTaskTimer(plugin, 0, 1);
    }

    public static BukkitTask runTaskTimer(Plugin plugin, Runnable runnable, long delay, long period) {
        return new BukkitRunnable() {
            private int counter = 1;
            private boolean first = true;

            @Override
            public void run() {
                if (TPSFreezeUtils.isFrozen()) return;
                if (counter >= (first ? delay : period)) {
                    first = false;
                    runnable.run();
                    counter = 1;
                    return;
                }
                counter++;
            }
        }.runTaskTimer(plugin, 0, 1);
    }

    public static void runTaskTimer(Plugin plugin, Consumer<BukkitTask> consumer, long delay, long period) {
        AtomicReference<BukkitTask> task = new AtomicReference<>();
        task.set(runTaskTimer(plugin, () -> consumer.accept(task.get()), delay, period));
    }
}