]>
Commit | Line | Data |
---|---|---|
1 | import * as Sequelize from 'sequelize' | |
2 | ||
3 | import { addMethodsToModel, getSort } from '../utils' | |
4 | import { | |
5 | BlacklistedVideoInstance, | |
6 | BlacklistedVideoAttributes, | |
7 | ||
8 | BlacklistedVideoMethods | |
9 | } from './video-blacklist-interface' | |
10 | ||
11 | let BlacklistedVideo: Sequelize.Model<BlacklistedVideoInstance, BlacklistedVideoAttributes> | |
12 | let toFormatedJSON: BlacklistedVideoMethods.ToFormatedJSON | |
13 | let countTotal: BlacklistedVideoMethods.CountTotal | |
14 | let list: BlacklistedVideoMethods.List | |
15 | let listForApi: BlacklistedVideoMethods.ListForApi | |
16 | let loadById: BlacklistedVideoMethods.LoadById | |
17 | let loadByVideoId: BlacklistedVideoMethods.LoadByVideoId | |
18 | ||
19 | export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) { | |
20 | BlacklistedVideo = sequelize.define<BlacklistedVideoInstance, BlacklistedVideoAttributes>('BlacklistedVideo', | |
21 | {}, | |
22 | { | |
23 | indexes: [ | |
24 | { | |
25 | fields: [ 'videoId' ], | |
26 | unique: true | |
27 | } | |
28 | ] | |
29 | } | |
30 | ) | |
31 | ||
32 | const classMethods = [ | |
33 | associate, | |
34 | ||
35 | countTotal, | |
36 | list, | |
37 | listForApi, | |
38 | loadById, | |
39 | loadByVideoId | |
40 | ] | |
41 | const instanceMethods = [ | |
42 | toFormatedJSON | |
43 | ] | |
44 | addMethodsToModel(BlacklistedVideo, classMethods, instanceMethods) | |
45 | ||
46 | return BlacklistedVideo | |
47 | } | |
48 | ||
49 | // ------------------------------ METHODS ------------------------------ | |
50 | ||
51 | toFormatedJSON = function (this: BlacklistedVideoInstance) { | |
52 | return { | |
53 | id: this.id, | |
54 | videoId: this.videoId, | |
55 | createdAt: this.createdAt | |
56 | } | |
57 | } | |
58 | ||
59 | // ------------------------------ STATICS ------------------------------ | |
60 | ||
61 | function associate (models) { | |
62 | BlacklistedVideo.belongsTo(models.Video, { | |
63 | foreignKey: { | |
64 | name: 'videoId', | |
65 | allowNull: false | |
66 | }, | |
67 | onDelete: 'CASCADE' | |
68 | }) | |
69 | } | |
70 | ||
71 | countTotal = function () { | |
72 | return BlacklistedVideo.count() | |
73 | } | |
74 | ||
75 | list = function () { | |
76 | return BlacklistedVideo.findAll() | |
77 | } | |
78 | ||
79 | listForApi = function (start: number, count: number, sort: string) { | |
80 | const query = { | |
81 | offset: start, | |
82 | limit: count, | |
83 | order: [ getSort(sort) ] | |
84 | } | |
85 | ||
86 | return BlacklistedVideo.findAndCountAll(query).then(({ rows, count }) => { | |
87 | return { | |
88 | data: rows, | |
89 | total: count | |
90 | } | |
91 | }) | |
92 | } | |
93 | ||
94 | loadById = function (id: number) { | |
95 | return BlacklistedVideo.findById(id) | |
96 | } | |
97 | ||
98 | loadByVideoId = function (id: number) { | |
99 | const query = { | |
100 | where: { | |
101 | videoId: id | |
102 | } | |
103 | } | |
104 | ||
105 | return BlacklistedVideo.findOne(query) | |
106 | } |