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