AbstractPlutoDingetjes. Display.with_js_link to request calculations …

…and data from Julia dynamically
fonsp · Nov 21, 2023 · a153ec7 · a153ec7
1 parent 6224601
commit a153ec7
Show file tree

Hide file tree

Showing 4 changed files with 75 additions and 0 deletions.
diff --git a/frontend/components/CellOutput.js b/frontend/components/CellOutput.js
@@ -395,6 +395,8 @@ const execute_scripttags = async ({ root_node, script_nodes, previous_results_ma
                                 // @ts-ignore
                                 getPublishedObject: (id) => cell.getPublishedObject(id),
 
+                                _internal_getJSLinkResponse: (cell_id, link_id) => (input) => pluto_actions.request_js_link_response(cell_id, link_id, input),
+
                                 getBoundElementValueLikePluto: get_input_value,
                                 setBoundElementValueLikePluto: set_input_value,
                                 getBoundElementEventNameLikePluto: eventof,

diff --git a/frontend/components/Editor.js b/frontend/components/Editor.js
@@ -651,6 +651,19 @@ export class Editor extends Component {
                     false
                 )
             },
+            request_js_link_response: (cell_id, link_id, input) => {
+                return this.client
+                    .send(
+                        "request_js_link_response",
+                        {
+                            cell_id,
+                            link_id,
+                            input,
+                        },
+                        { notebook_id: this.state.notebook.notebook_id }
+                    )
+                    .then((r) => r.message)
+            },
             /** This actions avoids pushing selected cells all the way down, which is too heavy to handle! */
             get_selected_cells: (cell_id, /** @type {boolean} */ allow_other_selected_cells) =>
                 allow_other_selected_cells ? this.state.selected_cells : [cell_id],

diff --git a/src/runner/PlutoRunner.jl b/src/runner/PlutoRunner.jl
@@ -541,6 +541,9 @@ function run_expression(
 
     # reset registered bonds
     cell_registered_bond_names[cell_id] = Set{Symbol}()
+
+    # reset JS links
+    cell_js_links[cell_id] = Dict{String,Any}()
 
     # If the cell contains macro calls, we want those macro calls to preserve their identity,
     # so we macroexpand this earlier (during expression explorer stuff), and then we find it here.
@@ -990,6 +993,7 @@ const default_iocontext = IOContext(devnull,
     :is_pluto => true, 
     :pluto_supported_integration_features => supported_integration_features,
     :pluto_published_to_js => (io, x) -> core_published_to_js(io, x),
+    :pluto_with_js_link => (io, callback) -> core_with_js_link(io, callback),
 )
 
 const default_stdout_iocontext = IOContext(devnull, 
@@ -1703,6 +1707,9 @@ const integrations = Integration[
                 if isdefined(AbstractPlutoDingetjes.Display, :published_to_js)
                     supported!(AbstractPlutoDingetjes.Display.published_to_js)
                 end
+                if isdefined(AbstractPlutoDingetjes.Display, :with_js_link)
+                    supported!(AbstractPlutoDingetjes.Display.with_js_link)
+                end
             end
 
         end,
@@ -2486,6 +2493,41 @@ function Base.show(io::IO, m::MIME"text/html", e::DivElement)
     Base.show(io, m, embed_display(e))
 end
 
+
+###
+# JS LINK
+###
+
+const cell_js_links = Dict{UUID,Dict{String,Any}}()
+
+function core_with_js_link(io, callback)
+
+    _notebook_id = get(io, :pluto_notebook_id, notebook_id[])::UUID
+    _cell_id = get(io, :pluto_cell_id, currently_running_cell_id[])::UUID
+
+    # TODO is this okay? prob not
+    # link_id = objectid2str(callback)
+    link_id = String(rand('a':'z', 16))
+
+    links = get!(() -> Dict{String,Any}(), cell_js_links, _cell_id)
+    links[link_id] = callback
+
+    write(io, "/* See the documentation for AbstractPlutoDingetjes.Display.with_js_link */ _internal_getJSLinkResponse(\"$(_cell_id)\", \"$(link_id)\")")
+end
+
+function evaluate_js_link(cell_id::UUID, link_id::String, input::Any)
+    links = get(() -> Dict{String,Any}(), cell_js_links, cell_id)
+    callback = get(links, link_id, nothing)
+    if callback === nothing
+        @error "🚨 AbstractPlutoDingetjes: JS link not found." link_id
+    else
+        result = callback(input)
+        assertpackable(result)
+
+        result
+    end
+end
+
 ###
 # LOGGING
 ###

diff --git a/src/webserver/Dynamic.jl b/src/webserver/Dynamic.jl
@@ -525,6 +525,24 @@ responses[:reshow_cell] = function response_reshow_cell(🙋::ClientRequest)
     send_notebook_changes!(🙋 |> without_initiator)
 end
 
+responses[:request_js_link_response] = function response_request_js_link_response(🙋::ClientRequest)
+    require_notebook(🙋)
+    @assert will_run_code(🙋.notebook)
+
+    result = WorkspaceManager.eval_fetch_in_workspace(
+        (🙋.session, 🙋.notebook), 
+        quote
+            PlutoRunner.evaluate_js_link(
+                $(UUID(🙋.body["cell_id"])),
+                $(🙋.body["link_id"]),
+                $(🙋.body["input"]),
+            )
+        end
+    )
+
+    putclientupdates!(🙋.session, 🙋.initiator, UpdateMessage(:🐤, result, nothing, nothing, 🙋.initiator))
+end
+
 responses[:nbpkg_available_versions] = function response_nbpkg_available_versions(🙋::ClientRequest)
     # require_notebook(🙋)
     all_versions = PkgCompat.package_versions(🙋.body["package_name"])