RangeValidator.java

package com.hypixel.hytale.codec.validation.validator;

import com.hypixel.hytale.codec.schema.SchemaContext;
import com.hypixel.hytale.codec.schema.config.IntegerSchema;
import com.hypixel.hytale.codec.schema.config.NumberSchema;
import com.hypixel.hytale.codec.schema.config.Schema;
import com.hypixel.hytale.codec.validation.ValidationResults;
import com.hypixel.hytale.codec.validation.Validator;
import com.hypixel.hytale.logger.HytaleLogger;
import java.util.logging.Level;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class RangeValidator<T extends Comparable<T>> implements Validator<T> {
   private static final HytaleLogger LOGGER = HytaleLogger.forEnclosingClass();
   private final T min;
   private final T max;
   private final boolean inclusive;

   public RangeValidator(T min, T max, boolean inclusive) {
      this.min = min;
      this.max = max;
      this.inclusive = inclusive;
   }

   public void accept(@Nullable T t, @Nonnull ValidationResults results) {
      if (t != null) {
         if (this.min != null) {
            int compare = t.compareTo(this.min);
            if (this.inclusive) {
               if (compare < 0) {
                  results.fail("Must be greater than or equal to " + String.valueOf(this.min));
               }
            } else if (compare < 0 || compare == 0) {
               results.fail("Must be greater than " + String.valueOf(this.min));
            }
         }

         if (this.max != null) {
            int compare = t.compareTo(this.max);
            if (this.inclusive) {
               if (compare > 0) {
                  results.fail("Must be less than or equal to " + String.valueOf(this.max));
               }
            } else if (compare > 0 || compare == 0) {
               results.fail("Must be less than " + String.valueOf(this.max));
            }
         }

      }
   }

   public void updateSchema(SchemaContext context, Schema target) {
      if (this.min != null && !(this.min instanceof Number)) {
         LOGGER.at(Level.WARNING).log("Can't handle: min is not a Number: %s, %s", this.min, this.min.getClass());
      } else if (this.max != null && !(this.max instanceof Number)) {
         LOGGER.at(Level.WARNING).log("Can't handle: max is not a Number: %s, %s", this.max, this.max.getClass());
      } else if (!(target instanceof NumberSchema) && !(target instanceof IntegerSchema)) {
         boolean failed = true;
         if (target.getAnyOf() != null) {
            for(Schema schema : target.getAnyOf()) {
               if (schema instanceof NumberSchema || schema instanceof IntegerSchema) {
                  this.updateSchema(schema);
                  failed = false;
               }
            }
         }

         if (failed) {
            LOGGER.at(Level.WARNING).log("Can't handle: %s as a range: %s", target.getHytale().getType(), target);
         }

      } else {
         this.updateSchema(target);
      }
   }

   private void updateSchema(Schema target) {
      if (target instanceof IntegerSchema i) {
         if (this.min != null) {
            Number v = (Number)this.min;
            if (this.inclusive) {
               i.setMinimum(v.intValue());
            } else {
               i.setExclusiveMinimum(v.intValue());
            }
         }

         if (this.max != null) {
            Number v = (Number)this.max;
            if (this.inclusive) {
               i.setMaximum(v.intValue());
            } else {
               i.setExclusiveMaximum(v.intValue());
            }
         }
      } else {
         NumberSchema i = (NumberSchema)target;
         if (this.min != null) {
            Number v = (Number)this.min;
            if (this.inclusive) {
               i.setMinimum(v.doubleValue());
            } else {
               i.setExclusiveMinimum(v.doubleValue());
            }
         }

         if (this.max != null) {
            Number v = (Number)this.max;
            if (this.inclusive) {
               i.setMaximum(v.doubleValue());
            } else {
               i.setExclusiveMaximum(v.doubleValue());
            }
         }
      }

   }
}