uptime-kuma/extra/reset-password.js

120 lines
3.7 KiB
JavaScript
Raw Normal View History

2021-08-09 05:34:44 +00:00
console.log("== Uptime Kuma Reset Password Tool ==");
const Database = require("../server/database");
const { R } = require("redbean-node");
const readline = require("readline");
const { initJWTSecret } = require("../server/util-server");
2022-04-18 07:21:58 +00:00
const User = require("../server/model/user");
const { io } = require("socket.io-client");
const { localWebSocketURL } = require("../server/config");
2021-09-20 08:29:18 +00:00
const args = require("args-parser")(process.argv);
2021-08-09 05:34:44 +00:00
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout
});
2021-10-15 17:33:44 +00:00
const main = async () => {
console.log("Connecting the database");
2021-09-20 08:29:18 +00:00
Database.init(args);
await Database.connect(false, false, true);
2021-08-09 05:34:44 +00:00
try {
2021-10-15 17:33:44 +00:00
// No need to actually reset the password for testing, just make sure no connection problem. It is ok for now.
if (!process.env.TEST_BACKEND) {
2021-10-16 05:02:04 +00:00
const user = await R.findOne("user");
2021-10-15 17:33:44 +00:00
if (! user) {
throw new Error("user not found, have you installed?");
}
2021-08-09 05:34:44 +00:00
2021-10-15 17:33:44 +00:00
console.log("Found user: " + user.username);
2021-08-09 05:34:44 +00:00
2021-10-15 17:33:44 +00:00
while (true) {
let password = await question("New Password: ");
let confirmPassword = await question("Confirm New Password: ");
2021-08-09 05:34:44 +00:00
2021-10-15 17:33:44 +00:00
if (password === confirmPassword) {
2022-04-18 07:21:58 +00:00
await User.resetPassword(user.id, password);
2021-08-09 05:34:44 +00:00
2021-10-15 17:33:44 +00:00
// Reset all sessions by reset jwt secret
await initJWTSecret();
2021-08-09 05:34:44 +00:00
// Disconnect all other socket clients of the user
await disconnectAllSocketClients(user.username, password);
2021-10-15 17:33:44 +00:00
break;
} else {
console.log("Passwords do not match, please try again.");
}
2021-08-09 05:34:44 +00:00
}
2021-10-15 17:33:44 +00:00
console.log("Password reset successfully.");
2021-08-09 05:34:44 +00:00
}
} catch (e) {
console.error("Error: " + e.message);
}
await Database.close();
2021-10-15 17:33:44 +00:00
rl.close();
2021-08-09 05:34:44 +00:00
2021-10-15 17:33:44 +00:00
console.log("Finished.");
};
2021-08-09 05:34:44 +00:00
/**
* Ask question of user
* @param {string} question Question to ask
* @returns {Promise<string>} Users response
*/
2021-08-09 05:34:44 +00:00
function question(question) {
return new Promise((resolve) => {
rl.question(question, (answer) => {
resolve(answer);
2021-10-15 17:33:44 +00:00
});
2021-08-09 05:34:44 +00:00
});
}
2021-10-15 17:33:44 +00:00
function disconnectAllSocketClients(username, password) {
return new Promise((resolve) => {
console.log("Connecting to " + localWebSocketURL + " to disconnect all other socket clients");
// Disconnect all socket connections
const socket = io(localWebSocketURL, {
reconnection: false,
timeout: 5000,
});
socket.on("connect", () => {
socket.emit("login", {
username,
password,
}, (res) => {
if (res.ok) {
console.log("Logged in.");
socket.emit("disconnectOtherSocketClients");
} else {
console.warn("Login failed.");
console.warn("Please restart the server to disconnect all sessions.");
}
socket.close();
});
});
socket.on("connect_error", function () {
// The localWebSocketURL is not guaranteed to be working for some complicated Uptime Kuma setup
// Ask the user to restart the server manually
console.warn("Failed to connect to " + localWebSocketURL);
console.warn("Please restart the server to disconnect all sessions manually.");
resolve();
});
socket.on("disconnect", () => {
resolve();
});
});
}
2021-10-15 17:33:44 +00:00
if (!process.env.TEST_BACKEND) {
main();
}
module.exports = {
main,
};