Implement client ban and unban endpoints

lib/matrix_server/room_server.ex

@@ -15,7 +15,7 @@ defmodule MatrixServer.RoomServer do
 
   alias MatrixServer.{Repo, Room, Event, StateResolution, Account, JoinedRoom}
   alias MatrixServer.StateResolution.Authorization
-  alias MatrixServerWeb.Client.Request.{CreateRoom, Kick}
+  alias MatrixServerWeb.Client.Request.{CreateRoom, Kick, Ban}
 
   @registry MatrixServer.RoomServer.Registry
   @supervisor MatrixServer.RoomServer.Supervisor
@@ -130,6 +130,22 @@ defmodule MatrixServer.RoomServer do
     GenServer.call(pid, {:kick, account, request})
   end
 
+  @doc """
+  Ban a user from this room.
+  """
+  @spec ban(pid(), Account.t(), Ban.t()) :: :ok | {:error, atom()}
+  def ban(pid, account, request) do
+    GenServer.call(pid, {:ban, account, request})
+  end
+
+  @doc """
+  Unban a user from this room.
+  """
+  @spec unban(pid(), Account.t(), String.t()) :: :ok | {:error, atom()}
+  def unban(pid, account, user_id) do
+    GenServer.call(pid, {:unban, account, user_id})
+  end
+
   ### Implementation
 
   @impl true
@@ -268,7 +284,29 @@ defmodule MatrixServer.RoomServer do
     end
   end
 
-  @spec insert_single_event(Room.t(), t(), Event.t()) :: {:ok, t(), Room.t()} | {:error, atom()}
+  def handle_call(
+        {:ban, account, %Kick{user_id: user_id, reason: reason}},
+        _from,
+        %{room: room, state_set: state_set} = state
+      ) do
+    ban_event = Event.ban(room, account, user_id, reason)
+
+    case insert_single_event(room, state_set, ban_event) do
+      {:ok, {state_set, room}} -> {:reply, :ok, %{state | state_set: state_set, room: room}}
+      {:error, reason} -> {:reply, {:error, reason}, state}
+    end
+  end
+
+  def handle_call({:unban, account, user_id},_from,%{room: room, state_set: state_set} = state) do
+    unban_event = Event.unban(room, account, user_id)
+
+    case insert_single_event(room, state_set, unban_event) do
+      {:ok, {state_set, room}} -> {:reply, :ok, %{state | state_set: state_set, room: room}}
+      {:error, reason} -> {:reply, {:error, reason}, state}
+    end
+  end
+
+  @spec insert_single_event(Room.t(), t(), Event.t()) :: {:ok, {t(), Room.t()}} | {:error, atom()}
   defp insert_single_event(room, state_set, event) do
     Repo.transaction(fn ->
       case finalize_and_insert_event(event, state_set, room) do

lib/matrix_server/schema/event.ex

@@ -226,6 +226,31 @@ defmodule MatrixServer.Event do
     }
   end
 
+  @spec ban(Room.t(), Account.t(), String.t(), String.t() | nil) :: t()
+  def ban(room, sender, user_id, reason \\ nil) do
+    content = %{"membership" => "ban"}
+    content = if reason, do: Map.put(content, "reason", reason), else: content
+
+    %Event{
+      new(room, sender)
+      | type: "m.room.member",
+        state_key: user_id,
+        content: content
+    }
+  end
+
+  @spec unban(Room.t(), Account.t(), String.t()) :: t()
+  def unban(room, sender, user_id) do
+    %Event{
+      new(room, sender)
+      | type: "m.room.member",
+        state_key: user_id,
+        content: %{
+          "membership" => "leave"
+        }
+    }
+  end
+
   @spec is_control_event(t()) :: boolean()
   def is_control_event(%Event{type: "m.room.power_levels", state_key: ""}), do: true
   def is_control_event(%Event{type: "m.room.join_rules", state_key: ""}), do: true