speckle-server/packages/server/modules/acc/services/cron.ts

import {
  queryAllPendingAccSyncItemsFactory,
  updateAccSyncItemStatusFactory
} from '@/modules/acc/repositories/accSyncItems'
import type { ScheduleExecution } from '@/modules/core/domain/scheduledTasks/operations'
import { db } from '@/db/knex'
import { getManifestByUrn, getToken } from '@/modules/acc/clients/autodesk'
import { isReadyForImport } from '@/modules/acc/domain/logic'
import type { Logger } from '@/observability/logging'
import { getProjectDbClient } from '@/modules/multiregion/utils/dbSelector'
import {
  getAutomationFactory,
  getLatestAutomationRevisionFactory,
  upsertAutomationRunFactory,
  getAutomationTokenFactory
} from '@/modules/automate/repositories/automations'
import {
  storeApiTokenFactory,
  storeTokenScopesFactory,
  storeTokenResourceAccessDefinitionsFactory,
  storeUserServerAppTokenFactory
} from '@/modules/core/repositories/tokens'
import { createAppTokenFactory } from '@/modules/core/services/tokens'
import { TIME_MS } from '@speckle/shared'
import type { AccRegion } from '@/modules/acc/domain/constants'
import { triggerSyncItemAutomationFactory } from '@/modules/acc/services/automate'

const queryAllPendingAccSyncItems = queryAllPendingAccSyncItemsFactory({ db })

export const schedulePendingSyncItemsCheck = (deps: {
  scheduleExecution: ScheduleExecution
}) => {
  const callback = async (_now: Date, { logger }: { logger: Logger }) => {
    const tokenData = await getToken()
    for await (const items of queryAllPendingAccSyncItems()) {
      for (const syncItem of items) {
        const manifest = await getManifestByUrn(
          {
            urn: syncItem.accFileVersionUrn,
            region: syncItem.accRegion as AccRegion
          },
          { token: tokenData.access_token }
        )
        const isReady = isReadyForImport(manifest)

        logger.info(
          { isReady, syncItem, manifest },
          'Checking pending sync item {syncItem.id} for import readiness.'
        )

        if (!isReady) continue

        const projectDb = await getProjectDbClient({ projectId: syncItem.projectId })

        await triggerSyncItemAutomationFactory({
          updateAccSyncItemStatus: updateAccSyncItemStatusFactory({ db }),
          getAutomation: getAutomationFactory({ db: projectDb }),
          getLatestAutomationRevision: getLatestAutomationRevisionFactory({
            db: projectDb
          }),
          upsertAutomationRun: upsertAutomationRunFactory({ db: projectDb }),
          createAppToken: createAppTokenFactory({
            storeApiToken: storeApiTokenFactory({ db }),
            storeTokenScopes: storeTokenScopesFactory({ db }),
            storeTokenResourceAccessDefinitions:
              storeTokenResourceAccessDefinitionsFactory({
                db
              }),
            storeUserServerAppToken: storeUserServerAppTokenFactory({ db })
          }),
          getAutomationToken: getAutomationTokenFactory({
            db: projectDb
          })
        })
      }
    }
  }

  return deps.scheduleExecution(
    '*/5 * * * *',
    'pendingAccSyncItemsCheck',
    callback,
    15 * TIME_MS.minute
  )
}