mirror of https://github.com/Chocobozzz/PeerTube
112 lines
3.5 KiB
TypeScript
112 lines
3.5 KiB
TypeScript
import { logger } from '@server/helpers/logger'
|
|
import { YoutubeDLWrapper } from '@server/helpers/youtube-dl'
|
|
import { CONFIG } from '@server/initializers/config'
|
|
import { buildYoutubeDLImport } from '@server/lib/video-pre-import'
|
|
import { UserModel } from '@server/models/user/user'
|
|
import { VideoImportModel } from '@server/models/video/video-import'
|
|
import { MChannel, MChannelAccountDefault, MChannelSync } from '@server/types/models'
|
|
import { VideoChannelSyncState, VideoPrivacy } from '@shared/models'
|
|
import { CreateJobArgument, JobQueue } from './job-queue'
|
|
import { ServerConfigManager } from './server-config-manager'
|
|
|
|
export async function synchronizeChannel (options: {
|
|
channel: MChannelAccountDefault
|
|
externalChannelUrl: string
|
|
videosCountLimit: number
|
|
channelSync?: MChannelSync
|
|
onlyAfter?: Date
|
|
}) {
|
|
const { channel, externalChannelUrl, videosCountLimit, onlyAfter, channelSync } = options
|
|
|
|
if (channelSync) {
|
|
channelSync.state = VideoChannelSyncState.PROCESSING
|
|
channelSync.lastSyncAt = new Date()
|
|
await channelSync.save()
|
|
}
|
|
|
|
try {
|
|
const user = await UserModel.loadByChannelActorId(channel.actorId)
|
|
const youtubeDL = new YoutubeDLWrapper(
|
|
externalChannelUrl,
|
|
ServerConfigManager.Instance.getEnabledResolutions('vod'),
|
|
CONFIG.TRANSCODING.ALWAYS_TRANSCODE_ORIGINAL_RESOLUTION
|
|
)
|
|
|
|
const targetUrls = await youtubeDL.getInfoForListImport({ latestVideosCount: videosCountLimit })
|
|
|
|
logger.info(
|
|
'Fetched %d candidate URLs for sync channel %s.',
|
|
targetUrls.length, channel.Actor.preferredUsername, { targetUrls }
|
|
)
|
|
|
|
if (targetUrls.length === 0) {
|
|
if (channelSync) {
|
|
channelSync.state = VideoChannelSyncState.SYNCED
|
|
await channelSync.save()
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
const children: CreateJobArgument[] = []
|
|
|
|
for (const targetUrl of targetUrls) {
|
|
if (await skipImport(channel, targetUrl, onlyAfter)) continue
|
|
|
|
const { job } = await buildYoutubeDLImport({
|
|
user,
|
|
channel,
|
|
targetUrl,
|
|
channelSync,
|
|
importDataOverride: {
|
|
privacy: VideoPrivacy.PUBLIC
|
|
}
|
|
})
|
|
|
|
children.push(job)
|
|
}
|
|
|
|
// Will update the channel sync status
|
|
const parent: CreateJobArgument = {
|
|
type: 'after-video-channel-import',
|
|
payload: {
|
|
channelSyncId: channelSync?.id
|
|
}
|
|
}
|
|
|
|
await JobQueue.Instance.createJobWithChildren(parent, children)
|
|
} catch (err) {
|
|
logger.error(`Failed to import ${externalChannelUrl} in channel ${channel.name}`, { err })
|
|
channelSync.state = VideoChannelSyncState.FAILED
|
|
await channelSync.save()
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
async function skipImport (channel: MChannel, targetUrl: string, onlyAfter?: Date) {
|
|
if (await VideoImportModel.urlAlreadyImported(channel.id, targetUrl)) {
|
|
logger.debug('%s is already imported for channel %s, skipping video channel synchronization.', targetUrl, channel.name)
|
|
return true
|
|
}
|
|
|
|
if (onlyAfter) {
|
|
const youtubeDL = new YoutubeDLWrapper(
|
|
targetUrl,
|
|
ServerConfigManager.Instance.getEnabledResolutions('vod'),
|
|
CONFIG.TRANSCODING.ALWAYS_TRANSCODE_ORIGINAL_RESOLUTION
|
|
)
|
|
|
|
const videoInfo = await youtubeDL.getInfoForDownload()
|
|
|
|
const onlyAfterWithoutTime = new Date(onlyAfter)
|
|
onlyAfterWithoutTime.setHours(0, 0, 0, 0)
|
|
|
|
if (videoInfo.originallyPublishedAtWithoutTime.getTime() < onlyAfterWithoutTime.getTime()) {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|