Add account recovery console script

recover.js

@@ -0,0 +1,62 @@
+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();