cat-bookmarker/deps/phoenix/lib/mix/tasks/phx.gen.channel.ex

112 lines
2.9 KiB
Elixir
Raw Normal View History

2024-03-10 18:52:04 +00:00
defmodule Mix.Tasks.Phx.Gen.Channel do
@shortdoc "Generates a Phoenix channel"
@moduledoc """
Generates a Phoenix channel.
$ mix phx.gen.channel Room
Accepts the module name for the channel
The generated files will contain:
For a regular application:
* a channel in `lib/my_app_web/channels`
* a channel test in `test/my_app_web/channels`
For an umbrella application:
* a channel in `apps/my_app_web/lib/app_name_web/channels`
* a channel test in `apps/my_app_web/test/my_app_web/channels`
"""
use Mix.Task
alias Mix.Tasks.Phx.Gen
@doc false
def run(args) do
if Mix.Project.umbrella?() do
Mix.raise(
"mix phx.gen.channel must be invoked from within your *_web application root directory"
)
end
[channel_name] = validate_args!(args)
context_app = Mix.Phoenix.context_app()
web_prefix = Mix.Phoenix.web_path(context_app)
web_test_prefix = Mix.Phoenix.web_test_path(context_app)
binding = Mix.Phoenix.inflect(channel_name)
binding = Keyword.put(binding, :module, "#{binding[:web_module]}.#{binding[:scoped]}")
Mix.Phoenix.check_module_name_availability!(binding[:module] <> "Channel")
test_path = Path.join(web_test_prefix, "channels/#{binding[:path]}_channel_test.exs")
case_path = Path.join(Path.dirname(web_test_prefix), "support/channel_case.ex")
maybe_case = if File.exists?(case_path) do
[]
else
[{:eex, "channel_case.ex", case_path}]
end
Mix.Phoenix.copy_from(paths(), "priv/templates/phx.gen.channel", binding, [
{:eex, "channel.ex", Path.join(web_prefix, "channels/#{binding[:path]}_channel.ex")},
{:eex, "channel_test.exs", test_path}
] ++ maybe_case)
user_socket_path = Mix.Phoenix.web_path(context_app, "channels/user_socket.ex")
if File.exists?(user_socket_path) do
Mix.shell().info("""
Add the channel to your `#{user_socket_path}` handler, for example:
channel "#{binding[:singular]}:lobby", #{binding[:module]}Channel
""")
else
Mix.shell().info("""
The default socket handler - #{binding[:web_module]}.UserSocket - was not found.
""")
if Mix.shell().yes?("Do you want to create it?") do
Gen.Socket.run(~w(User --from-channel #{channel_name}))
else
Mix.shell().info("""
To create it, please run the mix task:
mix phx.gen.socket User
Then add the channel to the newly created file, at `#{user_socket_path}`:
channel "#{binding[:singular]}:lobby", #{binding[:module]}Channel
""")
end
end
end
@spec raise_with_help() :: no_return()
defp raise_with_help do
Mix.raise("""
mix phx.gen.channel expects just the module name:
mix phx.gen.channel Room
""")
end
defp validate_args!(args) do
unless length(args) == 1 do
raise_with_help()
end
args
end
defp paths do
[".", :phoenix]
end
end