SillyTavern/recover.js

63 lines
1.7 KiB
JavaScript
Raw Normal View History

2024-04-12 20:31:43 +02:00
const yaml = require('yaml');
const fs = require('fs');
const storage = require('node-persist');
const users = require('./src/users');
const userAccount = process.argv[2];
const userPassword = process.argv[3];
if (!userAccount) {
console.error('A tool for recovering lost SillyTavern accounts. Uses a "dataRoot" setting from config.yaml file.');
console.error('Usage: node recover.js [account] (password)');
console.error('Example: node recover.js admin password');
process.exit(1);
}
async function initStorage() {
const config = yaml.parse(fs.readFileSync('config.yaml', 'utf8'));
const dataRoot = config.dataRoot;
if (!dataRoot) {
console.error('No "dataRoot" setting found in config.yaml file.');
process.exit(1);
}
await users.initUserStorage(dataRoot);
}
async function main() {
await initStorage();
/**
* @type {import('./src/users').User}
*/
const user = await storage.get(users.toKey(userAccount));
if (!user) {
console.error(`User "${userAccount}" not found.`);
process.exit(1);
}
if (!user.enabled) {
console.log('User is disabled. Enabling...');
user.enabled = true;
}
if (userPassword) {
console.log('Setting new password...');
const salt = users.getPasswordSalt();
const passwordHash = users.getPasswordHash(userPassword, salt);
user.password = passwordHash;
user.salt = salt;
} else {
console.log('Setting an empty password...');
user.password = '';
user.salt = '';
}
await storage.setItem(users.toKey(userAccount), user);
console.log('User recovered. A program will exit now.');
}
main();