77 lines
2.1 KiB
TypeScript
77 lines
2.1 KiB
TypeScript
'use client'
|
|
|
|
import { Fragment, ReactNode, useEffect } from 'react'
|
|
|
|
import { AppConfiguration, getUserHomePath } from '@janhq/core'
|
|
import { useSetAtom } from 'jotai'
|
|
|
|
import useAssistants from '@/hooks/useAssistants'
|
|
import useGetSystemResources from '@/hooks/useGetSystemResources'
|
|
import { useLoadTheme } from '@/hooks/useLoadTheme'
|
|
import useModels from '@/hooks/useModels'
|
|
import useThreads from '@/hooks/useThreads'
|
|
|
|
import { SettingScreenList } from '@/screens/Settings'
|
|
|
|
import { defaultJanDataFolderAtom } from '@/helpers/atoms/App.atom'
|
|
import {
|
|
janDataFolderPathAtom,
|
|
quickAskEnabledAtom,
|
|
} from '@/helpers/atoms/AppConfig.atom'
|
|
import { janSettingScreenAtom } from '@/helpers/atoms/Setting.atom'
|
|
|
|
type Props = {
|
|
children: ReactNode
|
|
}
|
|
|
|
const DataLoader: React.FC<Props> = ({ children }) => {
|
|
const setJanDataFolderPath = useSetAtom(janDataFolderPathAtom)
|
|
const setQuickAskEnabled = useSetAtom(quickAskEnabledAtom)
|
|
const setJanDefaultDataFolder = useSetAtom(defaultJanDataFolderAtom)
|
|
const setJanSettingScreen = useSetAtom(janSettingScreenAtom)
|
|
|
|
useThreads()
|
|
useAssistants()
|
|
useGetSystemResources()
|
|
useLoadTheme()
|
|
|
|
const { loadDataModel, isUpdated } = useModels()
|
|
useEffect(() => {
|
|
// Listen for model updates
|
|
if (isUpdated) {
|
|
loadDataModel()
|
|
}
|
|
}, [isUpdated, loadDataModel])
|
|
|
|
useEffect(() => {
|
|
window.core?.api
|
|
?.getAppConfigurations()
|
|
?.then((appConfig: AppConfiguration) => {
|
|
setJanDataFolderPath(appConfig.data_folder)
|
|
setQuickAskEnabled(appConfig.quick_ask)
|
|
})
|
|
}, [setJanDataFolderPath, setQuickAskEnabled])
|
|
|
|
useEffect(() => {
|
|
async function getDefaultJanDataFolder() {
|
|
const defaultJanDataFolder = await getUserHomePath()
|
|
|
|
setJanDefaultDataFolder(defaultJanDataFolder)
|
|
}
|
|
getDefaultJanDataFolder()
|
|
}, [setJanDefaultDataFolder])
|
|
|
|
useEffect(() => {
|
|
const janSettingScreen = SettingScreenList.filter(
|
|
(screen) => window.electronAPI || screen !== 'Extensions'
|
|
)
|
|
setJanSettingScreen(janSettingScreen)
|
|
}, [setJanSettingScreen])
|
|
|
|
console.debug('Load Data...')
|
|
|
|
return <Fragment>{children}</Fragment>
|
|
}
|
|
|
|
export default DataLoader
|