TimeCommand

package com.hypixel.hytale.server.core.modules.time.commands;

import com.hypixel.hytale.codec.validation.Validators;
import com.hypixel.hytale.common.util.FormatUtil;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.protocol.GameMode;
import com.hypixel.hytale.server.core.Message;
import com.hypixel.hytale.server.core.command.system.CommandContext;
import com.hypixel.hytale.server.core.command.system.arguments.system.RequiredArg;
import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
import com.hypixel.hytale.server.core.command.system.basecommands.AbstractCommandCollection;
import com.hypixel.hytale.server.core.command.system.basecommands.AbstractWorldCommand;
import com.hypixel.hytale.server.core.modules.time.WorldTimeResource;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.commands.worldconfig.WorldConfigPauseTimeCommand;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import it.unimi.dsi.fastutil.floats.Float2FloatFunction;
import java.time.LocalDateTime;
import java.time.temporal.ChronoField;
import javax.annotation.Nonnull;

public class TimeCommand extends AbstractWorldCommand {
   public TimeCommand() {
      super("time", "server.commands.time.get.desc");
      this.setPermissionGroup(GameMode.Creative);
      this.addAliases(new String[]{"daytime"});
      this.addUsageVariant(new TimeCommand.SetTimeHourCommand());

      for (TimeCommand.TimeOfDay value : TimeCommand.TimeOfDay.values()) {
         this.addSubCommand(new TimeCommand.SetTimePeriodCommand(value));
      }

      this.addSubCommand(new TimeCommand.TimeSetSubCommand());
      this.addSubCommand(new TimeCommand.TimePauseCommand());
      this.addSubCommand(new TimeCommand.TimeDilationCommand());
   }

   @Override
   public void execute(@Nonnull CommandContext context, @Nonnull World world, @Nonnull Store<EntityStore> store) {
      WorldTimeResource worldTimeResource = store.getResource(WorldTimeResource.getResourceType());
      LocalDateTime gameDateTime = worldTimeResource.getGameDateTime();
      Message pausedMessage = Message.translation(world.getWorldConfig().isGameTimePaused() ? "server.commands.time.paused" : "server.commands.time.unpaused");
      Message message = Message.translation("server.commands.time.info").param("worldName", world.getName()).param("timePaused", pausedMessage);
      context.sendMessage(
         message.param("time", worldTimeResource.getGameTime().toString())
            .param("dayOfWeek", FormatUtil.addNumberSuffix(gameDateTime.get(ChronoField.DAY_OF_WEEK)))
            .param("weekOfMonth", FormatUtil.addNumberSuffix(gameDateTime.get(ChronoField.ALIGNED_WEEK_OF_MONTH)))
            .param("weekOfYear", FormatUtil.addNumberSuffix(gameDateTime.get(ChronoField.ALIGNED_WEEK_OF_YEAR)))
            .param("dayOfYear", FormatUtil.addNumberSuffix(gameDateTime.getDayOfYear()))
            .param("moonPhase", FormatUtil.addNumberSuffix(worldTimeResource.getMoonPhase() + 1))
      );
   }

   private static class SetTimeHourCommand extends AbstractWorldCommand {
      @Nonnull
      private final RequiredArg<Float> timeArg = this.withRequiredArg("time", "server.commands.time.set.timeArg.desc", ArgTypes.FLOAT)
         .addValidator(Validators.range(0.0F, (float)WorldTimeResource.HOURS_PER_DAY));

      public SetTimeHourCommand() {
         super("server.commands.time.set.desc");
         this.setPermissionGroup(null);
      }

      @Override
      protected void execute(@Nonnull CommandContext context, @Nonnull World world, @Nonnull Store<EntityStore> store) {
         float time = this.timeArg.get(context);
         WorldTimeResource worldTimeResource = store.getResource(WorldTimeResource.getResourceType());
         worldTimeResource.setDayTime((double)(time / (float)WorldTimeResource.HOURS_PER_DAY), world, store);
         context.sendMessage(
            Message.translation("server.commands.time.set").param("worldName", world.getName()).param("time", worldTimeResource.getGameTime().toString())
         );
      }
   }

