const CouchDB = require("../../../db") const { generateGlobalUserID, getGlobalUserParams, StaticDatabases } = require("@budibase/auth").db const { hash, getGlobalUserByEmail } = require("@budibase/auth").utils const { UserStatus, EmailTemplatePurpose } = require("../../../constants") const { checkInviteCode } = require("../../../utilities/redis") const { sendEmail } = require("../../../utilities/email") const { user: userCache } = require("@budibase/auth/cache") const { invalidateSessions } = require("@budibase/auth/sessions") const GLOBAL_DB = StaticDatabases.GLOBAL.name async function allUsers() { const db = new CouchDB(GLOBAL_DB) const response = await db.allDocs( getGlobalUserParams(null, { include_docs: true, }) ) return response.rows.map(row => row.doc) } exports.save = async ctx => { const db = new CouchDB(GLOBAL_DB) const { email, password, _id } = ctx.request.body // make sure another user isn't using the same email let dbUser if (email) { dbUser = await getGlobalUserByEmail(email) if (dbUser != null && (dbUser._id !== _id || Array.isArray(dbUser))) { ctx.throw(400, "Email address already in use.") } } else { dbUser = await db.get(_id) } // get the password, make sure one is defined let hashedPassword if (password) { hashedPassword = await hash(password) } else if (dbUser) { hashedPassword = dbUser.password } else { ctx.throw(400, "Password must be specified.") } let user = { ...dbUser, ...ctx.request.body, _id: _id || generateGlobalUserID(), password: hashedPassword, } // make sure the roles object is always present if (!user.roles) { user.roles = {} } // add the active status to a user if its not provided if (user.status == null) { user.status = UserStatus.ACTIVE } try { const response = await db.put({ password: hashedPassword, ...user, }) await userCache.invalidateUser(response.id) ctx.body = { _id: response.id, _rev: response.rev, email, } } catch (err) { if (err.status === 409) { ctx.throw(400, "User exists already") } else { ctx.throw(err.status, err) } } } exports.adminUser = async ctx => { const db = new CouchDB(GLOBAL_DB) const response = await db.allDocs( getGlobalUserParams(null, { include_docs: true, }) ) if (response.rows.some(row => row.doc.admin)) { ctx.throw(403, "You cannot initialise once an admin user has been created.") } const { email, password } = ctx.request.body ctx.request.body = { email: email, password: password, roles: {}, builder: { global: true, }, admin: { global: true, }, } await exports.save(ctx) } exports.destroy = async ctx => { const db = new CouchDB(GLOBAL_DB) const dbUser = await db.get(ctx.params.id) await db.remove(dbUser._id, dbUser._rev) await userCache.invalidateUser(dbUser._id) await invalidateSessions(dbUser._id) ctx.body = { message: `User ${ctx.params.id} deleted.`, } } exports.removeAppRole = async ctx => { const { appId } = ctx.params const db = new CouchDB(GLOBAL_DB) const users = await allUsers() const bulk = [] const cacheInvalidations = [] for (let user of users) { if (user.roles[appId]) { cacheInvalidations.push(userCache.invalidateUser(user._id)) delete user.roles[appId] bulk.push(user) } } await db.bulkDocs(bulk) await Promise.all(cacheInvalidations) ctx.body = { message: "App role removed from all users", } } exports.getSelf = async ctx => { if (!ctx.user) { ctx.throw(403, "User not logged in") } ctx.params = { id: ctx.user._id, } // this will set the body await exports.find(ctx) } exports.updateSelf = async ctx => { const db = new CouchDB(GLOBAL_DB) const user = await db.get(ctx.user._id) if (ctx.request.body.password) { ctx.request.body.password = await hash(ctx.request.body.password) } // don't allow sending up an ID/Rev, always use the existing one delete ctx.request.body._id delete ctx.request.body._rev const response = await db.put({ ...user, ...ctx.request.body, }) await userCache.invalidateUser(user._id) ctx.body = { _id: response.id, _rev: response.rev, } } // called internally by app server user fetch exports.fetch = async ctx => { const users = await allUsers() // user hashed password shouldn't ever be returned for (let user of users) { if (user) { delete user.password } } ctx.body = users } // called internally by app server user find exports.find = async ctx => { const db = new CouchDB(GLOBAL_DB) let user try { user = await db.get(ctx.params.id) } catch (err) { // no user found, just return nothing user = {} } if (user) { delete user.password } ctx.body = user } exports.invite = async ctx => { const { email, userInfo } = ctx.request.body const existing = await getGlobalUserByEmail(email) if (existing) { ctx.throw(400, "Email address already in use.") } await sendEmail(email, EmailTemplatePurpose.INVITATION, { subject: "{{ company }} platform invitation", info: userInfo, }) ctx.body = { message: "Invitation has been sent.", } } exports.inviteAccept = async ctx => { const { inviteCode, password, firstName, lastName } = ctx.request.body try { // info is an extension of the user object that was stored by admin const { email, info } = await checkInviteCode(inviteCode) // only pass through certain props for accepting ctx.request.body = { firstName, lastName, password, email, ...info, } // this will flesh out the body response await exports.save(ctx) } catch (err) { ctx.throw(400, "Unable to create new user, invitation invalid.") } }