Init commit

server/core/models/video/video-source.ts

@@ -0,0 +1,168 @@
+import { getResolutionLabel } from '@peertube/peertube-core-utils'
+import { ActivityVideoUrlObject, type FileStorageType, type VideoSource } from '@peertube/peertube-models'
+import { DOWNLOAD_PATHS, WEBSERVER } from '@server/initializers/constants.js'
+import { getVideoFileMimeType } from '@server/lib/video-file.js'
+import { MVideoSource } from '@server/types/models/video/video-source.js'
+import { extname, join } from 'path'
+import { Transaction } from 'sequelize'
+import { AllowNull, BelongsTo, Column, CreatedAt, DataType, ForeignKey, Table, UpdatedAt } from 'sequelize-typescript'
+import { SequelizeModel, doesExist, getSort } from '../shared/index.js'
+import { VideoModel } from './video.js'
+
+@Table({
+  tableName: 'videoSource',
+  indexes: [
+    {
+      fields: [ 'videoId' ]
+    },
+    {
+      fields: [ { name: 'createdAt', order: 'DESC' } ]
+    },
+    {
+      fields: [ 'keptOriginalFilename' ],
+      unique: true
+    }
+  ]
+})
+export class VideoSourceModel extends SequelizeModel<VideoSourceModel> {
+  @CreatedAt
+  declare createdAt: Date
+
+  @UpdatedAt
+  declare updatedAt: Date
+
+  // The name of the uploaded file
+  @AllowNull(false)
+  @Column
+  declare inputFilename: string
+
+  // The name of the file stored on disk
+  // null means we don't have the file
+  @AllowNull(true)
+  @Column
+  declare keptOriginalFilename: string
+
+  @AllowNull(true)
+  @Column
+  declare resolution: number
+
+  @AllowNull(true)
+  @Column
+  declare width: number
+
+  @AllowNull(true)
+  @Column
+  declare height: number
+
+  @AllowNull(true)
+  @Column
+  declare fps: number
+
+  @AllowNull(true)
+  @Column(DataType.BIGINT)
+  declare size: number
+
+  @AllowNull(true)
+  @Column(DataType.JSONB)
+  declare metadata: any
+
+  @AllowNull(true)
+  @Column
+  declare storage: FileStorageType
+
+  @AllowNull(true)
+  @Column
+  declare fileUrl: string
+
+  @ForeignKey(() => VideoModel)
+  @Column
+  declare videoId: number
+
+  @BelongsTo(() => VideoModel, {
+    foreignKey: {
+      allowNull: false
+    },
+    onDelete: 'cascade'
+  })
+  declare Video: Awaited<VideoModel>
+
+  static loadLatest (videoId: number, transaction?: Transaction) {
+    return VideoSourceModel.findOne<MVideoSource>({
+      where: { videoId },
+      order: getSort('-createdAt'),
+      transaction
+    })
+  }
+
+  static loadByKeptOriginalFilename (keptOriginalFilename: string) {
+    return VideoSourceModel.findOne<MVideoSource>({
+      where: { keptOriginalFilename }
+    })
+  }
+
+  static listAll (videoId: number, transaction?: Transaction) {
+    return VideoSourceModel.findAll<MVideoSource>({
+      where: { videoId },
+      transaction
+    })
+  }
+
+  // ---------------------------------------------------------------------------
+
+  static async doesOwnedFileExist (filename: string, storage: FileStorageType) {
+    const query = 'SELECT 1 FROM "videoSource" ' +
+      'INNER JOIN "video" ON "video"."id" = "videoSource"."videoId" AND "video"."remote" IS FALSE ' +
+      `WHERE "keptOriginalFilename" = $filename AND "storage" = $storage LIMIT 1`
+
+    return doesExist({ sequelize: this.sequelize, query, bind: { filename, storage } })
+  }
+
+  // ---------------------------------------------------------------------------
+
+  getFileDownloadUrl () {
+    if (!this.keptOriginalFilename) return null
+
+    return WEBSERVER.URL + join(DOWNLOAD_PATHS.ORIGINAL_VIDEO_FILE, this.keptOriginalFilename)
+  }
+
+  toActivityPubObject (this: MVideoSource): ActivityVideoUrlObject {
+    const mimeType = getVideoFileMimeType(extname(this.inputFilename), false)
+
+    return {
+      type: 'Link',
+      mediaType: mimeType as ActivityVideoUrlObject['mediaType'],
+      href: null,
+      height: this.height || this.resolution,
+      width: this.width,
+      size: this.size,
+      fps: this.fps,
+      attachment: []
+    }
+  }
+
+  toFormattedJSON (this: MVideoSource): VideoSource {
+    return {
+      inputFilename: this.inputFilename,
+
+      fileUrl: this.fileUrl,
+      fileDownloadUrl: this.getFileDownloadUrl(),
+
+      resolution: {
+        id: this.resolution,
+        label: this.resolution !== null
+          ? getResolutionLabel({ resolution: this.resolution, height: this.height, width: this.width })
+          : null
+      },
+      size: this.size,
+
+      width: this.width,
+      height: this.height,
+
+      fps: this.fps,
+
+      metadata: this.metadata,
+
+      createdAt: this.createdAt.toISOString()
+    }
+  }
+}