ScheduleCommand.kt updated to 1.15.2 forge.

Signed-off-by: Pavel Erokhin (MairwunNx) <MairwunNx@gmail.com>

Author: MairwunNx

src/main/kotlin/com/mairwunnx/projectessentials/core/vanilla/commands/ScheduleCommand.kt

@@ -4,32 +4,50 @@
  * added check on permissions and aliases register.
  */
 
+@file:Suppress("FunctionName")
+
 package com.mairwunnx.projectessentials.core.vanilla.commands
 
 import com.mairwunnx.projectessentials.core.EntryPoint
 import com.mairwunnx.projectessentials.core.helpers.PERMISSION_LEVEL
 import com.mairwunnx.projectessentials.core.vanilla.utils.NativeCommandUtils
 import com.mojang.brigadier.CommandDispatcher
 import com.mojang.brigadier.arguments.IntegerArgumentType
+import com.mojang.brigadier.arguments.StringArgumentType
 import com.mojang.brigadier.context.CommandContext
 import com.mojang.brigadier.exceptions.CommandSyntaxException
+import com.mojang.brigadier.exceptions.DynamicCommandExceptionType
 import com.mojang.brigadier.exceptions.SimpleCommandExceptionType
+import com.mojang.brigadier.suggestion.SuggestionProvider
+import com.mojang.brigadier.suggestion.SuggestionsBuilder
 import com.mojang.datafixers.util.Either
 import net.minecraft.command.*
 import net.minecraft.command.arguments.FunctionArgument
 import net.minecraft.command.arguments.TimeArgument
+import net.minecraft.command.impl.FunctionCommand
 import net.minecraft.tags.Tag
 import net.minecraft.util.text.Style
 import net.minecraft.util.text.TranslationTextComponent
 import net.minecraft.util.text.event.HoverEvent
 import org.apache.logging.log4j.LogManager
+import java.util.function.Function
 
 internal object ScheduleCommand {
     private val logger = LogManager.getLogger()
 
-    private val field_218913_a = SimpleCommandExceptionType(
-        TranslationTextComponent("commands.schedule.same_tick")
+    private val field_218913_a =
+        SimpleCommandExceptionType(TranslationTextComponent("commands.schedule.same_tick"))
+    private val field_229811_b_ = DynamicCommandExceptionType(
+        Function { p_229818_0_: Any? ->
+            TranslationTextComponent("commands.schedule.cleared.failure", p_229818_0_)
+        }
     )
+    private val field_229812_c_ =
+        SuggestionProvider { p_229814_0_: CommandContext<CommandSource>, p_229814_1_: SuggestionsBuilder ->
+            ISuggestionProvider.suggest(
+                p_229814_0_.source.world.worldInfo.scheduledEvents.func_227574_a_(), p_229814_1_
+            )
+        }
 
     fun register(p_218909_0_: CommandDispatcher<CommandSource>) {
         logger.info("Replacing `/schedule` vanilla command")
@@ -43,15 +61,45 @@ internal object ScheduleCommand {
                     ).suggests(FunctionCommand.FUNCTION_SUGGESTER).then(
                         Commands.argument(
                             "time", TimeArgument.func_218091_a()
-                        ).executes { p_218911_0_: CommandContext<CommandSource> ->
-                            func_218908_a(
-                                p_218911_0_.source,
-                                FunctionArgument.func_218110_b(p_218911_0_, "function"),
-                                IntegerArgumentType.getInteger(p_218911_0_, "time")
+                        ).executes { p_229823_0_ ->
+                            func_229816_a_(
+                                p_229823_0_.source,
+                                FunctionArgument.func_218110_b(p_229823_0_, "function"),
+                                IntegerArgumentType.getInteger(p_229823_0_, "time"),
+                                true
                             )
-                        }
+                        }.then(
+                            Commands.literal("append").executes { p_229822_0_ ->
+                                func_229816_a_(
+                                    p_229822_0_.source,
+                                    FunctionArgument.func_218110_b(p_229822_0_, "function"),
+                                    IntegerArgumentType.getInteger(p_229822_0_, "time"),
+                                    false
+                                )
+                            }
+                        ).then(
+                            Commands.literal("replace").executes { p_229821_0_ ->
+                                func_229816_a_(
+                                    p_229821_0_.source,
+                                    FunctionArgument.func_218110_b(p_229821_0_, "function"),
+                                    IntegerArgumentType.getInteger(p_229821_0_, "time"),
+                                    true
+                                )
+                            }
+                        )
                     )
                 )
+            ).then(
+                Commands.literal("clear").then(
+                    Commands.argument(
+                        "function", StringArgumentType.greedyString()
+                    ).suggests(field_229812_c_).executes { p_229813_0_ ->
+                        func_229817_a_(
+                            p_229813_0_.source,
+                            StringArgumentType.getString(p_229813_0_, "function")
+                        )
+                    }
+                )
             )
         )
     }
