import express from 'express' import { Server } from 'http' import { join } from 'path' import { ffprobePromise } from '@server/helpers/ffmpeg/ffprobe-utils' import { buildLogger } from '@server/helpers/logger' import { CONFIG } from '@server/initializers/config' import { WEBSERVER } from '@server/initializers/constants' import { sequelizeTypescript } from '@server/initializers/database' import { AccountModel } from '@server/models/account/account' import { AccountBlocklistModel } from '@server/models/account/account-blocklist' import { getServerActor } from '@server/models/application/application' import { ServerModel } from '@server/models/server/server' import { ServerBlocklistModel } from '@server/models/server/server-blocklist' import { UserModel } from '@server/models/user/user' import { VideoModel } from '@server/models/video/video' import { VideoBlacklistModel } from '@server/models/video/video-blacklist' import { MPlugin, MVideo, UserNotificationModelForApi } from '@server/types/models' import { PeerTubeHelpers } from '@server/types/plugins' import { VideoBlacklistCreate, VideoStorage } from '@shared/models' import { addAccountInBlocklist, addServerInBlocklist, removeAccountFromBlocklist, removeServerFromBlocklist } from '../blocklist' import { PeerTubeSocket } from '../peertube-socket' import { ServerConfigManager } from '../server-config-manager' import { blacklistVideo, unblacklistVideo } from '../video-blacklist' import { VideoPathManager } from '../video-path-manager' function buildPluginHelpers (httpServer: Server, pluginModel: MPlugin, npmName: string): PeerTubeHelpers { const logger = buildPluginLogger(npmName) const database = buildDatabaseHelpers() const videos = buildVideosHelpers() const config = buildConfigHelpers() const server = buildServerHelpers(httpServer) const moderation = buildModerationHelpers() const plugin = buildPluginRelatedHelpers(pluginModel, npmName) const socket = buildSocketHelpers() const user = buildUserHelpers() return { logger, database, videos, config, moderation, plugin, server, socket, user } } export { buildPluginHelpers } // --------------------------------------------------------------------------- function buildPluginLogger (npmName: string) { return buildLogger(npmName) } function buildDatabaseHelpers () { return { query: sequelizeTypescript.query.bind(sequelizeTypescript) } } function buildServerHelpers (httpServer: Server) { return { getHTTPServer: () => httpServer, getServerActor: () => getServerActor() } } function buildVideosHelpers () { return { loadByUrl: (url: string) => { return VideoModel.loadByUrl(url) }, loadByIdOrUUID: (id: number | string) => { return VideoModel.load(id) }, removeVideo: (id: number) => { return sequelizeTypescript.transaction(async t => { const video = await VideoModel.loadFull(id, t) await video.destroy({ transaction: t }) }) }, ffprobe: (path: string) => { return ffprobePromise(path) }, getFiles: async (id: number | string) => { const video = await VideoModel.loadFull(id) if (!video) return undefined const webtorrentVideoFiles = (video.VideoFiles || []).map(f => ({ path: f.storage === VideoStorage.FILE_SYSTEM ? VideoPathManager.Instance.getFSVideoFileOutputPath(video, f) : null, url: f.getFileUrl(video), resolution: f.resolution, size: f.size, fps: f.fps })) const hls = video.getHLSPlaylist() const hlsVideoFiles = hls ? (video.getHLSPlaylist().VideoFiles || []).map(f => { return { path: f.storage === VideoStorage.FILE_SYSTEM ? VideoPathManager.Instance.getFSVideoFileOutputPath(hls, f) : null, url: f.getFileUrl(video), resolution: f.resolution, size: f.size, fps: f.fps } }) : [] const thumbnails = video.Thumbnails.map(t => ({ type: t.type, url: t.getFileUrl(video), path: t.getPath() })) return { webtorrent: { videoFiles: webtorrentVideoFiles }, hls: { videoFiles: hlsVideoFiles }, thumbnails } } } } function buildModerationHelpers () { return { blockServer: async (options: { byAccountId: number, hostToBlock: string }) => { const serverToBlock = await ServerModel.loadOrCreateByHost(options.hostToBlock) await addServerInBlocklist(options.byAccountId, serverToBlock.id) }, unblockServer: async (options: { byAccountId: number, hostToUnblock: string }) => { const serverBlock = await ServerBlocklistModel.loadByAccountAndHost(options.byAccountId, options.hostToUnblock) if (!serverBlock) return await removeServerFromBlocklist(serverBlock) }, blockAccount: async (options: { byAccountId: number, handleToBlock: string }) => { const accountToBlock = await AccountModel.loadByNameWithHost(options.handleToBlock) if (!accountToBlock) return await addAccountInBlocklist(options.byAccountId, accountToBlock.id) }, unblockAccount: async (options: { byAccountId: number, handleToUnblock: string }) => { const targetAccount = await AccountModel.loadByNameWithHost(options.handleToUnblock) if (!targetAccount) return const accountBlock = await AccountBlocklistModel.loadByAccountAndTarget(options.byAccountId, targetAccount.id) if (!accountBlock) return await removeAccountFromBlocklist(accountBlock) }, blacklistVideo: async (options: { videoIdOrUUID: number | string, createOptions: VideoBlacklistCreate }) => { const video = await VideoModel.loadFull(options.videoIdOrUUID) if (!video) return await blacklistVideo(video, options.createOptions) }, unblacklistVideo: async (options: { videoIdOrUUID: number | string }) => { const video = await VideoModel.loadFull(options.videoIdOrUUID) if (!video) return const videoBlacklist = await VideoBlacklistModel.loadByVideoId(video.id) if (!videoBlacklist) return await unblacklistVideo(videoBlacklist, video) } } } function buildConfigHelpers () { return { getWebserverUrl () { return WEBSERVER.URL }, getServerConfig () { return ServerConfigManager.Instance.getServerConfig() } } } function buildPluginRelatedHelpers (plugin: MPlugin, npmName: string) { return { getBaseStaticRoute: () => `/plugins/${plugin.name}/${plugin.version}/static/`, getBaseRouterRoute: () => `/plugins/${plugin.name}/${plugin.version}/router/`, getBaseWebSocketRoute: () => `/plugins/${plugin.name}/${plugin.version}/ws/`, getDataDirectoryPath: () => join(CONFIG.STORAGE.PLUGINS_DIR, 'data', npmName) } } function buildSocketHelpers () { return { sendNotification: (userId: number, notification: UserNotificationModelForApi) => { PeerTubeSocket.Instance.sendNotification(userId, notification) }, sendVideoLiveNewState: (video: MVideo) => { PeerTubeSocket.Instance.sendVideoLiveNewState(video) } } } function buildUserHelpers () { return { loadById: (id: number) => { return UserModel.loadByIdFull(id) }, getAuthUser: (res: express.Response) => { const user = res.locals.oauth?.token?.User if (!user) return undefined return UserModel.loadByIdFull(user.id) } } }