Add visible to SetCustomNameEntityModifier

Author: ErrorCraft

src/main/java/errorcraft/entitymodifiers/entity/modifier/modifiers/SetCustomNameEntityModifier.java

@@ -10,12 +10,16 @@
 import net.minecraft.loot.context.LootContext;
 import net.minecraft.loot.function.SetNameLootFunction;
 import net.minecraft.text.Text;
+import net.minecraft.util.JsonHelper;
+import org.jetbrains.annotations.Nullable;
 
 public class SetCustomNameEntityModifier implements EntityModifier {
 	private final Text customName;
+	private final @Nullable Boolean visible;
 
-	public SetCustomNameEntityModifier(Text customName) {
+	public SetCustomNameEntityModifier(Text customName, @Nullable Boolean visible) {
 		this.customName = customName;
+		this.visible = visible;
 	}
 
 	@Override
@@ -28,6 +32,9 @@ public Entity apply(Entity entity, LootContext lootContext) {
 		if (this.customName != null) {
 			entity.setCustomName(SetNameLootFunction.applySourceEntity(lootContext, LootContext.EntityTarget.THIS).apply(this.customName));
 		}
+		if (this.visible != null) {
+			entity.setCustomNameVisible(this.visible);
+		}
 		return entity;
 	}
 
@@ -42,7 +49,8 @@ public void toJson(JsonObject json, SetCustomNameEntityModifier object, JsonSeri
 		@Override
 		public SetCustomNameEntityModifier fromJson(JsonObject json, JsonDeserializationContext context) {
 			Text customName = Text.Serializer.fromJson(json.get("name"));
-			return new SetCustomNameEntityModifier(customName);
+			@Nullable Boolean visible = json.has("visible") ? JsonHelper.getBoolean(json, "visible") : null;
+			return new SetCustomNameEntityModifier(customName, visible);
 		}
 	}
 }