   private static class SetTimePeriodCommand extends AbstractWorldCommand {
      @Nonnull
      private final TimeCommand.TimeOfDay timeOfDay;

      public SetTimePeriodCommand(@Nonnull TimeCommand.TimeOfDay timeOfDay) {
         super(timeOfDay.name(), "server.commands.time.period." + timeOfDay.name().toLowerCase() + ".desc");
         this.setPermissionGroup(null);
         this.timeOfDay = timeOfDay;
         this.addAliases(timeOfDay.aliases);
      }

      @Override
      protected void execute(@Nonnull CommandContext context, @Nonnull World world, @Nonnull Store<EntityStore> store) {
         float daylightHours = (float)WorldTimeResource.HOURS_PER_DAY * 0.6F;
         float periodTime = Math.max(0.0F, (Float)this.timeOfDay.periodFunc.apply(daylightHours));
         WorldTimeResource worldTimeResource = store.getResource(WorldTimeResource.getResourceType());
         worldTimeResource.setDayTime((double)(periodTime / (float)WorldTimeResource.HOURS_PER_DAY), world, store);
         context.sendMessage(
            Message.translation("server.commands.time.set")
               .param("worldName", world.getName())
               .param("time", String.format("%s (%s)", worldTimeResource.getGameTime().toString(), this.timeOfDay.name()))
         );
      }
   }

   private static class TimeDilationCommand extends AbstractWorldCommand {
      private static final float TIME_DILATION_MIN = 0.01F;
      private static final float TIME_DILATION_MAX = 4.0F;
      @Nonnull
      private final RequiredArg<Float> timeDilationArg = this.withRequiredArg("timeDilation", "server.commands.time.dilation.timeDilation.desc", ArgTypes.FLOAT)
         .addValidator(Validators.greaterThan(0.01F))
         .addValidator(Validators.max(4.0F));

      public TimeDilationCommand() {
         super("dilation", "server.commands.time.dilation.desc");
         this.setPermissionGroup(null);
      }

      @Override
      protected void execute(@Nonnull CommandContext context, @Nonnull World world, @Nonnull Store<EntityStore> store) {
         float timeDilation = this.timeDilationArg.get(context);
         World.setTimeDilation(timeDilation, store);
         context.sendMessage(Message.translation("server.commands.time.dilation.set.success").param("timeDilation", timeDilation));
      }
   }

   public static enum TimeOfDay {
      Dawn(hoursOfDaylight -> ((float)WorldTimeResource.HOURS_PER_DAY - hoursOfDaylight) / 2.0F, "day", "morning"),
      Midday(hoursOfDaylight -> (float)WorldTimeResource.HOURS_PER_DAY / 2.0F, "noon"),
      Dusk(hoursOfDaylight -> ((float)WorldTimeResource.HOURS_PER_DAY - hoursOfDaylight) / 2.0F + hoursOfDaylight, "night"),
      Midnight(hoursOfDaylight -> 0.0F);

      @Nonnull
      private final Float2FloatFunction periodFunc;
      private final String[] aliases;

      private TimeOfDay(@Nonnull final Float2FloatFunction periodFunc, String... aliases) {
         this.periodFunc = periodFunc;
         this.aliases = aliases;
      }
   }

   private static class TimePauseCommand extends AbstractWorldCommand {
      public TimePauseCommand() {
         super("pause", "server.commands.pausetime.desc");
         this.setPermissionGroup(null);
         this.addAliases(new String[]{"stop"});
      }

      @Override
      protected void execute(@Nonnull CommandContext context, @Nonnull World world, @Nonnull Store<EntityStore> store) {
         WorldConfigPauseTimeCommand.pauseTime(context.sender(), world, store);
      }
   }

   private static class TimeSetSubCommand extends AbstractCommandCollection {
      public TimeSetSubCommand() {
         super("set", "server.commands.time.set.desc");
         this.setPermissionGroup(null);
         this.addUsageVariant(new TimeCommand.SetTimeHourCommand());

         for (TimeCommand.TimeOfDay value : TimeCommand.TimeOfDay.values()) {
            this.addSubCommand(new TimeCommand.SetTimePeriodCommand(value));
         }
      }
   }
}