]>
Commit | Line | Data |
---|---|---|
93e1258c C |
1 | import * as Sequelize from 'sequelize' |
2 | import { values } from 'lodash' | |
3 | ||
4 | import { CONSTRAINTS_FIELDS } from '../../initializers' | |
5 | import { | |
6 | isVideoFileResolutionValid, | |
7 | isVideoFileSizeValid, | |
8 | isVideoFileInfoHashValid | |
9 | } from '../../helpers' | |
10 | ||
11 | import { addMethodsToModel } from '../utils' | |
12 | import { | |
13 | VideoFileInstance, | |
14 | VideoFileAttributes | |
15 | } from './video-file-interface' | |
16 | ||
17 | let VideoFile: Sequelize.Model<VideoFileInstance, VideoFileAttributes> | |
18 | ||
19 | export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) { | |
20 | VideoFile = sequelize.define<VideoFileInstance, VideoFileAttributes>('VideoFile', | |
21 | { | |
22 | resolution: { | |
23 | type: DataTypes.INTEGER, | |
24 | allowNull: false, | |
25 | validate: { | |
26 | resolutionValid: value => { | |
27 | const res = isVideoFileResolutionValid(value) | |
28 | if (res === false) throw new Error('Video file resolution is not valid.') | |
29 | } | |
30 | } | |
31 | }, | |
32 | size: { | |
33 | type: DataTypes.INTEGER, | |
34 | allowNull: false, | |
35 | validate: { | |
36 | sizeValid: value => { | |
37 | const res = isVideoFileSizeValid(value) | |
38 | if (res === false) throw new Error('Video file size is not valid.') | |
39 | } | |
40 | } | |
41 | }, | |
42 | extname: { | |
43 | type: DataTypes.ENUM(values(CONSTRAINTS_FIELDS.VIDEOS.EXTNAME)), | |
44 | allowNull: false | |
45 | }, | |
46 | infoHash: { | |
47 | type: DataTypes.STRING, | |
48 | allowNull: false, | |
49 | validate: { | |
50 | infoHashValid: value => { | |
51 | const res = isVideoFileInfoHashValid(value) | |
52 | if (res === false) throw new Error('Video file info hash is not valid.') | |
53 | } | |
54 | } | |
55 | } | |
56 | }, | |
57 | { | |
58 | indexes: [ | |
59 | { | |
60 | fields: [ 'videoId' ] | |
61 | }, | |
62 | { | |
63 | fields: [ 'infoHash' ] | |
64 | } | |
65 | ] | |
66 | } | |
67 | ) | |
68 | ||
69 | const classMethods = [ | |
70 | associate | |
71 | ] | |
72 | addMethodsToModel(VideoFile, classMethods) | |
73 | ||
74 | return VideoFile | |
75 | } | |
76 | ||
77 | // ------------------------------ STATICS ------------------------------ | |
78 | ||
79 | function associate (models) { | |
80 | VideoFile.belongsTo(models.Video, { | |
81 | foreignKey: { | |
82 | name: 'videoId', | |
83 | allowNull: false | |
84 | }, | |
85 | onDelete: 'CASCADE' | |
86 | }) | |
87 | } | |
88 | ||
89 | // ------------------------------ METHODS ------------------------------ |