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
|