mirror of https://github.com/Chocobozzz/PeerTube
				
				
				
			
		
			
				
	
	
		
			125 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			125 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			TypeScript
		
	
	
import { Transaction } from 'sequelize'
 | 
						|
import { VideoPrivacy } from '../../../shared/models/videos'
 | 
						|
import { getServerActor } from '../../helpers/utils'
 | 
						|
import { VideoModel } from '../../models/video/video'
 | 
						|
import { VideoShareModel } from '../../models/video/video-share'
 | 
						|
import { sendUndoAnnounce, sendVideoAnnounce } from './send'
 | 
						|
import { getVideoAnnounceActivityPubUrl } from './url'
 | 
						|
import { VideoChannelModel } from '../../models/video/video-channel'
 | 
						|
import * as Bluebird from 'bluebird'
 | 
						|
import { doRequest } from '../../helpers/requests'
 | 
						|
import { getOrCreateActorAndServerAndModel } from './actor'
 | 
						|
import { logger } from '../../helpers/logger'
 | 
						|
import { CRAWL_REQUEST_CONCURRENCY } from '../../initializers'
 | 
						|
import { checkUrlsSameHost, getAPUrl } from '../../helpers/activitypub'
 | 
						|
 | 
						|
async function shareVideoByServerAndChannel (video: VideoModel, t: Transaction) {
 | 
						|
  if (video.privacy === VideoPrivacy.PRIVATE) return undefined
 | 
						|
 | 
						|
  return Promise.all([
 | 
						|
    shareByServer(video, t),
 | 
						|
    shareByVideoChannel(video, t)
 | 
						|
  ])
 | 
						|
}
 | 
						|
 | 
						|
async function changeVideoChannelShare (video: VideoModel, oldVideoChannel: VideoChannelModel, t: Transaction) {
 | 
						|
  logger.info('Updating video channel of video %s: %s -> %s.', video.uuid, oldVideoChannel.name, video.VideoChannel.name)
 | 
						|
 | 
						|
  await undoShareByVideoChannel(video, oldVideoChannel, t)
 | 
						|
 | 
						|
  await shareByVideoChannel(video, t)
 | 
						|
}
 | 
						|
 | 
						|
async function addVideoShares (shareUrls: string[], instance: VideoModel) {
 | 
						|
  await Bluebird.map(shareUrls, async shareUrl => {
 | 
						|
    try {
 | 
						|
      // Fetch url
 | 
						|
      const { body } = await doRequest({
 | 
						|
        uri: shareUrl,
 | 
						|
        json: true,
 | 
						|
        activityPub: true
 | 
						|
      })
 | 
						|
      if (!body || !body.actor) throw new Error('Body or body actor is invalid')
 | 
						|
 | 
						|
      const actorUrl = getAPUrl(body.actor)
 | 
						|
      if (checkUrlsSameHost(shareUrl, actorUrl) !== true) {
 | 
						|
        throw new Error(`Actor url ${actorUrl} has not the same host than the share url ${shareUrl}`)
 | 
						|
      }
 | 
						|
 | 
						|
      const actor = await getOrCreateActorAndServerAndModel(actorUrl)
 | 
						|
 | 
						|
      const entry = {
 | 
						|
        actorId: actor.id,
 | 
						|
        videoId: instance.id,
 | 
						|
        url: shareUrl
 | 
						|
      }
 | 
						|
 | 
						|
      await VideoShareModel.findOrCreate({
 | 
						|
        where: {
 | 
						|
          url: shareUrl
 | 
						|
        },
 | 
						|
        defaults: entry
 | 
						|
      })
 | 
						|
    } catch (err) {
 | 
						|
      logger.warn('Cannot add share %s.', shareUrl, { err })
 | 
						|
    }
 | 
						|
  }, { concurrency: CRAWL_REQUEST_CONCURRENCY })
 | 
						|
}
 | 
						|
 | 
						|
export {
 | 
						|
  changeVideoChannelShare,
 | 
						|
  addVideoShares,
 | 
						|
  shareVideoByServerAndChannel
 | 
						|
}
 | 
						|
 | 
						|
// ---------------------------------------------------------------------------
 | 
						|
 | 
						|
async function shareByServer (video: VideoModel, t: Transaction) {
 | 
						|
  const serverActor = await getServerActor()
 | 
						|
 | 
						|
  const serverShareUrl = getVideoAnnounceActivityPubUrl(serverActor, video)
 | 
						|
  return VideoShareModel.findOrCreate({
 | 
						|
    defaults: {
 | 
						|
      actorId: serverActor.id,
 | 
						|
      videoId: video.id,
 | 
						|
      url: serverShareUrl
 | 
						|
    },
 | 
						|
    where: {
 | 
						|
      url: serverShareUrl
 | 
						|
    },
 | 
						|
    transaction: t
 | 
						|
  }).then(([ serverShare, created ]) => {
 | 
						|
    if (created) return sendVideoAnnounce(serverActor, serverShare, video, t)
 | 
						|
 | 
						|
    return undefined
 | 
						|
  })
 | 
						|
}
 | 
						|
 | 
						|
async function shareByVideoChannel (video: VideoModel, t: Transaction) {
 | 
						|
  const videoChannelShareUrl = getVideoAnnounceActivityPubUrl(video.VideoChannel.Actor, video)
 | 
						|
  return VideoShareModel.findOrCreate({
 | 
						|
    defaults: {
 | 
						|
      actorId: video.VideoChannel.actorId,
 | 
						|
      videoId: video.id,
 | 
						|
      url: videoChannelShareUrl
 | 
						|
    },
 | 
						|
    where: {
 | 
						|
      url: videoChannelShareUrl
 | 
						|
    },
 | 
						|
    transaction: t
 | 
						|
  }).then(([ videoChannelShare, created ]) => {
 | 
						|
    if (created) return sendVideoAnnounce(video.VideoChannel.Actor, videoChannelShare, video, t)
 | 
						|
 | 
						|
    return undefined
 | 
						|
  })
 | 
						|
}
 | 
						|
 | 
						|
async function undoShareByVideoChannel (video: VideoModel, oldVideoChannel: VideoChannelModel, t: Transaction) {
 | 
						|
  // Load old share
 | 
						|
  const oldShare = await VideoShareModel.load(oldVideoChannel.actorId, video.id, t)
 | 
						|
  if (!oldShare) return new Error('Cannot find old video channel share ' + oldVideoChannel.actorId + ' for video ' + video.id)
 | 
						|
 | 
						|
  await sendUndoAnnounce(oldVideoChannel.Actor, oldShare, video, t)
 | 
						|
  await oldShare.destroy({ transaction: t })
 | 
						|
}
 |