143 lines
3.8 KiB
TypeScript
143 lines
3.8 KiB
TypeScript
import { userSettingsHelperAtom } from '@/atoms/settings';
|
|
import { GPUStackVersionAtom, UpdateCheckAtom } from '@/atoms/user';
|
|
import { setAtomStorage } from '@/atoms/utils';
|
|
import { DEFAULT_ENTER_PAGE, GPUSTACK_API_BASE_URL } from '@/config/settings';
|
|
import { COLOR_PRIMARY } from '@/config/theme/constants';
|
|
import { enterprisePluginReady } from '@/plugins/enterprise-ready';
|
|
import { GPUStackPluginManager } from '@/plugins/manager';
|
|
import { requestConfig } from '@/request-config';
|
|
import {
|
|
queryCurrentUserState,
|
|
queryVersionInfo,
|
|
updateCheck
|
|
} from '@/services/profile/apis';
|
|
import { fetchSystemConfig } from '@/services/system/query-system-config';
|
|
import { isOnline } from '@/utils';
|
|
import {
|
|
IS_FIRST_LOGIN,
|
|
readState,
|
|
writeState
|
|
} from '@/utils/localstore/index';
|
|
import '@gpustack/core-ui/style.css';
|
|
import { RequestConfig, history, request as umiRequest } from '@umijs/max';
|
|
import { message } from 'antd';
|
|
|
|
// only for the first login and access from http://localhost
|
|
|
|
const checkDefaultPage = async (userInfo: any) => {
|
|
const isFirstLogin = await readState(IS_FIRST_LOGIN);
|
|
if (isFirstLogin === null && isOnline()) {
|
|
writeState(IS_FIRST_LOGIN, true);
|
|
if (userInfo && userInfo?.is_admin) {
|
|
history.push(DEFAULT_ENTER_PAGE.adminForFirst);
|
|
}
|
|
}
|
|
};
|
|
|
|
// runtime configuration
|
|
export async function getInitialState(): Promise<{
|
|
fetchUserInfo: () => Promise<Global.UserInfo>;
|
|
currentUser?: Global.UserInfo;
|
|
pluginData?: Record<string, any>;
|
|
}> {
|
|
const { location } = history;
|
|
|
|
// In open-source builds the promise resolves immediately.
|
|
await enterprisePluginReady;
|
|
|
|
// initialize plugins and merge enterprise locales
|
|
let pluginData = {};
|
|
try {
|
|
pluginData = await GPUStackPluginManager.initialize({
|
|
request: umiRequest,
|
|
setUserSettings: (value) => setAtomStorage(userSettingsHelperAtom, value),
|
|
setStorageUserSettings: (value) =>
|
|
setAtomStorage(userSettingsHelperAtom, value),
|
|
defaultColorPrimary: COLOR_PRIMARY
|
|
});
|
|
} catch (error) {
|
|
console.error('Failed to initialize plugins:', error);
|
|
}
|
|
|
|
const getUpdateCheck = async () => {
|
|
try {
|
|
const data = await updateCheck();
|
|
|
|
setAtomStorage(UpdateCheckAtom, {
|
|
...data
|
|
});
|
|
return data;
|
|
} catch (error) {
|
|
console.error('updateCheck error', error);
|
|
}
|
|
};
|
|
|
|
const fetchUserInfo = async (config?: {
|
|
skipErrorHandler?: boolean;
|
|
}): Promise<Global.UserInfo> => {
|
|
try {
|
|
const data = await queryCurrentUserState({
|
|
skipErrorHandler: true
|
|
});
|
|
if (data.is_admin) {
|
|
getUpdateCheck();
|
|
fetchSystemConfig();
|
|
}
|
|
return data;
|
|
} catch (error: any) {
|
|
const data = error?.response?.data;
|
|
if (data?.code === 401 && data?.message.includes('deactivate')) {
|
|
message.error({
|
|
content: (
|
|
<div>
|
|
<span>{data?.message}</span>
|
|
</div>
|
|
),
|
|
duration: 5
|
|
});
|
|
}
|
|
history.push(DEFAULT_ENTER_PAGE.login);
|
|
}
|
|
return {} as Global.UserInfo;
|
|
};
|
|
|
|
const getAppVersionInfo = async () => {
|
|
try {
|
|
const data = await queryVersionInfo();
|
|
|
|
const isDev = data.version?.indexOf('0.0.0') > -1;
|
|
const isRc = data.version?.indexOf('rc') > -1;
|
|
|
|
setAtomStorage(GPUStackVersionAtom, {
|
|
...data,
|
|
isProd: !isDev && !isRc,
|
|
isDev,
|
|
isRc
|
|
});
|
|
} catch (error) {
|
|
console.error('queryVersionInfo error', error);
|
|
}
|
|
};
|
|
|
|
getAppVersionInfo();
|
|
|
|
if (![DEFAULT_ENTER_PAGE.login].includes(location.pathname)) {
|
|
const userInfo = await fetchUserInfo();
|
|
checkDefaultPage(userInfo);
|
|
return {
|
|
fetchUserInfo,
|
|
currentUser: userInfo,
|
|
pluginData
|
|
};
|
|
}
|
|
return {
|
|
fetchUserInfo,
|
|
pluginData
|
|
};
|
|
}
|
|
|
|
export const request: RequestConfig = {
|
|
baseURL: `/${GPUSTACK_API_BASE_URL}`,
|
|
...requestConfig
|
|
};
|