aboutsummaryrefslogtreecommitdiff
path: root/lib/pleroma/user/email_list.ex
blob: a483d91e0e344647913d0788c7a5e53790849c14 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# Pleroma: A lightweight social networking server
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only

defmodule Pleroma.User.EmailList do
  @moduledoc """
  Functions for generating email lists from local users.
  """
  import Ecto.Query

  alias Pleroma.Repo
  alias Pleroma.User

  @header_row ["Email Address", "Nickname", "Subscribe?"]

  defp query(:subscribers) do
    User.Query.build(%{
      local: true,
      active: true,
      accepts_email_list: true
    })
    |> where([u], not is_nil(u.email))
  end

  defp query(:unsubscribers) do
    User.Query.build(%{
      local: true,
      accepts_email_list: false
    })
    |> where([u], not is_nil(u.email))
  end

  defp query(:combined) do
    User.Query.build(%{
      local: true
    })
    |> where([u], not is_nil(u.email))
  end

  def generate_csv(audience) when is_atom(audience) do
    audience
    |> query()
    |> generate_csv()
  end

  def generate_csv(%Ecto.Query{} = query) do
    query
    |> Repo.all()
    |> Enum.map(&build_row/1)
    |> build_csv()
  end

  defp subscribe?(%User{} = user) do
    user.accepts_email_list && user.is_active && user.is_approved && user.is_confirmed
  end

  defp build_row(%User{} = user) do
    [
      user.email,
      user.nickname,
      subscribe?(user)
    ]
  end

  defp build_csv(lines) do
    [@header_row | lines]
    |> CSV.encode()
    |> Enum.join()
  end
end