@@ -84,52 +132,72 @@ internal object ScheduleCommand {
         }
     }
 
-    @Suppress("FunctionName")
     @Throws(CommandSyntaxException::class)
-    private fun func_218908_a(
-        p_218908_0_: CommandSource,
-        p_218908_1_: Either<FunctionObject, Tag<FunctionObject>>,
-        p_218908_2_: Int
+    private fun func_229816_a_(
+        p_229816_0_: CommandSource,
+        p_229816_1_: Either<FunctionObject, Tag<FunctionObject>>,
+        p_229816_2_: Int,
+        p_229816_3_: Boolean
     ): Int {
-        checkPermissions(p_218908_0_)
+        checkPermissions(p_229816_0_)
 
-        return if (p_218908_2_ == 0) {
+        return if (p_229816_2_ == 0) {
             throw field_218913_a.create()
         } else {
-            val i = p_218908_0_.world.gameTime + p_218908_2_.toLong()
-            p_218908_1_.ifLeft { p_218910_4_ ->
-                val resourcelocation = p_218910_4_.id
-                p_218908_0_.world.worldInfo.scheduledEvents.scheduleReplaceDuplicate(
-                    resourcelocation.toString(),
-                    i,
-                    TimedFunction(resourcelocation)
-                )
-                p_218908_0_.sendFeedback(
+            val i = p_229816_0_.world.gameTime + p_229816_2_.toLong()
+            val timercallbackmanager =
+                p_229816_0_.world.worldInfo.scheduledEvents
+
+            p_229816_1_.ifLeft { p_229820_6_ ->
+                val resourcelocation = p_229820_6_.id
+                val s = resourcelocation.toString()
+                if (p_229816_3_) {
+                    timercallbackmanager.func_227575_a_(s)
+                }
+                timercallbackmanager.func_227576_a_(s, i, TimedFunction(resourcelocation))
+                p_229816_0_.sendFeedback(
                     TranslationTextComponent(
                         "commands.schedule.created.function",
                         resourcelocation,
-                        p_218908_2_,
+                        p_229816_2_,
                         i
                     ), true
                 )
-            }.ifRight { p_218907_4_ ->
-                val resourcelocation = p_218907_4_.id
-                p_218908_0_.world.worldInfo.scheduledEvents
-                    .scheduleReplaceDuplicate(
-                        "#$resourcelocation",
-                        i,
-                        TimedFunctionTag(resourcelocation)
-                    )
-                p_218908_0_.sendFeedback(
+            }.ifRight { p_229819_6_ ->
+                val resourcelocation = p_229819_6_.id
+                val s = "#$resourcelocation"
+                if (p_229816_3_) {
+                    timercallbackmanager.func_227575_a_(s)
+                }
+                timercallbackmanager.func_227576_a_(s, i, TimedFunctionTag(resourcelocation))
+                p_229816_0_.sendFeedback(
                     TranslationTextComponent(
                         "commands.schedule.created.tag",
                         resourcelocation,
-                        p_218908_2_,
+                        p_229816_2_,
                         i
                     ), true
                 )
             }
             Math.floorMod(i, 2147483647L).toInt()
         }
     }
+
+
+    @Throws(CommandSyntaxException::class)
+    private fun func_229817_a_(p_229817_0_: CommandSource, p_229817_1_: String): Int {
+        val i = p_229817_0_.world.worldInfo.scheduledEvents.func_227575_a_(p_229817_1_)
+        return if (i == 0) {
+            throw field_229811_b_.create(p_229817_1_)
+        } else {
+            p_229817_0_.sendFeedback(
+                TranslationTextComponent(
+                    "commands.schedule.cleared.success",
+                    i,
+                    p_229817_1_
+                ), true
+            )
+            i
+        }
+    }
 }