1 import * as passwordGenerator from 'password-generator'
2 import { UserRole } from '../../shared'
3 import { logger } from '../helpers/logger'
4 import { createApplicationActor, createUserAccountAndChannelAndPlaylist } from '../lib/user'
5 import { UserModel } from '../models/account/user'
6 import { ApplicationModel } from '../models/application/application'
7 import { OAuthClientModel } from '../models/oauth/oauth-client'
8 import { applicationExist, clientsExist, usersExist } from './checker-after-init'
9 import { CACHE, CONFIG, HLS_STREAMING_PLAYLIST_DIRECTORY, LAST_MIGRATION_VERSION } from './constants'
10 import { sequelizeTypescript } from './database'
11 import { remove, ensureDir } from 'fs-extra'
13 async function installApplication () {
17 sequelizeTypescript.sync()
20 createApplicationIfNotExist(),
21 createOAuthClientIfNotExist(),
22 createOAuthAdminIfNotExist()
27 removeCacheDirectories()
28 .then(() => createDirectoriesIfNotExist())
31 logger.error('Cannot install application.', { err })
36 // ---------------------------------------------------------------------------
42 // ---------------------------------------------------------------------------
44 function removeCacheDirectories () {
45 const cacheDirectories = Object.keys(CACHE)
46 .map(k => CACHE[k].DIRECTORY)
48 const tasks: Promise<any>[] = []
51 for (const key of Object.keys(cacheDirectories)) {
52 const dir = cacheDirectories[key]
53 tasks.push(remove(dir))
56 return Promise.all(tasks)
59 function createDirectoriesIfNotExist () {
60 const storage = CONFIG.STORAGE
61 const cacheDirectories = Object.keys(CACHE)
62 .map(k => CACHE[k].DIRECTORY)
64 const tasks: Promise<void>[] = []
65 for (const key of Object.keys(storage)) {
66 const dir = storage[key]
67 tasks.push(ensureDir(dir))
71 for (const key of Object.keys(cacheDirectories)) {
72 const dir = cacheDirectories[key]
73 tasks.push(ensureDir(dir))
76 // Playlist directories
77 tasks.push(ensureDir(HLS_STREAMING_PLAYLIST_DIRECTORY))
79 return Promise.all(tasks)
82 async function createOAuthClientIfNotExist () {
83 const exist = await clientsExist()
84 // Nothing to do, clients already exist
85 if (exist === true) return undefined
87 logger.info('Creating a default OAuth Client.')
89 const id = passwordGenerator(32, false, /[a-z0-9]/)
90 const secret = passwordGenerator(32, false, /[a-zA-Z0-9]/)
91 const client = new OAuthClientModel({
94 grants: [ 'password', 'refresh_token' ],
98 const createdClient = await client.save()
99 logger.info('Client id: ' + createdClient.clientId)
100 logger.info('Client secret: ' + createdClient.clientSecret)
105 async function createOAuthAdminIfNotExist () {
106 const exist = await usersExist()
107 // Nothing to do, users already exist
108 if (exist === true) return undefined
110 logger.info('Creating the administrator.')
112 const username = 'root'
113 const role = UserRole.ADMINISTRATOR
114 const email = CONFIG.ADMIN.EMAIL
115 let validatePassword = true
118 // Do not generate a random password for tests
119 if (process.env.NODE_ENV === 'test') {
122 if (process.env.NODE_APP_INSTANCE) {
123 password += process.env.NODE_APP_INSTANCE
126 // Our password is weak so do not validate it
127 validatePassword = false
129 password = passwordGenerator(16, true)
138 nsfwPolicy: CONFIG.INSTANCE.DEFAULT_NSFW_POLICY,
142 const user = new UserModel(userData)
144 await createUserAccountAndChannelAndPlaylist(user, validatePassword)
145 logger.info('Username: ' + username)
146 logger.info('User password: ' + password)
149 async function createApplicationIfNotExist () {
150 const exist = await applicationExist()
151 // Nothing to do, application already exist
152 if (exist === true) return undefined
154 logger.info('Creating application account.')
156 const application = await ApplicationModel.create({
157 migrationVersion: LAST_MIGRATION_VERSION
160 return createApplicationActor(application.id)