Chocobozzz/PeerTube

import { CONFIG } from '@server/initializers/config.js'
import { MStreamingPlaylistVideo, MUserExport, MVideoFile } from '@server/types/models/index.js'
import { MVideoSource } from '@server/types/models/video/video-source.js'
import {
  generateHLSObjectStorageKey,
  generateOriginalVideoObjectStorageKey,
  generateUserExportObjectStorageKey,
  generateWebVideoObjectStorageKey
} from './keys.js'
import { buildKey, getClient } from './shared/index.js'
import { getObjectStoragePublicFileUrl } from './urls.js'

export async function generateWebVideoPresignedUrl (options: {
  file: MVideoFile
  downloadFilename: string
}) {
  const { file, downloadFilename } = options

  const url = await generatePresignedUrl({
    bucket: CONFIG.OBJECT_STORAGE.WEB_VIDEOS.BUCKET_NAME,
    key: buildKey(generateWebVideoObjectStorageKey(file.filename), CONFIG.OBJECT_STORAGE.WEB_VIDEOS),
    downloadFilename
  })

  return getObjectStoragePublicFileUrl(url, CONFIG.OBJECT_STORAGE.WEB_VIDEOS)
}

export async function generateHLSFilePresignedUrl (options: {
  streamingPlaylist: MStreamingPlaylistVideo
  file: MVideoFile
  downloadFilename: string
}) {
  const { streamingPlaylist, file, downloadFilename } = options

  const url = await generatePresignedUrl({
    bucket: CONFIG.OBJECT_STORAGE.STREAMING_PLAYLISTS.BUCKET_NAME,
    key: buildKey(generateHLSObjectStorageKey(streamingPlaylist, file.filename), CONFIG.OBJECT_STORAGE.STREAMING_PLAYLISTS),
    downloadFilename
  })

  return getObjectStoragePublicFileUrl(url, CONFIG.OBJECT_STORAGE.STREAMING_PLAYLISTS)
}

export async function generateUserExportPresignedUrl (options: {
  userExport: MUserExport
  downloadFilename: string
}) {
  const { userExport, downloadFilename } = options

  const url = await generatePresignedUrl({
    bucket: CONFIG.OBJECT_STORAGE.USER_EXPORTS.BUCKET_NAME,
    key: buildKey(generateUserExportObjectStorageKey(userExport.filename), CONFIG.OBJECT_STORAGE.USER_EXPORTS),
    downloadFilename
  })

  return getObjectStoragePublicFileUrl(url, CONFIG.OBJECT_STORAGE.USER_EXPORTS)
}

export async function generateOriginalFilePresignedUrl (options: {
  videoSource: MVideoSource
  downloadFilename: string
}) {
  const { videoSource, downloadFilename } = options

  const url = await generatePresignedUrl({
    bucket: CONFIG.OBJECT_STORAGE.ORIGINAL_VIDEO_FILES.BUCKET_NAME,
    key: buildKey(generateOriginalVideoObjectStorageKey(videoSource.keptOriginalFilename), CONFIG.OBJECT_STORAGE.ORIGINAL_VIDEO_FILES),
    downloadFilename
  })

  return getObjectStoragePublicFileUrl(url, CONFIG.OBJECT_STORAGE.ORIGINAL_VIDEO_FILES)
}

// ---------------------------------------------------------------------------
// Private
// ---------------------------------------------------------------------------

async function generatePresignedUrl (options: {
  bucket: string
  key: string
  downloadFilename: string
}) {
  const { bucket, downloadFilename, key } = options

  const { GetObjectCommand } = await import('@aws-sdk/client-s3')
  const { getSignedUrl } = await import('@aws-sdk/s3-request-presigner')

  const command = new GetObjectCommand({
    Bucket: bucket,
    Key: key,
    ResponseContentDisposition: `attachment; filename="${encodeURI(downloadFilename)}"`
  })

  return getSignedUrl(await getClient(), command, { expiresIn: 3600 * 24 })
}