]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blobdiff - server/helpers/custom-validators/accounts.ts
Add ability to set a name to a channel
[github/Chocobozzz/PeerTube.git] / server / helpers / custom-validators / accounts.ts
index fe0fc650ae89109823da480cee539e502b67eb12..191de1496eec4b545e04a0438a1b85f95881b986 100644 (file)
@@ -1,43 +1,77 @@
-import * as Promise from 'bluebird'
-import * as express from 'express'
+import * as Bluebird from 'bluebird'
+import { Response } from 'express'
 import 'express-validator'
 import * as validator from 'validator'
-import { database as db } from '../../initializers'
-import { AccountInstance } from '../../models'
-import { logger } from '../logger'
-import { isUserUsernameValid } from './users'
+import { AccountModel } from '../../models/account/account'
+import { isUserDescriptionValid, isUserUsernameValid } from './users'
+import { exists } from './misc'
+import { CONFIG } from '../../initializers'
 
 function isAccountNameValid (value: string) {
   return isUserUsernameValid(value)
 }
 
-function checkVideoAccountExists (id: string, res: express.Response, callback: () => void) {
-  let promise: Promise<AccountInstance>
-  if (validator.isInt(id)) {
-    promise = db.Account.load(+id)
+function isAccountIdValid (value: string) {
+  return exists(value)
+}
+
+function isAccountDescriptionValid (value: string) {
+  return isUserDescriptionValid(value)
+}
+
+function isAccountIdExist (id: number | string, res: Response, sendNotFound = true) {
+  let promise: Bluebird<AccountModel>
+
+  if (validator.isInt('' + id)) {
+    promise = AccountModel.load(+id)
   } else { // UUID
-    promise = db.Account.loadByUUID(id)
+    promise = AccountModel.loadByUUID('' + id)
   }
 
-  promise.then(account => {
-    if (!account) {
-      return res.status(404)
-        .json({ error: 'Video account not found' })
-        .end()
+  return isAccountExist(promise, res, sendNotFound)
+}
+
+function isLocalAccountNameExist (name: string, res: Response, sendNotFound = true) {
+  const promise = AccountModel.loadLocalByName(name)
+
+  return isAccountExist(promise, res, sendNotFound)
+}
+
+function isAccountNameWithHostExist (nameWithDomain: string, res: Response, sendNotFound = true) {
+  const [ accountName, host ] = nameWithDomain.split('@')
+
+  let promise: Bluebird<AccountModel>
+  if (!host || host === CONFIG.WEBSERVER.HOST) promise = AccountModel.loadLocalByName(accountName)
+  else promise = AccountModel.loadByNameAndHost(accountName, host)
+
+  return isAccountExist(promise, res, sendNotFound)
+}
+
+async function isAccountExist (p: Bluebird<AccountModel>, res: Response, sendNotFound: boolean) {
+  const account = await p
+
+  if (!account) {
+    if (sendNotFound === true) {
+      res.status(404)
+         .send({ error: 'Account not found' })
+         .end()
     }
 
-    res.locals.account = account
-    callback()
-  })
-  .catch(err => {
-    logger.error('Error in video account request validator.', err)
-    return res.sendStatus(500)
-  })
+    return false
+  }
+
+  res.locals.account = account
+
+  return true
 }
 
 // ---------------------------------------------------------------------------
 
 export {
-  checkVideoAccountExists,
+  isAccountIdValid,
+  isAccountIdExist,
+  isLocalAccountNameExist,
+  isAccountDescriptionValid,
+  isAccountNameWithHostExist,
   isAccountNameValid
 }