diff --git a/extra/remove-2fa.js b/extra/remove-2fa.js new file mode 100644 index 000000000..0f3f63462 --- /dev/null +++ b/extra/remove-2fa.js @@ -0,0 +1,60 @@ +console.log("== Uptime Kuma Remove 2FA Tool =="); +console.log("Loading the database"); + +const Database = require("../server/database"); +const { R } = require("redbean-node"); +const readline = require("readline"); +const TwoFA = require("../server/2fa"); +const args = require("args-parser")(process.argv); +const rl = readline.createInterface({ + input: process.stdin, + output: process.stdout +}); + +const main = async () => { + Database.init(args); + await Database.connect(); + + 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); + + let ans = await question("Are you sure want to remove 2FA? [y/N]"); + + if (ans.toLowerCase() === "y") { + await TwoFA.disable2FA(user.id); + console.log("2FA has been removed successfully."); + } + + } + } catch (e) { + console.error("Error: " + e.message); + } + + await Database.close(); + rl.close(); + + console.log("Finished."); +}; + +function question(question) { + return new Promise((resolve) => { + rl.question(question, (answer) => { + resolve(answer); + }); + }); +} + +if (!process.env.TEST_BACKEND) { + main(); +} + +module.exports = { + main, +}; diff --git a/package.json b/package.json index c7ec3d8ee..79f01b558 100644 --- a/package.json +++ b/package.json @@ -41,6 +41,7 @@ "update-version": "node extra/update-version.js", "mark-as-nightly": "node extra/mark-as-nightly.js", "reset-password": "node extra/reset-password.js", + "remove-2fa": "node extra/remove-2fa.js", "compile-install-script": "@powershell -NoProfile -ExecutionPolicy Unrestricted -Command ./extra/compile-install-script.ps1", "test-install-script-centos7": "npm run compile-install-script && docker build --progress plain -f test/test_install_script/centos7.dockerfile .", "test-install-script-alpine3": "npm run compile-install-script && docker build --progress plain -f test/test_install_script/alpine3.dockerfile .", diff --git a/server/2fa.js b/server/2fa.js new file mode 100644 index 000000000..bc8145cff --- /dev/null +++ b/server/2fa.js @@ -0,0 +1,14 @@ +const { checkLogin } = require("./util-server"); +const { R } = require("redbean-node"); + +class TwoFA { + + static async disable2FA(userID) { + return await R.exec("UPDATE `user` SET twofa_status = 0 WHERE id = ? ", [ + userID, + ]); + } + +} + +module.exports = TwoFA; diff --git a/server/server.js b/server/server.js index d1fd7ff29..78106515a 100644 --- a/server/server.js +++ b/server/server.js @@ -120,6 +120,7 @@ module.exports.io = io; const { sendNotificationList, sendHeartbeatList, sendImportantHeartbeatList, sendInfo } = require("./client"); const { statusPageSocketHandler } = require("./socket-handlers/status-page-socket-handler"); const databaseSocketHandler = require("./socket-handlers/database-socket-handler"); +const TwoFA = require("./2fa"); app.use(express.json()); @@ -420,10 +421,7 @@ exports.entryPage = "dashboard"; socket.on("disable2FA", async (callback) => { try { checkLogin(socket); - - await R.exec("UPDATE `user` SET twofa_status = 0 WHERE id = ? ", [ - socket.userID, - ]); + await TwoFA.disable2FA(socket.userID); callback({ ok: true,