Page Menu
Home
Phorge
Search
Configure Global Search
Log In
Files
F5811140
notification.ex
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Award Token
Flag For Later
Size
2 KB
Referenced Files
None
Subscribers
None
notification.ex
View Options
defmodule
Pleroma.Notification
do
use
Ecto.Schema
alias
Pleroma
.
{
User
,
Activity
,
Notification
,
Repo
}
import
Ecto.Query
schema
"notifications"
do
field
(
:seen
,
:boolean
,
default
:
false
)
belongs_to
(
:user
,
Pleroma.User
)
belongs_to
(
:activity
,
Pleroma.Activity
)
timestamps
()
end
# TODO: Make generic and unify (see activity_pub.ex)
defp
restrict_max
(
query
,
%{
"max_id"
=>
max_id
})
do
from
(
activity
in
query
,
where
:
activity
.
id
<
^
max_id
)
end
defp
restrict_max
(
query
,
_
),
do
:
query
defp
restrict_since
(
query
,
%{
"since_id"
=>
since_id
})
do
from
(
activity
in
query
,
where
:
activity
.
id
>
^
since_id
)
end
defp
restrict_since
(
query
,
_
),
do
:
query
def
for_user
(
user
,
opts
\\
%{})
do
query
=
from
(
n
in
Notification
,
where
:
n
.
user_id
==
^
user
.
id
,
order_by
:
[
desc
:
n
.
id
],
preload
:
[
:activity
],
limit
:
20
)
query
=
query
|>
restrict_since
(
opts
)
|>
restrict_max
(
opts
)
Repo
.
all
(
query
)
end
def
get
(%{
id
:
user_id
}
=
_user
,
id
)
do
query
=
from
(
n
in
Notification
,
where
:
n
.
id
==
^
id
,
preload
:
[
:activity
]
)
notification
=
Repo
.
one
(
query
)
case
notification
do
%{
user_id
:
^
user_id
}
->
{
:ok
,
notification
}
_
->
{
:error
,
"Cannot get notification"
}
end
end
def
clear
(
user
)
do
query
=
from
(
n
in
Notification
,
where
:
n
.
user_id
==
^
user
.
id
)
Repo
.
delete_all
(
query
)
end
def
dismiss
(%{
id
:
user_id
}
=
_user
,
id
)
do
notification
=
Repo
.
get
(
Notification
,
id
)
case
notification
do
%{
user_id
:
^
user_id
}
->
Repo
.
delete
(
notification
)
_
->
{
:error
,
"Cannot dismiss notification"
}
end
end
def
create_notifications
(%
Activity
{
id
:
_
,
data
:
%{
"to"
=>
_
,
"type"
=>
type
}}
=
activity
)
when
type
in
[
"Create"
,
"Like"
,
"Announce"
,
"Follow"
]
do
users
=
User
.
get_notified_from_activity
(
activity
)
notifications
=
Enum
.
map
(
users
,
fn
user
->
create_notification
(
activity
,
user
)
end
)
{
:ok
,
notifications
}
end
def
create_notifications
(
_
),
do
:
{
:ok
,
[]}
# TODO move to sql, too.
def
create_notification
(%
Activity
{}
=
activity
,
%
User
{}
=
user
)
do
unless
User
.
blocks?
(
user
,
%{
ap_id
:
activity
.
data
[
"actor"
]})
or
user
.
ap_id
==
activity
.
data
[
"actor"
]
do
notification
=
%
Notification
{
user_id
:
user
.
id
,
activity
:
activity
}
{
:ok
,
notification
}
=
Repo
.
insert
(
notification
)
Pleroma.Web.Streamer
.
stream
(
"user"
,
notification
)
notification
end
end
end
File Metadata
Details
Attached
Mime Type
text/x-ruby
Expires
Thu, Aug 14, 3:29 AM (18 h, 7 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
310952
Default Alt Text
notification.ex (2 KB)
Attached To
Mode
rPUBE pleroma-upstream
Attached
Detach File
Event Timeline
Log In to Comment