Source file `gather_leaf0.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
open! Core
open! Bonsai_private_base.Import
open! Bonsai_private_base
open Incr.Let_syntax

let f
  ~model
  ~static_action
  ~time_source
  ~apply_action
  ~reset
  ~action_name
  ~sexp_of_action
  ~here
  =
  let wrap_leaf inject = Action.static_leaf >>> inject in
  let run ~environment:_ ~fix_envs:_ ~path:_ ~model ~inject =
    annotate ~here Model model;
    (* It's important to create [inject_static] outside of the [let%mapn] so that it
       remains [phys_equal] when the [model] changes. *)
    let inject_static = Lazy_inject.make (wrap_leaf inject) in
    let result =
      let%map model in
      model, inject_static
    in
    Trampoline.return
      (Snapshot.create ~here ~result ~input:Input.static ~lifecycle:None, ())
  in
  let apply_action ~inject ~schedule_event _input model = function
    | Action.Leaf_dynamic _ ->
      Ui_metrics.Counters.observe Bonsai_leaf0_apply_action_got_dynamic_action;
      eprint_s
        [%message "BUG: state_machine0's apply_action was called with a dynamic action"];
      model
    | Leaf_static action ->
      let inject = Lazy_inject.make (wrap_leaf inject) in
      apply_action ~inject ~schedule_event ~time_source model action
  in
  let reset ~inject ~schedule_event model =
    reset ~inject:(wrap_leaf inject) ~schedule_event ~time_source model
  in
  Trampoline.return
    (Computation.T
       { model
       ; input = Meta.Input.unit
       ; action = Action.Type_id.leaf static_action ~action_name ~sexp_of_action
       ; apply_action
       ; reset
       ; run
       ; may_contain =
           May_contain.Unresolved.non_recursive ~path:No ~lifecycle:No ~input:No
       })
;;