Source file after_next_display.ml

open! Core
open Bonsai_web

module Action = struct
  type t =
    | Activate
    | Run_if_active
  [@@deriving sexp_of]
end

module Model = struct
  type t = { activated : bool } [@@deriving sexp, equal, compare]
end

let component effect =
  let open Bonsai.Let_syntax in
  let%sub _, run_action =
    Bonsai.state_machine1
      ~equal:[%equal: Model.t]
      ~sexp_of_action:[%sexp_of: Action.t]
      ~default_model:{ activated = false }
      ~apply_action:(fun context input model action ->
        match input with
        | Active input ->
          (match action with
           | Activate -> { activated = true }
           | Run_if_active ->
             if model.activated
             then (
               Bonsai.Apply_action_context.schedule_event context input;
               { activated = false })
             else model)
        | Inactive ->
          eprint_s
            [%message
              [%here]
                "An action sent to a [state_machine1] has been dropped because its input \
                 was not present. This happens when the [state_machine1] is inactive \
                 when it receives a message."
                (action : Action.t)];
          model)
      effect
  in
  let run_if_active =
    let%map run_action = run_action in
    run_action Run_if_active
  in
  let%sub () = Bonsai.Edge.after_display run_if_active in
  let%arr run_action = run_action in
  run_action Activate
;;