import { db } from '@/db/knex'
import { BasicTestUser, buildBasicTestUser, createTestUser } from '@/test/authHelper'
import {
  BasicTestWorkspace,
  buildBasicTestWorkspace,
  createTestWorkspace
} from '@/modules/workspaces/tests/helpers/creation'
import { Workspaces } from '@/modules/workspacesCore/helpers/db'
import {
  getWorkspaceFactory,
  getWorkspacesNonCompleteFactory
} from '@/modules/workspaces/repositories/workspaces'
import { expect } from 'chai'
import dayjs from 'dayjs'
import { deleteWorkspacesNonCompleteFactory } from '@/modules/workspaces/services/workspaceCreationState'
import { logger } from '@/observability/logging'
import {
  deleteStreamFactory,
  legacyGetStreamsFactory
} from '@/modules/core/repositories/streams'
import { deleteSsoProviderFactory } from '@/modules/workspaces/repositories/sso'
import { getEventBus } from '@/modules/shared/services/eventBus'
import { deleteAllResourceInvitesFactory } from '@/modules/serverinvites/repositories/serverInvites'
import { deleteWorkspaceFactory as repoDeleteWorkspaceFactory } from '@/modules/workspaces/repositories/workspaces'
import { deleteWorkspaceFactory } from '@/modules/workspaces/services/management'
import { queryAllProjectsFactory } from '@/modules/core/services/projects'

const updateAWorkspaceCreatedAt = async (
  workspaceId: string,
  createdAt: dayjs.Dayjs = dayjs()
) => {
  await db
    .table(Workspaces.name)
    .where({ [Workspaces.col.id]: workspaceId })
    .update({ createdAt })
}

describe('WorkspaceCreationState services', () => {
  const getWorkspacesNonComplete = getWorkspacesNonCompleteFactory({ db })
  const getWorkspace = getWorkspaceFactory({ db })
  const deleteWorkspacesNonComplete = deleteWorkspacesNonCompleteFactory({
    getWorkspacesNonComplete,
    deleteWorkspace: deleteWorkspaceFactory({
      deleteWorkspace: repoDeleteWorkspaceFactory({ db }),
      deleteProject: deleteStreamFactory({ db }),
      deleteAllResourceInvites: deleteAllResourceInvitesFactory({ db }),
      queryAllProjects: queryAllProjectsFactory({
        getStreams: legacyGetStreamsFactory({ db })
      }),
      deleteSsoProvider: deleteSsoProviderFactory({ db }),
      emitWorkspaceEvent: getEventBus().emit
    })
  })

  let adminUser: BasicTestUser
  let completeWorkspace: BasicTestWorkspace
  let nonCompleteWorkspace: BasicTestWorkspace

  before(async () => {
    adminUser = buildBasicTestUser()
    completeWorkspace = buildBasicTestWorkspace()
    nonCompleteWorkspace = buildBasicTestWorkspace()

    await createTestUser(adminUser)
    await createTestWorkspace(completeWorkspace, adminUser, {
      addCreationState: {
        state: {},
        completed: true
      }
    })
    await createTestWorkspace(nonCompleteWorkspace, adminUser, {
      addCreationState: {
        state: {},
        completed: false
      }
    })
  })

  it('does not show completed/impcompleted workspaces when they are recent', async () => {
    await updateAWorkspaceCreatedAt(completeWorkspace.id, dayjs())
    await updateAWorkspaceCreatedAt(nonCompleteWorkspace.id, dayjs())
    const thirtyMinutesAgo = dayjs().subtract(30, 'minutes')

    const workspaces = await getWorkspacesNonComplete({
      createdAtBefore: thirtyMinutesAgo.toDate()
    })

    expect(workspaces).to.have.lengthOf(0)
  })

  it('hits workspaces with complete false when they are 30 mins older', async () => {
    const fortyMinutesAgo = dayjs().subtract(40, 'minutes')
    const thirtyMinutesAgo = dayjs().subtract(30, 'minutes')

    await updateAWorkspaceCreatedAt(completeWorkspace.id, fortyMinutesAgo)
    await updateAWorkspaceCreatedAt(nonCompleteWorkspace.id, fortyMinutesAgo)

    const workspaces = await getWorkspacesNonComplete({
      createdAtBefore: thirtyMinutesAgo.toDate()
    })

    expect(workspaces).to.have.lengthOf(1)
    expect(workspaces).to.deep.eq([{ workspaceId: nonCompleteWorkspace.id }])
  })

  it('deletes only those workspaces that are not completed', async () => {
    const fortyMinutesAgo = dayjs().subtract(40, 'minutes')
    await updateAWorkspaceCreatedAt(completeWorkspace.id, fortyMinutesAgo)
    await updateAWorkspaceCreatedAt(nonCompleteWorkspace.id, fortyMinutesAgo)

    await deleteWorkspacesNonComplete({ logger })

    const workspace1 = await getWorkspace({ workspaceId: completeWorkspace.id })
    const workspace2 = await getWorkspace({ workspaceId: nonCompleteWorkspace.id })
    expect(workspace1).to.exist
    expect(workspace2).to.be.null
  })
})