mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-22 14:55:16 +00:00
135 lines
5.1 KiB
TypeScript
135 lines
5.1 KiB
TypeScript
import React from 'react';
|
|
import PropTypes from 'prop-types';
|
|
import Head from 'next/head';
|
|
|
|
import {ThemeProvider} from '@material-ui/core/styles';
|
|
|
|
import Dialog from '@material-ui/core/Dialog';
|
|
import DialogContent from '@material-ui/core/DialogContent';
|
|
import DialogContentText from '@material-ui/core/DialogContentText';
|
|
import DialogTitle from '@material-ui/core/DialogTitle';
|
|
import LinearProgress from '@material-ui/core/LinearProgress';
|
|
import Box from '@material-ui/core/Box';
|
|
|
|
import CssBaseline from '@material-ui/core/CssBaseline';
|
|
import theme from '../src/theme';
|
|
import '../styles/globals.css'
|
|
import {querySessions, querySyncStatus} from "../api/bbgo";
|
|
import {Sync} from "@material-ui/icons";
|
|
|
|
const SyncNotStarted = 0
|
|
const Syncing = 1
|
|
const SyncDone = 2
|
|
|
|
// session is configured, check if we're syncing data
|
|
let syncStatusPoller = null
|
|
|
|
export default function MyApp(props) {
|
|
const {Component, pageProps} = props;
|
|
|
|
const [loading, setLoading] = React.useState(true)
|
|
const [syncing, setSyncing] = React.useState(false)
|
|
|
|
React.useEffect(() => {
|
|
// Remove the server-side injected CSS.
|
|
const jssStyles = document.querySelector('#jss-server-side');
|
|
if (jssStyles) {
|
|
jssStyles.parentElement.removeChild(jssStyles);
|
|
}
|
|
|
|
querySessions((sessions) => {
|
|
if (sessions.length > 0) {
|
|
setSyncing(true)
|
|
|
|
const pollSyncStatus = () => {
|
|
querySyncStatus((status) => {
|
|
switch (status) {
|
|
case SyncNotStarted:
|
|
break
|
|
case Syncing:
|
|
setSyncing(true);
|
|
break;
|
|
case SyncDone:
|
|
clearInterval(syncStatusPoller);
|
|
setLoading(false);
|
|
setSyncing(false);
|
|
break;
|
|
}
|
|
}).catch((err) => {
|
|
console.error(err)
|
|
})
|
|
}
|
|
|
|
syncStatusPoller = setInterval(pollSyncStatus, 1000)
|
|
} else {
|
|
// no session found, so we can not sync any data
|
|
setLoading(false)
|
|
setSyncing(false)
|
|
}
|
|
}).catch((err) => {
|
|
console.error(err)
|
|
})
|
|
|
|
}, []);
|
|
|
|
return (
|
|
<React.Fragment>
|
|
<Head>
|
|
<title>BBGO</title>
|
|
<meta name="viewport" content="minimum-scale=1, initial-scale=1, width=device-width"/>
|
|
</Head>
|
|
<ThemeProvider theme={theme}>
|
|
{/* CssBaseline kickstart an elegant, consistent, and simple baseline to build upon. */}
|
|
<CssBaseline/>
|
|
{
|
|
loading ? (syncing ? (
|
|
<React.Fragment>
|
|
<Dialog
|
|
open={syncing}
|
|
aria-labelledby="alert-dialog-title"
|
|
aria-describedby="alert-dialog-description"
|
|
>
|
|
<DialogTitle id="alert-dialog-title">{"Syncing Trades"}</DialogTitle>
|
|
<DialogContent>
|
|
<DialogContentText id="alert-dialog-description">
|
|
The environment is syncing trades from the exchange sessions.
|
|
Please wait a moment...
|
|
</DialogContentText>
|
|
<Box m={2}>
|
|
<LinearProgress/>
|
|
</Box>
|
|
</DialogContent>
|
|
</Dialog>
|
|
</React.Fragment>
|
|
) : (
|
|
<React.Fragment>
|
|
<Dialog
|
|
open={loading}
|
|
aria-labelledby="alert-dialog-title"
|
|
aria-describedby="alert-dialog-description"
|
|
>
|
|
<DialogTitle id="alert-dialog-title">{"Loading"}</DialogTitle>
|
|
<DialogContent>
|
|
<DialogContentText id="alert-dialog-description">
|
|
Loading...
|
|
</DialogContentText>
|
|
<Box m={2}>
|
|
<LinearProgress/>
|
|
</Box>
|
|
</DialogContent>
|
|
</Dialog>
|
|
</React.Fragment>
|
|
)) : (
|
|
<Component {...pageProps}/>
|
|
)
|
|
}
|
|
</ThemeProvider>
|
|
</React.Fragment>
|
|
);
|
|
}
|
|
|
|
MyApp.propTypes = {
|
|
Component: PropTypes.elementType.isRequired,
|
|
pageProps: PropTypes.object.isRequired,
|
|
};
|