72 lines
2.1 KiB
TypeScript
72 lines
2.1 KiB
TypeScript
import { useEffect, useState } from 'react';
|
||
import { clientInitData } from '@/web/common/system/staticData';
|
||
import { useRouter } from 'next/router';
|
||
import { useSystemStore } from '@/web/common/system/useSystemStore';
|
||
import type { FastGPTFeConfigsType } from '@fastgpt/global/common/system/types/index.d';
|
||
import { useMemoizedFn, useMount } from 'ahooks';
|
||
import { TrackEventName } from '../common/system/constants';
|
||
|
||
export const useInitApp = () => {
|
||
const router = useRouter();
|
||
const { hiId, bd_vid, k } = router.query as { hiId?: string; bd_vid?: string; k?: string };
|
||
const { loadGitStar, setInitd, feConfigs } = useSystemStore();
|
||
const [scripts, setScripts] = useState<FastGPTFeConfigsType['scripts']>([]);
|
||
const [title, setTitle] = useState(process.env.SYSTEM_NAME || 'AI');
|
||
|
||
const initFetch = useMemoizedFn(async () => {
|
||
const {
|
||
feConfigs: { scripts, isPlus, systemTitle }
|
||
} = await clientInitData();
|
||
|
||
setTitle(systemTitle || 'FastGPT');
|
||
|
||
// log fastgpt
|
||
if (!isPlus) {
|
||
console.log(
|
||
'%cWelcome to FastGPT',
|
||
'font-family:Arial; color:#3370ff ; font-size:18px; font-weight:bold;',
|
||
`GitHub：https://github.com/labring/FastGPT`
|
||
);
|
||
}
|
||
|
||
loadGitStar();
|
||
|
||
setScripts(scripts || []);
|
||
setInitd();
|
||
});
|
||
|
||
useMount(() => {
|
||
initFetch();
|
||
|
||
const errorTrack = (event: ErrorEvent) => {
|
||
window.umami?.track(TrackEventName.windowError, {
|
||
device: {
|
||
userAgent: navigator.userAgent,
|
||
platform: navigator.platform,
|
||
appName: navigator.appName
|
||
},
|
||
error: event,
|
||
url: location.href
|
||
});
|
||
};
|
||
// add window error track
|
||
window.addEventListener('error', errorTrack);
|
||
|
||
return () => {
|
||
window.removeEventListener('error', errorTrack);
|
||
};
|
||
});
|
||
|
||
useEffect(() => {
|
||
hiId && localStorage.setItem('inviterId', hiId);
|
||
bd_vid && sessionStorage.setItem('bd_vid', bd_vid);
|
||
k && sessionStorage.setItem('fastgpt_sem', JSON.stringify({ keyword: k }));
|
||
}, [bd_vid, hiId, k]);
|
||
|
||
return {
|
||
feConfigs,
|
||
scripts,
|
||
title
|
||
};
|
||
};
|