]>
Commit | Line | Data |
---|---|---|
d38b8281 C |
1 | /* |
2 | User rates per video. | |
d38b8281 | 3 | */ |
65fcc311 | 4 | import { values } from 'lodash' |
e02643f3 | 5 | import * as Sequelize from 'sequelize' |
d38b8281 | 6 | |
74889a71 | 7 | import { VIDEO_RATE_TYPES } from '../../initializers' |
d38b8281 | 8 | |
74889a71 | 9 | import { addMethodsToModel } from '../utils' |
e02643f3 | 10 | import { |
e02643f3 C |
11 | UserVideoRateInstance, |
12 | UserVideoRateAttributes, | |
d38b8281 | 13 | |
e02643f3 C |
14 | UserVideoRateMethods |
15 | } from './user-video-rate-interface' | |
16 | ||
17 | let UserVideoRate: Sequelize.Model<UserVideoRateInstance, UserVideoRateAttributes> | |
18 | let load: UserVideoRateMethods.Load | |
19 | ||
127944aa C |
20 | export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) { |
21 | UserVideoRate = sequelize.define<UserVideoRateInstance, UserVideoRateAttributes>('UserVideoRate', | |
d38b8281 C |
22 | { |
23 | type: { | |
65fcc311 | 24 | type: DataTypes.ENUM(values(VIDEO_RATE_TYPES)), |
d38b8281 C |
25 | allowNull: false |
26 | } | |
27 | }, | |
28 | { | |
29 | indexes: [ | |
30 | { | |
31 | fields: [ 'videoId', 'userId', 'type' ], | |
32 | unique: true | |
33 | } | |
e02643f3 | 34 | ] |
d38b8281 C |
35 | } |
36 | ) | |
37 | ||
e02643f3 C |
38 | const classMethods = [ |
39 | associate, | |
40 | ||
41 | load | |
42 | ] | |
43 | addMethodsToModel(UserVideoRate, classMethods) | |
44 | ||
d38b8281 C |
45 | return UserVideoRate |
46 | } | |
47 | ||
48 | // ------------------------------ STATICS ------------------------------ | |
49 | ||
50 | function associate (models) { | |
e02643f3 | 51 | UserVideoRate.belongsTo(models.Video, { |
d38b8281 C |
52 | foreignKey: { |
53 | name: 'videoId', | |
54 | allowNull: false | |
55 | }, | |
56 | onDelete: 'CASCADE' | |
57 | }) | |
58 | ||
e02643f3 | 59 | UserVideoRate.belongsTo(models.User, { |
d38b8281 C |
60 | foreignKey: { |
61 | name: 'userId', | |
62 | allowNull: false | |
63 | }, | |
64 | onDelete: 'CASCADE' | |
65 | }) | |
66 | } | |
67 | ||
0a6658fd | 68 | load = function (userId: number, videoId: number, transaction: Sequelize.Transaction) { |
e02643f3 | 69 | const options: Sequelize.FindOptions = { |
d38b8281 C |
70 | where: { |
71 | userId, | |
72 | videoId | |
73 | } | |
74 | } | |
d38b8281 C |
75 | if (transaction) options.transaction = transaction |
76 | ||
6fcd19ba | 77 | return UserVideoRate.findOne(options) |
d38b8281 | 78 | } |