diff options
Diffstat (limited to 'lib/exirc/client.ex')
-rw-r--r-- | lib/exirc/client.ex | 90 |
1 files changed, 83 insertions, 7 deletions
diff --git a/lib/exirc/client.ex b/lib/exirc/client.ex index c86f58d..4c49106 100644 --- a/lib/exirc/client.ex +++ b/lib/exirc/client.ex @@ -33,7 +33,8 @@ defmodule ExIrc.Client do debug?: false, retries: 0, inet: :inet, - owner: nil + owner: nil, + whois_buffers: %{} end ################# @@ -167,6 +168,16 @@ defmodule ExIrc.Client do def names(client, channel) do GenServer.call(client, {:names, channel}, :infinity) end + + @doc """ + Ask the server for the user's informations. + """ + @spec whois(client :: pid, user :: binary) :: :ok | {:error, atom()} + def whois(client, user) do + GenServer.call(client, {:whois, user}, :infinity) + end + + @doc """ Change mode for a user or channel """ @@ -330,7 +341,7 @@ defmodule ExIrc.Client do # Set SSL mode state = %{state | ssl?: ssl} # Open a new connection - case Transport.connect(state, String.to_char_list(server), port, [:list, {:packet, :line}, {:keepalive, true}] ++ options) do + case Transport.connect(state, String.to_charlist(server), port, [:list, {:packet, :line}, {:keepalive, true}] ++ options) do {:ok, socket} -> send_event {:connected, server, port}, state {:reply, :ok, %{state | connected?: true, server: server, port: port, socket: socket}} @@ -397,6 +408,12 @@ defmodule ExIrc.Client do Transport.send(state, names!(channel)) {:reply, :ok, state} end + + def handle_call({:whois, user}, _from, state) do + Transport.send(state, whois!(user)) + {:reply, :ok, state} + end + # Handles a call to change mode for a user or channel def handle_call({:mode, channel_or_nick, flags, args}, _from, state) do Transport.send(state, mode!(channel_or_nick, flags, args)) @@ -551,7 +568,7 @@ defmodule ExIrc.Client do end # Called when the client enters a channel def handle_data(%IrcMessage{nick: nick, cmd: "JOIN"} = msg, %ClientState{nick: nick} = state) do - channel = msg.args |> List.first |> String.strip + channel = msg.args |> List.first |> String.trim if state.debug?, do: debug "JOINED A CHANNEL #{channel}" channels = Channels.join(state.channels, channel) new_state = %{state | channels: channels} @@ -561,7 +578,7 @@ defmodule ExIrc.Client do # Called when another user joins a channel the client is in def handle_data(%IrcMessage{nick: user_nick, cmd: "JOIN", host: host, user: user} = msg, state) do sender = %SenderInfo{nick: user_nick, host: host, user: user} - channel = msg.args |> List.first |> String.strip + channel = msg.args |> List.first |> String.trim if state.debug?, do: debug "ANOTHER USER JOINED A CHANNEL: #{channel} - #{user_nick}" channels = Channels.user_join(state.channels, channel, user_nick) new_state = %{state | channels: channels} @@ -589,6 +606,65 @@ defmodule ExIrc.Client do send_event {:topic_changed, channel, topic}, new_state {:noreply, new_state} end + + + ## WHOIS + + + def handle_data(%IrcMessage{cmd: @rpl_whoisuser, args: [_sender, nickname, username, hostname, _, realname]}, state) do + user = %{nickname: nickname, username: username, hostname: hostname, realname: realname} + {:noreply, %ClientState{state|whois_buffers: Map.put(state.whois_buffers, nickname, user)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoiscertfp, args: [_sender, nickname, "has client certificate fingerprint "<> fingerprint]}, state) do + {:noreply, %ClientState{state|whois_buffers: put_in(state.whois_buffers, [nickname, :certfp], fingerprint)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoisregnick, args: [_sender, nickname, _message]}, state) do + {:noreply, %ClientState{state|whois_buffers: put_in(state.whois_buffers, [nickname, :registered_nick?], true)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoishelpop, args: [_sender, nickname, _message]}, state) do + {:noreply, %ClientState{state|whois_buffers: put_in(state.whois_buffers, [nickname, :helpop?], true)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoischannels, args: [_sender, nickname, channels]}, state) do + chans = String.split(channels, " ") + {:noreply, %ClientState{state|whois_buffers: put_in(state.whois_buffers, [nickname, :channels], chans)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoisserver, args: [_sender, nickname, server_addr, server_name]}, state) do + new_buffer = state.whois_buffers + |> put_in([nickname, :server_name], server_name) + |> put_in([nickname, :server_address], server_addr) + {:noreply, %ClientState{state|whois_buffers: new_buffer}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoisoperator, args: [_sender, nickname, _message]}, state) do + {:noreply, %ClientState{state|whois_buffers: put_in(state.whois_buffers, [nickname, :ircop?], true)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoisaccount, args: [_sender, nickname, account_name, _message]}, state) do + {:noreply, %ClientState{state|whois_buffers: put_in(state.whois_buffers, [nickname, :account_name], account_name)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoissecure, args: [_sender, nickname, _message]}, state) do + {:noreply, %ClientState{state|whois_buffers: put_in(state.whois_buffers, [nickname, :tls?], true)}} + end + + def handle_data(%IrcMessage{cmd: @rpl_whoisidle, args: [_sender, nickname, idling_time, signon_time, _message]}, state) do + new_buffer = state.whois_buffers + |> put_in([nickname, :idling_time], idling_time) + |> put_in([nickname, :signon_time], signon_time) + {:noreply, %ClientState{state|whois_buffers: new_buffer}} + end + + def handle_data(%IrcMessage{cmd: @rpl_endofwhois, args: [_sender, nickname, _message]}, state) do + buffer = struct(Irc.Whois, state.whois_buffers[nickname]) + send_event {:whois, buffer}, state + {:noreply, %ClientState{state|whois_buffers: Map.delete(state.whois_buffers, nickname)}} + end + def handle_data(%IrcMessage{cmd: @rpl_notopic, args: [channel]}, state) do if state.debug? do debug "INITIAL TOPIC MSG" @@ -645,7 +721,7 @@ defmodule ExIrc.Client do end # Called when we leave a channel def handle_data(%IrcMessage{cmd: "PART", nick: nick} = msg, %ClientState{nick: nick} = state) do - channel = msg.args |> List.first |> String.strip + channel = msg.args |> List.first |> String.trim if state.debug?, do: debug "WE LEFT A CHANNEL: #{channel}" channels = Channels.part(state.channels, channel) new_state = %{state | channels: channels} @@ -655,7 +731,7 @@ defmodule ExIrc.Client do # Called when someone else in our channel leaves def handle_data(%IrcMessage{cmd: "PART", nick: from, host: host, user: user} = msg, state) do sender = %SenderInfo{nick: from, host: host, user: user} - channel = msg.args |> List.first |> String.strip + channel = msg.args |> List.first |> String.trim if state.debug?, do: debug "#{from} LEFT A CHANNEL: #{channel}" channels = Channels.user_part(state.channels, channel, from) new_state = %{state | channels: channels} @@ -729,7 +805,7 @@ defmodule ExIrc.Client do {:noreply, state} end # Called when a NOTICE is received by the client. - def handle_data(%IrcMessage{nick: from, cmd: "NOTICE", args: [target, message], host: host, user: user} = _msg, state) do + def handle_data(%IrcMessage{nick: from, cmd: "NOTICE", args: [_target, message], host: host, user: user} = _msg, state) do sender = %SenderInfo{nick: from, host: host, user: user} |