106 lines
2.5 KiB
JavaScript
106 lines
2.5 KiB
JavaScript
|
const { getDB } = require("../db")
|
||
|
|
const { SEPARATOR, StaticDatabases } = require("../db/constants")
|
||
|
|
const { getTenantId, DEFAULT_TENANT_ID, isMultiTenant } = require("./context")
|
||
|
|
const env = require("../environment")
|
||
|
|
|
||
|
|
const TENANT_DOC = StaticDatabases.PLATFORM_INFO.docs.tenants
|
||
|
|
const PLATFORM_INFO_DB = StaticDatabases.PLATFORM_INFO.name
|
||
|
|
|
||
|
|
exports.addTenantToUrl = url => {
|
||
|
|
const tenantId = getTenantId()
|
||
|
|
|
||
|
|
if (isMultiTenant()) {
|
||
|
|
const char = url.indexOf("?") === -1 ? "?" : "&"
|
||
|
|
url += `${char}tenantId=${tenantId}`
|
||
|
|
}
|
||
|
|
|
||
|
|
return url
|
||
|
|
}
|
||
|
|
|
||
|
|
exports.doesTenantExist = async tenantId => {
|
||
|
|
const db = getDB(PLATFORM_INFO_DB)
|
||
|
|
let tenants
|
||
|
|
try {
|
||
|
|
tenants = await db.get(TENANT_DOC)
|
||
|
|
} catch (err) {
|
||
|
|
// if theres an error the doc doesn't exist, no tenants exist
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return (
|
||
|
|
tenants &&
|
||
|
|
Array.isArray(tenants.tenantIds) &&
|
||
|
|
tenants.tenantIds.indexOf(tenantId) !== -1
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
exports.tryAddTenant = async (tenantId, userId, email) => {
|
||
|
|
const db = getDB(PLATFORM_INFO_DB)
|
||
|
|
const getDoc = async id => {
|
||
|
|
if (!id) {
|
||
|
|
return null
|
||
|
|
}
|
||
|
|
try {
|
||
|
|
return await db.get(id)
|
||
|
|
} catch (err) {
|
||
|
|
return { _id: id }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
let [tenants, userIdDoc, emailDoc] = await Promise.all([
|
||
|
|
getDoc(TENANT_DOC),
|
||
|
|
getDoc(userId),
|
||
|
|
getDoc(email),
|
||
|
|
])
|
||
|
|
if (!Array.isArray(tenants.tenantIds)) {
|
||
|
|
tenants = {
|
||
|
|
_id: TENANT_DOC,
|
||
|
|
tenantIds: [],
|
||
|
|
}
|
||
|
|
}
|
||
|
|
let promises = []
|
||
|
|
if (userIdDoc) {
|
||
|
|
userIdDoc.tenantId = tenantId
|
||
|
|
promises.push(db.put(userIdDoc))
|
||
|
|
}
|
||
|
|
if (emailDoc) {
|
||
|
|
emailDoc.tenantId = tenantId
|
||
|
|
promises.push(db.put(emailDoc))
|
||
|
|
}
|
||
|
|
if (tenants.tenantIds.indexOf(tenantId) === -1) {
|
||
|
|
tenants.tenantIds.push(tenantId)
|
||
|
|
promises.push(db.put(tenants))
|
||
|
|
}
|
||
|
|
await Promise.all(promises)
|
||
|
|
}
|
||
|
|
|
||
|
|
exports.getGlobalDB = (tenantId = null) => {
|
||
|
|
// tenant ID can be set externally, for example user API where
|
||
|
|
// new tenants are being created, this may be the case
|
||
|
|
if (!tenantId) {
|
||
|
|
tenantId = getTenantId()
|
||
|
|
}
|
||
|
|
|
||
|
|
let dbName
|
||
|
|
|
||
|
|
if (tenantId === DEFAULT_TENANT_ID) {
|
||
|
|
dbName = StaticDatabases.GLOBAL.name
|
||
|
|
} else {
|
||
|
|
dbName = `${tenantId}${SEPARATOR}${StaticDatabases.GLOBAL.name}`
|
||
|
|
}
|
||
|
|
|
||
|
|
return getDB(dbName)
|
||
|
|
}
|
||
|
|
|
||
|
|
exports.lookupTenantId = async userId => {
|
||
|
|
const db = getDB(StaticDatabases.PLATFORM_INFO.name)
|
||
|
|
let tenantId = env.MULTI_TENANCY ? DEFAULT_TENANT_ID : null
|
||
|
|
try {
|
||
|
|
const doc = await db.get(userId)
|
||
|
|
if (doc && doc.tenantId) {
|
||
|
|
tenantId = doc.tenantId
|
||
|
|
}
|
||
|
|
} catch (err) {
|
||
|
|
// just return the default
|
||
|
|
}
|
||
|
|
return tenantId
|
||
|
|
}
|