implements fabric-debug-api-v1 api

Author: wdog5

tenet-server/minecraft-patches/sources/net/minecraft/world/entity/Entity.java.patch

@@ -1,11 +1,12 @@
 --- a/net/minecraft/world/entity/Entity.java
 +++ b/net/minecraft/world/entity/Entity.java
-@@ -29,6 +_,8 @@
+@@ -29,6 +_,9 @@
  import java.util.function.Predicate;
  import java.util.stream.Stream;
  import javax.annotation.Nullable;
 +
 +import net.fabricmc.fabric.api.entity.event.v1.ServerEntityLevelChangeEvents;
++import net.fabricmc.fabric.impl.debug.EntityDebugSubscriptionRegistryImpl;
  import net.minecraft.BlockUtil;
  import net.minecraft.CrashReport;
  import net.minecraft.CrashReportCategory;
@@ -22,3 +23,11 @@
          } else {
              return null;
          }
+@@ -5320,6 +_,7 @@
+ 
+     @Override
+     public void registerDebugValues(ServerLevel level, DebugValueSource.Registration registrar) {
++        EntityDebugSubscriptionRegistryImpl.addDebugValues(this, registrar);
+     }
+ 
+     record EntityPathElement(Entity entity) implements ProblemReporter.PathElement {

tenet-server/src/main/java/net/fabricmc/fabric/api/debug/v1/DebugValueFactory.java

@@ -0,0 +1,27 @@
+/*
+ * Copyright (c) 2016, 2017, 2018, 2019 FabricMC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package net.fabricmc.fabric.api.debug.v1;
+
+/// A factory that creates a debug value of type [T] using data of type [D].
+///
+/// @param <D> the data passed for construction
+/// (e.g. [net.minecraft.world.entity.Entity]).
+/// @param <T> the debug value being constructed.
+@FunctionalInterface
+public interface DebugValueFactory<D, T> {
+	T create(D data);
+}

tenet-server/src/main/java/net/fabricmc/fabric/api/debug/v1/EntityDebugSubscriptionRegistry.java

@@ -0,0 +1,77 @@
+/*
+ * Copyright (c) 2016, 2017, 2018, 2019 FabricMC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package net.fabricmc.fabric.api.debug.v1;
+
+import java.util.Objects;
+import java.util.function.Predicate;
+
+import net.minecraft.util.debug.DebugSubscription;
+import net.minecraft.world.entity.Entity;
+
+import net.fabricmc.fabric.impl.debug.EntityDebugSubscriptionRegistryImpl;
+
+/// A server-side registry for [debug subscriptions][DebugSubscription] specific
+/// to entities or properties of entities.
+public final class EntityDebugSubscriptionRegistry {
+	/// Registers a [DebugSubscription] based on a given [Entity] and
+	/// [Predicate].
+	///
+	/// @param <T> the inner type of the [DebugSubscription].
+	/// @param <E> the type of [Entity] to check against.
+	/// @param debugSubscription the [DebugSubscription].
+	/// @param shouldSubscribe whether an [Entity] should subscribe to this
+	/// [DebugSubscription].
+	/// @param valueFactory the factory for the value of type [T].
+	public static <T, E extends Entity> void register(
+			DebugSubscription<T> debugSubscription,
+			Predicate<Entity> shouldSubscribe,
+			DebugValueFactory<E, T> valueFactory
+	) {
+		Objects.requireNonNull(debugSubscription);
+		EntityDebugSubscriptionRegistryImpl.register(
+				debugSubscription,
+				shouldSubscribe,
+				valueFactory
+		);
+	}
+
+	/// Registers a [DebugSubscription] based on a given [Entity] and
+	/// [Predicate] if `isEnabledFlag` is `true`.
+	///
+	/// @param <T> the inner type of the [DebugSubscription].
+	/// @param <E> the type of [Entity] to check against.
+	/// @param debugSubscription the [DebugSubscription].
+	/// @param shouldSubscribe whether an [Entity] should subscribe to this
+	/// [DebugSubscription].
+	/// @param valueFactory the factory for the value of type [T].
+	/// @param isEnabledFlag the flag determining whether to register this
+	/// [DebugSubscription].
+	public static <T, E extends Entity> void register(
+			DebugSubscription<T> debugSubscription,
+			Predicate<Entity> shouldSubscribe,
+			DebugValueFactory<E, T> valueFactory,
+			boolean isEnabledFlag
+	) {
+		if (isEnabledFlag) {
+			register(
+					debugSubscription,
+					shouldSubscribe,
+					valueFactory
+			);
+		}
+	}
+}

tenet-server/src/main/java/net/fabricmc/fabric/impl/debug/EntityDebugSubscriptionRegistryImpl.java

@@ -0,0 +1,84 @@
+/*
+ * Copyright (c) 2016, 2017, 2018, 2019 FabricMC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package net.fabricmc.fabric.impl.debug;
+
+import java.util.HashSet;
+import java.util.Objects;
+import java.util.Set;
+import java.util.function.Predicate;
+
+import net.minecraft.util.debug.DebugSubscription;
+import net.minecraft.util.debug.DebugValueSource;
+import net.minecraft.world.entity.Entity;
+
+import net.fabricmc.fabric.api.debug.v1.DebugValueFactory;
+
+public final class EntityDebugSubscriptionRegistryImpl {
+	public static final Set<Entry> ENTITY_DEBUG_SUBSCRIPTIONS =
+			new HashSet<>();
+
+	public static <T, E extends Entity> void register(
+			DebugSubscription<T> debugSubscription,
+			Predicate<Entity> shouldSubscribe,
+			DebugValueFactory<E, T> valueFactory
+	) {
+		//noinspection unchecked // Casts to super-type
+		ENTITY_DEBUG_SUBSCRIPTIONS.add(new Entry(
+				(DebugSubscription<Object>) debugSubscription,
+				shouldSubscribe,
+				(DebugValueFactory<Entity, T>) valueFactory
+		));
+	}
+
+	public static void addDebugValues(
+			Object entity,
+			DebugValueSource.Registration registration
+	) {
+		for (Entry entry : ENTITY_DEBUG_SUBSCRIPTIONS) {
+			if (entry.shouldSubscribe().test((Entity) entity)) {
+				registration.register(
+						entry.debugSubscription(),
+						() -> entry.valueFactory().create((Entity) entity)
+				);
+			}
+		}
+	}
+
+	public record Entry(
+			DebugSubscription<Object> debugSubscription,
+			Predicate<Entity> shouldSubscribe,
+			DebugValueFactory<Entity, ?> valueFactory
+	) {
+		// Ensure DebugSubscriptions with different values don't both
+		// get into the Set, causing undesirable behavior
+
+		@Override
+		public boolean equals(Object o) {
+			if (o == null || getClass() != o.getClass()) return false;
+			Entry entry = (Entry) o;
+			return Objects.equals(
+					debugSubscription,
+					entry.debugSubscription
+			);
+		}
+
+		@Override
+		public int hashCode() {
+			return Objects.hash(debugSubscription);
+		}
+	}
+}