Init commit

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

@@ -0,0 +1,214 @@
+import { FindAndCountOptions, Op, Transaction } from 'sequelize'
+import {
+  AllowNull,
+  BelongsTo,
+  Column,
+  CreatedAt,
+  DataType,
+  Default,
+  ForeignKey,
+  Table,
+  UpdatedAt
+} from 'sequelize-typescript'
+import { MLiveVideo } from '@server/types/models/index.js'
+import { UserModel } from '../user/user.js'
+import { SequelizeModel } from '../shared/index.js'
+
+export enum LiveVideoStatus {
+  SCHEDULED = 'scheduled',
+  CANCELLED = 'cancelled',
+  COMPLETED = 'completed'
+}
+
+@Table({
+  tableName: 'liveVideo',
+  indexes: [
+    {
+      fields: [ 'targetCamera' ]
+    },
+    {
+      fields: [ 'startTime' ]
+    },
+    {
+      fields: [ 'status' ]
+    }
+  ]
+})
+export class LiveVideoModel extends SequelizeModel<LiveVideoModel> {
+  @AllowNull(false)
+  @Column
+  declare title: string
+
+  @AllowNull(true)
+  @Column
+  declare thumbnailFilename: string | null
+
+  @AllowNull(true)
+  @Column(DataType.BLOB)
+  declare thumbnailData: Buffer | null
+
+  @AllowNull(true)
+  @Column
+  declare thumbnailMimeType: string | null
+
+  @AllowNull(true)
+  @Column(DataType.BIGINT)
+  declare thumbnailSize: number | null
+
+  @AllowNull(true)
+  @Column
+  declare thumbnailEtag: string | null
+
+  @AllowNull(false)
+  @Column
+  declare videoOriginalName: string
+
+  @AllowNull(false)
+  @Column(DataType.TEXT)
+  declare videoStoragePath: string
+
+  @AllowNull(false)
+  @Column
+  declare videoMimeType: string
+
+  @AllowNull(false)
+  @Column(DataType.BIGINT)
+  declare videoSize: number
+
+  @AllowNull(false)
+  @Column
+  declare startTime: Date
+
+  @AllowNull(false)
+  @Column
+  declare endTime: Date
+
+  @AllowNull(false)
+  @Column
+  declare targetCamera: string
+
+  @AllowNull(false)
+  @Default(LiveVideoStatus.SCHEDULED)
+  @Column
+  declare status: LiveVideoStatus
+
+  @ForeignKey(() => UserModel)
+  @Column
+  declare createdByUserId: number
+
+  @BelongsTo(() => UserModel, {
+    foreignKey: {
+      allowNull: false
+    },
+    onDelete: 'restrict'
+  })
+  declare CreatedBy: Awaited<UserModel>
+
+  @CreatedAt
+  declare createdAt: Date
+
+  @UpdatedAt
+  declare updatedAt: Date
+
+  static listForApi (options: {
+    start?: number
+    count?: number
+    targetCamera?: string
+    status?: string
+    transaction?: Transaction
+  }) {
+    const query: FindAndCountOptions = {
+      offset: options.start,
+      limit: options.count,
+      order: [ [ 'startTime', 'DESC' ] ],
+      transaction: options.transaction
+    }
+
+    const where: Record<string, any> = {}
+
+    if (options.targetCamera) where.targetCamera = options.targetCamera
+    if (options.status) where.status = options.status
+
+    if (Object.keys(where).length !== 0) query.where = where
+
+    return LiveVideoModel.findAndCountAll<MLiveVideo>(query)
+  }
+
+  static loadById (id: number, transaction?: Transaction) {
+    return LiveVideoModel.findByPk<MLiveVideo>(id, { transaction })
+  }
+
+  static loadByThumbnailFilename (thumbnailFilename: string, transaction?: Transaction) {
+    return LiveVideoModel.findOne<MLiveVideo>({
+      where: {
+        thumbnailFilename
+      },
+      transaction
+    })
+  }
+
+  static listConflicts (options: {
+    targetCamera: string
+    startTime: Date
+    endTime: Date
+    excludeId?: number
+    transaction?: Transaction
+  }) {
+    const where: Record<string, any> = {
+      targetCamera: options.targetCamera,
+      status: {
+        [Op.not]: LiveVideoStatus.CANCELLED
+      }
+    }
+
+    if (options.excludeId) {
+      where.id = {
+        [Op.ne]: options.excludeId
+      }
+    }
+
+    return LiveVideoModel.findAll<MLiveVideo>({
+      where,
+      transaction: options.transaction
+    })
+  }
+
+  static listForStarting (options: {
+    startBefore: Date
+    endAfter: Date
+    transaction?: Transaction
+  }) {
+    return LiveVideoModel.findAll<MLiveVideo>({
+      where: {
+        status: LiveVideoStatus.SCHEDULED,
+        startTime: {
+          [Op.lte]: options.startBefore
+        },
+        endTime: {
+          [Op.gt]: options.endAfter
+        }
+      },
+      order: [ [ 'startTime', 'ASC' ] ],
+      transaction: options.transaction
+    })
+  }
+
+  toFormattedJSON (this: LiveVideoModel) {
+    return {
+      id: this.id,
+      title: this.title,
+      thumbnailFilename: this.thumbnailFilename,
+      videoOriginalName: this.videoOriginalName,
+      videoStoragePath: this.videoStoragePath,
+      videoMimeType: this.videoMimeType,
+      videoSize: this.videoSize,
+      startTime: this.startTime.toISOString(),
+      endTime: this.endTime.toISOString(),
+      targetCamera: this.targetCamera,
+      status: this.status,
+      createdByUserId: this.createdByUserId,
+      createdAt: this.createdAt.toISOString(),
+      updatedAt: this.updatedAt.toISOString()
+    }
+  }
+}