diff options
Diffstat (limited to 'server/models/account')
-rw-r--r-- | server/models/account/account-interface.ts | 3 | ||||
-rw-r--r-- | server/models/account/account.ts | 27 | ||||
-rw-r--r-- | server/models/account/user.ts | 5 |
3 files changed, 30 insertions, 5 deletions
diff --git a/server/models/account/account-interface.ts b/server/models/account/account-interface.ts index b369766dc..46fe068e3 100644 --- a/server/models/account/account-interface.ts +++ b/server/models/account/account-interface.ts | |||
@@ -1,6 +1,7 @@ | |||
1 | import * as Bluebird from 'bluebird' | 1 | import * as Bluebird from 'bluebird' |
2 | import * as Sequelize from 'sequelize' | 2 | import * as Sequelize from 'sequelize' |
3 | import { Account as FormattedAccount, ActivityPubActor } from '../../../shared' | 3 | import { Account as FormattedAccount, ActivityPubActor } from '../../../shared' |
4 | import { AvatarInstance } from '../avatar' | ||
4 | import { ServerInstance } from '../server/server-interface' | 5 | import { ServerInstance } from '../server/server-interface' |
5 | import { VideoChannelInstance } from '../video/video-channel-interface' | 6 | import { VideoChannelInstance } from '../video/video-channel-interface' |
6 | 7 | ||
@@ -51,6 +52,7 @@ export interface AccountAttributes { | |||
51 | serverId?: number | 52 | serverId?: number |
52 | userId?: number | 53 | userId?: number |
53 | applicationId?: number | 54 | applicationId?: number |
55 | avatarId?: number | ||
54 | } | 56 | } |
55 | 57 | ||
56 | export interface AccountInstance extends AccountClass, AccountAttributes, Sequelize.Instance<AccountAttributes> { | 58 | export interface AccountInstance extends AccountClass, AccountAttributes, Sequelize.Instance<AccountAttributes> { |
@@ -68,6 +70,7 @@ export interface AccountInstance extends AccountClass, AccountAttributes, Sequel | |||
68 | 70 | ||
69 | Server: ServerInstance | 71 | Server: ServerInstance |
70 | VideoChannels: VideoChannelInstance[] | 72 | VideoChannels: VideoChannelInstance[] |
73 | Avatar: AvatarInstance | ||
71 | } | 74 | } |
72 | 75 | ||
73 | export interface AccountModel extends AccountClass, Sequelize.Model<AccountInstance, AccountAttributes> {} | 76 | export interface AccountModel extends AccountClass, Sequelize.Model<AccountInstance, AccountAttributes> {} |
diff --git a/server/models/account/account.ts b/server/models/account/account.ts index 61a88524c..15be1126b 100644 --- a/server/models/account/account.ts +++ b/server/models/account/account.ts | |||
@@ -1,4 +1,6 @@ | |||
1 | import { join } from 'path' | ||
1 | import * as Sequelize from 'sequelize' | 2 | import * as Sequelize from 'sequelize' |
3 | import { Avatar } from '../../../shared/models/avatars/avatar.model' | ||
2 | import { | 4 | import { |
3 | activityPubContextify, | 5 | activityPubContextify, |
4 | isAccountFollowersCountValid, | 6 | isAccountFollowersCountValid, |
@@ -8,8 +10,10 @@ import { | |||
8 | isUserUsernameValid | 10 | isUserUsernameValid |
9 | } from '../../helpers' | 11 | } from '../../helpers' |
10 | import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc' | 12 | import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc' |
13 | import { AVATARS_DIR } from '../../initializers' | ||
11 | import { CONFIG, CONSTRAINTS_FIELDS } from '../../initializers/constants' | 14 | import { CONFIG, CONSTRAINTS_FIELDS } from '../../initializers/constants' |
12 | import { sendDeleteAccount } from '../../lib/activitypub/send/send-delete' | 15 | import { sendDeleteAccount } from '../../lib/activitypub/send/send-delete' |
16 | import { AvatarModel } from '../avatar' | ||
13 | import { addMethodsToModel } from '../utils' | 17 | import { addMethodsToModel } from '../utils' |
14 | import { AccountAttributes, AccountInstance, AccountMethods } from './account-interface' | 18 | import { AccountAttributes, AccountInstance, AccountMethods } from './account-interface' |
15 | 19 | ||
@@ -252,6 +256,14 @@ function associate (models) { | |||
252 | as: 'followers', | 256 | as: 'followers', |
253 | onDelete: 'cascade' | 257 | onDelete: 'cascade' |
254 | }) | 258 | }) |
259 | |||
260 | Account.hasOne(models.Avatar, { | ||
261 | foreignKey: { | ||
262 | name: 'avatarId', | ||
263 | allowNull: true | ||
264 | }, | ||
265 | onDelete: 'cascade' | ||
266 | }) | ||
255 | } | 267 | } |
256 | 268 | ||
257 | function afterDestroy (account: AccountInstance) { | 269 | function afterDestroy (account: AccountInstance) { |
@@ -265,6 +277,15 @@ function afterDestroy (account: AccountInstance) { | |||
265 | toFormattedJSON = function (this: AccountInstance) { | 277 | toFormattedJSON = function (this: AccountInstance) { |
266 | let host = CONFIG.WEBSERVER.HOST | 278 | let host = CONFIG.WEBSERVER.HOST |
267 | let score: number | 279 | let score: number |
280 | let avatar: Avatar = null | ||
281 | |||
282 | if (this.Avatar) { | ||
283 | avatar = { | ||
284 | path: join(AVATARS_DIR.ACCOUNT, this.Avatar.filename), | ||
285 | createdAt: this.Avatar.createdAt, | ||
286 | updatedAt: this.Avatar.updatedAt | ||
287 | } | ||
288 | } | ||
268 | 289 | ||
269 | if (this.Server) { | 290 | if (this.Server) { |
270 | host = this.Server.host | 291 | host = this.Server.host |
@@ -273,11 +294,15 @@ toFormattedJSON = function (this: AccountInstance) { | |||
273 | 294 | ||
274 | const json = { | 295 | const json = { |
275 | id: this.id, | 296 | id: this.id, |
297 | uuid: this.uuid, | ||
276 | host, | 298 | host, |
277 | score, | 299 | score, |
278 | name: this.name, | 300 | name: this.name, |
301 | followingCount: this.followingCount, | ||
302 | followersCount: this.followersCount, | ||
279 | createdAt: this.createdAt, | 303 | createdAt: this.createdAt, |
280 | updatedAt: this.updatedAt | 304 | updatedAt: this.updatedAt, |
305 | avatar | ||
281 | } | 306 | } |
282 | 307 | ||
283 | return json | 308 | return json |
diff --git a/server/models/account/user.ts b/server/models/account/user.ts index 8f7c9b013..3705947c0 100644 --- a/server/models/account/user.ts +++ b/server/models/account/user.ts | |||
@@ -157,10 +157,7 @@ toFormattedJSON = function (this: UserInstance) { | |||
157 | roleLabel: USER_ROLE_LABELS[this.role], | 157 | roleLabel: USER_ROLE_LABELS[this.role], |
158 | videoQuota: this.videoQuota, | 158 | videoQuota: this.videoQuota, |
159 | createdAt: this.createdAt, | 159 | createdAt: this.createdAt, |
160 | account: { | 160 | account: this.Account.toFormattedJSON() |
161 | id: this.Account.id, | ||
162 | uuid: this.Account.uuid | ||
163 | } | ||
164 | } | 161 | } |
165 | 162 | ||
166 | if (Array.isArray(this.Account.VideoChannels) === true) { | 163 | if (Array.isArray(this.Account.VideoChannels) === true) { |