2020-12-19 07:19:43 +13:00
|
|
|
const CouchDB = require("../../db")
|
2021-01-13 05:49:11 +13:00
|
|
|
const {
|
|
|
|
generateDatasourceID,
|
|
|
|
getDatasourceParams,
|
|
|
|
getQueryParams,
|
2021-06-12 04:52:08 +12:00
|
|
|
DocumentTypes,
|
2021-01-13 05:49:11 +13:00
|
|
|
} = require("../../db/utils")
|
2021-06-02 02:00:28 +12:00
|
|
|
const { integrations } = require("../../integrations")
|
2021-06-04 04:56:04 +12:00
|
|
|
const plusIntegrations = require("../../integrations/plus")
|
2020-12-19 07:19:43 +13:00
|
|
|
|
2021-05-03 19:31:09 +12:00
|
|
|
exports.fetch = async function (ctx) {
|
2021-03-30 05:32:05 +13:00
|
|
|
const database = new CouchDB(ctx.appId)
|
2021-03-06 01:11:44 +13:00
|
|
|
ctx.body = (
|
2020-12-19 07:19:43 +13:00
|
|
|
await database.allDocs(
|
|
|
|
getDatasourceParams(null, {
|
|
|
|
include_docs: true,
|
|
|
|
})
|
|
|
|
)
|
2021-05-04 22:32:22 +12:00
|
|
|
).rows.map(row => row.doc)
|
2020-12-19 07:19:43 +13:00
|
|
|
}
|
|
|
|
|
2021-05-03 19:31:09 +12:00
|
|
|
exports.save = async function (ctx) {
|
2021-03-30 05:32:05 +13:00
|
|
|
const db = new CouchDB(ctx.appId)
|
2021-06-12 04:52:08 +12:00
|
|
|
const plus = ctx.request.body.plus
|
2020-12-19 07:19:43 +13:00
|
|
|
|
|
|
|
const datasource = {
|
2021-06-12 04:52:08 +12:00
|
|
|
_id: generateDatasourceID({ plus }),
|
|
|
|
type: plus ? DocumentTypes.DATASOURCE_PLUS : DocumentTypes.DATASOURCE,
|
2020-12-19 07:19:43 +13:00
|
|
|
...ctx.request.body,
|
|
|
|
}
|
|
|
|
|
2021-06-12 04:52:08 +12:00
|
|
|
// update the schema
|
|
|
|
if (ctx.query.refresh) {
|
|
|
|
const PlusConnector = plusIntegrations[datasource.source].integration
|
|
|
|
|
|
|
|
const connector = new PlusConnector(ctx.request.body.config)
|
|
|
|
await connector.init()
|
|
|
|
|
|
|
|
datasource.entities = connector.tables
|
|
|
|
}
|
|
|
|
|
2021-03-11 00:56:52 +13:00
|
|
|
const response = await db.post(datasource)
|
|
|
|
datasource._rev = response.rev
|
2020-12-19 07:19:43 +13:00
|
|
|
|
2021-06-02 02:00:28 +12:00
|
|
|
// Drain connection pools when configuration is changed
|
2021-06-04 04:56:04 +12:00
|
|
|
if (datasource.source) {
|
|
|
|
const source = integrations[datasource.source]
|
|
|
|
if (source && source.pool) {
|
|
|
|
await source.pool.end()
|
|
|
|
}
|
2021-06-02 02:00:28 +12:00
|
|
|
}
|
|
|
|
|
2020-12-19 07:19:43 +13:00
|
|
|
ctx.status = 200
|
2021-03-11 00:56:52 +13:00
|
|
|
ctx.message = "Datasource saved successfully."
|
|
|
|
ctx.body = datasource
|
2020-12-19 07:19:43 +13:00
|
|
|
}
|
|
|
|
|
2021-05-03 19:31:09 +12:00
|
|
|
exports.destroy = async function (ctx) {
|
2021-03-30 05:32:05 +13:00
|
|
|
const db = new CouchDB(ctx.appId)
|
2021-01-13 05:49:11 +13:00
|
|
|
|
|
|
|
// Delete all queries for the datasource
|
|
|
|
const rows = await db.allDocs(getQueryParams(ctx.params.datasourceId, null))
|
2021-05-04 22:32:22 +12:00
|
|
|
await db.bulkDocs(rows.rows.map(row => ({ ...row.doc, _deleted: true })))
|
2021-01-13 05:49:11 +13:00
|
|
|
|
|
|
|
// delete the datasource
|
2021-01-13 06:45:43 +13:00
|
|
|
await db.remove(ctx.params.datasourceId, ctx.params.revId)
|
2021-01-13 05:49:11 +13:00
|
|
|
|
2020-12-19 07:19:43 +13:00
|
|
|
ctx.message = `Datasource deleted.`
|
|
|
|
ctx.status = 200
|
|
|
|
}
|
|
|
|
|
2021-05-03 19:31:09 +12:00
|
|
|
exports.find = async function (ctx) {
|
2021-03-30 05:32:05 +13:00
|
|
|
const database = new CouchDB(ctx.appId)
|
2021-03-11 00:56:52 +13:00
|
|
|
ctx.body = await database.get(ctx.params.datasourceId)
|
2020-12-19 07:19:43 +13:00
|
|
|
}
|
2021-06-04 04:56:04 +12:00
|
|
|
|
2021-06-04 05:48:04 +12:00
|
|
|
// dynamic query functionality
|
|
|
|
exports.query = async function (ctx) {
|
2021-06-05 01:53:49 +12:00
|
|
|
const queryJson = ctx.request.body
|
|
|
|
const datasourceId = queryJson.endpoint.datasourceId
|
|
|
|
const database = new CouchDB(ctx.appId)
|
|
|
|
const datasource = await database.get(datasourceId)
|
2021-06-05 02:16:15 +12:00
|
|
|
const Integration = integrations[datasource.source]
|
2021-06-05 01:53:49 +12:00
|
|
|
// query is the opinionated function
|
2021-06-05 02:16:15 +12:00
|
|
|
if (Integration.prototype.query) {
|
|
|
|
const integration = new Integration(datasource.config)
|
|
|
|
ctx.body = await integration.query(queryJson)
|
2021-06-05 01:53:49 +12:00
|
|
|
} else {
|
|
|
|
ctx.throw(400, "Datasource does not support query.")
|
|
|
|
}
|
2021-06-04 05:48:04 +12:00
|
|
|
}
|