From 48e4bb87b558634e38b71b32a3c133fa30f92544 Mon Sep 17 00:00:00 2001
From: Mathieu Rochette <mathieu@texthtml.net>
Date: Sun, 9 Feb 2025 19:38:30 +0000
Subject: [PATCH] Add Reaper::forget()

---
 src/Reaper.php | 37 +++++++++++++++++++++++++++++++++----
 1 file changed, 33 insertions(+), 4 deletions(-)

diff --git a/src/Reaper.php b/src/Reaper.php
index e70e6d3..3fda323 100644
--- a/src/Reaper.php
+++ b/src/Reaper.php
@@ -7,7 +7,30 @@ final class Reaper
     /** @var \WeakMap<object,list<self>> */
     private static \WeakMap $dd;
 
-    private function __construct(private readonly \Closure $callback) {}
+    private bool $active = true;
+
+    private function __construct(
+        /** @var \Closure(): void */
+        private readonly \Closure $callback,
+    ) {}
+
+    /**
+     * Detach the reaper from its target
+     *
+     * ```php
+     * $a = (object) [];
+     *
+     * $reaper = Reaper::watch($a, function () { echo "Good Bye"; });
+     * $reaper->forget();
+     *
+     * unset($b);
+     * // prints nothing
+     * ```
+     */
+    public function forget(): void
+    {
+        $this->active = false;
+    }
 
     /**
      * Attach a callback to an object to be called when that object is destroyed.
@@ -52,16 +75,22 @@ private function __construct(private readonly \Closure $callback) {}
      *
      * @param callable(): void $callback
      */
-    public static function watch(object $subject, callable $callback): void {
+    public static function watch(object $subject, callable $callback): self
+    {
         self::$dd ??= new \WeakMap();
 
         self::$dd[$subject] ??= [];
 
-        self::$dd[$subject][] = new self(\Closure::fromCallable($callback));
+        return self::$dd[$subject][] = new self(\Closure::fromCallable($callback));
     }
 
     /** @nodoc */
-    public function __destruct() {
+    public function __destruct()
+    {
+        if (!$this->active) {
+            return;
+        }
+
         ($this->callback)();
     }
 }