]>
Commit | Line | Data |
---|---|---|
1 | import { values } from 'lodash' | |
2 | import { AllowNull, BelongsTo, Column, CreatedAt, DataType, Default, ForeignKey, Is, Model, Table, UpdatedAt } from 'sequelize-typescript' | |
3 | import { | |
4 | isVideoFileInfoHashValid, | |
5 | isVideoFileResolutionValid, | |
6 | isVideoFileSizeValid, | |
7 | isVideoFPSResolutionValid | |
8 | } from '../../helpers/custom-validators/videos' | |
9 | import { CONSTRAINTS_FIELDS } from '../../initializers' | |
10 | import { throwIfNotValid } from '../utils' | |
11 | import { VideoModel } from './video' | |
12 | import * as Sequelize from 'sequelize' | |
13 | ||
14 | @Table({ | |
15 | tableName: 'videoFile', | |
16 | indexes: [ | |
17 | { | |
18 | fields: [ 'videoId' ] | |
19 | }, | |
20 | { | |
21 | fields: [ 'infoHash' ] | |
22 | }, | |
23 | { | |
24 | fields: [ 'videoId', 'resolution', 'fps' ], | |
25 | unique: true | |
26 | } | |
27 | ] | |
28 | }) | |
29 | export class VideoFileModel extends Model<VideoFileModel> { | |
30 | @CreatedAt | |
31 | createdAt: Date | |
32 | ||
33 | @UpdatedAt | |
34 | updatedAt: Date | |
35 | ||
36 | @AllowNull(false) | |
37 | @Is('VideoFileResolution', value => throwIfNotValid(value, isVideoFileResolutionValid, 'resolution')) | |
38 | @Column | |
39 | resolution: number | |
40 | ||
41 | @AllowNull(false) | |
42 | @Is('VideoFileSize', value => throwIfNotValid(value, isVideoFileSizeValid, 'size')) | |
43 | @Column(DataType.BIGINT) | |
44 | size: number | |
45 | ||
46 | @AllowNull(false) | |
47 | @Column(DataType.ENUM(values(CONSTRAINTS_FIELDS.VIDEOS.EXTNAME))) | |
48 | extname: string | |
49 | ||
50 | @AllowNull(false) | |
51 | @Is('VideoFileSize', value => throwIfNotValid(value, isVideoFileInfoHashValid, 'info hash')) | |
52 | @Column | |
53 | infoHash: string | |
54 | ||
55 | @AllowNull(true) | |
56 | @Default(null) | |
57 | @Is('VideoFileFPS', value => throwIfNotValid(value, isVideoFPSResolutionValid, 'fps')) | |
58 | @Column | |
59 | fps: number | |
60 | ||
61 | @ForeignKey(() => VideoModel) | |
62 | @Column | |
63 | videoId: number | |
64 | ||
65 | @BelongsTo(() => VideoModel, { | |
66 | foreignKey: { | |
67 | allowNull: false | |
68 | }, | |
69 | onDelete: 'CASCADE' | |
70 | }) | |
71 | Video: VideoModel | |
72 | ||
73 | static isInfohashExists (infoHash: string) { | |
74 | const query = 'SELECT 1 FROM "videoFile" WHERE "infoHash" = $infoHash LIMIT 1' | |
75 | const options = { | |
76 | type: Sequelize.QueryTypes.SELECT, | |
77 | bind: { infoHash }, | |
78 | raw: true | |
79 | } | |
80 | ||
81 | return VideoModel.sequelize.query(query, options) | |
82 | .then(results => { | |
83 | return results.length === 1 | |
84 | }) | |
85 | } | |
86 | } |