server/controllers/api/users/me.ts

   1 import 'multer'
   2 import * as express from 'express'
   3 import { auditLoggerFactory, getAuditIdFromRes, UserAuditView } from '@server/helpers/audit-logger'
   4 import { Hooks } from '@server/lib/plugins/hooks'
   5 import { ActorImageType, UserUpdateMe, UserVideoRate as FormattedUserVideoRate } from '../../../../shared'
   6 import { HttpStatusCode } from '../../../../shared/core-utils/miscs/http-error-codes'
   7 import { UserVideoQuota } from '../../../../shared/models/users/user-video-quota.model'
   8 import { createReqFiles } from '../../../helpers/express-utils'
   9 import { getFormattedObjects } from '../../../helpers/utils'
  10 import { CONFIG } from '../../../initializers/config'
  11 import { MIMETYPES } from '../../../initializers/constants'
  12 import { sequelizeTypescript } from '../../../initializers/database'
  13 import { sendUpdateActor } from '../../../lib/activitypub/send'
  14 import { deleteLocalActorImageFile, updateLocalActorImageFile } from '../../../lib/actor-image'
  15 import { getOriginalVideoFileTotalDailyFromUser, getOriginalVideoFileTotalFromUser, sendVerifyUserEmail } from '../../../lib/user'
  16 import {
  17   asyncMiddleware,
  18   asyncRetryTransactionMiddleware,
  19   authenticate,
  20   paginationValidator,
  21   setDefaultPagination,
  22   setDefaultSort,
  23   setDefaultVideosSort,
  24   usersUpdateMeValidator,
  25   usersVideoRatingValidator
  26 } from '../../../middlewares'
  27 import { deleteMeValidator, videoImportsSortValidator, videosSortValidator } from '../../../middlewares/validators'
  28 import { updateAvatarValidator } from '../../../middlewares/validators/actor-image'
  29 import { AccountModel } from '../../../models/account/account'
  30 import { AccountVideoRateModel } from '../../../models/account/account-video-rate'
  31 import { UserModel } from '../../../models/account/user'
  32 import { VideoModel } from '../../../models/video/video'
  33 import { VideoImportModel } from '../../../models/video/video-import'
  34
  35 const auditLogger = auditLoggerFactory('users')
  36
  37 const reqAvatarFile = createReqFiles([ 'avatarfile' ], MIMETYPES.IMAGE.MIMETYPE_EXT, { avatarfile: CONFIG.STORAGE.TMP_DIR })
  38
  39 const meRouter = express.Router()
  40
  41 meRouter.get('/me',
  42   authenticate,
  43   asyncMiddleware(getUserInformation)
  44 )
  45 meRouter.delete('/me',
  46   authenticate,
  47   deleteMeValidator,
  48   asyncMiddleware(deleteMe)
  49 )
  50
  51 meRouter.get('/me/video-quota-used',
  52   authenticate,
  53   asyncMiddleware(getUserVideoQuotaUsed)
  54 )
  55
  56 meRouter.get('/me/videos/imports',
  57   authenticate,
  58   paginationValidator,
  59   videoImportsSortValidator,
  60   setDefaultSort,
  61   setDefaultPagination,
  62   asyncMiddleware(getUserVideoImports)
  63 )
  64
  65 meRouter.get('/me/videos',
  66   authenticate,
  67   paginationValidator,
  68   videosSortValidator,
  69   setDefaultVideosSort,
  70   setDefaultPagination,
  71   asyncMiddleware(getUserVideos)
  72 )
  73
  74 meRouter.get('/me/videos/:videoId/rating',
  75   authenticate,
  76   asyncMiddleware(usersVideoRatingValidator),
  77   asyncMiddleware(getUserVideoRating)
  78 )
  79
  80 meRouter.put('/me',
  81   authenticate,
  82   asyncMiddleware(usersUpdateMeValidator),
  83   asyncRetryTransactionMiddleware(updateMe)
  84 )
  85
  86 meRouter.post('/me/avatar/pick',
  87   authenticate,
  88   reqAvatarFile,
  89   updateAvatarValidator,
  90   asyncRetryTransactionMiddleware(updateMyAvatar)
  91 )
  92
  93 meRouter.delete('/me/avatar',
  94   authenticate,
  95   asyncRetryTransactionMiddleware(deleteMyAvatar)
  96 )
  97
  98 // ---------------------------------------------------------------------------
  99
 100 export {
 101   meRouter
 102 }
 103
 104 // ---------------------------------------------------------------------------
 105
 106 async function getUserVideos (req: express.Request, res: express.Response) {
 107   const user = res.locals.oauth.token.User
 108
 109   const apiOptions = await Hooks.wrapObject({
 110     accountId: user.Account.id,
 111     start: req.query.start,
 112     count: req.query.count,
 113     sort: req.query.sort,
 114     search: req.query.search,
 115     isLive: req.query.isLive
 116   }, 'filter:api.user.me.videos.list.params')
 117
 118   const resultList = await Hooks.wrapPromiseFun(
 119     VideoModel.listUserVideosForApi,
 120     apiOptions,
 121     'filter:api.user.me.videos.list.result'
 122   )
 123
 124   const additionalAttributes = {
 125     waitTranscoding: true,
 126     state: true,
 127     scheduledUpdate: true,
 128     blacklistInfo: true
 129   }
 130   return res.json(getFormattedObjects(resultList.data, resultList.total, { additionalAttributes }))
 131 }
 132
 133 async function getUserVideoImports (req: express.Request, res: express.Response) {
 134   const user = res.locals.oauth.token.User
 135   const resultList = await VideoImportModel.listUserVideoImportsForApi(
 136     user.id,
 137     req.query.start as number,
 138     req.query.count as number,
 139     req.query.sort
 140   )
 141
 142   return res.json(getFormattedObjects(resultList.data, resultList.total))
 143 }
 144
 145 async function getUserInformation (req: express.Request, res: express.Response) {
 146   // We did not load channels in res.locals.user
 147   const user = await UserModel.loadForMeAPI(res.locals.oauth.token.user.id)
 148
 149   return res.json(user.toMeFormattedJSON())
 150 }
 151
 152 async function getUserVideoQuotaUsed (req: express.Request, res: express.Response) {
 153   const user = res.locals.oauth.token.user
 154   const videoQuotaUsed = await getOriginalVideoFileTotalFromUser(user)
 155   const videoQuotaUsedDaily = await getOriginalVideoFileTotalDailyFromUser(user)
 156
 157   const data: UserVideoQuota = {
 158     videoQuotaUsed,
 159     videoQuotaUsedDaily
 160   }
 161   return res.json(data)
 162 }
 163
 164 async function getUserVideoRating (req: express.Request, res: express.Response) {
 165   const videoId = res.locals.videoId.id
 166   const accountId = +res.locals.oauth.token.User.Account.id
 167
 168   const ratingObj = await AccountVideoRateModel.load(accountId, videoId, null)
 169   const rating = ratingObj ? ratingObj.type : 'none'
 170
 171   const json: FormattedUserVideoRate = {
 172     videoId,
 173     rating
 174   }
 175   return res.json(json)
 176 }
 177
 178 async function deleteMe (req: express.Request, res: express.Response) {
 179   const user = await UserModel.loadByIdWithChannels(res.locals.oauth.token.User.id)
 180
 181   auditLogger.delete(getAuditIdFromRes(res), new UserAuditView(user.toFormattedJSON()))
 182
 183   await user.destroy()
 184
 185   return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
 186 }
 187
 188 async function updateMe (req: express.Request, res: express.Response) {
 189   const body: UserUpdateMe = req.body
 190   let sendVerificationEmail = false
 191
 192   const user = res.locals.oauth.token.user
 193
 194   if (body.password !== undefined) user.password = body.password
 195   if (body.nsfwPolicy !== undefined) user.nsfwPolicy = body.nsfwPolicy
 196   if (body.webTorrentEnabled !== undefined) user.webTorrentEnabled = body.webTorrentEnabled
 197   if (body.autoPlayVideo !== undefined) user.autoPlayVideo = body.autoPlayVideo
 198   if (body.autoPlayNextVideo !== undefined) user.autoPlayNextVideo = body.autoPlayNextVideo
 199   if (body.autoPlayNextVideoPlaylist !== undefined) user.autoPlayNextVideoPlaylist = body.autoPlayNextVideoPlaylist
 200   if (body.videosHistoryEnabled !== undefined) user.videosHistoryEnabled = body.videosHistoryEnabled
 201   if (body.videoLanguages !== undefined) user.videoLanguages = body.videoLanguages
 202   if (body.theme !== undefined) user.theme = body.theme
 203   if (body.noInstanceConfigWarningModal !== undefined) user.noInstanceConfigWarningModal = body.noInstanceConfigWarningModal
 204   if (body.noWelcomeModal !== undefined) user.noWelcomeModal = body.noWelcomeModal
 205
 206   if (body.email !== undefined) {
 207     if (CONFIG.SIGNUP.REQUIRES_EMAIL_VERIFICATION) {
 208       user.pendingEmail = body.email
 209       sendVerificationEmail = true
 210     } else {
 211       user.email = body.email
 212     }
 213   }
 214
 215   await sequelizeTypescript.transaction(async t => {
 216     await user.save({ transaction: t })
 217
 218     if (body.displayName !== undefined || body.description !== undefined) {
 219       const userAccount = await AccountModel.load(user.Account.id, t)
 220
 221       if (body.displayName !== undefined) userAccount.name = body.displayName
 222       if (body.description !== undefined) userAccount.description = body.description
 223       await userAccount.save({ transaction: t })
 224
 225       await sendUpdateActor(userAccount, t)
 226     }
 227   })
 228
 229   if (sendVerificationEmail === true) {
 230     await sendVerifyUserEmail(user, true)
 231   }
 232
 233   return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
 234 }
 235
 236 async function updateMyAvatar (req: express.Request, res: express.Response) {
 237   const avatarPhysicalFile = req.files['avatarfile'][0]
 238   const user = res.locals.oauth.token.user
 239
 240   const userAccount = await AccountModel.load(user.Account.id)
 241
 242   const avatar = await updateLocalActorImageFile(userAccount, avatarPhysicalFile, ActorImageType.AVATAR)
 243
 244   return res.json({ avatar: avatar.toFormattedJSON() })
 245 }
 246
 247 async function deleteMyAvatar (req: express.Request, res: express.Response) {
 248   const user = res.locals.oauth.token.user
 249
 250   const userAccount = await AccountModel.load(user.Account.id)
 251   await deleteLocalActorImageFile(userAccount, ActorImageType.AVATAR)
 252
 253   return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
 254 }