+
+defmodule Pleroma.Web.ChatChannel.ChatChannelState do
+ use Agent
+ @max_messages 20
+
+ def start_link do
+ Agent.start_link(fn -> %{max_id: 1, messages: []} end, name: __MODULE__)
+ end
+
+ def add_message(message) do
+ Agent.get_and_update(__MODULE__, fn state ->
+ id = state[:max_id] + 1
+ message = Map.put(message, "id", id)
+ messages = [message | state[:messages]] |> Enum.take(@max_messages)
+ {message, %{max_id: id, messages: messages}}
+ end)
+ end
+
+ def messages() do
+ Agent.get(__MODULE__, fn state -> state[:messages] |> Enum.reverse end)
+ end
+end