Daniel Gak Anagrov
e67c27914c
* added a chron job that updates all workspaces properties to mixpanel
82 lines
2.9 KiB
TypeScript
82 lines
2.9 KiB
TypeScript
import cron from 'node-cron'
|
|
import { moduleLogger } from '@/observability/logging'
|
|
import { getFeatureFlags } from '@/modules/shared/helpers/envHelper'
|
|
import { registerOrUpdateScopeFactory } from '@/modules/shared/repositories/scopes'
|
|
import db from '@/db/knex'
|
|
import { Optional, SpeckleModule } from '@/modules/shared/helpers/typeHelper'
|
|
import { workspaceRoles } from '@/modules/workspaces/roles'
|
|
import { workspaceScopes } from '@/modules/workspaces/scopes'
|
|
import { registerOrUpdateRole } from '@/modules/shared/repositories/roles'
|
|
import { initializeEventListenersFactory } from '@/modules/workspaces/events/eventListener'
|
|
import { validateModuleLicense } from '@/modules/gatekeeper/services/validateLicense'
|
|
import { getSsoRouter } from '@/modules/workspaces/rest/sso'
|
|
import { InvalidLicenseError } from '@/modules/gatekeeper/errors/license'
|
|
import { scheduleExecutionFactory } from '@/modules/core/services/taskScheduler'
|
|
import {
|
|
acquireTaskLockFactory,
|
|
releaseTaskLockFactory
|
|
} from '@/modules/core/repositories/scheduledTasks'
|
|
import { scheduleUpdateAllWorkspacesTracking } from '@/modules/workspaces/services/tracking'
|
|
import { getClient } from '@/modules/shared/utils/mixpanel'
|
|
|
|
const {
|
|
FF_WORKSPACES_MODULE_ENABLED,
|
|
FF_WORKSPACES_SSO_ENABLED,
|
|
FF_BILLING_INTEGRATION_ENABLED
|
|
} = getFeatureFlags()
|
|
|
|
let quitListeners: Optional<() => void> = undefined
|
|
let scheduledTasks: cron.ScheduledTask[] = []
|
|
|
|
const initScopes = async () => {
|
|
const registerFunc = registerOrUpdateScopeFactory({ db })
|
|
await Promise.all(workspaceScopes.map((scope) => registerFunc({ scope })))
|
|
}
|
|
|
|
const initRoles = async () => {
|
|
const registerFunc = registerOrUpdateRole({ db })
|
|
await Promise.all(workspaceRoles.map((role) => registerFunc({ role })))
|
|
}
|
|
|
|
const workspacesModule: SpeckleModule = {
|
|
async init({ app, isInitial }) {
|
|
if (!FF_WORKSPACES_MODULE_ENABLED) return
|
|
const isWorkspaceLicenseValid = await validateModuleLicense({
|
|
requiredModules: ['workspaces']
|
|
})
|
|
|
|
if (!isWorkspaceLicenseValid)
|
|
throw new InvalidLicenseError(
|
|
'The workspaces module needs a valid license to run, contact Speckle to get one.'
|
|
)
|
|
moduleLogger.info('⚒️ Init workspaces module')
|
|
|
|
if (FF_WORKSPACES_SSO_ENABLED) app.use(getSsoRouter())
|
|
|
|
if (isInitial) {
|
|
const mixpanel = getClient()
|
|
const scheduleExecution = scheduleExecutionFactory({
|
|
acquireTaskLock: acquireTaskLockFactory({ db }),
|
|
releaseTaskLock: releaseTaskLockFactory({ db })
|
|
})
|
|
|
|
if (FF_BILLING_INTEGRATION_ENABLED && mixpanel)
|
|
scheduledTasks = [
|
|
scheduleUpdateAllWorkspacesTracking({ scheduleExecution, mixpanel })
|
|
]
|
|
|
|
quitListeners = initializeEventListenersFactory({ db })()
|
|
}
|
|
await Promise.all([initScopes(), initRoles()])
|
|
},
|
|
shutdown() {
|
|
if (!FF_WORKSPACES_MODULE_ENABLED) return
|
|
quitListeners?.()
|
|
scheduledTasks.forEach((task) => {
|
|
task.stop()
|
|
})
|
|
}
|
|
}
|
|
|
|
export = workspacesModule
|