git.squeep.com Git - akkoma/blob - lib/pleroma/web/chat_channel.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.ChatChannel do
   6   use Phoenix.Channel
   7   alias Pleroma.User
   8   alias Pleroma.Web.ChatChannel.ChatChannelState
   9
  10   def join("chat:public", _message, socket) do
  11     send(self(), :after_join)
  12     {:ok, socket}
  13   end
  14
  15   def handle_info(:after_join, socket) do
  16     push(socket, "messages", %{messages: ChatChannelState.messages()})
  17     {:noreply, socket}
  18   end
  19
  20   def handle_in("new_msg", %{"text" => text}, %{assigns: %{user_name: user_name}} = socket) do
  21     text = String.trim(text)
  22
  23     if String.length(text) in 1..Pleroma.Config.get([:instance, :chat_limit]) do
  24       author = User.get_cached_by_nickname(user_name)
  25
  26       author =
  27         Pleroma.Web.MastodonAPI.AccountView.render("show.json",
  28           user: author,
  29           skip_relationships: true
  30         )
  31
  32       message = ChatChannelState.add_message(%{text: text, author: author})
  33
  34       broadcast!(socket, "new_msg", message)
  35     end
  36
  37     {:noreply, socket}
  38   end
  39 end
  40
  41 defmodule Pleroma.Web.ChatChannel.ChatChannelState do
  42   use Agent
  43
  44   @max_messages 20
  45
  46   def start_link(_) do
  47     Agent.start_link(fn -> %{max_id: 1, messages: []} end, name: __MODULE__)
  48   end
  49
  50   def add_message(message) do
  51     Agent.get_and_update(__MODULE__, fn state ->
  52       id = state[:max_id] + 1
  53       message = Map.put(message, "id", id)
  54       messages = [message | state[:messages]] |> Enum.take(@max_messages)
  55       {message, %{max_id: id, messages: messages}}
  56     end)
  57   end
  58
  59   def messages do
  60     Agent.get(__MODULE__, fn state -> state[:messages] |> Enum.reverse() end)
  61   end
  62 end