130 lines
3.9 KiB
TypeScript
130 lines
3.9 KiB
TypeScript
import { Database } from "../backend/database";
|
|
import { R } from "redbean-node";
|
|
import readline from "readline";
|
|
import { User } from "../backend/models/user";
|
|
import { DockgeServer } from "../backend/dockge-server";
|
|
import { log } from "../backend/log";
|
|
import { io } from "socket.io-client";
|
|
import { BaseRes } from "../common/util-common";
|
|
|
|
console.log("== Dockge Reset Password Tool ==");
|
|
|
|
const rl = readline.createInterface({
|
|
input: process.stdin,
|
|
output: process.stdout
|
|
});
|
|
|
|
const server = new DockgeServer();
|
|
|
|
export const main = async () => {
|
|
// Check if
|
|
console.log("Connecting the database");
|
|
try {
|
|
await Database.init(server);
|
|
} catch (e) {
|
|
if (e instanceof Error) {
|
|
log.error("server", "Failed to connect to your database: " + e.message);
|
|
}
|
|
process.exit(1);
|
|
}
|
|
|
|
try {
|
|
// 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) {
|
|
const user = await R.findOne("user");
|
|
if (! user) {
|
|
throw new Error("user not found, have you installed?");
|
|
}
|
|
|
|
console.log("Found user: " + user.username);
|
|
|
|
while (true) {
|
|
let password = await question("New Password: ");
|
|
let confirmPassword = await question("Confirm New Password: ");
|
|
|
|
if (password === confirmPassword) {
|
|
await User.resetPassword(user.id, password);
|
|
|
|
// Reset all sessions by reset jwt secret
|
|
await server.initJWTSecret();
|
|
|
|
console.log("Password reset successfully.");
|
|
|
|
// Disconnect all other socket clients of the user
|
|
await disconnectAllSocketClients(user.username, password);
|
|
|
|
break;
|
|
} else {
|
|
console.log("Passwords do not match, please try again.");
|
|
}
|
|
}
|
|
}
|
|
} catch (e) {
|
|
if (e instanceof Error) {
|
|
console.error("Error: " + e.message);
|
|
}
|
|
}
|
|
|
|
await Database.close();
|
|
rl.close();
|
|
|
|
console.log("Finished.");
|
|
};
|
|
|
|
/**
|
|
* Ask question of user
|
|
* @param question Question to ask
|
|
* @returns Users response
|
|
*/
|
|
function question(question : string) : Promise<string> {
|
|
return new Promise((resolve) => {
|
|
rl.question(question, (answer) => {
|
|
resolve(answer);
|
|
});
|
|
});
|
|
}
|
|
|
|
function disconnectAllSocketClients(username : string, password : string) : Promise<void> {
|
|
return new Promise((resolve) => {
|
|
const url = server.getLocalWebSocketURL();
|
|
|
|
console.log("Connecting to " + url + " to disconnect all other socket clients");
|
|
|
|
// Disconnect all socket connections
|
|
const socket = io(url, {
|
|
reconnection: false,
|
|
timeout: 5000,
|
|
});
|
|
socket.on("connect", () => {
|
|
socket.emit("login", {
|
|
username,
|
|
password,
|
|
}, (res : BaseRes) => {
|
|
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 " + url);
|
|
console.warn("Please restart the server to disconnect all sessions manually.");
|
|
resolve();
|
|
});
|
|
socket.on("disconnect", () => {
|
|
resolve();
|
|
});
|
|
});
|
|
}
|
|
|
|
if (!process.env.TEST_BACKEND) {
|
|
main();
|
|
}
|