X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fmodels%2Fvideo%2Fvideo-import.ts;h=ec10085d644db923579fef6837896ce6afa1a050;hb=74dc3bca2b14f5fd3fe80c394dfc34177a46db77;hp=d6c02e5ac9af65dfded8a7b91dd2eb1e324e2a6d;hpb=990b6a0b0c4fbebc165e5cf7cec8fbc1cbaa6c66;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/models/video/video-import.ts b/server/models/video/video-import.ts index d6c02e5ac..ec10085d6 100644 --- a/server/models/video/video-import.ts +++ b/server/models/video/video-import.ts @@ -13,36 +13,23 @@ import { Table, UpdatedAt } from 'sequelize-typescript' -import { CONSTRAINTS_FIELDS, VIDEO_IMPORT_STATES } from '../../initializers' +import { CONSTRAINTS_FIELDS, VIDEO_IMPORT_STATES } from '../../initializers/constants' import { getSort, throwIfNotValid } from '../utils' -import { VideoModel } from './video' +import { ScopeNames as VideoModelScopeNames, VideoModel } from './video' import { isVideoImportStateValid, isVideoImportTargetUrlValid } from '../../helpers/custom-validators/video-imports' import { VideoImport, VideoImportState } from '../../../shared' -import { VideoChannelModel } from './video-channel' -import { AccountModel } from '../account/account' -import { TagModel } from './tag' import { isVideoMagnetUriValid } from '../../helpers/custom-validators/videos' +import { UserModel } from '../account/user' @DefaultScope({ include: [ { - model: () => VideoModel, - required: false, - include: [ - { - model: () => VideoChannelModel, - required: true, - include: [ - { - model: () => AccountModel, - required: true - } - ] - }, - { - model: () => TagModel - } - ] + model: () => UserModel.unscoped(), + required: true + }, + { + model: () => VideoModel.scope([ VideoModelScopeNames.WITH_ACCOUNT_DETAILS, VideoModelScopeNames.WITH_TAGS]), + required: false } ] }) @@ -53,6 +40,9 @@ import { isVideoMagnetUriValid } from '../../helpers/custom-validators/videos' { fields: [ 'videoId' ], unique: true + }, + { + fields: [ 'userId' ] } ] }) @@ -91,6 +81,18 @@ export class VideoImportModel extends Model { @Column(DataType.TEXT) error: string + @ForeignKey(() => UserModel) + @Column + userId: number + + @BelongsTo(() => UserModel, { + foreignKey: { + allowNull: false + }, + onDelete: 'cascade' + }) + User: UserModel + @ForeignKey(() => VideoModel) @Column videoId: number @@ -113,44 +115,27 @@ export class VideoImportModel extends Model { } static loadAndPopulateVideo (id: number) { - return VideoImportModel.findById(id) + return VideoImportModel.findByPk(id) } - static listUserVideoImportsForApi (accountId: number, start: number, count: number, sort: string) { + static listUserVideoImportsForApi (userId: number, start: number, count: number, sort: string) { const query = { distinct: true, - offset: start, - limit: count, - order: getSort(sort), include: [ { - model: VideoModel, - required: false, - include: [ - { - model: VideoChannelModel, - required: true, - include: [ - { - model: AccountModel, - required: true, - where: { - id: accountId - } - } - ] - }, - { - model: TagModel, - required: false - } - ] + model: UserModel.unscoped(), // FIXME: Without this, sequelize try to COUNT(DISTINCT(*)) which is an invalid SQL query + required: true } - ] + ], + offset: start, + limit: count, + order: getSort(sort), + where: { + userId + } } - return VideoImportModel.unscoped() - .findAndCountAll(query) + return VideoImportModel.findAndCountAll(query) .then(({ rows, count }) => { return { data: rows, @@ -159,14 +144,17 @@ export class VideoImportModel extends Model { }) } + getTargetIdentifier () { + return this.targetUrl || this.magnetUri || this.torrentName + } + toFormattedJSON (): VideoImport { const videoFormatOptions = { + completeDescription: true, additionalAttributes: { state: true, waitTranscoding: true, scheduledUpdate: true } } const video = this.Video - ? Object.assign(this.Video.toFormattedJSON(videoFormatOptions), { - tags: this.Video.Tags.map(t => t.name) - }) + ? Object.assign(this.Video.toFormattedJSON(videoFormatOptions), { tags: this.Video.Tags.map(t => t.name) }) : undefined return { @@ -186,6 +174,7 @@ export class VideoImportModel extends Model { video } } + private static getStateLabel (id: number) { return VIDEO_IMPORT_STATES[id] || 'Unknown' }