server/controllers/api/server/debug.ts

   1 import express from 'express'
   2 import { InboxManager } from '@server/lib/activitypub/inbox-manager'
   3 import { RemoveDanglingResumableUploadsScheduler } from '@server/lib/schedulers/remove-dangling-resumable-uploads-scheduler'
   4 import { VideoViewsBufferScheduler } from '@server/lib/schedulers/video-views-buffer-scheduler'
   5 import { VideoViewsManager } from '@server/lib/views/video-views-manager'
   6 import { Debug, SendDebugCommand } from '@shared/models'
   7 import { HttpStatusCode } from '../../../../shared/models/http/http-error-codes'
   8 import { UserRight } from '../../../../shared/models/users'
   9 import { authenticate, ensureUserHasRight } from '../../../middlewares'
  10
  11 const debugRouter = express.Router()
  12
  13 debugRouter.get('/debug',
  14   authenticate,
  15   ensureUserHasRight(UserRight.MANAGE_DEBUG),
  16   getDebug
  17 )
  18
  19 debugRouter.post('/debug/run-command',
  20   authenticate,
  21   ensureUserHasRight(UserRight.MANAGE_DEBUG),
  22   runCommand
  23 )
  24
  25 // ---------------------------------------------------------------------------
  26
  27 export {
  28   debugRouter
  29 }
  30
  31 // ---------------------------------------------------------------------------
  32
  33 function getDebug (req: express.Request, res: express.Response) {
  34   return res.json({
  35     ip: req.ip,
  36     activityPubMessagesWaiting: InboxManager.Instance.getActivityPubMessagesWaiting()
  37   } as Debug)
  38 }
  39
  40 async function runCommand (req: express.Request, res: express.Response) {
  41   const body: SendDebugCommand = req.body
  42
  43   const processors: { [id in SendDebugCommand['command']]: () => Promise<any> } = {
  44     'remove-dandling-resumable-uploads': () => RemoveDanglingResumableUploadsScheduler.Instance.execute(),
  45     'process-video-views-buffer': () => VideoViewsBufferScheduler.Instance.execute(),
  46     'process-video-viewers': () => VideoViewsManager.Instance.processViewerStats()
  47   }
  48
  49   await processors[body.command]()
  50
  51   return res.status(HttpStatusCode.NO_CONTENT_204).end()
  52 }