Alek Ratzloff
73c5f58243
Since modules can now intercept all messages in the channels they're listening for, it'd be cool to have modules handling things like making sure the Omnibot.State stays updated as appropriate, and that pings are ponged, etc. This will probably deprecate the router, since it's been reduced to a single function call, but we'll see about that. Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
121 lines
2.9 KiB
Elixir
121 lines
2.9 KiB
Elixir
alias Omnibot.Irc
|
|
alias Omnibot.Util
|
|
|
|
defmodule Omnibot.Irc.Msg do
|
|
defmodule Prefix do
|
|
defstruct [:nick, :user, :host]
|
|
|
|
@prefix_regex ~r/(?<nick>[^!]+)(!(?<user>[^@]+)(@(?<host>.+))?)?/
|
|
def parse(prefix) do
|
|
cap = Regex.named_captures(@prefix_regex, prefix)
|
|
|
|
if cap do
|
|
%{
|
|
"nick" => nick,
|
|
"user" => user,
|
|
"host" => host
|
|
} = cap
|
|
|
|
%Irc.Msg.Prefix{
|
|
nick: nick,
|
|
user: if(user == "", do: nil, else: user),
|
|
host: if(host == "", do: nil, else: host)
|
|
}
|
|
else
|
|
nil
|
|
end
|
|
end
|
|
end
|
|
|
|
@enforce_keys [:command]
|
|
defstruct prefix: nil, command: nil, params: []
|
|
|
|
@msg_regex ~r/
|
|
^(:(?P<prefix>[^ ]+)\ )?
|
|
(?<command>[a-zA-Z]+|[0-9]{3})
|
|
(?<params>(\ [^: \r\n]+)*)
|
|
(?<trailing>\ :[^\r\n]+)?
|
|
/x
|
|
|
|
def parse(msg) do
|
|
%{
|
|
"prefix" => prefix,
|
|
"command" => command,
|
|
"params" => params,
|
|
"trailing" => trailing
|
|
} = Regex.named_captures(@msg_regex, msg)
|
|
|
|
prefix = Irc.Msg.Prefix.parse(prefix)
|
|
|
|
params =
|
|
String.slice(params, 1..-1)
|
|
|> String.split(" ")
|
|
|> Enum.filter(fn s -> String.length(s) > 0 end)
|
|
|
|
trailing =
|
|
trailing
|
|
|> String.slice(2..-1)
|
|
|> Util.string_or_nil()
|
|
|
|
params = if trailing, do: params ++ [trailing], else: params
|
|
|
|
%Irc.Msg{
|
|
prefix: prefix,
|
|
command: command,
|
|
params: params,
|
|
}
|
|
end
|
|
|
|
@doc "Gets the channel that the given message is targeting, if any."
|
|
def channel(msg) do
|
|
case String.upcase(msg.command) do
|
|
"PRIVMSG" -> Enum.at(msg.params, 0)
|
|
"JOIN" -> Enum.at(msg.params, 0)
|
|
"PART" -> Enum.at(msg.params, 0)
|
|
"KICK" -> Enum.at(msg.params, 0)
|
|
_ -> nil
|
|
end
|
|
end
|
|
end
|
|
|
|
defimpl String.Chars, for: Irc.Msg.Prefix do
|
|
def to_string(prefix) do
|
|
nick = prefix.nick || ""
|
|
user = if prefix.user, do: "!#{prefix.user}", else: ""
|
|
host = if prefix.host, do: "@#{prefix.host}", else: ""
|
|
"#{nick}#{user}#{host}"
|
|
end
|
|
end
|
|
|
|
defimpl String.Chars, for: Irc.Msg do
|
|
def to_string(msg) do
|
|
prefix =
|
|
case String.Chars.to_string(msg.prefix) do
|
|
"" -> ""
|
|
p -> ":#{p}"
|
|
end
|
|
|
|
# Figure out where "trailing" parameters begin, e.g.
|
|
# ["privmsg", "param", "some message", "with another param"]
|
|
#
|
|
# becomes
|
|
#
|
|
# {["privmsg", "param"], ["some message", "with another param"]}
|
|
#
|
|
{params, trailing} = Enum.split_while(msg.params, fn param -> !String.contains?(param, " ") end)
|
|
|
|
# If trailing parameters exist, then join them with a space character and prefix with a colon,
|
|
# appending it as another list item to the non-trailing parameter list
|
|
#
|
|
# If trailing parameters don't exist, don't append anything
|
|
params = params ++
|
|
if length(trailing) > 0,
|
|
do: [":" <> Enum.join(trailing, " ")],
|
|
else: []
|
|
|
|
([prefix, msg.command] ++ params)
|
|
|> Enum.filter(fn n -> String.length(n) > 0 end)
|
|
|> Enum.join(" ")
|
|
end
|
|
end
|