server/controllers/api/pods.ts

   1 import * as express from 'express'
   2
   3 import { database as db } from '../../initializers/database'
   4 import { logger, getFormattedObjects } from '../../helpers'
   5 import {
   6   makeFriends,
   7   quitFriends,
   8   removeFriend
   9 } from '../../lib'
  10 import {
  11   authenticate,
  12   ensureIsAdmin,
  13   makeFriendsValidator,
  14   setBodyHostsPort,
  15   podRemoveValidator,
  16   paginationValidator,
  17   setPagination,
  18   setPodsSort,
  19   podsSortValidator,
  20   asyncMiddleware
  21 } from '../../middlewares'
  22 import { PodInstance } from '../../models'
  23
  24 const podsRouter = express.Router()
  25
  26 podsRouter.get('/',
  27   paginationValidator,
  28   podsSortValidator,
  29   setPodsSort,
  30   setPagination,
  31   asyncMiddleware(listPods)
  32 )
  33 podsRouter.post('/make-friends',
  34   authenticate,
  35   ensureIsAdmin,
  36   makeFriendsValidator,
  37   setBodyHostsPort,
  38   asyncMiddleware(makeFriendsController)
  39 )
  40 podsRouter.get('/quit-friends',
  41   authenticate,
  42   ensureIsAdmin,
  43   asyncMiddleware(quitFriendsController)
  44 )
  45 podsRouter.delete('/:id',
  46   authenticate,
  47   ensureIsAdmin,
  48   podRemoveValidator,
  49   asyncMiddleware(removeFriendController)
  50 )
  51
  52 // ---------------------------------------------------------------------------
  53
  54 export {
  55   podsRouter
  56 }
  57
  58 // ---------------------------------------------------------------------------
  59
  60 async function listPods (req: express.Request, res: express.Response, next: express.NextFunction) {
  61   const resultList = await db.Pod.listForApi(req.query.start, req.query.count, req.query.sort)
  62
  63   return res.json(getFormattedObjects(resultList.data, resultList.total))
  64 }
  65
  66 async function makeFriendsController (req: express.Request, res: express.Response, next: express.NextFunction) {
  67   const hosts = req.body.hosts as string[]
  68
  69   // Don't wait the process that could be long
  70   makeFriends(hosts)
  71     .then(() => logger.info('Made friends!'))
  72     .catch(err => logger.error('Could not make friends.', err))
  73
  74   return res.type('json').status(204).end()
  75 }
  76
  77 async function quitFriendsController (req: express.Request, res: express.Response, next: express.NextFunction) {
  78   await quitFriends()
  79
  80   return res.type('json').status(204).end()
  81 }
  82
  83 async function removeFriendController (req: express.Request, res: express.Response, next: express.NextFunction) {
  84   const pod = res.locals.pod as PodInstance
  85
  86   await removeFriend(pod)
  87
  88   return res.type('json').status(204).end()
  89 }