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