]>
Commit | Line | Data |
---|---|---|
1 | import * as Sequelize from 'sequelize' | |
2 | import * as uuidv4 from 'uuid/v4' | |
3 | import { ActivityPubActorType } from '../../shared/models/activitypub' | |
4 | import { SERVER_ACTOR_NAME } from '../initializers/constants' | |
5 | import { AccountModel } from '../models/account/account' | |
6 | import { UserModel } from '../models/account/user' | |
7 | import { buildActorInstance, getAccountActivityPubUrl, setAsyncActorKeys } from './activitypub' | |
8 | import { createVideoChannel } from './video-channel' | |
9 | import { VideoChannelModel } from '../models/video/video-channel' | |
10 | import { ActorModel } from '../models/activitypub/actor' | |
11 | import { UserNotificationSettingModel } from '../models/account/user-notification-setting' | |
12 | import { UserNotificationSetting, UserNotificationSettingValue } from '../../shared/models/users' | |
13 | import { createWatchLaterPlaylist } from './video-playlist' | |
14 | import { sequelizeTypescript } from '../initializers/database' | |
15 | ||
16 | type ChannelNames = { name: string, displayName: string } | |
17 | async function createUserAccountAndChannelAndPlaylist (userToCreate: UserModel, channelNames?: ChannelNames, validateUser = true) { | |
18 | const { user, account, videoChannel } = await sequelizeTypescript.transaction(async t => { | |
19 | const userOptions = { | |
20 | transaction: t, | |
21 | validate: validateUser | |
22 | } | |
23 | ||
24 | const userCreated = await userToCreate.save(userOptions) | |
25 | userCreated.NotificationSetting = await createDefaultUserNotificationSettings(userCreated, t) | |
26 | ||
27 | const accountCreated = await createLocalAccountWithoutKeys(userCreated.username, userCreated.id, null, t) | |
28 | userCreated.Account = accountCreated | |
29 | ||
30 | const channelAttributes = await buildChannelAttributes(userCreated, channelNames) | |
31 | const videoChannel = await createVideoChannel(channelAttributes, accountCreated, t) | |
32 | ||
33 | const videoPlaylist = await createWatchLaterPlaylist(accountCreated, t) | |
34 | ||
35 | return { user: userCreated, account: accountCreated, videoChannel, videoPlaylist } | |
36 | }) | |
37 | ||
38 | const [ accountKeys, channelKeys ] = await Promise.all([ | |
39 | setAsyncActorKeys(account.Actor), | |
40 | setAsyncActorKeys(videoChannel.Actor) | |
41 | ]) | |
42 | ||
43 | account.Actor = accountKeys | |
44 | videoChannel.Actor = channelKeys | |
45 | ||
46 | return { user, account, videoChannel } as { user: UserModel, account: AccountModel, videoChannel: VideoChannelModel } | |
47 | } | |
48 | ||
49 | async function createLocalAccountWithoutKeys ( | |
50 | name: string, | |
51 | userId: number | null, | |
52 | applicationId: number | null, | |
53 | t: Sequelize.Transaction | undefined, | |
54 | type: ActivityPubActorType= 'Person' | |
55 | ) { | |
56 | const url = getAccountActivityPubUrl(name) | |
57 | ||
58 | const actorInstance = buildActorInstance(type, url, name) | |
59 | const actorInstanceCreated = await actorInstance.save({ transaction: t }) | |
60 | ||
61 | const accountInstance = new AccountModel({ | |
62 | name, | |
63 | userId, | |
64 | applicationId, | |
65 | actorId: actorInstanceCreated.id | |
66 | }) | |
67 | ||
68 | const accountInstanceCreated = await accountInstance.save({ transaction: t }) | |
69 | accountInstanceCreated.Actor = actorInstanceCreated | |
70 | ||
71 | return accountInstanceCreated | |
72 | } | |
73 | ||
74 | async function createApplicationActor (applicationId: number) { | |
75 | const accountCreated = await createLocalAccountWithoutKeys(SERVER_ACTOR_NAME, null, applicationId, undefined, 'Application') | |
76 | ||
77 | accountCreated.Actor = await setAsyncActorKeys(accountCreated.Actor) | |
78 | ||
79 | return accountCreated | |
80 | } | |
81 | ||
82 | // --------------------------------------------------------------------------- | |
83 | ||
84 | export { | |
85 | createApplicationActor, | |
86 | createUserAccountAndChannelAndPlaylist, | |
87 | createLocalAccountWithoutKeys | |
88 | } | |
89 | ||
90 | // --------------------------------------------------------------------------- | |
91 | ||
92 | function createDefaultUserNotificationSettings (user: UserModel, t: Sequelize.Transaction | undefined) { | |
93 | const values: UserNotificationSetting & { userId: number } = { | |
94 | userId: user.id, | |
95 | newVideoFromSubscription: UserNotificationSettingValue.WEB, | |
96 | newCommentOnMyVideo: UserNotificationSettingValue.WEB, | |
97 | myVideoImportFinished: UserNotificationSettingValue.WEB, | |
98 | myVideoPublished: UserNotificationSettingValue.WEB, | |
99 | videoAbuseAsModerator: UserNotificationSettingValue.WEB | UserNotificationSettingValue.EMAIL, | |
100 | videoAutoBlacklistAsModerator: UserNotificationSettingValue.WEB | UserNotificationSettingValue.EMAIL, | |
101 | blacklistOnMyVideo: UserNotificationSettingValue.WEB | UserNotificationSettingValue.EMAIL, | |
102 | newUserRegistration: UserNotificationSettingValue.WEB, | |
103 | commentMention: UserNotificationSettingValue.WEB, | |
104 | newFollow: UserNotificationSettingValue.WEB, | |
105 | newInstanceFollower: UserNotificationSettingValue.WEB | |
106 | } | |
107 | ||
108 | return UserNotificationSettingModel.create(values, { transaction: t }) | |
109 | } | |
110 | ||
111 | async function buildChannelAttributes (user: UserModel, channelNames?: ChannelNames) { | |
112 | if (channelNames) return channelNames | |
113 | ||
114 | let channelName = user.username + '_channel' | |
115 | ||
116 | // Conflict, generate uuid instead | |
117 | const actor = await ActorModel.loadLocalByName(channelName) | |
118 | if (actor) channelName = uuidv4() | |
119 | ||
120 | const videoChannelDisplayName = `Main ${user.username} channel` | |
121 | ||
122 | return { | |
123 | name: channelName, | |
124 | displayName: videoChannelDisplayName | |
125 | } | |
126 | } |