1
0
Fork 0
mirror of synced 2024-06-15 00:44:41 +12:00
budibase/packages/server/src/threads/query.js

157 lines
3.7 KiB
JavaScript
Raw Normal View History

require("./utils").threadSetup()
const ScriptRunner = require("../utilities/scriptRunner")
const { integrations } = require("../integrations")
2021-12-17 05:58:15 +13:00
const { processStringSync } = require("@budibase/string-templates")
async function getDynamicVariable() {}
2021-12-17 05:58:15 +13:00
async function addDatasourceVariables(datasource, parameters) {
if (!datasource || !datasource.config) {
return parameters
}
const staticVars = datasource.config.staticVariables || {}
const dynamicVars = datasource.config.dynamicVariables || []
for (let [key, value] of Object.entries(staticVars)) {
if (!parameters[key]) {
parameters[key] = value
}
}
const dynamics = dynamicVars.map(dynVar =>
getDynamicVariable(datasource, dynVar, parameters)
)
for (let i = 0; i < dynamicVars.length; i++) {
const variable = dynamicVars[i]
parameters[variable.name] = dynamics[i]
2021-12-17 05:58:15 +13:00
}
return parameters
}
function enrichQueryFields(fields, parameters = {}) {
const enrichedQuery = {}
// enrich the fields with dynamic parameters
for (let key of Object.keys(fields)) {
if (fields[key] == null) {
continue
}
if (typeof fields[key] === "object") {
// enrich nested fields object
enrichedQuery[key] = enrichQueryFields(fields[key], parameters)
} else if (typeof fields[key] === "string") {
// enrich string value as normal
enrichedQuery[key] = processStringSync(fields[key], parameters, {
noHelpers: true,
})
} else {
enrichedQuery[key] = fields[key]
}
}
if (
enrichedQuery.json ||
enrichedQuery.customData ||
enrichedQuery.requestBody
) {
try {
enrichedQuery.json = JSON.parse(
enrichedQuery.json ||
enrichedQuery.customData ||
enrichedQuery.requestBody
)
} catch (err) {
// no json found, ignore
}
delete enrichedQuery.customData
}
return enrichedQuery
}
function formatResponse(resp) {
if (typeof resp === "string") {
try {
resp = JSON.parse(resp)
} catch (err) {
resp = { response: resp }
}
}
return resp
}
function hasExtraData(response) {
return (
typeof response === "object" &&
!Array.isArray(response) &&
response.data != null &&
response.info != null
)
}
2021-12-17 05:58:15 +13:00
async function runAndTransform(
datasource,
queryVerb,
fields,
parameters,
transformer
) {
// pre-query, make sure datasource variables are added to parameters
parameters = await addDatasourceVariables(datasource, parameters)
const query = enrichQueryFields(fields, parameters)
const Integration = integrations[datasource.source]
if (!Integration) {
throw "Integration type does not exist."
}
const integration = new Integration(datasource.config)
let output = formatResponse(await integration[queryVerb](query))
let rows = output,
2021-12-07 07:35:44 +13:00
info = undefined,
extra = undefined
if (hasExtraData(output)) {
rows = output.data
info = output.info
extra = output.extra
}
// transform as required
if (transformer) {
const runner = new ScriptRunner(transformer, { data: rows })
rows = runner.execute()
}
// needs to an array for next step
if (!Array.isArray(rows)) {
rows = [rows]
}
// map into JSON if just raw primitive here
if (rows.find(row => typeof row !== "object")) {
rows = rows.map(value => ({ value }))
}
// get all the potential fields in the schema
let keys = rows.flatMap(Object.keys)
if (integration.end) {
integration.end()
}
return { rows, keys, info, extra }
}
module.exports = (input, callback) => {
runAndTransform(
input.datasource,
input.queryVerb,
2021-12-17 05:58:15 +13:00
input.fields,
input.parameters,
input.transformer
)
.then(response => {
callback(null, response)
})
.catch(err => {
callback(err)
})
}