-import * as passwordGenerator from 'password-generator'
-import { UserRole } from '../../shared'
+import { ensureDir, readdir, remove } from 'fs-extra'
+import passwordGenerator from 'password-generator'
+import { join } from 'path'
+import { UserRole } from '@shared/models'
import { logger } from '../helpers/logger'
-import { createApplicationActor, createUserAccountAndChannelAndPlaylist } from '../lib/user'
-import { UserModel } from '../models/account/user'
+import { buildUser, createApplicationActor, createUserAccountAndChannelAndPlaylist } from '../lib/user'
import { ApplicationModel } from '../models/application/application'
import { OAuthClientModel } from '../models/oauth/oauth-client'
import { applicationExist, clientsExist, usersExist } from './checker-after-init'
-import { FILES_CACHE, HLS_STREAMING_PLAYLIST_DIRECTORY, LAST_MIGRATION_VERSION } from './constants'
-import { sequelizeTypescript } from './database'
-import { ensureDir, remove } from 'fs-extra'
import { CONFIG } from './config'
+import { FILES_CACHE, HLS_STREAMING_PLAYLIST_DIRECTORY, LAST_MIGRATION_VERSION, RESUMABLE_UPLOAD_DIRECTORY } from './constants'
+import { sequelizeTypescript } from './database'
async function installApplication () {
try {
// Cache directories
for (const key of Object.keys(cacheDirectories)) {
const dir = cacheDirectories[key]
- tasks.push(remove(dir))
+ tasks.push(removeDirectoryOrContent(dir))
}
- tasks.push(remove(CONFIG.STORAGE.TMP_DIR))
+ tasks.push(removeDirectoryOrContent(CONFIG.STORAGE.TMP_DIR))
return Promise.all(tasks)
}
+async function removeDirectoryOrContent (dir: string) {
+ try {
+ await remove(dir)
+ } catch (err) {
+ logger.debug('Cannot remove directory %s. Removing content instead.', dir, { err })
+
+ const files = await readdir(dir)
+
+ for (const file of files) {
+ await remove(join(dir, file))
+ }
+ }
+}
+
function createDirectoriesIfNotExist () {
const storage = CONFIG.STORAGE
const cacheDirectories = Object.keys(FILES_CACHE)
// Playlist directories
tasks.push(ensureDir(HLS_STREAMING_PLAYLIST_DIRECTORY))
+ // Resumable upload directory
+ tasks.push(ensureDir(RESUMABLE_UPLOAD_DIRECTORY))
+
return Promise.all(tasks)
}
// Our password is weak so do not validate it
validatePassword = false
+ } else if (process.env.PT_INITIAL_ROOT_PASSWORD) {
+ password = process.env.PT_INITIAL_ROOT_PASSWORD
} else {
password = passwordGenerator(16, true)
}
- const userData = {
+ const user = buildUser({
username,
email,
password,
role,
- verified: true,
- nsfwPolicy: CONFIG.INSTANCE.DEFAULT_NSFW_POLICY,
+ emailVerified: true,
videoQuota: -1,
videoQuotaDaily: -1
- }
- const user = new UserModel(userData)
+ })
- await createUserAccountAndChannelAndPlaylist(user, validatePassword)
+ await createUserAccountAndChannelAndPlaylist({ userToCreate: user, channelNames: undefined, validateUser: validatePassword })
logger.info('Username: ' + username)
logger.info('User password: ' + password)
}