Move licensing middleware to pro repo
This commit is contained in:
parent
5e94c8a931
commit
ad4a268a69
|
@ -15,6 +15,8 @@ module.exports = {
|
||||||
auth: require("../auth"),
|
auth: require("../auth"),
|
||||||
constants: require("../constants"),
|
constants: require("../constants"),
|
||||||
migrations: require("../migrations"),
|
migrations: require("../migrations"),
|
||||||
licensing: require("./licensing"),
|
|
||||||
errors: require("./errors"),
|
errors: require("./errors"),
|
||||||
|
env: require("./environment"),
|
||||||
|
accounts: require("./cloud/accounts"),
|
||||||
|
tenancy: require("./tenancy"),
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,35 +0,0 @@
|
||||||
const redis = require("./redis")
|
|
||||||
const env = require("../../environment")
|
|
||||||
const accounts = require("../../cloud/accounts")
|
|
||||||
|
|
||||||
const EXPIRY_SECONDS = 3600
|
|
||||||
|
|
||||||
const populateLicense = async tenantId => {
|
|
||||||
if (env.SELF_HOSTED) {
|
|
||||||
// get license key
|
|
||||||
} else {
|
|
||||||
return accounts.getLicense(tenantId)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
exports.getLicense = async (tenantId, opts = { populateLicense: null }) => {
|
|
||||||
// try cache
|
|
||||||
const client = await redis.getClient()
|
|
||||||
let license = await client.get(tenantId)
|
|
||||||
if (!license) {
|
|
||||||
const populate = opts.populateLicense
|
|
||||||
? opts.populateLicense
|
|
||||||
: populateLicense
|
|
||||||
license = await populate(tenantId)
|
|
||||||
if (license) {
|
|
||||||
client.store(tenantId, license, EXPIRY_SECONDS)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return license
|
|
||||||
}
|
|
||||||
|
|
||||||
exports.invalidate = async tenantId => {
|
|
||||||
const client = await redis.getClient()
|
|
||||||
await client.delete(tenantId)
|
|
||||||
}
|
|
|
@ -1,25 +0,0 @@
|
||||||
const Redis = require("../../redis")
|
|
||||||
const utils = require("../../redis/utils")
|
|
||||||
|
|
||||||
let client
|
|
||||||
|
|
||||||
const init = async () => {
|
|
||||||
client = await new Redis(utils.Databases.LICENSES).init()
|
|
||||||
}
|
|
||||||
|
|
||||||
const shutdown = async () => {
|
|
||||||
if (client) {
|
|
||||||
await client.finish()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
process.on("exit", async () => {
|
|
||||||
await shutdown()
|
|
||||||
})
|
|
||||||
|
|
||||||
exports.getClient = async () => {
|
|
||||||
if (!client) {
|
|
||||||
await init()
|
|
||||||
}
|
|
||||||
return client
|
|
||||||
}
|
|
|
@ -1,9 +0,0 @@
|
||||||
const middleware = require("./middleware")
|
|
||||||
const cache = require("./cache")
|
|
||||||
const usage = require("./usage")
|
|
||||||
|
|
||||||
module.exports = {
|
|
||||||
middleware,
|
|
||||||
cache,
|
|
||||||
usage,
|
|
||||||
}
|
|
|
@ -1,17 +0,0 @@
|
||||||
const cache = require("../cache")
|
|
||||||
|
|
||||||
const buildLicensingMiddleware = opts => {
|
|
||||||
return async (ctx, next) => {
|
|
||||||
if (ctx.user) {
|
|
||||||
const tenantId = ctx.user.tenantId
|
|
||||||
const license = await cache.getLicense(tenantId, opts)
|
|
||||||
if (license) {
|
|
||||||
ctx.user.license = license
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return next()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
module.exports = buildLicensingMiddleware
|
|
|
@ -1,41 +0,0 @@
|
||||||
const { getTenantId } = require("../../context")
|
|
||||||
const cache = require("../cache")
|
|
||||||
const env = require("../../environment")
|
|
||||||
const utils = require("../../utils")
|
|
||||||
const { UsageLimitError } = require("../../errors")
|
|
||||||
|
|
||||||
const UsageType = {
|
|
||||||
MONTHLY: "monthly",
|
|
||||||
STATIC: "static",
|
|
||||||
}
|
|
||||||
|
|
||||||
const StaticUsageLimits = {
|
|
||||||
MAX_DEVELOPERS: "maxDevelopers",
|
|
||||||
QUERY_TIMEOUT_SECONDS: "queryTimeoutSeconds",
|
|
||||||
}
|
|
||||||
|
|
||||||
// eslint-disable-next-line no-unused-vars
|
|
||||||
const MonthlyUsageLimits = {
|
|
||||||
MAX_QUERY_COUNT: "maxQueryCount",
|
|
||||||
}
|
|
||||||
|
|
||||||
exports.checkMaxDevelopers = async () => {
|
|
||||||
const developerCount = await utils.getBuildersCount()
|
|
||||||
await checkUsageLimit(
|
|
||||||
developerCount,
|
|
||||||
UsageType.STATIC,
|
|
||||||
StaticUsageLimits.MAX_DEVELOPERS
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
const checkUsageLimit = async (currentUsage, usageType, usageLimit) => {
|
|
||||||
const tenantId = getTenantId()
|
|
||||||
const license = await cache.getLicense(tenantId)
|
|
||||||
const limit = license.usageLimits[usageType][usageLimit]
|
|
||||||
if (currentUsage >= limit.value) {
|
|
||||||
throw new UsageLimitError(
|
|
||||||
`Licensed ${limit.name} has been exceeded. To upgrade, visit ${env.ACCOUNT_PORTAL_URL}/portal/plans`,
|
|
||||||
limit.name
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -20,7 +20,6 @@ const { hash } = require("./hashing")
|
||||||
const userCache = require("./cache/user")
|
const userCache = require("./cache/user")
|
||||||
const env = require("./environment")
|
const env = require("./environment")
|
||||||
const { getUserSessions, invalidateSessions } = require("./security/sessions")
|
const { getUserSessions, invalidateSessions } = require("./security/sessions")
|
||||||
const { usage } = require("./licensing")
|
|
||||||
|
|
||||||
const APP_PREFIX = DocumentTypes.APP + SEPARATOR
|
const APP_PREFIX = DocumentTypes.APP + SEPARATOR
|
||||||
|
|
||||||
|
@ -182,11 +181,11 @@ exports.saveUser = async (
|
||||||
hashPassword = true,
|
hashPassword = true,
|
||||||
requirePassword = true
|
requirePassword = true
|
||||||
) => {
|
) => {
|
||||||
// new user
|
// // new user
|
||||||
// check license restrictions
|
// // check license restrictions
|
||||||
if (!user._id && user.builder) {
|
// if (!user._id && user.builder) {
|
||||||
await usage.checkMaxDevelopers()
|
// await limits.checkMaxDevelopers()
|
||||||
}
|
// }
|
||||||
|
|
||||||
if (!tenantId) {
|
if (!tenantId) {
|
||||||
throw "No tenancy specified."
|
throw "No tenancy specified."
|
||||||
|
|
|
@ -11,7 +11,7 @@ const zlib = require("zlib")
|
||||||
const { mainRoutes, staticRoutes } = require("./routes")
|
const { mainRoutes, staticRoutes } = require("./routes")
|
||||||
const pkg = require("../../package.json")
|
const pkg = require("../../package.json")
|
||||||
const env = require("../environment")
|
const env = require("../environment")
|
||||||
const { licensing } = require("@budibase/backend-core")
|
const { middleware: licensing } = require("@budibase/pro")
|
||||||
|
|
||||||
const router = new Router()
|
const router = new Router()
|
||||||
|
|
||||||
|
@ -55,7 +55,7 @@ router
|
||||||
.use(currentApp)
|
.use(currentApp)
|
||||||
// this middleware will try to use the app ID to determine the tenancy
|
// this middleware will try to use the app ID to determine the tenancy
|
||||||
.use(buildAppTenancyMiddleware())
|
.use(buildAppTenancyMiddleware())
|
||||||
.use(licensing.middleware())
|
.use(licensing())
|
||||||
.use(auditLog)
|
.use(auditLog)
|
||||||
|
|
||||||
// error handling middleware
|
// error handling middleware
|
||||||
|
|
|
@ -8,7 +8,7 @@ const {
|
||||||
buildTenancyMiddleware,
|
buildTenancyMiddleware,
|
||||||
buildCsrfMiddleware,
|
buildCsrfMiddleware,
|
||||||
} = require("@budibase/backend-core/auth")
|
} = require("@budibase/backend-core/auth")
|
||||||
const { licensing } = require("@budibase/backend-core")
|
const { middleware: licensing } = require("@budibase/pro")
|
||||||
const { errors } = require("@budibase/backend-core")
|
const { errors } = require("@budibase/backend-core")
|
||||||
|
|
||||||
const PUBLIC_ENDPOINTS = [
|
const PUBLIC_ENDPOINTS = [
|
||||||
|
@ -93,7 +93,7 @@ router
|
||||||
.use(buildAuthMiddleware(PUBLIC_ENDPOINTS))
|
.use(buildAuthMiddleware(PUBLIC_ENDPOINTS))
|
||||||
.use(buildTenancyMiddleware(PUBLIC_ENDPOINTS, NO_TENANCY_ENDPOINTS))
|
.use(buildTenancyMiddleware(PUBLIC_ENDPOINTS, NO_TENANCY_ENDPOINTS))
|
||||||
.use(buildCsrfMiddleware({ noCsrfPatterns: NO_CSRF_ENDPOINTS }))
|
.use(buildCsrfMiddleware({ noCsrfPatterns: NO_CSRF_ENDPOINTS }))
|
||||||
.use(licensing.middleware())
|
.use(licensing())
|
||||||
// for now no public access is allowed to worker (bar health check)
|
// for now no public access is allowed to worker (bar health check)
|
||||||
.use((ctx, next) => {
|
.use((ctx, next) => {
|
||||||
if (ctx.publicEndpoint) {
|
if (ctx.publicEndpoint) {
|
||||||
|
|
Loading…
Reference in New Issue