2023-02-16 05:28:20 +13:00
|
|
|
if (process.env.DD_APM_ENABLED) {
|
|
|
|
require("./ddApm")
|
|
|
|
}
|
2022-08-30 21:59:27 +12:00
|
|
|
|
2023-02-16 05:28:20 +13:00
|
|
|
// need to load environment first
|
|
|
|
import env from "./environment"
|
2022-01-20 21:15:46 +13:00
|
|
|
import Application from "koa"
|
2022-04-20 22:30:03 +12:00
|
|
|
import { bootstrap } from "global-agent"
|
2022-11-15 11:55:47 +13:00
|
|
|
import * as db from "./db"
|
2023-02-28 03:41:28 +13:00
|
|
|
import { sdk as proSdk } from "@budibase/pro"
|
2023-02-28 02:42:51 +13:00
|
|
|
import {
|
|
|
|
auth,
|
|
|
|
logging,
|
|
|
|
events,
|
|
|
|
middleware,
|
2023-02-28 03:51:33 +13:00
|
|
|
queue,
|
2023-02-28 02:42:51 +13:00
|
|
|
env as coreEnv,
|
2023-03-28 07:38:49 +13:00
|
|
|
timers,
|
2023-11-11 05:17:18 +13:00
|
|
|
redis,
|
2023-02-28 02:42:51 +13:00
|
|
|
} from "@budibase/backend-core"
|
2023-11-11 05:17:18 +13:00
|
|
|
db.init()
|
2022-11-24 07:25:20 +13:00
|
|
|
import Koa from "koa"
|
|
|
|
import koaBody from "koa-body"
|
|
|
|
import http from "http"
|
2022-11-29 06:54:04 +13:00
|
|
|
import api from "./api"
|
2023-10-17 22:11:08 +13:00
|
|
|
|
2021-06-28 02:46:04 +12:00
|
|
|
const koaSession = require("koa-session")
|
2023-10-17 22:11:08 +13:00
|
|
|
import { userAgent } from "koa-useragent"
|
2023-02-22 08:14:57 +13:00
|
|
|
|
2022-12-16 00:35:22 +13:00
|
|
|
import destroyable from "server-destroy"
|
2023-03-14 02:25:19 +13:00
|
|
|
import { initPro } from "./initPro"
|
2023-03-16 00:23:55 +13:00
|
|
|
import { handleScimBody } from "./middleware/handleScimBody"
|
2020-12-17 08:50:02 +13:00
|
|
|
|
2023-02-28 02:42:51 +13:00
|
|
|
if (coreEnv.ENABLE_SSO_MAINTENANCE_MODE) {
|
2023-02-21 21:23:53 +13:00
|
|
|
console.warn(
|
|
|
|
"Warning: ENABLE_SSO_MAINTENANCE_MODE is set. It is recommended this flag is disabled if maintenance is not in progress"
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2022-04-20 22:30:03 +12:00
|
|
|
// this will setup http and https proxies form env variables
|
|
|
|
bootstrap()
|
|
|
|
|
2022-01-20 21:15:46 +13:00
|
|
|
const app: Application = new Koa()
|
2020-12-17 08:50:02 +13:00
|
|
|
|
2021-07-06 04:16:45 +12:00
|
|
|
app.keys = ["secret", "key"]
|
2021-06-28 02:46:04 +12:00
|
|
|
|
2020-12-17 08:50:02 +13:00
|
|
|
// set up top level koa middleware
|
2023-03-16 00:23:55 +13:00
|
|
|
app.use(handleScimBody)
|
2023-03-16 01:34:17 +13:00
|
|
|
app.use(koaBody({ multipart: true }))
|
2023-03-16 00:23:55 +13:00
|
|
|
|
2021-06-28 02:46:04 +12:00
|
|
|
app.use(koaSession(app))
|
2023-04-05 02:08:46 +12:00
|
|
|
app.use(middleware.correlation)
|
|
|
|
app.use(middleware.pino)
|
2023-02-22 08:14:57 +13:00
|
|
|
app.use(userAgent)
|
2020-12-17 08:50:02 +13:00
|
|
|
|
2021-04-02 08:34:43 +13:00
|
|
|
// authentication
|
2022-11-24 07:25:20 +13:00
|
|
|
app.use(auth.passport.initialize())
|
|
|
|
app.use(auth.passport.session())
|
2021-04-02 08:34:43 +13:00
|
|
|
|
2020-12-17 08:50:02 +13:00
|
|
|
// api routes
|
|
|
|
app.use(api.routes())
|
|
|
|
|
|
|
|
const server = http.createServer(app.callback())
|
|
|
|
destroyable(server)
|
|
|
|
|
2022-05-31 21:16:22 +12:00
|
|
|
let shuttingDown = false,
|
|
|
|
errCode = 0
|
2021-05-25 05:45:43 +12:00
|
|
|
server.on("close", async () => {
|
2022-05-31 08:22:06 +12:00
|
|
|
if (shuttingDown) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
shuttingDown = true
|
2022-08-31 21:47:41 +12:00
|
|
|
console.log("Server Closed")
|
2023-03-28 07:38:49 +13:00
|
|
|
timers.cleanup()
|
2023-11-11 05:17:18 +13:00
|
|
|
await redis.invite.shutdown()
|
|
|
|
await redis.passwordReset.shutdown()
|
2022-05-10 21:33:59 +12:00
|
|
|
await events.shutdown()
|
2023-02-28 03:41:28 +13:00
|
|
|
await queue.shutdown()
|
2022-05-31 21:16:22 +12:00
|
|
|
if (!env.isTest()) {
|
|
|
|
process.exit(errCode)
|
|
|
|
}
|
2021-05-25 05:45:43 +12:00
|
|
|
})
|
2020-12-17 08:50:02 +13:00
|
|
|
|
2022-05-31 08:22:06 +12:00
|
|
|
const shutdown = () => {
|
|
|
|
server.close()
|
|
|
|
server.destroy()
|
|
|
|
}
|
|
|
|
|
2023-01-11 22:37:37 +13:00
|
|
|
export default server.listen(parseInt(env.PORT || "4002"), async () => {
|
2020-12-18 01:39:55 +13:00
|
|
|
console.log(`Worker running on ${JSON.stringify(server.address())}`)
|
2023-03-14 02:25:19 +13:00
|
|
|
await initPro()
|
2023-11-11 05:17:18 +13:00
|
|
|
await redis.invite.init()
|
|
|
|
await redis.passwordReset.init()
|
2023-11-04 03:55:56 +13:00
|
|
|
// configure events to use the pro audit log write
|
|
|
|
// can't integrate directly into backend-core due to cyclic issues
|
|
|
|
await events.processors.init(proSdk.auditLogs.write)
|
2020-12-17 08:50:02 +13:00
|
|
|
})
|
|
|
|
|
|
|
|
process.on("uncaughtException", err => {
|
2022-05-31 21:16:22 +12:00
|
|
|
errCode = -1
|
2022-11-24 07:25:20 +13:00
|
|
|
logging.logAlert("Uncaught exception.", err)
|
2022-05-31 08:22:06 +12:00
|
|
|
shutdown()
|
2020-12-17 08:50:02 +13:00
|
|
|
})
|
|
|
|
|
|
|
|
process.on("SIGTERM", () => {
|
2022-05-31 08:22:06 +12:00
|
|
|
shutdown()
|
2020-12-17 08:50:02 +13:00
|
|
|
})
|
2022-08-26 06:41:47 +12:00
|
|
|
|
|
|
|
process.on("SIGINT", () => {
|
|
|
|
shutdown()
|
|
|
|
})
|