Spaces:
Sleeping
Sleeping
const connectDb = require('@librechat/backend/lib/db/connectDb'); | |
const migrateDb = require('@librechat/backend/lib/db/migrateDb'); | |
const { registerUser } = require('@librechat/backend/server/services/auth.service'); | |
const { askQuestion, silentExit } = require('./helpers'); | |
const User = require('@librechat/backend/models/User'); | |
(async () => { | |
/** | |
* Connect to the database | |
* - If it takes a while, we'll warn the user | |
*/ | |
// Warn the user if this is taking a while | |
let timeout = setTimeout(() => { | |
console.orange( | |
'This is taking a while... You may need to check your connection if this fails.', | |
); | |
timeout = setTimeout(() => { | |
console.orange('Still going... Might as well assume the connection failed...'); | |
timeout = setTimeout(() => { | |
console.orange('Error incoming in 3... 2... 1...'); | |
}, 13000); | |
}, 10000); | |
}, 5000); | |
// Attempt to connect to the database | |
try { | |
console.orange('Warming up the engines...'); | |
await connectDb(); | |
clearTimeout(timeout); | |
await migrateDb(); | |
} catch (e) { | |
console.error(e); | |
silentExit(1); | |
} | |
/** | |
* Show the welcome / help menu | |
*/ | |
console.purple('--------------------------'); | |
console.purple('Create a new user account!'); | |
console.purple('--------------------------'); | |
// If we don't have enough arguments, show the help menu | |
if (process.argv.length < 5) { | |
console.orange('Usage: npm run create-user <email> <name> <username>'); | |
console.orange('Note: if you do not pass in the arguments, you will be prompted for them.'); | |
console.orange( | |
'If you really need to pass in the password, you can do so as the 4th argument (not recommended for security).', | |
); | |
console.purple('--------------------------'); | |
} | |
/** | |
* Set up the variables we need and get the arguments if they were passed in | |
*/ | |
let email = ''; | |
let password = ''; | |
let name = ''; | |
let username = ''; | |
// If we have the right number of arguments, lets use them | |
if (process.argv.length >= 4) { | |
email = process.argv[2]; | |
name = process.argv[3]; | |
if (process.argv.length >= 5) { | |
username = process.argv[4]; | |
} | |
if (process.argv.length >= 6) { | |
console.red('Warning: password passed in as argument, this is not secure!'); | |
password = process.argv[5]; | |
} | |
} | |
/** | |
* If we don't have the right number of arguments, lets prompt the user for them | |
*/ | |
if (!email) { | |
email = await askQuestion('Email:'); | |
} | |
// Validate the email | |
if (!email.includes('@')) { | |
console.red('Error: Invalid email address!'); | |
silentExit(1); | |
} | |
const defaultName = email.split('@')[0]; | |
if (!name) { | |
name = await askQuestion('Name: (default is: ' + defaultName + ')'); | |
if (!name) { | |
name = defaultName; | |
} | |
} | |
if (!username) { | |
username = await askQuestion('Username: (default is: ' + defaultName + ')'); | |
if (!username) { | |
username = defaultName; | |
} | |
} | |
if (!password) { | |
password = await askQuestion('Password: (leave blank, to generate one)'); | |
if (!password) { | |
// Make it a random password, length 18 | |
password = Math.random().toString(36).slice(-18); | |
console.orange('Your password is: ' + password); | |
} | |
} | |
// Validate the user doesn't already exist | |
const userExists = await User.findOne({ $or: [{ email }, { username }] }); | |
if (userExists) { | |
console.red('Error: A user with that email or username already exists!'); | |
silentExit(1); | |
} | |
/** | |
* Now that we have all the variables we need, lets create the user | |
*/ | |
const user = { email, password, name, username, confirm_password: password }; | |
let result; | |
try { | |
result = await registerUser(user); | |
} catch (error) { | |
console.red('Error: ' + error.message); | |
silentExit(1); | |
} | |
// Check the result | |
if (result.status !== 200) { | |
console.red('Error: ' + result.message); | |
silentExit(1); | |
} | |
// Done! | |
console.green('User created successfully!'); | |
silentExit(0); | |
})(); | |