mirror of
https://github.com/louislam/uptime-kuma.git
synced 2025-06-06 13:12:33 +02:00
Add enviourment variable for remote user authentication
This commit is contained in:
parent
ce9a97a107
commit
54cec2754d
2 changed files with 51 additions and 18 deletions
|
@ -1,9 +1,11 @@
|
|||
const basicAuth = require("express-basic-auth")
|
||||
const basicAuth = require("express-basic-auth");
|
||||
const passwordHash = require("./password-hash");
|
||||
const { R } = require("redbean-node");
|
||||
const { setting } = require("./util-server");
|
||||
const { debug } = require("../src/util");
|
||||
|
||||
const remoteUserHeader = process.env.REMOTE_USER_HEADER;
|
||||
|
||||
/**
|
||||
*
|
||||
* @param username : string
|
||||
|
@ -13,7 +15,7 @@ const { debug } = require("../src/util");
|
|||
exports.login = async function (username, password) {
|
||||
let user = await R.findOne("user", " username = ? AND active = 1 ", [
|
||||
username,
|
||||
])
|
||||
]);
|
||||
|
||||
if (user && passwordHash.verify(password, user.password)) {
|
||||
// Upgrade the hash to bcrypt
|
||||
|
@ -27,25 +29,38 @@ exports.login = async function (username, password) {
|
|||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
function myAuthorizer(username, password, callback) {
|
||||
};
|
||||
|
||||
function basicAuthHandler(username, password, callback) {
|
||||
setting("disableAuth").then((result) => {
|
||||
|
||||
if (result) {
|
||||
callback(null, true)
|
||||
callback(null, true);
|
||||
} else {
|
||||
exports.login(username, password).then((user) => {
|
||||
callback(null, user != null)
|
||||
})
|
||||
callback(null, user != null);
|
||||
});
|
||||
}
|
||||
})
|
||||
|
||||
});
|
||||
}
|
||||
|
||||
exports.basicAuth = basicAuth({
|
||||
authorizer: myAuthorizer,
|
||||
authorizeAsync: true,
|
||||
challenge: true,
|
||||
});
|
||||
async function authMiddleware(req, res, next) {
|
||||
if (remoteUserHeader !== undefined) {
|
||||
const remoteUser = req.headers[remoteUserHeader.toLowerCase()];
|
||||
if (remoteUser !== undefined) {
|
||||
let user = await R.findOne("user", " username = ? AND active = 1 ", [
|
||||
remoteUser,
|
||||
]);
|
||||
if (user) {
|
||||
next();
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
return basicAuth({
|
||||
authorizer: basicAuthHandler,
|
||||
authorizeAsync: true,
|
||||
challenge: true,
|
||||
})(req, res, next);
|
||||
}
|
||||
|
||||
exports.basicAuth = authMiddleware;
|
||||
|
|
|
@ -64,6 +64,9 @@ const port = parseInt(process.env.PORT || args.port || 3001);
|
|||
const sslKey = process.env.SSL_KEY || args["ssl-key"] || undefined;
|
||||
const sslCert = process.env.SSL_CERT || args["ssl-cert"] || undefined;
|
||||
|
||||
// Header AUTH
|
||||
const remoteUserHeader = process.env.REMOTE_USER_HEADER;
|
||||
|
||||
// Demo Mode?
|
||||
const demoMode = args["demo"] || false;
|
||||
|
||||
|
@ -180,7 +183,6 @@ exports.entryPage = "dashboard";
|
|||
|
||||
console.log("Adding socket handler");
|
||||
io.on("connection", async (socket) => {
|
||||
|
||||
socket.emit("info", {
|
||||
version: checkVersion.version,
|
||||
latestVersion: checkVersion.latestVersion,
|
||||
|
@ -1189,10 +1191,26 @@ exports.entryPage = "dashboard";
|
|||
// ***************************
|
||||
|
||||
debug("check auto login");
|
||||
console.log(socket.handshake.headers);
|
||||
if (await setting("disableAuth")) {
|
||||
console.log("Disabled Auth: auto login to admin");
|
||||
afterLogin(socket, await R.findOne("user"));
|
||||
socket.emit("autoLogin");
|
||||
} else if (remoteUserHeader !== undefined) {
|
||||
const remoteUser = socket.handshake.headers[remoteUserHeader.toLowerCase()];
|
||||
if (remoteUser !== undefined) {
|
||||
const user = await R.findOne("user", " username = ? AND active = 1 ", [
|
||||
remoteUser,
|
||||
]);
|
||||
if (user) {
|
||||
afterLogin(socket, user);
|
||||
socket.emit("autoLogin");
|
||||
} else {
|
||||
debug(`remote user ${remoteUser} doesnt exist`);
|
||||
}
|
||||
} else {
|
||||
debug("remote user header set but not found in headers");
|
||||
}
|
||||
} else {
|
||||
debug("need auth");
|
||||
}
|
||||
|
@ -1398,7 +1416,7 @@ function finalFunction() {
|
|||
gracefulShutdown(server, {
|
||||
signals: "SIGINT SIGTERM",
|
||||
timeout: 30000, // timeout: 30 secs
|
||||
development: false, // not in dev mode
|
||||
development: true, // not in dev mode
|
||||
forceExit: true, // triggers process.exit() at the end of shutdown process
|
||||
onShutdown: shutdownFunction, // shutdown function (async) - e.g. for cleanup DB, ...
|
||||
finally: finalFunction, // finally function (sync) - e.g. for logging
|
||||
|
|
Loading…
Add table
Reference in a new issue