server/controllers/api/users/me.ts

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