util: WithPos

aya-prover · Jan 4, 2024 · e18a6f8 · e18a6f8
1 parent e14478c
commit e18a6f8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tools/src/main/java/org/aya/util/error/WithPos.java b/tools/src/main/java/org/aya/util/error/WithPos.java
@@ -7,8 +7,9 @@
 
 import java.util.function.Function;
 
-public record WithPos<T>(@NotNull SourcePos sourcePos, T data) {
+public record WithPos<T>(@NotNull SourcePos sourcePos, T data) implements SourceNode {
   public <U> @Contract("_->new") WithPos<U> map(@NotNull Function<T, U> mapper) {
-    return new WithPos<>(sourcePos, mapper.apply(data));
+    var result = mapper.apply(data);
+    return result == data ? ((WithPos<U>) this) : new WithPos<>(sourcePos, mapper.apply(data));
   }
 }