import { afterAll, beforeAll, beforeEach, describe, expect, it } from 'vitest'

import mongoose from 'mongoose'
import plugin from '../src/index'
import { server } from './mongo/server'

import { ObjectId } from 'bson'
import { StoryModel } from './models/Story'
import { UserModel } from './models/User'

import type CacheMongoose from '../src/index'

describe('cache-redis', async () => {
  const instance = server('cache-redis')
  let cache: CacheMongoose

  beforeAll(async () => {
    cache = plugin.init(mongoose, {
      engine: 'redis',
      engineOptions: {
        host: 'localhost',
        port: 6379,
      },
      defaultTTL: '10 seconds',
    })

    await instance.create()
  })

  afterAll(async () => {
    await cache.clear()
    await cache.close()
    await instance.destroy()
  })

  beforeEach(async () => {
    await mongoose.connection.collection('users').deleteMany({})
    await mongoose.connection.collection('stories').deleteMany({})
  })

  it('should use cache', async () => {
    const user = await UserModel.create({
      name: 'John Doe',
      role: 'admin',
    })

    const user1 = await UserModel.findById(user._id).cache().exec()
    await UserModel.findOneAndUpdate({ _id: user._id }, { name: 'John Doe 2' }).exec()
    const user2 = await UserModel.findById(user._id).cache().exec()

    expect(user1).not.toBeNull()
    expect(user2).not.toBeNull()
    expect(user1?._id.toString()).toBe(user2?._id.toString())
    expect(user1?.name).toEqual(user2?.name)
  })

  it('should clear cache', async () => {
    const user = await UserModel.create({
      name: 'John Doe',
      role: 'admin',
    })

    const cache1 = await UserModel.findById(user._id).cache().exec()
    await UserModel.findByIdAndUpdate(user._id, { name: 'Steve' }).exec()
    await cache.clear()
    const cache2 = await UserModel.findById(user._id).cache().exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache1?._id.toString()).toBe(cache2?._id.toString())
    expect(cache1?.name).not.toEqual(cache2?.name)
  })

  it('should use cache with custom-key-1', async () => {
    const user = await UserModel.create({
      name: 'John Doe',
      role: 'admin',
    })

    const cache1 = await UserModel.findById(user._id).cache('30 seconds', 'custom-key-1').exec()
    await UserModel.findOneAndUpdate({ _id: user._id }, { name: 'John Doe 2' }).exec()
    const cache2 = await UserModel.findById(user._id).cache('30 seconds', 'custom-key-1').exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache1?._id.toString()).toBe(cache2?._id.toString())
    expect(cache1?.name).toEqual(cache2?.name)
  })

  it('should clear custom-key-2', async () => {
    const user = await UserModel.create({
      name: 'John Doe',
      role: 'admin',
    })

    const cache1 = await UserModel.findById(user._id).cache('30 seconds', 'custom-key-2').exec()
    await UserModel.updateOne({ _id: user._id }, { name: 'Steve' }).exec()
    await cache.clear('custom-key-2')
    const cache2 = await UserModel.findById(user._id).cache('30 seconds', 'custom-key-2').exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache1?._id.toString()).toBe(cache2?._id.toString())
    expect(cache1?.name).not.toEqual(cache2?.name)
  })

  it('should use cache without cache options', async () => {
    expect(cache).toBeDefined()

    const user = await UserModel.create({
      name: 'John Doe',
      role: 'admin',
    })

    const cache1 = await UserModel.findById(user._id).cache().exec()
    await UserModel.updateOne({ _id: user._id }, { name: 'John Doe 2' }).exec()
    const cache2 = await UserModel.findById(user._id).cache().exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache1?._id.toString()).toBe(cache2?._id.toString())
    expect(cache1?.name).toEqual(cache2?.name)

    await UserModel.create([
      {
        name: 'Alice',
        role: 'admin',
      },
      {
        name: 'Bob',
        role: 'admin',
      },
    ])

    const cache3 = await UserModel.find({ role: 'admin' }).cache().exec()
    await UserModel.updateMany({ role: 'admin' }, { name: 'Steve' }).exec()
    const cache4 = await UserModel.find({ role: 'admin' }).cache().exec()

    expect(cache3).not.toBeNull()
    expect(cache4).not.toBeNull()
    expect(cache3?.length).toEqual(cache4?.length)
    expect(cache3?.[0].name).toEqual(cache4?.[0].name)
  })

  it('should use cache on aggregate', async () => {
    await UserModel.create([
      { name: 'John', role: 'admin' },
      { name: 'Bob', role: 'admin' },
      { name: 'Alice', role: 'user' },
    ])

    const cache1 = await UserModel.aggregate([{ $match: { role: 'admin' } }, { $group: { _id: '$role', count: { $sum: 1 } } }])
      .cache('30 seconds')
      .exec()

    await UserModel.create({ name: 'Mark', role: 'admin' })

    const cache2 = await UserModel.aggregate([{ $match: { role: 'admin' } }, { $group: { _id: '$role', count: { $sum: 1 } } }])
      .cache('30 seconds')
      .exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache1?.[0].count).toEqual(cache2?.[0].count)
  })

  it('should use cache on aggregate with custom-key', async () => {
    await UserModel.create([
      { name: 'John', role: 'admin' },
      { name: 'Bob', role: 'admin' },
      { name: 'Alice', role: 'user' },
    ])

    const cache1 = await UserModel.aggregate([{ $match: { role: 'admin' } }, { $group: { _id: '$role', count: { $sum: 1 } } }])
      .cache('30 seconds', 'aggregate-custom-key')
      .exec()

    await UserModel.create({ name: 'Mark', role: 'admin' })

    const cache2 = await UserModel.aggregate([{ $match: { role: 'admin' } }, { $group: { _id: '$role', count: { $sum: 1 } } }])
      .cache('30 seconds', 'aggregate-custom-key')
      .exec()

    // Don't use cache key
    const cache3 = await UserModel.aggregate([{ $match: { role: 'admin' } }, { $group: { _id: '$role', count: { $sum: 1 } } }]).exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache3).not.toBeNull()
    expect(cache1?.[0].count).toEqual(cache2?.[0].count)
    expect(cache1?.[0].count).not.toEqual(cache3?.[0].count)
  })

  it('should test lean cache', async () => {
    await UserModel.create([
      { name: 'John', role: 'admin' },
      { name: 'Bob', role: 'admin' },
      { name: 'Alice', role: 'user' },
    ])

    const cache1 = await UserModel.find({ role: 'admin' }).lean().cache('30 seconds').exec()
    await UserModel.create({ name: 'Mark', role: 'admin' })
    const cache2 = await UserModel.find({ role: 'admin' }).lean().cache('30 seconds').exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache1?.length).toEqual(cache2?.length)
  })

  it('should cache with regex query', async () => {
    await UserModel.create([
      { name: 'John', role: 'admin' },
      { name: 'Alice', role: 'user' },
      { name: 'Andy', role: 'user' },
    ])

    const cache1 = await UserModel.find({ name: /^J/ }).cache('30 seconds').exec()
    await UserModel.create({ name: 'Jenifer', role: 'admin' })
    const cache2 = await UserModel.find({ name: /^J/ }).cache('30 seconds').exec()

    expect(cache1).not.toBeNull()
    expect(cache2).not.toBeNull()
    expect(cache1?.length).toEqual(cache2?.length)

    const cache3 = await UserModel.find({ name: /^A/ }).cache('30 seconds').exec()
    await UserModel.create({ name: 'Alex', role: 'admin' })
    const cache4 = await UserModel.find({ name: /^A/ }).cache('30 seconds').exec()

    expect(cache3).not.toBeNull()
    expect(cache4).not.toBeNull()
    expect(cache3?.length).toEqual(cache4?.length)
  })

  it('should findOne', async () => {
    await UserModel.create([
      { name: 'C', role: 'admin' },
      { name: 'V', role: 'user' },
      { name: 'G', role: 'user' },
    ])

    const miss = await UserModel.findOne({ name: 'G' }).lean().cache('30 seconds').exec()
    expect(miss).not.toBeNull()

    expect(typeof miss?._id).toBe('object')
    expect(miss?._id instanceof mongoose.Types.ObjectId).toBeTruthy()

    expect(miss).toHaveProperty('name', 'G')

    const hit = await UserModel.findOne({ name: 'G' }).lean().cache('30 seconds').exec()
    expect(hit).not.toBeNull()

    expect(typeof hit?._id).toBe('object')
    expect(hit?._id instanceof ObjectId).toBeTruthy()

    expect(hit).toHaveProperty('name', 'G')

    expect(miss?._id.toString()).toBe(hit?._id.toString())
    expect(miss?.name).toEqual(hit?.name)
    expect(miss?.role).toEqual(hit?.role)
    expect(miss?.createdAt).toEqual(hit?.createdAt)
    expect(miss?.updatedAt).toEqual(hit?.updatedAt)
  })

  it('should distinct("_id") and distinct("role") and distinct("createdAt")', async () => {
    await UserModel.create({ name: 'i', role: 'admin' })
    await UserModel.create({ name: 'p', role: 'user' })
    await UserModel.create({ name: 'm', role: 'user' })

    const miss = await UserModel.distinct('_id').cache('30 seconds').exec()
    expect(miss).not.toBeNull()
    expect(miss?.length).toBe(3)

    expect(typeof miss?.[0]).toBe('object')
    expect(miss?.[0] instanceof mongoose.Types.ObjectId).toBeTruthy()

    const hit = await UserModel.distinct('_id').cache('30 seconds').exec()
    expect(hit).not.toBeNull()
    expect(hit?.length).toBe(3)

    expect(typeof hit?.[0]).toBe('object')
    expect(hit?.[0] instanceof ObjectId).toBeTruthy()

    const cache4 = await UserModel.distinct('role').cache('30 seconds').exec()
    expect(cache4).not.toBeNull()
    expect(cache4?.length).toBe(2)
    expect(cache4).toEqual(['admin', 'user'])

    const cache5 = await UserModel.distinct('role').cache('30 seconds').exec()
    expect(cache5).not.toBeNull()
    expect(cache5?.length).toBe(2)
    expect(cache5).toEqual(['admin', 'user'])

    const cache6 = await UserModel.distinct('name').cache('30 seconds').exec()
    expect(cache6).not.toBeNull()
    expect(cache6?.length).toBe(3)

    const cache7 = await UserModel.distinct('name').cache('30 seconds').exec()

    expect(miss.map((id) => id.toString())).toEqual(hit.map((id) => id.toString()))
    expect(cache4).toEqual(cache5)
    expect(cache6).toEqual(cache7)
  })

  it('should test exceptions', async () => {
    const user = await UserModel.create({ name: 'i', role: 'admin' })
    const story1 = await StoryModel.create({ title: '1', userId: user._id })
    const story2 = await StoryModel.create({ title: '2', userId: user._id })

    const miss = await UserModel.findOne({ name: 'i' }).populate({ path: 'stories' }).lean().cache('30 seconds').exec()
    const hit = await UserModel.findOne({ name: 'i' }).populate({ path: 'stories' }).lean().cache('30 seconds').exec()

    expect(miss).not.toBeNull()

    expect(typeof miss?._id).toBe('object')
    expect(miss?._id instanceof mongoose.Types.ObjectId).toBeTruthy()

    expect(miss?.name).toBe('i')
    expect(miss?.stories).not.toBeNull()
    expect(miss?.stories?.length).toBe(2)

    expect(miss?.stories?.[0]._id.toString()).toBe(story1._id.toString())

    expect(typeof miss?.stories?.[0]._id).toBe('object')
    expect(miss?.stories?.[0]._id instanceof mongoose.Types.ObjectId).toBeTruthy()

    expect(typeof miss?.stories?.[0].createdAt).toBe('object')
    expect(miss?.stories?.[0].createdAt instanceof Date).toBeTruthy()

    expect(miss?.stories?.[1]._id.toString()).toBe(story2._id.toString())

    expect(typeof miss?.stories?.[1]._id).toBe('object')
    expect(miss?.stories?.[1]._id instanceof mongoose.Types.ObjectId).toBeTruthy()

    expect(typeof miss?.stories?.[1].createdAt).toBe('object')
    expect(miss?.stories?.[1].createdAt instanceof Date).toBeTruthy()

    expect(hit).not.toBeNull()

    expect(typeof hit?._id).toBe('object')
    expect(hit?._id instanceof ObjectId).toBeTruthy()

    expect(hit?.name).toBe('i')
    expect(hit?.stories).not.toBeNull()
    expect(hit?.stories?.length).toBe(2)

    expect(hit?.stories?.[0]._id.toString()).toBe(story1._id.toString())

    expect(typeof hit?.stories?.[0]._id).toBe('object')
    expect(hit?.stories?.[0]._id instanceof ObjectId).toBeTruthy()

    expect(hit?.stories?.[0].createdAt instanceof Date).toBeTruthy()
    expect(typeof hit?.stories?.[0].createdAt).toBe('object')

    expect(hit?.stories?.[1]._id.toString()).toBe(story2._id.toString())

    expect(typeof hit?.stories?.[1]._id).toBe('object')
    expect(hit?.stories?.[1]._id instanceof ObjectId).toBeTruthy()

    expect(hit?.stories?.[1].createdAt instanceof Date).toBeTruthy()
    expect(typeof hit?.stories?.[1].createdAt).toBe('object')

    expect(miss?._id.toString()).toBe(hit?._id.toString())
    expect(miss?.name).toBe(hit?.name)
    expect(miss?.role).toBe(hit?.role)
    expect(miss?.createdAt?.toString()).toBe(hit?.createdAt?.toString())
    expect(miss?.updatedAt?.toString()).toBe(hit?.updatedAt?.toString())

    expect(miss?.stories?.[0]._id.toString()).toBe(hit?.stories?.[0]._id.toString())
    expect(miss?.stories?.[0].title).toBe(hit?.stories?.[0].title)
    expect(miss?.stories?.[0].userId.toString()).toBe(hit?.stories?.[0].userId.toString())
    expect(miss?.stories?.[0].createdAt?.toString()).toBe(hit?.stories?.[0].createdAt?.toString())
    expect(miss?.stories?.[0].updatedAt?.toString()).toBe(hit?.stories?.[0].updatedAt?.toString())

    expect(miss?.stories?.[1]._id.toString()).toBe(hit?.stories?.[1]._id.toString())
    expect(miss?.stories?.[1].title).toBe(hit?.stories?.[1].title)
    expect(miss?.stories?.[1].userId.toString()).toBe(hit?.stories?.[1].userId.toString())
    expect(miss?.stories?.[1].createdAt?.toString()).toBe(hit?.stories?.[1].createdAt?.toString())
    expect(miss?.stories?.[1].updatedAt?.toString()).toBe(hit?.stories?.[1].updatedAt?.toString())
  })

  it('should not misclassify certain fields as objectIds', async () => {
    // ObjectId.isValid will return true for multiple scenarios.
    // A string being a potentially valid objectId should not be the
    // determining factor on wether or not deserialize it as objectId.
    const user = await UserModel.create({
      name: '12CharString',
      role: '660ef695677786928202dc1f',
    })
    const pureLean = await UserModel.findOne({ _id: user._id }).lean()

    const miss = await UserModel.findOne({ _id: user._id }).lean().cache('30 seconds')
    const hit = await UserModel.findOne({ _id: user._id }).lean().cache('30 seconds')

    expect(pureLean).not.toBeNull()
    expect(typeof pureLean?._id).toBe('object')
    expect(typeof pureLean?.createdAt).toBe('object')

    expect(miss).not.toBeNull()
    expect(typeof miss?._id).toBe('object')
    expect(typeof miss?.createdAt).toBe('object')

    expect(hit).not.toBeNull()
    expect(typeof hit?._id).toBe('object')
    expect(typeof hit?.createdAt).toBe('object')

    expect(miss?._id.toString()).toBe(hit?._id.toString())
    expect(miss?.role).toEqual(hit?.role)
    expect(miss?.createdAt).toEqual(hit?.createdAt)

    const distinctMiss = await UserModel.distinct('_id').cache('30 seconds').lean().exec()
    expect(distinctMiss).not.toBeNull()
    expect(distinctMiss?.length).toBe(1)
    expect(distinctMiss).toEqual([pureLean?._id])

    const distinctHit = await UserModel.distinct('_id').cache('30 seconds').lean().exec()
    expect(distinctHit).not.toBeNull()
    expect(distinctHit?.length).toBe(1)
    expect(distinctHit.map((id) => id.toString())).toEqual([pureLean?._id.toString()])

    const distinctCreatedAtMiss = await UserModel.distinct('createdAt').cache('30 seconds').lean().exec()
    expect(distinctCreatedAtMiss).not.toBeNull()
    expect(distinctCreatedAtMiss?.length).toBe(1)
    expect(typeof distinctCreatedAtMiss?.[0]).toBe('object')
    expect(distinctCreatedAtMiss?.[0] instanceof Date).toBeTruthy()
    expect(distinctCreatedAtMiss).toEqual([pureLean?.createdAt])

    const distinctCreatedAtHit = await UserModel.distinct('createdAt').cache('30 seconds').lean().exec()
    expect(distinctCreatedAtHit).not.toBeNull()
    expect(distinctCreatedAtHit?.length).toBe(1)
    expect(typeof distinctCreatedAtMiss?.[0]).toBe('object')
    expect(distinctCreatedAtMiss?.[0] instanceof Date).toBeTruthy()
    expect(distinctCreatedAtHit).toEqual([pureLean?.createdAt])

    expect(miss?._id.toString()).toBe(hit?._id.toString())
    expect(miss?.name).toBe(hit?.name)
    expect(miss?.role).toBe(hit?.role)
    expect(miss?.createdAt?.toString()).toBe(hit?.createdAt?.toString())
    expect(miss?.updatedAt?.toString()).toBe(hit?.updatedAt?.toString())
  })

  it('should hydrate populated objects from cache', async () => {
    const user = await UserModel.create({ name: 'Alex', role: 'user' })
    const story1 = await StoryModel.create({ title: 'Ticket 1', userId: user._id })
    const story2 = await StoryModel.create({ title: 'Ticket 2', userId: user._id })

    const populatedOriginal = await UserModel.findOne({ name: 'Alex' }).populate('stories').lean().cache('1 minute').exec()

    expect(populatedOriginal).not.toBeNull()

    expect(typeof populatedOriginal?._id).toBe('object')
    expect(populatedOriginal?._id instanceof mongoose.Types.ObjectId).toBeTruthy()

    expect(populatedOriginal?.name).toBe('Alex')
    expect(populatedOriginal?.stories).not.toBeNull()
    expect(populatedOriginal?.stories?.length).toBe(2)

    expect(populatedOriginal?.stories?.[0]._id.toString()).toBe(story1._id.toString())

    expect(typeof populatedOriginal?.stories?.[0]._id).toBe('object')
    expect(populatedOriginal?.stories?.[0]._id instanceof mongoose.Types.ObjectId).toBeTruthy()

    expect(typeof populatedOriginal?.stories?.[0].createdAt).toBe('object')
    expect(populatedOriginal?.stories?.[0].createdAt instanceof Date).toBeTruthy()

    expect(populatedOriginal?.stories?.[1]._id.toString()).toBe(story2._id.toString())

    expect(typeof populatedOriginal?.stories?.[1]._id).toBe('object')
    expect(populatedOriginal?.stories?.[1]._id instanceof mongoose.Types.ObjectId).toBeTruthy()

    expect(typeof populatedOriginal?.stories?.[1].createdAt).toBe('object')
    expect(populatedOriginal?.stories?.[1].createdAt instanceof Date).toBeTruthy()

    // Deleting user and stories, to ensure that the cache is used
    await UserModel.deleteOne({ _id: user._id }).exec()
    await StoryModel.deleteMany({ userId: user._id }).exec()

    const populatedCache = await UserModel.findOne({ name: 'Alex' }).populate('stories').lean().cache('1 minute').exec()

    expect(populatedCache).not.toBeNull()

    expect(typeof populatedCache?._id).toBe('object')
    expect(populatedCache?._id instanceof ObjectId).toBeTruthy()

    expect(populatedCache?.name).toBe('Alex')
    expect(populatedCache?.stories).not.toBeNull()
    expect(populatedCache?.stories?.length).toBe(2)

    expect(populatedCache?.stories?.[0]._id.toString()).toBe(story1._id.toString())

    expect(typeof populatedCache?.stories?.[0]._id).toBe('object')
    expect(populatedCache?.stories?.[0]._id instanceof ObjectId).toBeTruthy()

    expect(typeof populatedCache?.stories?.[0].createdAt).toBe('object')
    expect(populatedCache?.stories?.[0].createdAt instanceof Date).toBeTruthy()

    expect(populatedCache?.stories?.[1]._id.toString()).toBe(story2._id.toString())

    expect(typeof populatedCache?.stories?.[1]._id).toBe('object')
    expect(populatedCache?.stories?.[1]._id instanceof ObjectId).toBeTruthy()

    expect(typeof populatedCache?.stories?.[1].createdAt).toBe('object')
    expect(populatedCache?.stories?.[1].createdAt instanceof Date).toBeTruthy()

    expect(populatedOriginal?._id.toString()).toBe(populatedCache?._id.toString())
    expect(populatedOriginal?.name).toBe(populatedCache?.name)
    expect(populatedOriginal?.role).toBe(populatedCache?.role)
    expect(populatedOriginal?.createdAt?.toString()).toBe(populatedCache?.createdAt?.toString())
    expect(populatedOriginal?.updatedAt?.toString()).toBe(populatedCache?.updatedAt?.toString())

    expect(populatedOriginal?.stories?.[0]._id.toString()).toBe(populatedCache?.stories?.[0]._id.toString())
    expect(populatedOriginal?.stories?.[0].title).toBe(populatedCache?.stories?.[0].title)
    expect(populatedOriginal?.stories?.[0].userId.toString()).toBe(populatedCache?.stories?.[0].userId.toString())
    expect(populatedOriginal?.stories?.[0].createdAt?.toString()).toBe(populatedCache?.stories?.[0].createdAt?.toString())
    expect(populatedOriginal?.stories?.[0].updatedAt?.toString()).toBe(populatedCache?.stories?.[0].updatedAt?.toString())

    expect(populatedOriginal?.stories?.[1]._id.toString()).toBe(populatedCache?.stories?.[1]._id.toString())
    expect(populatedOriginal?.stories?.[1].title).toBe(populatedCache?.stories?.[1].title)
    expect(populatedOriginal?.stories?.[1].userId.toString()).toBe(populatedCache?.stories?.[1].userId.toString())
    expect(populatedOriginal?.stories?.[1].createdAt?.toString()).toBe(populatedCache?.stories?.[1].createdAt?.toString())
    expect(populatedOriginal?.stories?.[1].updatedAt?.toString()).toBe(populatedCache?.stories?.[1].updatedAt?.toString())
  })
})