aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/models/user
diff options
context:
space:
mode:
authorChocobozzz <florian.bigard@gmail.com>2017-11-09 17:51:58 +0100
committerChocobozzz <florian.bigard@gmail.com>2017-11-27 19:40:51 +0100
commite4f97babf701481b55cc10fb3448feab5f97c867 (patch)
treeaf37402a594dc5ff09f71ecb0687e8cfe4cdb471 /server/models/user
parent343ad675f2a26c15b86150a9a3552e619d5d44f4 (diff)
downloadPeerTube-e4f97babf701481b55cc10fb3448feab5f97c867.tar.gz
PeerTube-e4f97babf701481b55cc10fb3448feab5f97c867.tar.zst
PeerTube-e4f97babf701481b55cc10fb3448feab5f97c867.zip
Begin activitypub
Diffstat (limited to 'server/models/user')
-rw-r--r--server/models/user/index.ts2
-rw-r--r--server/models/user/user-interface.ts69
-rw-r--r--server/models/user/user-video-rate-interface.ts26
-rw-r--r--server/models/user/user-video-rate.ts78
-rw-r--r--server/models/user/user.ts312
5 files changed, 0 insertions, 487 deletions
diff --git a/server/models/user/index.ts b/server/models/user/index.ts
deleted file mode 100644
index ed3689518..000000000
--- a/server/models/user/index.ts
+++ /dev/null
@@ -1,2 +0,0 @@
1export * from './user-video-rate-interface'
2export * from './user-interface'
diff --git a/server/models/user/user-interface.ts b/server/models/user/user-interface.ts
deleted file mode 100644
index 49c75aa3b..000000000
--- a/server/models/user/user-interface.ts
+++ /dev/null
@@ -1,69 +0,0 @@
1import * as Sequelize from 'sequelize'
2import * as Promise from 'bluebird'
3
4// Don't use barrel, import just what we need
5import { User as FormattedUser } from '../../../shared/models/users/user.model'
6import { ResultList } from '../../../shared/models/result-list.model'
7import { AuthorInstance } from '../video/author-interface'
8import { UserRight } from '../../../shared/models/users/user-right.enum'
9import { UserRole } from '../../../shared/models/users/user-role'
10
11export namespace UserMethods {
12 export type HasRight = (this: UserInstance, right: UserRight) => boolean
13 export type IsPasswordMatch = (this: UserInstance, password: string) => Promise<boolean>
14
15 export type ToFormattedJSON = (this: UserInstance) => FormattedUser
16 export type IsAbleToUploadVideo = (this: UserInstance, videoFile: Express.Multer.File) => Promise<boolean>
17
18 export type CountTotal = () => Promise<number>
19
20 export type GetByUsername = (username: string) => Promise<UserInstance>
21
22 export type ListForApi = (start: number, count: number, sort: string) => Promise< ResultList<UserInstance> >
23
24 export type LoadById = (id: number) => Promise<UserInstance>
25
26 export type LoadByUsername = (username: string) => Promise<UserInstance>
27 export type LoadByUsernameAndPopulateChannels = (username: string) => Promise<UserInstance>
28
29 export type LoadByUsernameOrEmail = (username: string, email: string) => Promise<UserInstance>
30}
31
32export interface UserClass {
33 isPasswordMatch: UserMethods.IsPasswordMatch,
34 toFormattedJSON: UserMethods.ToFormattedJSON,
35 hasRight: UserMethods.HasRight,
36 isAbleToUploadVideo: UserMethods.IsAbleToUploadVideo,
37
38 countTotal: UserMethods.CountTotal,
39 getByUsername: UserMethods.GetByUsername,
40 listForApi: UserMethods.ListForApi,
41 loadById: UserMethods.LoadById,
42 loadByUsername: UserMethods.LoadByUsername,
43 loadByUsernameAndPopulateChannels: UserMethods.LoadByUsernameAndPopulateChannels,
44 loadByUsernameOrEmail: UserMethods.LoadByUsernameOrEmail
45}
46
47export interface UserAttributes {
48 id?: number
49 password: string
50 username: string
51 email: string
52 displayNSFW?: boolean
53 role: UserRole
54 videoQuota: number
55
56 Author?: AuthorInstance
57}
58
59export interface UserInstance extends UserClass, UserAttributes, Sequelize.Instance<UserAttributes> {
60 id: number
61 createdAt: Date
62 updatedAt: Date
63
64 isPasswordMatch: UserMethods.IsPasswordMatch
65 toFormattedJSON: UserMethods.ToFormattedJSON
66 hasRight: UserMethods.HasRight
67}
68
69export interface UserModel extends UserClass, Sequelize.Model<UserInstance, UserAttributes> {}
diff --git a/server/models/user/user-video-rate-interface.ts b/server/models/user/user-video-rate-interface.ts
deleted file mode 100644
index ea0fdc4d9..000000000
--- a/server/models/user/user-video-rate-interface.ts
+++ /dev/null
@@ -1,26 +0,0 @@
1import * as Sequelize from 'sequelize'
2import * as Promise from 'bluebird'
3
4import { VideoRateType } from '../../../shared/models/videos/video-rate.type'
5
6export namespace UserVideoRateMethods {
7 export type Load = (userId: number, videoId: number, transaction: Sequelize.Transaction) => Promise<UserVideoRateInstance>
8}
9
10export interface UserVideoRateClass {
11 load: UserVideoRateMethods.Load
12}
13
14export interface UserVideoRateAttributes {
15 type: VideoRateType
16 userId: number
17 videoId: number
18}
19
20export interface UserVideoRateInstance extends UserVideoRateClass, UserVideoRateAttributes, Sequelize.Instance<UserVideoRateAttributes> {
21 id: number
22 createdAt: Date
23 updatedAt: Date
24}
25
26export interface UserVideoRateModel extends UserVideoRateClass, Sequelize.Model<UserVideoRateInstance, UserVideoRateAttributes> {}
diff --git a/server/models/user/user-video-rate.ts b/server/models/user/user-video-rate.ts
deleted file mode 100644
index 7d6dd7281..000000000
--- a/server/models/user/user-video-rate.ts
+++ /dev/null
@@ -1,78 +0,0 @@
1/*
2 User rates per video.
3*/
4import { values } from 'lodash'
5import * as Sequelize from 'sequelize'
6
7import { VIDEO_RATE_TYPES } from '../../initializers'
8
9import { addMethodsToModel } from '../utils'
10import {
11 UserVideoRateInstance,
12 UserVideoRateAttributes,
13
14 UserVideoRateMethods
15} from './user-video-rate-interface'
16
17let UserVideoRate: Sequelize.Model<UserVideoRateInstance, UserVideoRateAttributes>
18let load: UserVideoRateMethods.Load
19
20export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
21 UserVideoRate = sequelize.define<UserVideoRateInstance, UserVideoRateAttributes>('UserVideoRate',
22 {
23 type: {
24 type: DataTypes.ENUM(values(VIDEO_RATE_TYPES)),
25 allowNull: false
26 }
27 },
28 {
29 indexes: [
30 {
31 fields: [ 'videoId', 'userId', 'type' ],
32 unique: true
33 }
34 ]
35 }
36 )
37
38 const classMethods = [
39 associate,
40
41 load
42 ]
43 addMethodsToModel(UserVideoRate, classMethods)
44
45 return UserVideoRate
46}
47
48// ------------------------------ STATICS ------------------------------
49
50function associate (models) {
51 UserVideoRate.belongsTo(models.Video, {
52 foreignKey: {
53 name: 'videoId',
54 allowNull: false
55 },
56 onDelete: 'CASCADE'
57 })
58
59 UserVideoRate.belongsTo(models.User, {
60 foreignKey: {
61 name: 'userId',
62 allowNull: false
63 },
64 onDelete: 'CASCADE'
65 })
66}
67
68load = function (userId: number, videoId: number, transaction: Sequelize.Transaction) {
69 const options: Sequelize.FindOptions<UserVideoRateAttributes> = {
70 where: {
71 userId,
72 videoId
73 }
74 }
75 if (transaction) options.transaction = transaction
76
77 return UserVideoRate.findOne(options)
78}
diff --git a/server/models/user/user.ts b/server/models/user/user.ts
deleted file mode 100644
index b974418d4..000000000
--- a/server/models/user/user.ts
+++ /dev/null
@@ -1,312 +0,0 @@
1import * as Sequelize from 'sequelize'
2import * as Promise from 'bluebird'
3
4import { getSort, addMethodsToModel } from '../utils'
5import {
6 cryptPassword,
7 comparePassword,
8 isUserPasswordValid,
9 isUserUsernameValid,
10 isUserDisplayNSFWValid,
11 isUserVideoQuotaValid,
12 isUserRoleValid
13} from '../../helpers'
14import { UserRight, USER_ROLE_LABELS, hasUserRight } from '../../../shared'
15
16import {
17 UserInstance,
18 UserAttributes,
19
20 UserMethods
21} from './user-interface'
22
23let User: Sequelize.Model<UserInstance, UserAttributes>
24let isPasswordMatch: UserMethods.IsPasswordMatch
25let hasRight: UserMethods.HasRight
26let toFormattedJSON: UserMethods.ToFormattedJSON
27let countTotal: UserMethods.CountTotal
28let getByUsername: UserMethods.GetByUsername
29let listForApi: UserMethods.ListForApi
30let loadById: UserMethods.LoadById
31let loadByUsername: UserMethods.LoadByUsername
32let loadByUsernameAndPopulateChannels: UserMethods.LoadByUsernameAndPopulateChannels
33let loadByUsernameOrEmail: UserMethods.LoadByUsernameOrEmail
34let isAbleToUploadVideo: UserMethods.IsAbleToUploadVideo
35
36export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
37 User = sequelize.define<UserInstance, UserAttributes>('User',
38 {
39 password: {
40 type: DataTypes.STRING,
41 allowNull: false,
42 validate: {
43 passwordValid: value => {
44 const res = isUserPasswordValid(value)
45 if (res === false) throw new Error('Password not valid.')
46 }
47 }
48 },
49 username: {
50 type: DataTypes.STRING,
51 allowNull: false,
52 validate: {
53 usernameValid: value => {
54 const res = isUserUsernameValid(value)
55 if (res === false) throw new Error('Username not valid.')
56 }
57 }
58 },
59 email: {
60 type: DataTypes.STRING(400),
61 allowNull: false,
62 validate: {
63 isEmail: true
64 }
65 },
66 displayNSFW: {
67 type: DataTypes.BOOLEAN,
68 allowNull: false,
69 defaultValue: false,
70 validate: {
71 nsfwValid: value => {
72 const res = isUserDisplayNSFWValid(value)
73 if (res === false) throw new Error('Display NSFW is not valid.')
74 }
75 }
76 },
77 role: {
78 type: DataTypes.INTEGER,
79 allowNull: false,
80 validate: {
81 roleValid: value => {
82 const res = isUserRoleValid(value)
83 if (res === false) throw new Error('Role is not valid.')
84 }
85 }
86 },
87 videoQuota: {
88 type: DataTypes.BIGINT,
89 allowNull: false,
90 validate: {
91 videoQuotaValid: value => {
92 const res = isUserVideoQuotaValid(value)
93 if (res === false) throw new Error('Video quota is not valid.')
94 }
95 }
96 }
97 },
98 {
99 indexes: [
100 {
101 fields: [ 'username' ],
102 unique: true
103 },
104 {
105 fields: [ 'email' ],
106 unique: true
107 }
108 ],
109 hooks: {
110 beforeCreate: beforeCreateOrUpdate,
111 beforeUpdate: beforeCreateOrUpdate
112 }
113 }
114 )
115
116 const classMethods = [
117 associate,
118
119 countTotal,
120 getByUsername,
121 listForApi,
122 loadById,
123 loadByUsername,
124 loadByUsernameAndPopulateChannels,
125 loadByUsernameOrEmail
126 ]
127 const instanceMethods = [
128 hasRight,
129 isPasswordMatch,
130 toFormattedJSON,
131 isAbleToUploadVideo
132 ]
133 addMethodsToModel(User, classMethods, instanceMethods)
134
135 return User
136}
137
138function beforeCreateOrUpdate (user: UserInstance) {
139 if (user.changed('password')) {
140 return cryptPassword(user.password)
141 .then(hash => {
142 user.password = hash
143 return undefined
144 })
145 }
146}
147
148// ------------------------------ METHODS ------------------------------
149
150hasRight = function (this: UserInstance, right: UserRight) {
151 return hasUserRight(this.role, right)
152}
153
154isPasswordMatch = function (this: UserInstance, password: string) {
155 return comparePassword(password, this.password)
156}
157
158toFormattedJSON = function (this: UserInstance) {
159 const json = {
160 id: this.id,
161 username: this.username,
162 email: this.email,
163 displayNSFW: this.displayNSFW,
164 role: this.role,
165 roleLabel: USER_ROLE_LABELS[this.role],
166 videoQuota: this.videoQuota,
167 createdAt: this.createdAt,
168 author: {
169 id: this.Author.id,
170 uuid: this.Author.uuid
171 }
172 }
173
174 if (Array.isArray(this.Author.VideoChannels) === true) {
175 const videoChannels = this.Author.VideoChannels
176 .map(c => c.toFormattedJSON())
177 .sort((v1, v2) => {
178 if (v1.createdAt < v2.createdAt) return -1
179 if (v1.createdAt === v2.createdAt) return 0
180
181 return 1
182 })
183
184 json['videoChannels'] = videoChannels
185 }
186
187 return json
188}
189
190isAbleToUploadVideo = function (this: UserInstance, videoFile: Express.Multer.File) {
191 if (this.videoQuota === -1) return Promise.resolve(true)
192
193 return getOriginalVideoFileTotalFromUser(this).then(totalBytes => {
194 return (videoFile.size + totalBytes) < this.videoQuota
195 })
196}
197
198// ------------------------------ STATICS ------------------------------
199
200function associate (models) {
201 User.hasOne(models.Author, {
202 foreignKey: 'userId',
203 onDelete: 'cascade'
204 })
205
206 User.hasMany(models.OAuthToken, {
207 foreignKey: 'userId',
208 onDelete: 'cascade'
209 })
210}
211
212countTotal = function () {
213 return this.count()
214}
215
216getByUsername = function (username: string) {
217 const query = {
218 where: {
219 username: username
220 },
221 include: [ { model: User['sequelize'].models.Author, required: true } ]
222 }
223
224 return User.findOne(query)
225}
226
227listForApi = function (start: number, count: number, sort: string) {
228 const query = {
229 offset: start,
230 limit: count,
231 order: [ getSort(sort) ],
232 include: [ { model: User['sequelize'].models.Author, required: true } ]
233 }
234
235 return User.findAndCountAll(query).then(({ rows, count }) => {
236 return {
237 data: rows,
238 total: count
239 }
240 })
241}
242
243loadById = function (id: number) {
244 const options = {
245 include: [ { model: User['sequelize'].models.Author, required: true } ]
246 }
247
248 return User.findById(id, options)
249}
250
251loadByUsername = function (username: string) {
252 const query = {
253 where: {
254 username
255 },
256 include: [ { model: User['sequelize'].models.Author, required: true } ]
257 }
258
259 return User.findOne(query)
260}
261
262loadByUsernameAndPopulateChannels = function (username: string) {
263 const query = {
264 where: {
265 username
266 },
267 include: [
268 {
269 model: User['sequelize'].models.Author,
270 required: true,
271 include: [ User['sequelize'].models.VideoChannel ]
272 }
273 ]
274 }
275
276 return User.findOne(query)
277}
278
279loadByUsernameOrEmail = function (username: string, email: string) {
280 const query = {
281 include: [ { model: User['sequelize'].models.Author, required: true } ],
282 where: {
283 [Sequelize.Op.or]: [ { username }, { email } ]
284 }
285 }
286
287 // FIXME: https://github.com/DefinitelyTyped/DefinitelyTyped/issues/18387
288 return (User as any).findOne(query)
289}
290
291// ---------------------------------------------------------------------------
292
293function getOriginalVideoFileTotalFromUser (user: UserInstance) {
294 // Don't use sequelize because we need to use a sub query
295 const query = 'SELECT SUM("size") AS "total" FROM ' +
296 '(SELECT MAX("VideoFiles"."size") AS "size" FROM "VideoFiles" ' +
297 'INNER JOIN "Videos" ON "VideoFiles"."videoId" = "Videos"."id" ' +
298 'INNER JOIN "VideoChannels" ON "VideoChannels"."id" = "Videos"."channelId" ' +
299 'INNER JOIN "Authors" ON "VideoChannels"."authorId" = "Authors"."id" ' +
300 'INNER JOIN "Users" ON "Authors"."userId" = "Users"."id" ' +
301 'WHERE "Users"."id" = $userId GROUP BY "Videos"."id") t'
302
303 const options = {
304 bind: { userId: user.id },
305 type: Sequelize.QueryTypes.SELECT
306 }
307 return User['sequelize'].query(query, options).then(([ { total } ]) => {
308 if (total === null) return 0
309
310 return parseInt(total, 10)
311 })
312}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-09 08:47:19 +0000