114 lines
2.8 KiB
JavaScript
114 lines
2.8 KiB
JavaScript
const { processString } = require("@budibase/string-templates")
|
|
const CouchDB = require("../../db")
|
|
const { generateQueryID, getQueryParams } = require("../../db/utils")
|
|
const { integrations } = require("../../integrations")
|
|
|
|
function formatResponse(resp) {
|
|
if (typeof resp === "string") {
|
|
resp = JSON.parse(resp)
|
|
}
|
|
if (!Array.isArray(resp)) {
|
|
resp = [resp]
|
|
}
|
|
return resp
|
|
}
|
|
|
|
exports.fetch = async function(ctx) {
|
|
const db = new CouchDB(ctx.user.appId)
|
|
|
|
const body = await db.allDocs(
|
|
getQueryParams(null, {
|
|
include_docs: true,
|
|
})
|
|
)
|
|
ctx.body = body.rows.map(row => row.doc)
|
|
}
|
|
|
|
exports.save = async function(ctx) {
|
|
const db = new CouchDB(ctx.user.appId)
|
|
const query = ctx.request.body
|
|
|
|
if (!query._id) {
|
|
query._id = generateQueryID(query.datasourceId)
|
|
}
|
|
|
|
const response = await db.put(query)
|
|
query._rev = response.rev
|
|
|
|
ctx.body = query
|
|
ctx.message = `Query ${query.name} saved successfully.`
|
|
}
|
|
|
|
async function enrichQueryFields(fields, parameters) {
|
|
const enrichedQuery = {}
|
|
|
|
// enrich the fields with dynamic parameters
|
|
for (let key of Object.keys(fields)) {
|
|
enrichedQuery[key] = await processString(fields[key], parameters)
|
|
}
|
|
|
|
if (enrichedQuery.json || enrichedQuery.customData) {
|
|
try {
|
|
enrichedQuery.json = JSON.parse(
|
|
enrichedQuery.json || enrichedQuery.customData
|
|
)
|
|
} catch (err) {
|
|
throw { message: `JSON Invalid - error: ${err}` }
|
|
}
|
|
delete enrichedQuery.customData
|
|
}
|
|
|
|
return enrichedQuery
|
|
}
|
|
|
|
exports.preview = async function(ctx) {
|
|
const db = new CouchDB(ctx.user.appId)
|
|
|
|
const datasource = await db.get(ctx.request.body.datasourceId)
|
|
|
|
const Integration = integrations[datasource.source]
|
|
|
|
if (!Integration) {
|
|
ctx.throw(400, "Integration type does not exist.")
|
|
return
|
|
}
|
|
|
|
const { fields, parameters, queryVerb } = ctx.request.body
|
|
|
|
const enrichedQuery = await enrichQueryFields(fields, parameters)
|
|
|
|
ctx.body = formatResponse(
|
|
await new Integration(datasource.config)[queryVerb](enrichedQuery)
|
|
)
|
|
}
|
|
|
|
exports.execute = async function(ctx) {
|
|
const db = new CouchDB(ctx.user.appId)
|
|
|
|
const query = await db.get(ctx.params.queryId)
|
|
const datasource = await db.get(query.datasourceId)
|
|
|
|
const Integration = integrations[datasource.source]
|
|
|
|
if (!Integration) {
|
|
ctx.throw(400, "Integration type does not exist.")
|
|
return
|
|
}
|
|
|
|
const enrichedQuery = await enrichQueryFields(
|
|
query.fields,
|
|
ctx.request.body.parameters
|
|
)
|
|
|
|
// call the relevant CRUD method on the integration class
|
|
ctx.body = formatResponse(
|
|
await new Integration(datasource.config)[query.queryVerb](enrichedQuery)
|
|
)
|
|
}
|
|
|
|
exports.destroy = async function(ctx) {
|
|
const db = new CouchDB(ctx.user.appId)
|
|
await db.remove(ctx.params.queryId, ctx.params.revId)
|
|
ctx.message = `Query deleted.`
|
|
ctx.status = 200
|
|
}
|