-
Notifications
You must be signed in to change notification settings - Fork 2.9k
/
phx.gen.socket.ex
116 lines (84 loc) · 2.94 KB
/
phx.gen.socket.ex
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
defmodule Mix.Tasks.Phx.Gen.Socket do
@shortdoc "Generates a Phoenix socket handler"
@moduledoc """
Generates a Phoenix socket handler.
$ mix phx.gen.socket User
Accepts the module name for the socket
The generated files will contain:
For a regular application:
* a client in `assets/js`
* a socket in `lib/my_app_web/channels`
For an umbrella application:
* a client in `apps/my_app_web/assets/js`
* a socket in `apps/my_app_web/lib/app_name_web/channels`
You can then generated channels with `mix phx.gen.channel`.
"""
use Mix.Task
@doc false
def run(args) do
if Mix.Project.umbrella?() do
Mix.raise(
"mix phx.gen.socket must be invoked from within your *_web application root directory"
)
end
[socket_name, pre_existing_channel] = validate_args!(args)
context_app = Mix.Phoenix.context_app()
web_prefix = Mix.Phoenix.web_path(context_app)
binding = Mix.Phoenix.inflect(socket_name)
existing_channel =
if pre_existing_channel do
channel_binding = Mix.Phoenix.inflect(pre_existing_channel)
Keyword.put(
channel_binding,
:module,
"#{channel_binding[:web_module]}.#{channel_binding[:scoped]}"
)
end
binding =
binding
|> Keyword.put(:module, "#{binding[:web_module]}.#{binding[:scoped]}")
|> Keyword.put(:endpoint_module, Module.concat([binding[:web_module], Endpoint]))
|> Keyword.put(:web_prefix, web_prefix)
|> Keyword.put(:existing_channel, existing_channel)
Mix.Phoenix.check_module_name_availability!(binding[:module] <> "Socket")
Mix.Phoenix.copy_from(paths(), "priv/templates/phx.gen.socket", binding, [
{:eex, "socket.ex", Path.join(web_prefix, "channels/#{binding[:path]}_socket.ex")},
{:eex, "socket.js", "assets/js/#{binding[:path]}_socket.js"}
])
Mix.shell().info("""
Add the socket handler to your `#{Mix.Phoenix.web_path(context_app, "endpoint.ex")}`, for example:
socket "/socket", #{binding[:module]}Socket,
websocket: true,
longpoll: false
For the front-end integration, you need to import the `#{binding[:path]}_socket.js`
in your `assets/js/app.js` file:
import "./#{binding[:path]}_socket.js"
""")
end
@spec raise_with_help() :: no_return()
defp raise_with_help do
Mix.raise("""
mix phx.gen.socket expects the module name:
mix phx.gen.socket User
""")
end
defp validate_args!([name, "--from-channel", pre_existing_channel]) do
unless valid_name?(name) and valid_name?(pre_existing_channel) do
raise_with_help()
end
[name, pre_existing_channel]
end
defp validate_args!([name]) do
unless valid_name?(name) do
raise_with_help()
end
[name, nil]
end
defp validate_args!(_), do: raise_with_help()
defp valid_name?(name) do
name =~ ~r/^[A-Z]\w*(\.[A-Z]\w*)*$/
end
defp paths do
[".", :phoenix]
end
end