application supports multiple concurrent client DB

This commit is contained in:
Michael Shanks 2020-06-10 21:39:30 +01:00
parent 8f7248a0d6
commit 71016f5a1d
8 changed files with 108 additions and 17 deletions

View File

@ -11,7 +11,7 @@ const { exec } = require("child_process")
const sqrl = require("squirrelly") const sqrl = require("squirrelly")
exports.fetch = async function(ctx) { exports.fetch = async function(ctx) {
const db = new CouchDB(ClientDb.name(env.CLIENT_ID)) const db = new CouchDB(ClientDb.name(getClientId(ctx)))
const body = await db.query("client/by_type", { const body = await db.query("client/by_type", {
include_docs: true, include_docs: true,
key: ["app"], key: ["app"],
@ -21,16 +21,30 @@ exports.fetch = async function(ctx) {
} }
exports.fetchAppPackage = async function(ctx) { exports.fetchAppPackage = async function(ctx) {
const db = new CouchDB(ClientDb.name(env.CLIENT_ID)) const clientId = await lookupClientId(ctx.params.applicationId)
const db = new CouchDB(ClientDb.name(clientId))
const application = await db.get(ctx.params.applicationId) const application = await db.get(ctx.params.applicationId)
ctx.body = await getPackageForBuilder(ctx.config, application) ctx.body = await getPackageForBuilder(ctx.config, application)
} }
exports.create = async function(ctx) { exports.create = async function(ctx) {
const db = new CouchDB(ClientDb.name(env.CLIENT_ID)) const clientId =
(ctx.request.body && ctx.request.body.clientId) || env.CLIENT_ID
if (!clientId) {
ctx.throw(400, "ClientId not suplied")
}
const appId = newid()
// insert an appId -> clientId lookup
const masterDb = new CouchDB("master")
await masterDb.put({
_id: appId,
clientId,
})
const db = new CouchDB(ClientDb.name(clientId))
const newApplication = { const newApplication = {
_id: newid(), _id: appId,
type: "app", type: "app",
instances: [], instances: [],
userInstanceMap: {}, userInstanceMap: {},
@ -47,11 +61,10 @@ exports.create = async function(ctx) {
const createInstCtx = { const createInstCtx = {
params: { params: {
clientId: env.CLIENT_ID,
applicationId: newApplication._id, applicationId: newApplication._id,
}, },
request: { request: {
body: { name: `dev-${env.CLIENT_ID}` }, body: { name: `dev-${clientId}` },
}, },
} }
await instanceController.create(createInstCtx) await instanceController.create(createInstCtx)
@ -61,6 +74,7 @@ exports.create = async function(ctx) {
await runNpmInstall(newAppFolder) await runNpmInstall(newAppFolder)
} }
ctx.status = 200
ctx.body = newApplication ctx.body = newApplication
ctx.message = `Application ${ctx.request.body.name} created successfully` ctx.message = `Application ${ctx.request.body.name} created successfully`
} }
@ -79,7 +93,6 @@ const createEmptyAppPackage = async (ctx, app) => {
if (await exists(newAppFolder)) { if (await exists(newAppFolder)) {
ctx.throw(400, "App folder already exists for this application") ctx.throw(400, "App folder already exists for this application")
return
} }
await copy(templateFolder, newAppFolder) await copy(templateFolder, newAppFolder)
@ -99,6 +112,24 @@ const createEmptyAppPackage = async (ctx, app) => {
return newAppFolder return newAppFolder
} }
const lookupClientId = async appId => {
const masterDb = new CouchDB("master")
const { clientId } = await masterDb.get(appId)
return clientId
}
const getClientId = ctx => {
const clientId =
(ctx.request.body && ctx.request.body.clientId) ||
(ctx.query && ctx.query.clientId) ||
env.CLIENT_ID
if (!clientId) {
ctx.throw(400, "ClientId not suplied")
}
return clientId
}
const updateJsonFile = async (filePath, app) => { const updateJsonFile = async (filePath, app) => {
const json = await readFile(filePath, "utf8") const json = await readFile(filePath, "utf8")
const newJson = sqrl.Render(json, app) const newJson = sqrl.Render(json, app)

View File

@ -2,7 +2,6 @@ const jwt = require("jsonwebtoken")
const CouchDB = require("../../db") const CouchDB = require("../../db")
const ClientDb = require("../../db/clientDb") const ClientDb = require("../../db/clientDb")
const bcrypt = require("../../utilities/bcrypt") const bcrypt = require("../../utilities/bcrypt")
const env = require("../../environment")
exports.authenticate = async ctx => { exports.authenticate = async ctx => {
const { username, password } = ctx.request.body const { username, password } = ctx.request.body
@ -10,8 +9,14 @@ exports.authenticate = async ctx => {
if (!username) ctx.throw(400, "Username Required.") if (!username) ctx.throw(400, "Username Required.")
if (!password) ctx.throw(400, "Password Required") if (!password) ctx.throw(400, "Password Required")
const masterDb = new CouchDB("master")
const { clientId } = await masterDb.get(ctx.params.appId)
if (!clientId) {
ctx.throw(400, "ClientId not suplied")
}
// find the instance that the user is associated with // find the instance that the user is associated with
const db = new CouchDB(ClientDb.name(env.CLIENT_ID)) const db = new CouchDB(ClientDb.name(clientId))
const appId = ctx.params.appId const appId = ctx.params.appId
const app = await db.get(appId) const app = await db.get(appId)
const instanceId = app.userInstanceMap[username] const instanceId = app.userInstanceMap[username]

View File

@ -6,13 +6,26 @@ exports.getClientId = async function(ctx) {
} }
exports.create = async function(ctx) { exports.create = async function(ctx) {
await create(env.CLIENT_ID) const clientId = getClientId(ctx)
await create(clientId)
ctx.status = 200 ctx.status = 200
ctx.message = `Client Database ${env.CLIENT_ID} successfully provisioned.` ctx.message = `Client Database ${clientId} successfully provisioned.`
} }
exports.destroy = async function(ctx) { exports.destroy = async function(ctx) {
await destroy(env.CLIENT_ID) const clientId = getClientId(ctx)
await destroy(clientId)
ctx.status = 200 ctx.status = 200
ctx.message = `Client Database ${env.CLIENT_ID} successfully deleted.` ctx.message = `Client Database ${clientId} successfully deleted.`
}
const getClientId = ctx => {
const clientId =
(ctx.query && ctx.query.clientId) ||
(ctx.body && ctx.body.clientId) ||
env.CLIENT_ID
if (!clientId) {
ctx.throw(400, "ClientId not suplied")
}
return clientId
} }

View File

@ -8,7 +8,9 @@ const {
const env = require("../../environment") const env = require("../../environment")
exports.fetchAppComponentDefinitions = async function(ctx) { exports.fetchAppComponentDefinitions = async function(ctx) {
const db = new CouchDB(ClientDb.name(env.CLIENT_ID)) const masterDb = new CouchDB("master")
const { clientId } = await masterDb.get(ctx.params.appId)
const db = new CouchDB(ClientDb.name(clientId))
const app = await db.get(ctx.params.appId) const app = await db.get(ctx.params.appId)
const componentDefinitions = app.componentLibraries.reduce( const componentDefinitions = app.componentLibraries.reduce(

View File

@ -8,7 +8,10 @@ exports.create = async function(ctx) {
const appShortId = ctx.params.applicationId.substring(0, 7) const appShortId = ctx.params.applicationId.substring(0, 7)
const instanceId = `inst_${appShortId}_${newid()}` const instanceId = `inst_${appShortId}_${newid()}`
const { applicationId } = ctx.params const { applicationId } = ctx.params
const clientId = env.CLIENT_ID
const masterDb = new CouchDB("master")
const { clientId } = await masterDb.get(applicationId)
const db = new CouchDB(instanceId) const db = new CouchDB(instanceId)
await db.put({ await db.put({
_id: "_design/database", _id: "_design/database",

View File

@ -42,8 +42,11 @@ exports.create = async function(ctx) {
const response = await database.post(user) const response = await database.post(user)
const masterDb = new CouchDB("master")
const { clientId } = await masterDb.get(appId)
// the clientDB needs to store a map of users against the app // the clientDB needs to store a map of users against the app
const db = new CouchDB(clientDb.name(env.CLIENT_ID)) const db = new CouchDB(clientDb.name(clientId))
const app = await db.get(appId) const app = await db.get(appId)
app.userInstanceMap = { app.userInstanceMap = {

View File

@ -5,6 +5,7 @@ const {
destroyClientDatabase, destroyClientDatabase,
builderEndpointShouldBlockNormalUsers, builderEndpointShouldBlockNormalUsers,
supertest, supertest,
TEST_CLIENT_ID,
defaultHeaders, defaultHeaders,
} = require("./couchTestUtils") } = require("./couchTestUtils")
@ -51,6 +52,7 @@ describe("/applications", () => {
body: { name: "My App" } body: { name: "My App" }
}) })
}) })
}) })
describe("fetch", () => { describe("fetch", () => {
@ -68,6 +70,37 @@ describe("/applications", () => {
expect(res.body.length).toBe(2) expect(res.body.length).toBe(2)
}) })
it("lists only applications in requested client databse", async () => {
await createApplication(request, "app1")
await createClientDatabase("new_client")
const blah = await request
.post("/api/applications")
.send({ name: "app2", clientId: "new_client"})
.set(defaultHeaders)
.expect('Content-Type', /json/)
//.expect(200)
const client1Res = await request
.get(`/api/applications?clientId=${TEST_CLIENT_ID}`)
.set(defaultHeaders)
.expect('Content-Type', /json/)
.expect(200)
expect(client1Res.body.length).toBe(1)
expect(client1Res.body[0].name).toBe("app1")
const client2Res = await request
.get(`/api/applications?clientId=new_client`)
.set(defaultHeaders)
.expect('Content-Type', /json/)
.expect(200)
expect(client2Res.body.length).toBe(1)
expect(client2Res.body[0].name).toBe("app2")
})
it("should apply authorization to endpoint", async () => { it("should apply authorization to endpoint", async () => {
const otherApplication = await createApplication(request) const otherApplication = await createApplication(request)
const instance = await createInstance(request, otherApplication._id) const instance = await createInstance(request, otherApplication._id)

View File

@ -9,6 +9,7 @@ const {
const TEST_CLIENT_ID = "test-client-id" const TEST_CLIENT_ID = "test-client-id"
exports.TEST_CLIENT_ID = TEST_CLIENT_ID
exports.supertest = async () => { exports.supertest = async () => {
let request let request
let port = 4002 let port = 4002
@ -59,7 +60,7 @@ exports.createView = async (request, instanceId, view) => {
return res.body return res.body
} }
exports.createClientDatabase = async () => await create(TEST_CLIENT_ID) exports.createClientDatabase = async id => await create(id || TEST_CLIENT_ID)
exports.createApplication = async (request, name = "test_application") => { exports.createApplication = async (request, name = "test_application") => {
const res = await request const res = await request