import path from "path" import { getRecordApi, getCollectionApi, getIndexApi, getActionsApi, } from "../src" import memory from "./memory" import { setupDatastore } from "../src/appInitialise" import { configFolder, fieldDefinitions, templateDefinitions, joinKey, isSomething, crypto as nodeCrypto, } from "../src/common" import { getNewIndexTemplate } from "../src/templateApi/createNodes" import { indexTypes } from "../src/templateApi/indexes" import getTemplateApi from "../src/templateApi" import getAuthApi from "../src/authApi" import { createEventAggregator } from "../src/appInitialise/eventAggregator" import { filter, find } from "lodash/fp" import { createBehaviourSources } from "../src/actionsApi/buildBehaviourSource" import { createAction, createTrigger } from "../src/templateApi/createActions" import { initialiseActions } from "../src/actionsApi/initialise" import { setCleanupFunc } from "../src/transactions/setCleanupFunc" import { permission } from "../src/authApi/permissions" import { generateFullPermissions } from "../src/authApi/generateFullPermissions" import { initialiseData } from "../src/appInitialise/initialiseData" export const testFileArea = testNameArea => path.join("test", "fs_test_area", testNameArea) export const testConfigFolder = testAreaName => path.join(testFileArea(testAreaName), configFolder) export const testFieldDefinitionsPath = testAreaName => path.join(testFileArea(testAreaName), fieldDefinitions) export const testTemplatesPath = testAreaName => path.join(testFileArea(testAreaName), templateDefinitions) export const getMemoryStore = () => setupDatastore(memory({})) export const getMemoryTemplateApi = store => { const app = { datastore: store || getMemoryStore(), publish: () => { }, getEpochTime: async () => new Date().getTime(), user: { name: "", permissions: [permission.writeTemplates.get()] }, } app.removePermission = removePermission(app) app.withOnlyThisPermission = withOnlyThisPermission(app) app.withNoPermissions = withNoPermissions(app) const templateApi = getTemplateApi(app) templateApi._eventAggregator = createEventAggregator() templateApi._storeHandle = app.datastore return { templateApi, app } } // TODO: subscribe actions export const appFromTempalteApi = async ( templateApi, disableCleanupTransactions = false ) => { const appDef = await templateApi.getApplicationDefinition() const app = { hierarchy: appDef.hierarchy, datastore: templateApi._storeHandle, publish: templateApi._eventAggregator.publish, _eventAggregator: templateApi._eventAggregator, getEpochTime: async () => new Date().getTime(), crypto: nodeCrypto, user: { name: "bob", permissions: [] }, actions: {}, } app.removePermission = removePermission(app) app.withOnlyThisPermission = withOnlyThisPermission(app) app.withNoPermissions = withNoPermissions(app) const fullPermissions = generateFullPermissions(app) app.user.permissions = fullPermissions if (disableCleanupTransactions) setCleanupFunc(app, async () => { }) else setCleanupFunc(app) return app } const removePermission = app => perm => { app.user.permissions = filter( p => p.type !== perm.type || (isSomething(perm.nodeKey) && perm.nodeKey !== p.nodeKey) )(app.user.permissions) } const withOnlyThisPermission = app => perm => (app.user.permissions = [perm]) const withNoPermissions = app => () => (app.user.permissions = []) export const getRecordApiFromTemplateApi = async ( templateApi, disableCleanupTransactions = false ) => { const app = await appFromTempalteApi(templateApi, disableCleanupTransactions) const recordapi = getRecordApi(app) recordapi._storeHandle = app.datastore return recordapi } export const getCollectionApiFromTemplateApi = async ( templateApi, disableCleanupTransactions = false ) => getCollectionApi( await appFromTempalteApi(templateApi, disableCleanupTransactions) ) export const getIndexApiFromTemplateApi = async ( templateApi, disableCleanupTransactions = false ) => getIndexApi(await appFromTempalteApi(templateApi, disableCleanupTransactions)) export const getAuthApiFromTemplateApi = async ( templateApi, disableCleanupTransactions = false ) => getAuthApi(await appFromTempalteApi(templateApi, disableCleanupTransactions)) export const findIndex = (parentNode, name) => find(i => i.name === name)(parentNode.indexes) export const findCollectionDefaultIndex = recordCollectionNode => findIndex(recordCollectionNode.parent(), recordCollectionNode.name + "_index") export const hierarchyFactory = (...additionalFeatures) => templateApi => { const root = templateApi.getNewRootLevel() const settingsRecord = templateApi.getNewSingleRecordTemplate(root) settingsRecord.name = "settings" const customerRecord = templateApi.getNewModelTemplate(root, "customer") customerRecord.collectionName = "customers" findCollectionDefaultIndex(customerRecord).map = "return {surname:record.surname, isalive:record.isalive, partner:record.partner};" const partnerRecord = templateApi.getNewModelTemplate(root, "partner") partnerRecord.collectionName = "partners" const partnerInvoiceRecord = templateApi.getNewModelTemplate( partnerRecord, "invoice" ) partnerInvoiceRecord.collectionName = "invoices" findCollectionDefaultIndex(partnerInvoiceRecord).name = "partnerInvoices_index" const invoiceRecord = templateApi.getNewModelTemplate( customerRecord, "invoice" ) invoiceRecord.collectionName = "invoices" findCollectionDefaultIndex(invoiceRecord).map = "return {createdDate: record.createdDate, totalIncVat: record.totalIncVat};" const chargeRecord = templateApi.getNewModelTemplate(invoiceRecord, "charge") chargeRecord.collectionName = "charges" const hierarchy = { root, customerRecord, invoiceRecord, partnerRecord, partnerInvoiceRecord, chargeRecord, settingsRecord, } for (let feature of additionalFeatures) { feature(hierarchy, templateApi) } return hierarchy } export const basicAppHierarchyCreator = templateApis => hierarchyFactory()(templateApis) export const withFields = (hierarchy, templateApi) => { const { customerRecord, invoiceRecord, partnerInvoiceRecord, chargeRecord, partnerRecord, settingsRecord, root, } = hierarchy getNewFieldAndAdd(templateApi, settingsRecord)("appName", "string", "") const newCustomerField = getNewFieldAndAdd(templateApi, customerRecord) const partnersReferenceIndex = templateApi.getNewIndexTemplate(root) partnersReferenceIndex.name = "partnersReference" partnersReferenceIndex.map = "return {name:record.businessName, phone:record.phone};" partnersReferenceIndex.allowedModelNodeIds = [partnerRecord.nodeId] const partnerCustomersReverseIndex = templateApi.getNewIndexTemplate( partnerRecord, indexTypes.reference ) partnerCustomersReverseIndex.name = "partnerCustomers" partnerCustomersReverseIndex.map = "return {...record};" partnerCustomersReverseIndex.filter = "record.isalive === true" partnerCustomersReverseIndex.allowedModelNodeIds = [customerRecord.nodeId] hierarchy.partnerCustomersReverseIndex = partnerCustomersReverseIndex newCustomerField("surname", "string") newCustomerField("isalive", "bool", "true") newCustomerField("createddate", "datetime") newCustomerField("age", "number") newCustomerField("profilepic", "file") newCustomerField("partner", "reference", undefined, { indexNodeKey: "/partnersReference", displayValue: "name", reverseIndexNodeKeys: [ joinKey(partnerRecord.nodeKey(), "partnerCustomers"), ], }) const referredToCustomersReverseIndex = templateApi.getNewIndexTemplate( customerRecord, indexTypes.reference ) referredToCustomersReverseIndex.name = "referredToCustomers" referredToCustomersReverseIndex.map = "return {...record};" referredToCustomersReverseIndex.getShardName = "return !record.surname ? 'null' : record.surname.substring(0,1);" referredToCustomersReverseIndex.allowedModelNodeIds = [customerRecord.nodeId] hierarchy.referredToCustomersReverseIndex = referredToCustomersReverseIndex const customerReferredByField = newCustomerField( "referredBy", "reference", undefined, { indexNodeKey: "/customer_index", displayValue: "surname", reverseIndexNodeKeys: [ joinKey(customerRecord.nodeKey(), "referredToCustomers"), ], } ) hierarchy.customerReferredByField = customerReferredByField const newInvoiceField = getNewFieldAndAdd(templateApi, invoiceRecord) const invoiceTotalIncVatField = newInvoiceField("totalIncVat", "number") invoiceTotalIncVatField.typeOptions.decimalPlaces = 2 newInvoiceField("createdDate", "datetime") newInvoiceField("paidAmount", "number") newInvoiceField("invoiceType", "string") newInvoiceField("isWrittenOff", "bool") const newPartnerField = getNewFieldAndAdd(templateApi, partnerRecord) newPartnerField("businessName", "string") newPartnerField("phone", "string") const newPartnerInvoiceField = getNewFieldAndAdd( templateApi, partnerInvoiceRecord ) const partnerInvoiceTotalIncVatVield = newPartnerInvoiceField( "totalIncVat", "number" ) partnerInvoiceTotalIncVatVield.typeOptions.decimalPlaces = 2 newPartnerInvoiceField("createdDate", "datetime") newPartnerInvoiceField("paidAmount", "number") const newChargeField = getNewFieldAndAdd(templateApi, chargeRecord) newChargeField("amount", "number") newChargeField("partnerInvoice", "reference", undefined, { reverseIndexNodeKeys: [ joinKey(partnerInvoiceRecord.nodeKey(), "partnerCharges"), ], displayValue: "createdDate", indexNodeKey: joinKey(partnerRecord.nodeKey(), "partnerInvoices_index"), }) const partnerChargesReverseIndex = templateApi.getNewIndexTemplate( partnerInvoiceRecord, indexTypes.reference ) partnerChargesReverseIndex.name = "partnerCharges" partnerChargesReverseIndex.map = "return {...record};" partnerChargesReverseIndex.allowedModelNodeIds = [chargeRecord] hierarchy.partnerChargesReverseIndex = partnerChargesReverseIndex const customersReferenceIndex = templateApi.getNewIndexTemplate( hierarchy.root ) customersReferenceIndex.name = "customersReference" customersReferenceIndex.map = "return {name:record.surname}" customersReferenceIndex.filter = "record.isalive === true" customersReferenceIndex.allowedModelNodeIds = [customerRecord.nodeId] newInvoiceField("customer", "reference", undefined, { indexNodeKey: "/customersReference", reverseIndexNodeKeys: [findCollectionDefaultIndex(invoiceRecord).nodeKey()], displayValue: "name", }) } export const withIndexes = (hierarchy, templateApi) => { const { root, customerRecord, partnerInvoiceRecord, invoiceRecord, partnerRecord, chargeRecord, } = hierarchy const deceasedCustomersIndex = getNewIndexTemplate(root) deceasedCustomersIndex.name = "deceased" deceasedCustomersIndex.map = "return {surname: record.surname, age:record.age};" deceasedCustomersIndex.filter = "record.isalive === false" findCollectionDefaultIndex(customerRecord).map = "return record;" deceasedCustomersIndex.allowedModelNodeIds = [customerRecord.nodeId] findCollectionDefaultIndex(invoiceRecord).allowedModelNodeIds = [ invoiceRecord.nodeId, ] findCollectionDefaultIndex(customerRecord).allowedModelNodeIds = [ customerRecord.nodeId, ] findCollectionDefaultIndex(partnerRecord).allowedModelNodeIds = [ partnerRecord.nodeId, ] findIndex(partnerRecord, "partnerInvoices_index").allowedModelNodeIds = [ partnerInvoiceRecord.nodeId, ] findCollectionDefaultIndex(chargeRecord).allowedModelNodeIds = [ chargeRecord.nodeId, ] const customerInvoicesIndex = getNewIndexTemplate(root) customerInvoicesIndex.name = "customer_invoices" customerInvoicesIndex.map = "return record;" customerInvoicesIndex.filter = "record.type === 'invoice'" customerInvoicesIndex.allowedModelNodeIds = [invoiceRecord.nodeId] const outstandingInvoicesIndex = getNewIndexTemplate(root) outstandingInvoicesIndex.name = "Outstanding Invoices" outstandingInvoicesIndex.filter = "record.type === 'invoice' && record.paidAmount < record.totalIncVat" outstandingInvoicesIndex.map = "return {...record};" outstandingInvoicesIndex.allowedModelNodeIds = [ invoiceRecord.nodeId, partnerInvoiceRecord.nodeId, ] const allInvoicesAggregateGroup = templateApi.getNewAggregateGroupTemplate( outstandingInvoicesIndex ) allInvoicesAggregateGroup.name = "all_invoices" const allInvoicesByType = templateApi.getNewAggregateGroupTemplate( outstandingInvoicesIndex ) allInvoicesByType.groupBy = "return record.invoiceType" allInvoicesByType.name = "all_invoices_by_type" const allInvoicesTotalAmountAggregate = templateApi.getNewAggregateTemplate( allInvoicesByType ) allInvoicesTotalAmountAggregate.name = "totalIncVat" allInvoicesTotalAmountAggregate.aggregatedValue = "return record.totalIncVat" const allInvoicesPaidAmountAggregate = templateApi.getNewAggregateTemplate( allInvoicesByType ) allInvoicesPaidAmountAggregate.name = "paidAmount" allInvoicesPaidAmountAggregate.aggregatedValue = "return record.paidAmount" const writtenOffInvoicesByType = templateApi.getNewAggregateGroupTemplate( outstandingInvoicesIndex ) writtenOffInvoicesByType.groupBy = "return record.invoiceType" writtenOffInvoicesByType.name = "written_off" writtenOffInvoicesByType.condition = "record.isWrittenOff === true" const writtenOffInvoicesTotalAmountAggregate = templateApi.getNewAggregateTemplate( writtenOffInvoicesByType ) writtenOffInvoicesTotalAmountAggregate.name = "totalIncVat" writtenOffInvoicesTotalAmountAggregate.aggregatedValue = "return record.totalIncVat" const customersBySurnameIndex = templateApi.getNewIndexTemplate(root) customersBySurnameIndex.name = "customersBySurname" customersBySurnameIndex.map = "return {...record};" customersBySurnameIndex.filter = "" customersBySurnameIndex.allowedModelNodeIds = [customerRecord.nodeId] customersBySurnameIndex.getShardName = "return !record.surname ? 'null' : record.surname.substring(0,1);" const customersDefaultIndex = findCollectionDefaultIndex(customerRecord) const customersNoGroupaggregateGroup = templateApi.getNewAggregateGroupTemplate( customersDefaultIndex ) customersNoGroupaggregateGroup.name = "Customers Summary" const allCustomersAgeFunctions = templateApi.getNewAggregateTemplate( customersNoGroupaggregateGroup ) allCustomersAgeFunctions.aggregatedValue = "return record.age" allCustomersAgeFunctions.name = "all customers - age breakdown" const invoicesByOutstandingIndex = templateApi.getNewIndexTemplate( customerRecord ) invoicesByOutstandingIndex.name = "invoicesByOutstanding" invoicesByOutstandingIndex.map = "return {...record};" invoicesByOutstandingIndex.filter = "" invoicesByOutstandingIndex.getShardName = "return (record.totalIncVat > record.paidAmount ? 'outstanding' : 'paid');" invoicesByOutstandingIndex.allowedModelNodeIds = [ partnerInvoiceRecord.nodeId, invoiceRecord.nodeId, ] const allInvoicesByType_Sharded = templateApi.getNewAggregateGroupTemplate( invoicesByOutstandingIndex ) allInvoicesByType_Sharded.groupBy = "return record.invoiceType" allInvoicesByType_Sharded.name = "all_invoices_by_type" const allInvoicesTotalAmountAggregate_Sharded = templateApi.getNewAggregateTemplate( allInvoicesByType_Sharded ) allInvoicesTotalAmountAggregate_Sharded.name = "totalIncVat" allInvoicesTotalAmountAggregate_Sharded.aggregatedValue = "return record.totalIncVat" hierarchy.allInvoicesByType = allInvoicesByType hierarchy.allInvoicesTotalAmountAggregate = allInvoicesTotalAmountAggregate hierarchy.allInvoicesPaidAmountAggregate = allInvoicesPaidAmountAggregate hierarchy.customersDefaultIndex = customersDefaultIndex hierarchy.allCustomersAgeFunctions = allCustomersAgeFunctions hierarchy.customersNoGroupaggregateGroup = customersNoGroupaggregateGroup hierarchy.invoicesByOutstandingIndex = invoicesByOutstandingIndex hierarchy.customersBySurnameIndex = customersBySurnameIndex hierarchy.outstandingInvoicesIndex = outstandingInvoicesIndex hierarchy.deceasedCustomersIndex = deceasedCustomersIndex hierarchy.customerInvoicesIndex = customerInvoicesIndex } export const basicAppHierarchyCreator_WithFields = templateApi => hierarchyFactory(withFields)(templateApi) export const basicAppHierarchyCreator_WithFields_AndIndexes = templateApi => hierarchyFactory(withFields, withIndexes)(templateApi) export const setupApphierarchy = async ( creator, disableCleanupTransactions = false ) => { const { templateApi } = getMemoryTemplateApi() const hierarchy = creator(templateApi) await initialiseData(templateApi._storeHandle, { hierarchy: hierarchy.root, actions: [], triggers: [], }) await templateApi.saveApplicationHierarchy(hierarchy.root) const app = await appFromTempalteApi(templateApi, disableCleanupTransactions) const collectionApi = getCollectionApi(app) const indexApi = getIndexApi(app) const authApi = getAuthApi(app) const actionsApi = getActionsApi(app) const recordApi = await getRecordApi(app) recordApi._storeHandle = app.datastore actionsApi._app = app const apis = { recordApi, collectionApi, templateApi, indexApi, authApi, actionsApi, appHierarchy: hierarchy, subscribe: templateApi._eventAggregator.subscribe, app, } return apis } export const getNewFieldAndAdd = (templateApi, record) => ( name, type, initial, typeOptions ) => { const field = templateApi.getNewField(type) field.name = name field.getInitialValue = !initial ? "default" : initial if (typeOptions) field.typeOptions = typeOptions templateApi.addField(record, field) return field } export const stubEventHandler = () => { const events = [] return { handle: (name, context) => { events.push({ name, context }) }, events, getEvents: n => filter(e => e.name === n)(events), } } export const createValidActionsAndTriggers = () => { const logMessage = createAction() logMessage.name = "logMessage" logMessage.behaviourName = "log" logMessage.behaviourSource = "budibase-behaviours" const measureCallTime = createAction() measureCallTime.name = "measureCallTime" measureCallTime.behaviourName = "call_timer" measureCallTime.behaviourSource = "budibase-behaviours" const sendEmail = createAction() sendEmail.name = "sendEmail" sendEmail.behaviourName = "send_email" sendEmail.behaviourSource = "my-custom-lib" const logOnErrorTrigger = createTrigger() logOnErrorTrigger.actionName = "logMessage" logOnErrorTrigger.eventName = "recordApi:save:onError" logOnErrorTrigger.optionsCreator = "return context.error.message;" const timeCustomerSaveTrigger = createTrigger() timeCustomerSaveTrigger.actionName = "measureCallTime" timeCustomerSaveTrigger.eventName = "recordApi:save:onComplete" timeCustomerSaveTrigger.optionsCreator = "return 999;" timeCustomerSaveTrigger.condition = "context.record.type === 'customer'" const allActions = [logMessage, measureCallTime, sendEmail] const allTriggers = [logOnErrorTrigger, timeCustomerSaveTrigger] const behaviourSources = createBehaviourSources() const logs = [] const call_timers = [] const emails = [] behaviourSources.register("budibase-behaviours", { log: message => logs.push(message), call_timer: opts => call_timers.push(opts), }) behaviourSources.register("my-custom-lib", { send_email: em => emails.push(em), }) return { logMessage, measureCallTime, sendEmail, logOnErrorTrigger, timeCustomerSaveTrigger, allActions, allTriggers, behaviourSources, logs, call_timers, emails, } } export const createAppDefinitionWithActionsAndTriggers = async () => { const { appHierarchy, templateApi, app, actionsApi, } = await setupApphierarchy(basicAppHierarchyCreator_WithFields) // adding validation rule so it can fail when we save it templateApi.addRecordValidationRule(appHierarchy.customerRecord)( templateApi.commonRecordValidationRules.fieldNotEmpty("surname") ) await templateApi.saveApplicationHierarchy(appHierarchy.root) const actionsAndTriggers = createValidActionsAndTriggers() const { allActions, allTriggers, behaviourSources } = actionsAndTriggers await templateApi.saveActionsAndTriggers(allActions, allTriggers) app.actions = initialiseActions( templateApi._eventAggregator.subscribe, behaviourSources, allActions, allTriggers ) app.user.permissions = generateFullPermissions(app) app.behaviourSources = behaviourSources const appDefinition = await templateApi.getApplicationDefinition() return { templateApi, appDefinition, ...actionsAndTriggers, ...appHierarchy, app, actionsApi, } } export const validUser = async ( app, authApi, password, enabled = true, accessLevels = null ) => { const access = await authApi.getNewAccessLevel(app) access.name = "admin" permission.setPassword.add(access) const access2 = await authApi.getNewAccessLevel(app) access2.name = "admin2" permission.setPassword.add(access) await authApi.saveAccessLevels({ version: 0, levels: [access, access2] }) const u = authApi.getNewUser(app) u.name = "bob" if (accessLevels === null) u.accessLevels = ["admin"] else u.accessLevels = accessLevels u.enabled = enabled await authApi.createUser(u, password) return u }