import { renderHook, act } from '@testing-library/react'
import { useAtomValue, useSetAtom } from 'jotai'
import useFactoryReset, { FactoryResetState } from './useFactoryReset'
import { useActiveModel } from './useActiveModel'
import { fs } from '@janhq/core'

// Mock the dependencies
jest.mock('jotai', () => ({
  atom: jest.fn(),
  useAtomValue: jest.fn(),
  useSetAtom: jest.fn(),
}))
jest.mock('./useActiveModel', () => ({
  useActiveModel: jest.fn(),
}))
jest.mock('@janhq/core', () => ({
  fs: {
    rm: jest.fn(),
  },
  EngineManager: {
    instance: jest.fn().mockReturnValue({
      get: jest.fn(),
      engines: {
        values: jest.fn().mockReturnValue([])
      }
    }),
  },
}))

describe('useFactoryReset', () => {
  const mockStopModel = jest.fn()
  const mockSetFactoryResetState = jest.fn()
  const mockGetAppConfigurations = jest.fn()
  const mockUpdateAppConfiguration = jest.fn()
  const mockRelaunch = jest.fn()

  beforeEach(() => {
    jest.clearAllMocks()
    ;(useAtomValue as jest.Mock).mockReturnValue('/default/jan/data/folder')
    ;(useSetAtom as jest.Mock).mockReturnValue(mockSetFactoryResetState)
    ;(useActiveModel as jest.Mock).mockReturnValue({ stopModel: mockStopModel })
    global.window ??= Object.create(window)
    global.window.core = {
      api: {
        getAppConfigurations: mockGetAppConfigurations,
        updateAppConfiguration: mockUpdateAppConfiguration,
        relaunch: mockRelaunch,
      },
    }
    mockGetAppConfigurations.mockResolvedValue({
      data_folder: '/current/jan/data/folder',
      quick_ask: false,
    })
    jest.spyOn(global, 'setTimeout')
  })

  it('should reset all correctly', async () => {
    const { result } = renderHook(() => useFactoryReset())

    await act(async () => {
      await result.current.resetAll()
    })

    expect(mockSetFactoryResetState).toHaveBeenCalledWith(
      FactoryResetState.Starting
    )
    expect(mockSetFactoryResetState).toHaveBeenCalledWith(
      FactoryResetState.StoppingModel
    )
    expect(mockStopModel).toHaveBeenCalled()
    expect(setTimeout).toHaveBeenCalledWith(expect.any(Function), 4000)
    expect(mockSetFactoryResetState).toHaveBeenCalledWith(
      FactoryResetState.DeletingData
    )
    expect(fs.rm).toHaveBeenCalledWith('/current/jan/data/folder')
    expect(mockUpdateAppConfiguration).toHaveBeenCalledWith({
      data_folder: '/default/jan/data/folder',
      quick_ask: false,
    })
    expect(mockSetFactoryResetState).toHaveBeenCalledWith(
      FactoryResetState.ClearLocalStorage
    )
    expect(mockRelaunch).toHaveBeenCalled()
  })

  it('should keep current folder when specified', async () => {
    const { result } = renderHook(() => useFactoryReset())

    await act(async () => {
      await result.current.resetAll(true)
    })

    expect(mockUpdateAppConfiguration).not.toHaveBeenCalled()
  })

  // Add more tests as needed for error cases, edge cases, etc.
})