Add new follow, mention and user registered notifs
[oweals/peertube.git] / server / controllers / api / users / my-notifications.ts
1 import * as express from 'express'
2 import 'multer'
3 import {
4   asyncMiddleware,
5   asyncRetryTransactionMiddleware,
6   authenticate,
7   paginationValidator,
8   setDefaultPagination,
9   setDefaultSort,
10   userNotificationsSortValidator
11 } from '../../../middlewares'
12 import { UserModel } from '../../../models/account/user'
13 import { getFormattedObjects } from '../../../helpers/utils'
14 import { UserNotificationModel } from '../../../models/account/user-notification'
15 import { meRouter } from './me'
16 import {
17   listUserNotificationsValidator,
18   markAsReadUserNotificationsValidator,
19   updateNotificationSettingsValidator
20 } from '../../../middlewares/validators/user-notifications'
21 import { UserNotificationSetting } from '../../../../shared/models/users'
22 import { UserNotificationSettingModel } from '../../../models/account/user-notification-setting'
23
24 const myNotificationsRouter = express.Router()
25
26 meRouter.put('/me/notification-settings',
27   authenticate,
28   updateNotificationSettingsValidator,
29   asyncRetryTransactionMiddleware(updateNotificationSettings)
30 )
31
32 myNotificationsRouter.get('/me/notifications',
33   authenticate,
34   paginationValidator,
35   userNotificationsSortValidator,
36   setDefaultSort,
37   setDefaultPagination,
38   listUserNotificationsValidator,
39   asyncMiddleware(listUserNotifications)
40 )
41
42 myNotificationsRouter.post('/me/notifications/read',
43   authenticate,
44   markAsReadUserNotificationsValidator,
45   asyncMiddleware(markAsReadUserNotifications)
46 )
47
48 export {
49   myNotificationsRouter
50 }
51
52 // ---------------------------------------------------------------------------
53
54 async function updateNotificationSettings (req: express.Request, res: express.Response) {
55   const user: UserModel = res.locals.oauth.token.User
56   const body = req.body
57
58   const query = {
59     where: {
60       userId: user.id
61     }
62   }
63
64   const values: UserNotificationSetting = {
65     newVideoFromSubscription: body.newVideoFromSubscription,
66     newCommentOnMyVideo: body.newCommentOnMyVideo,
67     videoAbuseAsModerator: body.videoAbuseAsModerator,
68     blacklistOnMyVideo: body.blacklistOnMyVideo,
69     myVideoPublished: body.myVideoPublished,
70     myVideoImportFinished: body.myVideoImportFinished,
71     newFollow: body.newFollow,
72     newUserRegistration: body.newUserRegistration,
73     commentMention: body.commentMention,
74   }
75
76   await UserNotificationSettingModel.update(values, query)
77
78   return res.status(204).end()
79 }
80
81 async function listUserNotifications (req: express.Request, res: express.Response) {
82   const user: UserModel = res.locals.oauth.token.User
83
84   const resultList = await UserNotificationModel.listForApi(user.id, req.query.start, req.query.count, req.query.sort, req.query.unread)
85
86   return res.json(getFormattedObjects(resultList.data, resultList.total))
87 }
88
89 async function markAsReadUserNotifications (req: express.Request, res: express.Response) {
90   const user: UserModel = res.locals.oauth.token.User
91
92   await UserNotificationModel.markAsRead(user.id, req.body.ids)
93
94   return res.status(204).end()
95 }