encapsulate the results in a list of structs

bitwalker · Jan 19, 2018 · 2b2de8b · 2b2de8b
1 parent 7802cd8
commit 2b2de8b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/lib/exirc/client.ex b/lib/exirc/client.ex
@@ -703,7 +703,11 @@ defmodule ExIRC.Client do
   end
 
   def handle_data(%ExIRC.Message{:cmd => "315", :args => [_, channel, _]}, state) do
-    buffer = Map.get(state.who_buffers, channel, [])
+    buffer = state
+             |> Map.get(:who_buffers)
+             |> Map.get(channel)
+             |> Enum.map(fn user -> struct(ExIRC.Who, user) end)
+
     send_event {:who, channel, buffer}, state
     {:noreply, %ClientState{state | who_buffers: Map.delete(state.who_buffers, channel)}}
   end

diff --git a/lib/exirc/who.ex b/lib/exirc/who.ex
@@ -1,9 +1,17 @@
 defmodule ExIRC.Who do
 
-  defstruct [name: nil,
+  defstruct [
+             admin?: nil,
+             away?: nil,
+             founder?: nil,
+             half_operator?: nil,
+             hops: nil,
+             name: nil,
              nickname: nil,
              operator?: nil,
              server: nil,
-             user: nil
+             server_operator?: nil,
+             user: nil,
+             voiced?: nil
             ]
 end