server/controllers/bots.ts

   1 import * as express from 'express'
   2 import { asyncMiddleware } from '../middlewares'
   3 import { CONFIG, ROUTE_CACHE_LIFETIME } from '../initializers'
   4 import * as sitemapModule from 'sitemap'
   5 import { logger } from '../helpers/logger'
   6 import { VideoModel } from '../models/video/video'
   7 import { VideoChannelModel } from '../models/video/video-channel'
   8 import { AccountModel } from '../models/account/account'
   9 import { cacheRoute } from '../middlewares/cache'
  10 import { buildNSFWFilter } from '../helpers/express-utils'
  11 import { truncate } from 'lodash'
  12
  13 const botsRouter = express.Router()
  14
  15 // Special route that add OpenGraph and oEmbed tags
  16 // Do not use a template engine for a so little thing
  17 botsRouter.use('/sitemap.xml',
  18   asyncMiddleware(cacheRoute(ROUTE_CACHE_LIFETIME.SITEMAP)),
  19   asyncMiddleware(getSitemap)
  20 )
  21
  22 // ---------------------------------------------------------------------------
  23
  24 export {
  25   botsRouter
  26 }
  27
  28 // ---------------------------------------------------------------------------
  29
  30 async function getSitemap (req: express.Request, res: express.Response) {
  31   let urls = getSitemapBasicUrls()
  32
  33   urls = urls.concat(await getSitemapLocalVideoUrls())
  34   urls = urls.concat(await getSitemapVideoChannelUrls())
  35   urls = urls.concat(await getSitemapAccountUrls())
  36
  37   const sitemap = sitemapModule.createSitemap({
  38     hostname: CONFIG.WEBSERVER.URL,
  39     urls: urls
  40   })
  41
  42   sitemap.toXML((err, xml) => {
  43     if (err) {
  44       logger.error('Cannot generate sitemap.', { err })
  45       return res.sendStatus(500)
  46     }
  47
  48     res.header('Content-Type', 'application/xml')
  49     res.send(xml)
  50   })
  51 }
  52
  53 async function getSitemapVideoChannelUrls () {
  54   const rows = await VideoChannelModel.listLocalsForSitemap('createdAt')
  55
  56   return rows.map(channel => ({
  57     url: CONFIG.WEBSERVER.URL + '/video-channels/' + channel.Actor.preferredUsername
  58   }))
  59 }
  60
  61 async function getSitemapAccountUrls () {
  62   const rows = await AccountModel.listLocalsForSitemap('createdAt')
  63
  64   return rows.map(channel => ({
  65     url: CONFIG.WEBSERVER.URL + '/accounts/' + channel.Actor.preferredUsername
  66   }))
  67 }
  68
  69 async function getSitemapLocalVideoUrls () {
  70   const resultList = await VideoModel.listForApi({
  71     start: 0,
  72     count: undefined,
  73     sort: 'createdAt',
  74     includeLocalVideos: true,
  75     nsfw: buildNSFWFilter(),
  76     filter: 'local',
  77     withFiles: false
  78   })
  79
  80   return resultList.data.map(v => ({
  81     url: CONFIG.WEBSERVER.URL + '/videos/watch/' + v.uuid,
  82     video: [
  83       {
  84         title: v.name,
  85         // Sitemap description should be < 2000 characters
  86         description: truncate(v.description || v.name, { length: 2000, omission: '...' }),
  87         player_loc: CONFIG.WEBSERVER.URL + '/videos/embed/' + v.uuid,
  88         thumbnail_loc: CONFIG.WEBSERVER.URL + v.getThumbnailStaticPath()
  89       }
  90     ]
  91   }))
  92 }
  93
  94 function getSitemapBasicUrls () {
  95   const paths = [
  96     '/about/instance',
  97     '/videos/local'
  98   ]
  99
 100   return paths.map(p => ({ url: CONFIG.WEBSERVER.URL + p }))
 101 }