package com.keuin.kbackupfabric;

import com.keuin.kbackupfabric.backup.suggestion.BackupNameSuggestionProvider;
import com.keuin.kbackupfabric.ui.KBCommands;
import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.StringArgumentType;
import me.lucko.fabric.api.permissions.v0.Permissions;
import net.minecraft.server.command.CommandManager;
import net.minecraft.server.command.ServerCommandSource;

public final class KBCommandsRegister {

    public static final int DEFAULT_REQUIRED_LEVEL = 2;

    // First make method to register
    public static void registerCommands(CommandDispatcher<ServerCommandSource> dispatcher, boolean dedicated) {

        // register /kb and /kb help for help menu
        dispatcher.register(CommandManager.literal("kb")
                .requires(Permissions.require("kb.root", true))
                .executes(KBCommands::kb));
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("help")
                        .requires(Permissions.require("kb.help", true))
                        .executes(KBCommands::help)));

        // register /kb list for showing the backup list. OP is required.
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("list")
                        .requires(Permissions.require("kb.list", DEFAULT_REQUIRED_LEVEL))
                        .executes(KBCommands::list)));

        // register /kb backup [name] for performing backup. OP is required.
        dispatcher.register(CommandManager.literal("kb").then(CommandManager.literal("backup").then(
                        CommandManager.argument("backupName", StringArgumentType.greedyString())
                                .requires(Permissions.require("kb.backup", DEFAULT_REQUIRED_LEVEL))
                                .executes(KBCommands::primitiveBackup)
                ).requires(Permissions.require("kb.backup", DEFAULT_REQUIRED_LEVEL))
                .executes(KBCommands::primitiveBackupWithDefaultName)));

        // register /kb incbak [name] for performing incremental backup. OP is required.
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("incbak")
                        .then(CommandManager.argument("backupName", StringArgumentType.greedyString())
                                .requires(Permissions.require("kb.incbak", DEFAULT_REQUIRED_LEVEL))
                                .executes(KBCommands::incrementalBackup))
                        .requires(Permissions.require("kb.incbak", DEFAULT_REQUIRED_LEVEL))
                        .executes(KBCommands::incrementalBackupWithDefaultName)
                ));

        // register /kb restore <name> for performing restore. OP is required.
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("restore")
                        .then(CommandManager.argument("backupName", StringArgumentType.greedyString())
                                .suggests(BackupNameSuggestionProvider.getProvider())
                                .requires(Permissions.require("kb.restore", DEFAULT_REQUIRED_LEVEL))
                                .executes(KBCommands::restore))
                        .requires(Permissions.require("kb.list", DEFAULT_REQUIRED_LEVEL))
                        .executes(KBCommands::list)));

        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("cow-info")
                        .executes(KBCommands::cowInfo)));

        // register /kb delete [name] for deleting an existing backup. OP is required.
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("delete")
                        .then(CommandManager.argument("backupName", StringArgumentType.greedyString())
                                .suggests(BackupNameSuggestionProvider.getProvider())
                                .requires(Permissions.require("kb.delete", DEFAULT_REQUIRED_LEVEL))
                                .executes(KBCommands::delete))
                        .requires(Permissions.require("kb.delete", DEFAULT_REQUIRED_LEVEL))));

        // register /kb confirm for confirming the execution. OP is required.
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("confirm")
                        .requires(Permissions.require("kb.confirm", DEFAULT_REQUIRED_LEVEL))
                        .executes(KBCommands::confirm)));

        // register /kb cancel for cancelling the execution to be confirmed. OP is required.
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("cancel")
                        .requires(Permissions.require("kb.cancel", DEFAULT_REQUIRED_LEVEL))
                        .executes(KBCommands::cancel)));

        // register /kb prev for showing the latest backup. OP is required.
        dispatcher.register(CommandManager.literal("kb")
                .then(CommandManager.literal("prev")
                        .requires(Permissions.require("kb.prev", DEFAULT_REQUIRED_LEVEL))
                        .executes(KBCommands::prev)));

    }
}