forked from tsai/budibase
Rory Powell
5 years ago
22 changed files with 345 additions and 440 deletions
@ -1,172 +0,0 @@ |
|||||
import { quotas, StaticQuotaName, QuotaUsageType } from "@budibase/pro" |
|
||||
const { getUniqueRows } = require("../utilities/usageQuota/rows") |
|
||||
const { |
|
||||
isExternalTable, |
|
||||
isRowId: isExternalRowId, |
|
||||
} = require("../integrations/utils") |
|
||||
const { getAppDB } = require("@budibase/backend-core/context") |
|
||||
|
|
||||
// currently only counting new writes and deletes
|
|
||||
const METHOD_MAP: any = { |
|
||||
POST: 1, |
|
||||
DELETE: -1, |
|
||||
} |
|
||||
|
|
||||
const DOMAIN_MAP: any = { |
|
||||
rows: { |
|
||||
name: StaticQuotaName.ROWS, |
|
||||
type: QuotaUsageType.STATIC, |
|
||||
}, |
|
||||
applications: { |
|
||||
name: StaticQuotaName.APPS, |
|
||||
type: QuotaUsageType.STATIC, |
|
||||
}, |
|
||||
} |
|
||||
|
|
||||
function getQuotaInfo(url: string) { |
|
||||
for (let domain of Object.keys(DOMAIN_MAP)) { |
|
||||
if (url.indexOf(domain) !== -1) { |
|
||||
return DOMAIN_MAP[domain] |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
module.exports = async (ctx: any, next: any) => { |
|
||||
if (!quotas.useQuotas()) { |
|
||||
return next() |
|
||||
} |
|
||||
|
|
||||
let usage = METHOD_MAP[ctx.req.method] |
|
||||
const quotaInfo = getQuotaInfo(ctx.req.url) |
|
||||
if (usage == null || quotaInfo == null) { |
|
||||
return next() |
|
||||
} |
|
||||
// post request could be a save of a pre-existing entry
|
|
||||
if (ctx.request.body && ctx.request.body._id && ctx.request.body._rev) { |
|
||||
const usageId = ctx.request.body._id |
|
||||
try { |
|
||||
if (ctx.appId) { |
|
||||
const db = getAppDB() |
|
||||
await db.get(usageId) |
|
||||
} |
|
||||
return next() |
|
||||
} catch (err) { |
|
||||
if ( |
|
||||
isExternalTable(usageId) || |
|
||||
(ctx.request.body.tableId && |
|
||||
isExternalTable(ctx.request.body.tableId)) || |
|
||||
isExternalRowId(usageId) |
|
||||
) { |
|
||||
return next() |
|
||||
} else { |
|
||||
ctx.throw(404, `${usageId} does not exist`) |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
try { |
|
||||
await performRequest(ctx, next, quotaInfo, usage) |
|
||||
} catch (err) { |
|
||||
ctx.throw(400, err) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
const performRequest = async ( |
|
||||
ctx: any, |
|
||||
next: any, |
|
||||
quotaInfo: any, |
|
||||
usage: number |
|
||||
) => { |
|
||||
const usageContext = { |
|
||||
skipNext: false, |
|
||||
skipUsage: false, |
|
||||
[StaticQuotaName.APPS]: {}, |
|
||||
} |
|
||||
|
|
||||
const quotaName = quotaInfo.name |
|
||||
|
|
||||
if (usage === -1) { |
|
||||
if (PRE_DELETE[quotaName]) { |
|
||||
await PRE_DELETE[quotaName](ctx, usageContext) |
|
||||
} |
|
||||
} else { |
|
||||
if (PRE_CREATE[quotaName]) { |
|
||||
await PRE_CREATE[quotaName](ctx, usageContext) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
// run the request
|
|
||||
if (!usageContext.skipNext) { |
|
||||
await quotas.updateUsage(usage, quotaName, quotaInfo.type, { |
|
||||
dryRun: true, |
|
||||
}) |
|
||||
await next() |
|
||||
} |
|
||||
|
|
||||
if (usage === -1) { |
|
||||
if (POST_DELETE[quotaName]) { |
|
||||
await POST_DELETE[quotaName](ctx, usageContext) |
|
||||
} |
|
||||
} else { |
|
||||
if (POST_CREATE[quotaName]) { |
|
||||
await POST_CREATE[quotaName](ctx) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
// update the usage
|
|
||||
if (!usageContext.skipUsage) { |
|
||||
await quotas.updateUsage(usage, quotaName, quotaInfo.type) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
const appPreDelete = async (ctx: any, usageContext: any) => { |
|
||||
if (ctx.query.unpublish) { |
|
||||
// don't run usage decrement for unpublish
|
|
||||
usageContext.skipUsage = true |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
// store the row count to delete
|
|
||||
const rows = await getUniqueRows([ctx.appId]) |
|
||||
if (rows.length) { |
|
||||
usageContext[StaticQuotaName.APPS] = { rowCount: rows.length } |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
const appPostDelete = async (ctx: any, usageContext: any) => { |
|
||||
// delete the app rows from usage
|
|
||||
const rowCount = usageContext[StaticQuotaName.APPS].rowCount |
|
||||
if (rowCount) { |
|
||||
await quotas.updateUsage( |
|
||||
-rowCount, |
|
||||
StaticQuotaName.ROWS, |
|
||||
QuotaUsageType.STATIC |
|
||||
) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
const appPostCreate = async (ctx: any) => { |
|
||||
// app import & template creation
|
|
||||
if (ctx.request.body.useTemplate === "true") { |
|
||||
const rows = await getUniqueRows([ctx.response.body.appId]) |
|
||||
const rowCount = rows ? rows.length : 0 |
|
||||
await quotas.updateUsage( |
|
||||
rowCount, |
|
||||
StaticQuotaName.ROWS, |
|
||||
QuotaUsageType.STATIC |
|
||||
) |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
const PRE_DELETE: any = { |
|
||||
[StaticQuotaName.APPS]: appPreDelete, |
|
||||
} |
|
||||
|
|
||||
const POST_DELETE: any = { |
|
||||
[StaticQuotaName.APPS]: appPostDelete, |
|
||||
} |
|
||||
|
|
||||
const PRE_CREATE: any = {} |
|
||||
|
|
||||
const POST_CREATE: any = { |
|
||||
[StaticQuotaName.APPS]: appPostCreate, |
|
||||
} |
|
||||
@ -1,8 +1,3 @@ |
|||||
import { quotas } from "@budibase/pro" |
|
||||
|
|
||||
export const runQuotaMigration = async (migration: Function) => { |
export const runQuotaMigration = async (migration: Function) => { |
||||
if (!quotas.useQuotas()) { |
|
||||
return |
|
||||
} |
|
||||
await migration() |
await migration() |
||||
} |
} |
||||
|
|||||
@ -1,3 +1,5 @@ |
|||||
declare module "@budibase/backend-core" |
declare module "@budibase/backend-core" |
||||
declare module "@budibase/backend-core/tenancy" |
declare module "@budibase/backend-core/tenancy" |
||||
declare module "@budibase/backend-core/db" |
declare module "@budibase/backend-core/db" |
||||
|
declare module "@budibase/backend-core/context" |
||||
|
declare module "@budibase/backend-core/cache" |
||||
|
|||||
Loading…
Reference in new issue