From 41ab37c06ab05eaf781bbe47c9e49e7230dd04fe Mon Sep 17 00:00:00 2001
From: glennsl <glenn@slotte.net>
Date: Sun, 13 Mar 2022 13:36:42 +0100
Subject: [PATCH] dom: add create_portal

---
 lib/dom.mli     |  9 +++++++++
 test/test_ml.ml | 30 ++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+)

diff --git a/lib/dom.mli b/lib/dom.mli
index d8b4fa13..3763cfda 100755
--- a/lib/dom.mli
+++ b/lib/dom.mli
@@ -45,6 +45,15 @@ val render_to_element : id:string -> Core.element -> unit
               ^ " found in the HTML."))
       | Some element -> render react_element element]
 
+val create_portal : Core.element -> dom_element -> Core.element
+  [@@js.custom
+    val create_portal_internal :
+      Imports.react_dom -> Core.element -> dom_element -> Core.element
+      [@@js.call "createPortal"]
+
+    let create_portal element dom_element =
+      create_portal_internal Imports.react_dom element dom_element]
+
 type dom_ref = private Ojs.t
 
 module Ref : sig
diff --git a/test/test_ml.ml b/test/test_ml.ml
index 326f082a..fb1051fd 100644
--- a/test/test_ml.ml
+++ b/test/test_ml.ml
@@ -27,6 +27,21 @@ let withContainer f =
   Dom.removeChild doc##.body container;
   result
 
+let withContainers2 f =
+  let container1 = Html.createDiv doc in
+  Dom.appendChild doc##.body container1;
+  let container2 = Html.createDiv doc in
+  Dom.appendChild doc##.body container2;
+
+  let result = f container1 container2 in
+
+  ignore (React.Dom.unmount_component_at_node container1);
+  Dom.removeChild doc##.body container1;
+  ignore (React.Dom.unmount_component_at_node container2);
+  Dom.removeChild doc##.body container2;
+
+  result
+
 let testDom () =
   doc##.title := Js.string "Testing";
   let p = Html.createP doc in
@@ -43,6 +58,20 @@ let testReact () =
             (Html.element c));
       assert_equal c##.textContent (Js.Opt.return (Js.string "Hello world!")))
 
+let testPortal () =
+  withContainers2 (fun c1 c2 ->
+      act (fun () ->
+          React.Dom.render
+            (div [||]
+               [ string "Hello"
+               ; React.Dom.create_portal
+                   (div [||] [ string "world!" ])
+                   (Html.element c2)
+               ])
+            (Html.element c1));
+      assert_equal c1##.innerHTML (Js.string "<div>Hello</div>");
+      assert_equal c2##.innerHTML (Js.string "<div>world!</div>"))
+
 let testKeys () =
   withContainer (fun c ->
       act (fun () ->
@@ -766,6 +795,7 @@ let basic =
   "basic"
   >::: [ "testDom" >:: testDom
        ; "testReact" >:: testReact
+       ; "testPortal" >:: testPortal
        ; "testKey" >:: testKeys
        ; "testOptionalPropsUppercase" >:: testOptionalPropsUppercase
        ; "testOptionalPropsLowercase" >:: testOptionalPropsLowercase