refactor(Modifier): use the new API internally

RedPRL · Sep 21, 2023 · e1c84aa · e1c84aa
1 parent e94564b
commit e1c84aa
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/Modifier.ml b/src/Modifier.ml
@@ -19,12 +19,6 @@ struct
     | Shadow : {context : context option; path : Trie.bwd_path; former : data * tag; latter : data * tag} -> (data * tag) Effect.t
     | Hook : {context : context option; prefix : Trie.bwd_path; hook : hook; input : (data, tag) Trie.t} -> (data, tag) Trie.t Effect.t
 
-  let () =
-    Printexc.register_printer @@
-    function
-    | Effect.Unhandled (NotFound _ | Shadow _ | Hook _) -> Some "Unhandled yuujinchou effect; use Yuujinchou.Modifier.run"
-    | _ -> None
-
   module type Perform = Perform with module Param := Param
 
   module Perform : Perform =
@@ -90,4 +84,6 @@ struct
     | Effect.Unhandled (Shadow {context; path; former; latter}) -> f (`Shadow (context, path, former, latter))
     | Effect.Unhandled (Hook {context; prefix; hook; input}) -> f (`Hook (context, prefix, hook, input))
     | _ -> None
+
+  let () = register_printer @@ fun _ -> Some "Unhandled yuujinchou effect; use Yuujinchou.Modifier.run"
 end