-import { database as db } from '../initializers'
-import { UserInstance } from '../models'
-import { addVideoAuthorToFriends } from './friends'
+import * as Sequelize from 'sequelize'
+import * as uuidv4 from 'uuid/v4'
+import { ActivityPubActorType } from '../../shared/models/activitypub'
+import { sequelizeTypescript, SERVER_ACTOR_NAME } from '../initializers'
+import { AccountModel } from '../models/account/account'
+import { UserModel } from '../models/account/user'
+import { buildActorInstance, getAccountActivityPubUrl, setAsyncActorKeys } from './activitypub'
import { createVideoChannel } from './video-channel'
+import { VideoChannelModel } from '../models/video/video-channel'
+import { FilteredModelAttributes } from 'sequelize-typescript/lib/models/Model'
+import { ActorModel } from '../models/activitypub/actor'
+import { UserNotificationSettingModel } from '../models/account/user-notification-setting'
+import { UserNotificationSetting, UserNotificationSettingValue } from '../../shared/models/users'
-function createUserAuthorAndChannel (user: UserInstance, validateUser = true) {
- return db.sequelize.transaction(t => {
+async function createUserAccountAndChannel (userToCreate: UserModel, validateUser = true) {
+ const { user, account, videoChannel } = await sequelizeTypescript.transaction(async t => {
const userOptions = {
transaction: t,
validate: validateUser
}
- return user.save(userOptions)
- .then(user => {
- const author = db.Author.build({
- name: user.username,
- podId: null, // It is our pod
- userId: user.id
- })
-
- return author.save({ transaction: t })
- .then(author => ({ author, user }))
- })
- .then(({ author, user }) => {
- const remoteVideoAuthor = author.toAddRemoteJSON()
-
- // Now we'll add the video channel's meta data to our friends
- return addVideoAuthorToFriends(remoteVideoAuthor, t)
- .then(() => ({ author, user }))
- })
- .then(({ author, user }) => {
- const videoChannelInfo = {
- name: `Default ${user.username} channel`
- }
-
- return createVideoChannel(videoChannelInfo, author, t)
- .then(videoChannel => ({ author, user, videoChannel }))
- })
+ const userCreated = await userToCreate.save(userOptions)
+ userCreated.NotificationSetting = await createDefaultUserNotificationSettings(userCreated, t)
+
+ const accountCreated = await createLocalAccountWithoutKeys(userCreated.username, userCreated.id, null, t)
+ userCreated.Account = accountCreated
+
+ let channelName = userCreated.username + '_channel'
+
+ // Conflict, generate uuid instead
+ const actor = await ActorModel.loadLocalByName(channelName)
+ if (actor) channelName = uuidv4()
+
+ const videoChannelDisplayName = `Main ${userCreated.username} channel`
+ const videoChannelInfo = {
+ name: channelName,
+ displayName: videoChannelDisplayName
+ }
+ const videoChannel = await createVideoChannel(videoChannelInfo, accountCreated, t)
+
+ return { user: userCreated, account: accountCreated, videoChannel }
})
+
+ const [ accountKeys, channelKeys ] = await Promise.all([
+ setAsyncActorKeys(account.Actor),
+ setAsyncActorKeys(videoChannel.Actor)
+ ])
+
+ account.Actor = accountKeys
+ videoChannel.Actor = channelKeys
+
+ return { user, account, videoChannel } as { user: UserModel, account: AccountModel, videoChannel: VideoChannelModel }
+}
+
+async function createLocalAccountWithoutKeys (
+ name: string,
+ userId: number | null,
+ applicationId: number | null,
+ t: Sequelize.Transaction | undefined,
+ type: ActivityPubActorType= 'Person'
+) {
+ const url = getAccountActivityPubUrl(name)
+
+ const actorInstance = buildActorInstance(type, url, name)
+ const actorInstanceCreated = await actorInstance.save({ transaction: t })
+
+ const accountInstance = new AccountModel({
+ name,
+ userId,
+ applicationId,
+ actorId: actorInstanceCreated.id
+ } as FilteredModelAttributes<AccountModel>)
+
+ const accountInstanceCreated = await accountInstance.save({ transaction: t })
+ accountInstanceCreated.Actor = actorInstanceCreated
+
+ return accountInstanceCreated
+}
+
+async function createApplicationActor (applicationId: number) {
+ const accountCreated = await createLocalAccountWithoutKeys(SERVER_ACTOR_NAME, null, applicationId, undefined, 'Application')
+
+ accountCreated.Actor = await setAsyncActorKeys(accountCreated.Actor)
+
+ return accountCreated
}
// ---------------------------------------------------------------------------
export {
- createUserAuthorAndChannel
+ createApplicationActor,
+ createUserAccountAndChannel,
+ createLocalAccountWithoutKeys
+}
+
+// ---------------------------------------------------------------------------
+
+function createDefaultUserNotificationSettings (user: UserModel, t: Sequelize.Transaction | undefined) {
+ const values: UserNotificationSetting & { userId: number } = {
+ userId: user.id,
+ newVideoFromSubscription: UserNotificationSettingValue.WEB_NOTIFICATION,
+ newCommentOnMyVideo: UserNotificationSettingValue.WEB_NOTIFICATION,
+ myVideoImportFinished: UserNotificationSettingValue.WEB_NOTIFICATION,
+ myVideoPublished: UserNotificationSettingValue.WEB_NOTIFICATION,
+ videoAbuseAsModerator: UserNotificationSettingValue.WEB_NOTIFICATION_AND_EMAIL,
+ blacklistOnMyVideo: UserNotificationSettingValue.WEB_NOTIFICATION_AND_EMAIL,
+ newUserRegistration: UserNotificationSettingValue.WEB_NOTIFICATION,
+ commentMention: UserNotificationSettingValue.WEB_NOTIFICATION,
+ newFollow: UserNotificationSettingValue.WEB_NOTIFICATION
+ }
+
+ return UserNotificationSettingModel.create(values, { transaction: t })
}