server/controllers/api/users/me.ts

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