]>
Commit | Line | Data |
---|---|---|
1 | import * as passwordGenerator from 'password-generator' | |
2 | import { UserRole } from '../../shared' | |
3 | import { logger } from '../helpers/logger' | |
4 | import { createApplicationActor, createUserAccountAndChannel } 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' | |
9 | import { CACHE, CONFIG, LAST_MIGRATION_VERSION } from './constants' | |
10 | import { sequelizeTypescript } from './database' | |
11 | import { remove, ensureDir } from 'fs-extra' | |
12 | ||
13 | async function installApplication () { | |
14 | try { | |
15 | await sequelizeTypescript.sync() | |
16 | await removeCacheDirectories() | |
17 | await createDirectoriesIfNotExist() | |
18 | await createApplicationIfNotExist() | |
19 | await createOAuthClientIfNotExist() | |
20 | await createOAuthAdminIfNotExist() | |
21 | } catch (err) { | |
22 | logger.error('Cannot install application.', { err }) | |
23 | process.exit(-1) | |
24 | } | |
25 | } | |
26 | ||
27 | // --------------------------------------------------------------------------- | |
28 | ||
29 | export { | |
30 | installApplication | |
31 | } | |
32 | ||
33 | // --------------------------------------------------------------------------- | |
34 | ||
35 | function removeCacheDirectories () { | |
36 | const cacheDirectories = Object.keys(CACHE) | |
37 | .map(k => CACHE[k].DIRECTORY) | |
38 | ||
39 | const tasks: Promise<any>[] = [] | |
40 | ||
41 | // Cache directories | |
42 | for (const key of Object.keys(cacheDirectories)) { | |
43 | const dir = cacheDirectories[key] | |
44 | tasks.push(remove(dir)) | |
45 | } | |
46 | ||
47 | return Promise.all(tasks) | |
48 | } | |
49 | ||
50 | function createDirectoriesIfNotExist () { | |
51 | const storage = CONFIG.STORAGE | |
52 | const cacheDirectories = Object.keys(CACHE) | |
53 | .map(k => CACHE[k].DIRECTORY) | |
54 | ||
55 | const tasks: Promise<void>[] = [] | |
56 | for (const key of Object.keys(storage)) { | |
57 | const dir = storage[key] | |
58 | tasks.push(ensureDir(dir)) | |
59 | } | |
60 | ||
61 | // Cache directories | |
62 | for (const key of Object.keys(cacheDirectories)) { | |
63 | const dir = cacheDirectories[key] | |
64 | tasks.push(ensureDir(dir)) | |
65 | } | |
66 | ||
67 | return Promise.all(tasks) | |
68 | } | |
69 | ||
70 | async function createOAuthClientIfNotExist () { | |
71 | const exist = await clientsExist() | |
72 | // Nothing to do, clients already exist | |
73 | if (exist === true) return undefined | |
74 | ||
75 | logger.info('Creating a default OAuth Client.') | |
76 | ||
77 | const id = passwordGenerator(32, false, /[a-z0-9]/) | |
78 | const secret = passwordGenerator(32, false, /[a-zA-Z0-9]/) | |
79 | const client = new OAuthClientModel({ | |
80 | clientId: id, | |
81 | clientSecret: secret, | |
82 | grants: [ 'password', 'refresh_token' ], | |
83 | redirectUris: null | |
84 | }) | |
85 | ||
86 | const createdClient = await client.save() | |
87 | logger.info('Client id: ' + createdClient.clientId) | |
88 | logger.info('Client secret: ' + createdClient.clientSecret) | |
89 | ||
90 | return undefined | |
91 | } | |
92 | ||
93 | async function createOAuthAdminIfNotExist () { | |
94 | const exist = await usersExist() | |
95 | // Nothing to do, users already exist | |
96 | if (exist === true) return undefined | |
97 | ||
98 | logger.info('Creating the administrator.') | |
99 | ||
100 | const username = 'root' | |
101 | const role = UserRole.ADMINISTRATOR | |
102 | const email = CONFIG.ADMIN.EMAIL | |
103 | let validatePassword = true | |
104 | let password = '' | |
105 | ||
106 | // Do not generate a random password for tests | |
107 | if (process.env.NODE_ENV === 'test') { | |
108 | password = 'test' | |
109 | ||
110 | if (process.env.NODE_APP_INSTANCE) { | |
111 | password += process.env.NODE_APP_INSTANCE | |
112 | } | |
113 | ||
114 | // Our password is weak so do not validate it | |
115 | validatePassword = false | |
116 | } else { | |
117 | password = passwordGenerator(16, true) | |
118 | } | |
119 | ||
120 | const userData = { | |
121 | username, | |
122 | email, | |
123 | password, | |
124 | role, | |
125 | nsfwPolicy: CONFIG.INSTANCE.DEFAULT_NSFW_POLICY, | |
126 | videoQuota: -1, | |
127 | videoQuotaDaily: -1 | |
128 | } | |
129 | const user = new UserModel(userData) | |
130 | ||
131 | await createUserAccountAndChannel(user, validatePassword) | |
132 | logger.info('Username: ' + username) | |
133 | logger.info('User password: ' + password) | |
134 | } | |
135 | ||
136 | async function createApplicationIfNotExist () { | |
137 | const exist = await applicationExist() | |
138 | // Nothing to do, application already exist | |
139 | if (exist === true) return undefined | |
140 | ||
141 | logger.info('Creating application account.') | |
142 | ||
143 | const application = await ApplicationModel.create({ | |
144 | migrationVersion: LAST_MIGRATION_VERSION | |
145 | }) | |
146 | ||
147 | return createApplicationActor(application.id) | |
148 | } |