import { Optional, SpeckleModule } from '@/modules/shared/helpers/typeHelper'
import { publishNotification } from '@/modules/notifications/services/publication'
import { activitiesLogger, moduleLogger } from '@/logging/logging'
import { weeklyEmailDigestEnabled } from '@/modules/shared/helpers/envHelper'
import { EventBus, getEventBus } from '@/modules/shared/services/eventBus'
import { sendActivityNotificationsFactory } from '@/modules/activitystream/services/summary'
import {
  getActiveUserStreamsFactory,
  saveActivityFactory
} from '@/modules/activitystream/repositories'
import { db } from '@/db/knex'
import { addStreamCreatedActivityFactory } from '@/modules/activitystream/services/streamActivity'
import { getStreamFactory } from '@/modules/core/repositories/streams'
import { ScheduleExecution } from '@/modules/core/domain/scheduledTasks/operations'
import { scheduleExecutionFactory } from '@/modules/core/services/taskScheduler'
import {
  acquireTaskLockFactory,
  releaseTaskLockFactory
} from '@/modules/core/repositories/scheduledTasks'
import { Knex } from 'knex'
import { publish } from '@/modules/shared/utils/subscriptions'
import { ProjectEvents } from '@/modules/core/domain/projects/events'
import { reportUserActivityFactory } from '@/modules/activitystream/events/userListeners'
import { reportAccessRequestActivityFactory } from '@/modules/activitystream/events/accessRequestListeners'
import { reportBranchActivityFactory } from '@/modules/activitystream/events/branchListeners'
import { reportCommitActivityFactory } from '@/modules/activitystream/events/commitListeners'
import { reportCommentActivityFactory } from '@/modules/activitystream/events/commentListeners'
import { reportStreamInviteActivityFactory } from '@/modules/activitystream/events/streamInviteListeners'
import { getProjectInviteProjectFactory } from '@/modules/serverinvites/services/projectInviteManagement'

let scheduledTask: ReturnType<ScheduleExecution> | null = null
let quitEventListeners: Optional<() => void> = undefined

/**
 * Initialize event listener for tracking various Speckle events and responding
 * to them by creating activitystream entries
 */
const initializeEventListeners = ({
  eventBus,
  db
}: {
  eventBus: EventBus
  db: Knex
}) => {
  const saveActivity = saveActivityFactory({ db })
  const reportUserActivity = reportUserActivityFactory({
    eventListen: eventBus.listen,
    saveActivity
  })
  const reportAccessRequestActivity = reportAccessRequestActivityFactory({
    eventListen: eventBus.listen,
    saveActivity
  })
  const reportBranchActivity = reportBranchActivityFactory({
    eventListen: eventBus.listen,
    saveActivity
  })
  const reportCommitActivity = reportCommitActivityFactory({
    eventListen: eventBus.listen,
    saveActivity
  })
  const reportCommentActivity = reportCommentActivityFactory({
    eventListen: eventBus.listen,
    saveActivity
  })
  const reportStreamInviteActivity = reportStreamInviteActivityFactory({
    eventListen: eventBus.listen,
    saveActivity,
    getProjectInviteProject: getProjectInviteProjectFactory({
      getStream: getStreamFactory({ db })
    })
  })

  const quitCbs = [
    reportUserActivity(),
    reportAccessRequestActivity(),
    reportBranchActivity(),
    reportCommitActivity(),
    reportCommentActivity(),
    reportStreamInviteActivity(),
    eventBus.listen(
      ProjectEvents.Created,
      async ({ payload: { ownerId, project } }) => {
        await addStreamCreatedActivityFactory({
          saveActivity: saveActivityFactory({ db }),
          publish
        })({
          streamId: project.id,
          creatorId: ownerId,
          stream: project,
          input: project
        })
      }
    )
  ]

  return () => quitCbs.forEach((quit) => quit())
}

const scheduleWeeklyActivityNotifications = () => {
  const scheduleExecution = scheduleExecutionFactory({
    acquireTaskLock: acquireTaskLockFactory({ db }),
    releaseTaskLock: releaseTaskLockFactory({ db })
  })

  // just to test stuff
  // every 1000 seconds
  // const cronExpression = '*/1000 * * * * *'
  // at 00 minutest, 10 (am) hours, every month, every year,
  // every 1st day of the week (monday)
  // cheat sheet https://crontab.guru
  const cronExpression = '00 10 * * 1'
  // configure the number of days, the activities are scraped for
  const numberOfDays = 7
  return scheduleExecution(
    cronExpression,
    'weeklyActivityNotification',
    //task should be locked for 10 minutes
    async (now: Date) => {
      activitiesLogger.info('Sending weekly activity digests notifications.')
      const end = now
      const start = new Date(end.getTime())
      start.setDate(start.getDate() - numberOfDays)
      await sendActivityNotificationsFactory({
        publishNotification,
        getActiveUserStreams: getActiveUserStreamsFactory({ db })
      })(start, end)
    },
    10 * 60 * 1000
  )
}

const activityModule: SpeckleModule = {
  init: async (_, isInitial) => {
    moduleLogger.info('🤺 Init activity module')
    if (isInitial) {
      quitEventListeners = initializeEventListeners({
        db,
        eventBus: getEventBus()
      })

      if (weeklyEmailDigestEnabled())
        scheduledTask = scheduleWeeklyActivityNotifications()
    }
  },
  shutdown: () => {
    scheduledTask?.stop()
    quitEventListeners?.()
  }
}

export = {
  ...activityModule
}