mirror of https://github.com/Chocobozzz/PeerTube
				
				
				
			
		
			
				
	
	
		
			82 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			82 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			TypeScript
		
	
	
| import { logger, loggerTagsFactory } from '@server/helpers/logger'
 | |
| import { PromiseCache } from '@server/helpers/promise-cache'
 | |
| import { PeerTubeRequestError } from '@server/helpers/requests'
 | |
| import { ActorLoadByUrlType } from '@server/lib/model-loaders'
 | |
| import { ActorModel } from '@server/models/actor/actor'
 | |
| import { MActorAccountChannelId, MActorFull } from '@server/types/models'
 | |
| import { HttpStatusCode } from '@shared/models'
 | |
| import { fetchRemoteActor } from './shared'
 | |
| import { APActorUpdater } from './updater'
 | |
| import { getUrlFromWebfinger } from './webfinger'
 | |
| 
 | |
| type RefreshResult <T> = Promise<{ actor: T | MActorFull, refreshed: boolean }>
 | |
| 
 | |
| type RefreshOptions <T> = {
 | |
|   actor: T
 | |
|   fetchedType: ActorLoadByUrlType
 | |
| }
 | |
| 
 | |
| const promiseCache = new PromiseCache(doRefresh, (options: RefreshOptions<MActorFull | MActorAccountChannelId>) => options.actor.url)
 | |
| 
 | |
| function refreshActorIfNeeded <T extends MActorFull | MActorAccountChannelId> (options: RefreshOptions<T>): RefreshResult <T> {
 | |
|   const actorArg = options.actor
 | |
|   if (!actorArg.isOutdated()) return Promise.resolve({ actor: actorArg, refreshed: false })
 | |
| 
 | |
|   return promiseCache.run(options)
 | |
| }
 | |
| 
 | |
| export {
 | |
|   refreshActorIfNeeded
 | |
| }
 | |
| 
 | |
| // ---------------------------------------------------------------------------
 | |
| 
 | |
| async function doRefresh <T extends MActorFull | MActorAccountChannelId> (options: RefreshOptions<T>): RefreshResult <MActorFull> {
 | |
|   const { actor: actorArg, fetchedType } = options
 | |
| 
 | |
|   // We need more attributes
 | |
|   const actor = fetchedType === 'all'
 | |
|     ? actorArg as MActorFull
 | |
|     : await ActorModel.loadByUrlAndPopulateAccountAndChannel(actorArg.url)
 | |
| 
 | |
|   const lTags = loggerTagsFactory('ap', 'actor', 'refresh', actor.url)
 | |
| 
 | |
|   logger.info('Refreshing actor %s.', actor.url, lTags())
 | |
| 
 | |
|   try {
 | |
|     const actorUrl = await getActorUrl(actor)
 | |
|     const { actorObject } = await fetchRemoteActor(actorUrl)
 | |
| 
 | |
|     if (actorObject === undefined) {
 | |
|       logger.warn('Cannot fetch remote actor in refresh actor.')
 | |
|       return { actor, refreshed: false }
 | |
|     }
 | |
| 
 | |
|     const updater = new APActorUpdater(actorObject, actor)
 | |
|     await updater.update()
 | |
| 
 | |
|     return { refreshed: true, actor }
 | |
|   } catch (err) {
 | |
|     if ((err as PeerTubeRequestError).statusCode === HttpStatusCode.NOT_FOUND_404) {
 | |
|       logger.info('Deleting actor %s because there is a 404 in refresh actor.', actor.url, lTags())
 | |
| 
 | |
|       actor.Account
 | |
|         ? await actor.Account.destroy()
 | |
|         : await actor.VideoChannel.destroy()
 | |
| 
 | |
|       return { actor: undefined, refreshed: false }
 | |
|     }
 | |
| 
 | |
|     logger.warn('Cannot refresh actor %s.', actor.url, { err, ...lTags() })
 | |
|     return { actor, refreshed: false }
 | |
|   }
 | |
| }
 | |
| 
 | |
| function getActorUrl (actor: MActorFull) {
 | |
|   return getUrlFromWebfinger(actor.preferredUsername + '@' + actor.getHost())
 | |
|     .catch(err => {
 | |
|       logger.warn('Cannot get actor URL from webfinger, keeping the old one.', err)
 | |
|       return actor.url
 | |
|     })
 | |
| }
 |