summaryrefslogtreecommitdiff
path: root/src/main/java/com/keuin/kbackupfabric/KBPluginEvents.java
blob: 20e90586bf3137d5b05bd32fbcf6cee9d728ebdc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package com.keuin.kbackupfabric;

import com.keuin.kbackupfabric.backup.BackupFilesystemUtil;
import com.keuin.kbackupfabric.backup.suggestion.BackupNameSuggestionProvider;
import com.keuin.kbackupfabric.config.KBackupConfig;
import com.keuin.kbackupfabric.event.OnPlayerConnect;
import com.keuin.kbackupfabric.metadata.BackupMetadata;
import com.keuin.kbackupfabric.metadata.MetadataHolder;
import com.keuin.kbackupfabric.notification.DistinctNotifiable;
import com.keuin.kbackupfabric.notification.NotificationManager;
import com.keuin.kbackupfabric.ui.KBCommands;
import com.keuin.kbackupfabric.util.DateUtil;
import com.keuin.kbackupfabric.util.PrintUtil;
import net.fabricmc.api.ModInitializer;
import net.fabricmc.fabric.api.command.v1.CommandRegistrationCallback;
import net.fabricmc.fabric.api.event.lifecycle.v1.ServerLifecycleEvents;
import net.minecraft.server.MinecraftServer;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.ObjectInputStream;

import static org.apache.commons.io.FileUtils.forceDelete;

/**
 * This is the Main file of this plugin.
 * It contains all events, including the init event.
 */
public final class KBPluginEvents implements ModInitializer {

    //private static final Logger LOGGER = LogManager.getLogger();

    @Override
    public void onInitialize() {
        PrintUtil.info("Reading config...");
        try {
            KBackupConfig.load();
        } catch (IOException e) {
            PrintUtil.error("Failed to read config file, using default values: " + e + e.getMessage());
        }
        boolean incCow = KBackupConfig.getInstance().getIncbakCow();
        PrintUtil.info("Incremental backup CoW: " + (incCow ? "enabled" : "disabled"));
        PrintUtil.info("Binding events and commands...");
        CommandRegistrationCallback.EVENT.register(KBCommandsRegister::registerCommands);
        OnPlayerConnect.ON_PLAYER_CONNECT.register((connection, player)
                -> NotificationManager.INSTANCE.notifyPlayer(DistinctNotifiable.fromServerPlayerEntity(player)));
        ServerLifecycleEvents.SERVER_STARTED.register(this::onStartServer);
    }

    public void onStartServer(MinecraftServer server) {
        // Initialize player manager reference
        PrintUtil.setPlayerManager(server.getPlayerManager());

        // Initialize backup manager server reference
        KBCommands.setServer(server);

        // Update backup suggestion list
        BackupNameSuggestionProvider.setBackupSaveDirectory(BackupFilesystemUtil.getBackupSaveDirectory(server).getPath());

        // Check if we have just recovered from a previous backup. If so, print message.
        try {
            File levelDirectory = new File(server.getRunDirectory(), server.getLevelName());
            File metadataFile = new File(levelDirectory, BackupMetadata.metadataFileName);
            if (metadataFile.exists()) {
                // Metadata exists. Deserialize.
                BackupMetadata metadata;
                FileInputStream fileInputStream = new FileInputStream(metadataFile);
                ObjectInputStream in = new ObjectInputStream(fileInputStream);
                metadata = (BackupMetadata) in.readObject();
                in.close();
                fileInputStream.close();

                // Print metadata
                MetadataHolder.setMetadata(metadata);
                PrintUtil.info("Restored world from a previous backup:");
                PrintUtil.info("Backup Name: " + metadata.getBackupName());
                PrintUtil.info("Create Time: " + DateUtil.fromEpochMillis(metadata.getBackupTime()));

                // Delete metadata file
                if (!metadataFile.delete()) {
                    System.gc();
                    forceDelete(metadataFile);
                }
            }
        } catch (IOException | ClassNotFoundException ignored) {
        }

        PrintUtil.info("KBackup is a free software. Project home: https://github.com/keuin/KBackup-Fabric");
    }
}