livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2024-09-20 10:05:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle module plugs in proxy

Browse source
This commit is contained in:
Wojtek Mach 2024-06-20 12:37:43 +02:00
parent 28336da83c
commit 220a4a2cdb
1 changed files with 27 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
lib/livebook/proxy/handler.ex
View file

@ -1,7 +1,7 @@
defmodule Livebook.Proxy.Handler do
@moduledoc false
# Handles request `conn` with the configured function.
# Handles request `conn` with the configured plug.
#
# The handler forwards all actual communication to the parent
# `Livebook.Proxy.Server` via `Livebook.Proxy.Adapter`.
@ -15,8 +15,11 @@ defmodule Livebook.Proxy.Handler do
@doc """
Returns a child spec to setup the handler supervision tree.
Expects the `:listen` option to be provided, and be a function with
the signature `Plug.Conn.t() -> Plug.Conn.t()`.
Expects the `:listen` option to be provided, and be one of:
* a function plug: `Plug.Conn.t() -> Plug.Conn.t()`
* a module plug: `module` atom or `{module, options}` tuple.
"""
@spec child_spec(keyword()) :: Supervisor.child_spec()
def child_spec(opts) do
@ -39,8 +42,27 @@ defmodule Livebook.Proxy.Handler do
conn =
struct!(Plug.Conn, %{conn_attrs | adapter: {Livebook.Proxy.Adapter, {parent_pid, ref}}})
listen = :persistent_term.get({__MODULE__, :listen})
listen.(conn)
case :persistent_term.get({__MODULE__, :listen}) do
fun when is_function(fun, 1) ->
fun.(conn)
mod when is_atom(mod) ->
mod.call(conn, mod.init([]))
{mod, opts} when is_atom(mod) ->
mod.call(conn, mod.init(opts))
other ->
raise """
expected plug to be one of:
* fun(conn)
* module
* {module, options}
got: #{inspect(other)}
"""
end
end
@doc """