-import * as express from 'express'
-import { asyncMiddleware } from '../middlewares'
+import { getServerActor } from '@server/models/application/application'
+import { logger } from '@uploadx/core'
+import express from 'express'
+import { truncate } from 'lodash'
+import { SitemapStream, streamToPromise, ErrorLevel } from 'sitemap'
+import { buildNSFWFilter } from '../helpers/express-utils'
import { ROUTE_CACHE_LIFETIME, WEBSERVER } from '../initializers/constants'
-import * as sitemapModule from 'sitemap'
-import { logger } from '../helpers/logger'
+import { asyncMiddleware } from '../middlewares'
+import { cacheRoute } from '../middlewares/cache/cache'
+import { AccountModel } from '../models/account/account'
import { VideoModel } from '../models/video/video'
import { VideoChannelModel } from '../models/video/video-channel'
-import { AccountModel } from '../models/account/account'
-import { cacheRoute } from '../middlewares/cache'
-import { buildNSFWFilter } from '../helpers/express-utils'
-import { truncate } from 'lodash'
const botsRouter = express.Router()
// Special route that add OpenGraph and oEmbed tags
// Do not use a template engine for a so little thing
botsRouter.use('/sitemap.xml',
- asyncMiddleware(cacheRoute(ROUTE_CACHE_LIFETIME.SITEMAP)),
+ cacheRoute(ROUTE_CACHE_LIFETIME.SITEMAP),
asyncMiddleware(getSitemap)
)
urls = urls.concat(await getSitemapVideoChannelUrls())
urls = urls.concat(await getSitemapAccountUrls())
- const sitemap = sitemapModule.createSitemap({
+ const sitemapStream = new SitemapStream({
hostname: WEBSERVER.URL,
- urls: urls
- })
+ errorHandler: (err: Error, level: ErrorLevel) => {
+ if (level === 'warn') {
+ logger.warn('Warning in sitemap generation.', { err })
+ } else if (level === 'throw') {
+ logger.error('Error in sitemap generation.', { err })
- sitemap.toXML((err, xml) => {
- if (err) {
- logger.error('Cannot generate sitemap.', { err })
- return res.sendStatus(500)
+ throw err
+ }
}
-
- res.header('Content-Type', 'application/xml')
- res.send(xml)
})
+
+ for (const urlObj of urls) {
+ sitemapStream.write(urlObj)
+ }
+ sitemapStream.end()
+
+ const xml = await streamToPromise(sitemapStream)
+
+ res.header('Content-Type', 'application/xml')
+ res.send(xml)
}
async function getSitemapVideoChannelUrls () {
}
async function getSitemapLocalVideoUrls () {
- const resultList = await VideoModel.listForApi({
+ const serverActor = await getServerActor()
+
+ const { data } = await VideoModel.listForApi({
start: 0,
count: undefined,
sort: 'createdAt',
- includeLocalVideos: true,
+ displayOnlyForFollower: {
+ actorId: serverActor.id,
+ orLocalVideos: true
+ },
+ isLocal: true,
nsfw: buildNSFWFilter(),
- filter: 'local',
- withFiles: false
+ countVideos: false
})
- return resultList.data.map(v => ({
- url: WEBSERVER.URL + '/videos/watch/' + v.uuid,
+ return data.map(v => ({
+ url: WEBSERVER.URL + v.getWatchStaticPath(),
video: [
{
- title: v.name,
+ // Sitemap title should be < 100 characters
+ title: truncate(v.name, { length: 100, omission: '...' }),
// Sitemap description should be < 2000 characters
description: truncate(v.description || v.name, { length: 2000, omission: '...' }),
- player_loc: WEBSERVER.URL + '/videos/embed/' + v.uuid,
- thumbnail_loc: WEBSERVER.URL + v.getThumbnailStaticPath()
+ player_loc: WEBSERVER.URL + v.getEmbedStaticPath(),
+ thumbnail_loc: WEBSERVER.URL + v.getMiniatureStaticPath()
}
]
}))