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.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.java.JavaPlugin;

import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.attribute.BasicFileAttributes;
import java.nio.file.attribute.FileTime;
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);
        fixBauSystem();

        // 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();

        // 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);
                }
            }
        }));
    }

    private void fixBauSystem() {
        if (!new File(world.getWorldFolder(), "sections.yml").exists()) {
            try {
                Path path = new File(world.getWorldFolder(), "region").toPath();
                BasicFileAttributes attributes = Files.readAttributes(path, BasicFileAttributes.class);
                FileTime creationTime = attributes.creationTime();
                long millis = creationTime.toMillis();
                if (millis < 1611081960) {
                    createLink("sections3.yml", "sections.yml");
                } else {
                    createLink("sections4.yml", "sections.yml");
                }
            } catch (IOException e) {
                Bukkit.getLogger().log(Level.SEVERE, e.getMessage(), e);
                Bukkit.shutdown();
                return;
            }
        } else {
            return;
        }
        int number = -1;
        try {
            String string = new File(world.getWorldFolder(), "sections.yml").getCanonicalPath();
            if (string.endsWith("/sections2.yml")) number = 2;
            if (string.endsWith("/sections3.yml")) number = 3;
            if (string.endsWith("/sections4.yml")) number = 4;
            Bukkit.getLogger().log(Level.INFO, "SectionFile: " + string);
        } catch (IOException e) {
            Bukkit.getLogger().log(Level.SEVERE, e.getMessage(), e);
            Bukkit.shutdown();
            return;
        }
        if (number == -1) {
            return;
        }
        if (!new File(world.getWorldFolder(),"prototypes.yapion").exists()) {
            createLink("prototypes" + number + ".yapion", "prototypes.yapion");
        }
        if (!new File(world.getWorldFolder(),"regions.yapion").exists()) {
            createLink("regions" + number + ".yapion", "regions.yapion");
        }
    }

    private void createLink(String source, String destination) {
        try {
            Bukkit.getLogger().log(Level.INFO, "Executing: ln -s /home/minecraft/server/Bau15/{0} {1}", new String[]{source, destination});
            ProcessBuilder processBuilder = new ProcessBuilder("ln", "-s", "/home/minecraft/server/Bau15/" + source, destination);
            processBuilder.directory(world.getWorldFolder());
            processBuilder.inheritIO();

            Process process = processBuilder.start();
            process.waitFor();
        } catch (IOException | InterruptedException e) {
            Thread.currentThread().interrupt();
            Bukkit.shutdown();
        }
    }
}