69 lines
1.8 KiB
TypeScript
69 lines
1.8 KiB
TypeScript
import * as redis from "../redis/init"
|
|
import * as tenancy from "../tenancy"
|
|
import * as context from "../context"
|
|
import * as platform from "../platform"
|
|
import env from "../environment"
|
|
import * as accounts from "../accounts"
|
|
|
|
const EXPIRY_SECONDS = 3600
|
|
|
|
/**
|
|
* The default populate user function
|
|
*/
|
|
async function populateFromDB(userId: string, tenantId: string) {
|
|
const db = tenancy.getTenantDB(tenantId)
|
|
const user = await db.get<any>(userId)
|
|
user.budibaseAccess = true
|
|
if (!env.SELF_HOSTED && !env.DISABLE_ACCOUNT_PORTAL) {
|
|
const account = await accounts.getAccount(user.email)
|
|
if (account) {
|
|
user.account = account
|
|
user.accountPortalAccess = true
|
|
}
|
|
}
|
|
|
|
return user
|
|
}
|
|
|
|
/**
|
|
* Get the requested user by id.
|
|
* Use redis cache to first read the user.
|
|
* If not present fallback to loading the user directly and re-caching.
|
|
* @param {*} userId the id of the user to get
|
|
* @param {*} tenantId the tenant of the user to get
|
|
* @param {*} populateUser function to provide the user for re-caching. default to couch db
|
|
* @returns
|
|
*/
|
|
export async function getUser(
|
|
userId: string,
|
|
tenantId?: string,
|
|
populateUser?: any
|
|
) {
|
|
if (!populateUser) {
|
|
populateUser = populateFromDB
|
|
}
|
|
if (!tenantId) {
|
|
try {
|
|
tenantId = context.getTenantId()
|
|
} catch (err) {
|
|
tenantId = await platform.users.lookupTenantId(userId)
|
|
}
|
|
}
|
|
const client = await redis.getUserClient()
|
|
// try cache
|
|
let user = await client.get(userId)
|
|
if (!user) {
|
|
user = await populateUser(userId, tenantId)
|
|
await client.store(userId, user, EXPIRY_SECONDS)
|
|
}
|
|
if (user && !user.tenantId && tenantId) {
|
|
// make sure the tenant ID is always correct/set
|
|
user.tenantId = tenantId
|
|
}
|
|
return user
|
|
}
|
|
|
|
export async function invalidateUser(userId: string) {
|
|
const client = await redis.getUserClient()
|
|
await client.delete(userId)
|
|
}
|