96 lines
2.3 KiB
TypeScript

'use client'
import { PropsWithChildren, useEffect, useState } from 'react'
import { Toaster } from 'react-hot-toast'
import { TooltipProvider } from '@janhq/uikit'
import { PostHogProvider } from 'posthog-js/react'
import GPUDriverPrompt from '@/containers/GPUDriverPromptModal'
import EventListenerWrapper from '@/containers/Providers/EventListener'
import JotaiWrapper from '@/containers/Providers/Jotai'
import ThemeWrapper from '@/containers/Providers/Theme'
import FeatureToggleWrapper from '@/context/FeatureToggle'
import { setupCoreServices } from '@/services/coreService'
import {
isCoreExtensionInstalled,
setupBaseExtensions,
} from '@/services/extensionService'
import { instance } from '@/utils/posthog'
import KeyListener from './KeyListener'
import { extensionManager } from '@/extension'
const Providers = (props: PropsWithChildren) => {
const { children } = props
const [setupCore, setSetupCore] = useState(false)
const [activated, setActivated] = useState(false)
async function setupExtensions() {
// Register all active extensions
await extensionManager.registerActive()
setTimeout(async () => {
if (!isCoreExtensionInstalled()) {
setupBaseExtensions()
return
}
extensionManager.load()
setActivated(true)
}, 500)
}
// Services Setup
useEffect(() => {
setupCoreServices()
setSetupCore(true)
return () => {
extensionManager.unload()
}
}, [])
useEffect(() => {
if (setupCore) {
// Electron
if (window && window.core?.api) {
setupExtensions()
} else {
// Host
setActivated(true)
}
}
}, [setupCore])
return (
<PostHogProvider client={instance}>
<JotaiWrapper>
<ThemeWrapper>
{setupCore && activated && (
<KeyListener>
<FeatureToggleWrapper>
<EventListenerWrapper>
<TooltipProvider delayDuration={0}>
{children}
</TooltipProvider>
{!isMac && <GPUDriverPrompt />}
</EventListenerWrapper>
<Toaster />
</FeatureToggleWrapper>
</KeyListener>
)}
</ThemeWrapper>
</JotaiWrapper>
</PostHogProvider>
)
}
export default Providers