-
Notifications
You must be signed in to change notification settings - Fork 3.3k
/
process.ex
158 lines (121 loc) · 3.93 KB
/
process.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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
defmodule Mix.Shell.Process do
@moduledoc """
Mix shell that uses the current process mailbox for communication.
This module provides a Mix shell implementation that uses
the current process mailbox for communication instead of IO.
As an example, when `Mix.shell.info("hello")` is called,
the following message will be sent to the calling process:
{:mix_shell, :info, ["hello"]}
This is mainly useful in tests, allowing us to assert
if given messages were received or not instead of performing
checks on some captured IO. Since we need to guarantee a clean
slate between tests, there is also a `flush/1` function
responsible for flushing all `:mix_shell` related messages
from the process inbox.
## Examples
Mix.shell.info "hello"
receive do {:mix_shell, :info, [msg]} -> msg end
#=> "hello"
send self(), {:mix_shell_input, :prompt, "Pretty cool"}
Mix.shell.prompt?("How cool was that?!")
#=> "Pretty cool"
"""
@behaviour Mix.Shell
@doc """
Flushes all `:mix_shell` and `:mix_shell_input` messages from the current process.
If a callback is given, it is invoked for each received message.
## Examples
flush &IO.inspect(&1)
"""
def flush(callback \\ fn(x) -> x end) do
receive do
{:mix_shell, _, _} = message ->
callback.(message)
flush(callback)
{:mix_shell_input, _, _} = message ->
callback.(message)
flush(callback)
after
0 -> :done
end
end
@doc """
Prints the current application if it
was not printed yet.
"""
def print_app do
if name = Mix.Shell.printable_app_name do
send self(), {:mix_shell, :info, ["==> #{name}"]}
end
end
@doc """
Executes the given command and forwards its messages to
the current process.
"""
def cmd(command, opts \\ []) do
print_app? = Keyword.get(opts, :print_app, true)
Mix.Shell.cmd(command, opts, fn(data) ->
if print_app?, do: print_app()
send self(), {:mix_shell, :run, [data]}
end)
end
@doc """
Forwards the message to the current process.
"""
def info(message) do
print_app()
send self(), {:mix_shell, :info, [format(message)]}
end
@doc """
Forwards the error to the current process.
"""
def error(message) do
print_app()
send self(), {:mix_shell, :error, [format(message)]}
end
defp format(message) do
message |> IO.ANSI.format(false) |> IO.iodata_to_binary
end
@doc """
Forwards the message to the current process.
It also checks the inbox for an input message matching:
{:mix_shell_input, :prompt, value}
If one does not exist, it will abort since there was no shell
process inputs given. `value` must be a string.
## Examples
The following will answer with `"Meg"` to the prompt
`"What's your name?"`:
# The response is sent before calling prompt/1 so that prompt/1 can read it
send self(), {:mix_shell_input, :prompt, "Meg"}
Mix.shell.prompt("What's your name?")
"""
def prompt(message) do
print_app()
send self(), {:mix_shell, :prompt, [message]}
receive do
{:mix_shell_input, :prompt, response} -> response
after
0 -> raise "no shell process input given for prompt/1"
end
end
@doc """
Forwards the message to the current process.
It also checks the inbox for an input message matching:
{:mix_shell_input, :yes?, value}
If one does not exist, it will abort since there was no shell
process inputs given. `value` must be `true` or `false`.
## Example
# Send the response to self() first so that yes?/1 will be able to read it
send self(), {:mix_shell_input, :yes?, true}
Mix.shell.yes?("Are you sure you want to continue?")
"""
def yes?(message) do
print_app()
send self(), {:mix_shell, :yes?, [message]}
receive do
{:mix_shell_input, :yes?, response} -> response
after
0 -> raise "no shell process input given for yes?/1"
end
end
end