import { makeStyles, Theme, createStyles } from '@material-ui/core/styles';
import { LinearProgress, Typography } from '@material-ui/core';

import { WebSocketControllerProps } from '.';

const useStyles = makeStyles((theme: Theme) =>
  createStyles({
    loadingSettings: {
      margin: theme.spacing(0.5)
    },
    loadingSettingsDetails: {
      margin: theme.spacing(4),
      textAlign: 'center'
    }
  })
);

export type WebSocketFormProps<D> = Omit<
  WebSocketControllerProps<D>,
  'connected'
> & { data: D };

interface WebSocketFormLoaderProps<D> extends WebSocketControllerProps<D> {
  render: (props: WebSocketFormProps<D>) => JSX.Element;
}

export default function WebSocketFormLoader<D>(
  props: WebSocketFormLoaderProps<D>
) {
  const { connected, render, data, ...rest } = props;
  const classes = useStyles();
  if (!connected || !data) {
    return (
      <div className={classes.loadingSettings}>
        <LinearProgress className={classes.loadingSettingsDetails} />
        <Typography variant="h6" className={classes.loadingSettingsDetails}>
          Connecting to WebSocket...
        </Typography>
      </div>
    );
  }
  return render({ ...rest, data });
}