anvil/striker-ui/components/Display/FullSize.tsx

import { useState, useRef, useEffect, Dispatch, SetStateAction } from 'react';
import { RFB } from 'novnc-node';
import { Box, Menu, MenuItem, Typography } from '@material-ui/core';
import { makeStyles } from '@material-ui/core/styles';
import CloseIcon from '@material-ui/icons/Close';
import KeyboardIcon from '@material-ui/icons/Keyboard';
import IconButton from '@material-ui/core/IconButton';
import VncDisplay from './VncDisplay';
import { Panel } from '../Panels';
import { BLACK, RED, TEXT } from '../../lib/consts/DEFAULT_THEME';
import keyCombinations from './keyCombinations';
import putJSON from '../../lib/fetchers/putJSON';
import { HeaderText } from '../Text';
import Spinner from '../Spinner';

const useStyles = makeStyles(() => ({
  displayBox: {
    paddingTop: '1em',
    paddingBottom: 0,
    width: '75%',
    height: '80%',
  },
  vncBox: {
    width: '75%',
    height: '80%',
  },
  closeButton: {
    borderRadius: 8,
    backgroundColor: RED,
    '&:hover': {
      backgroundColor: RED,
    },
  },
  keyboardButton: {
    borderRadius: 8,
    backgroundColor: TEXT,
    '&:hover': {
      backgroundColor: TEXT,
    },
  },
  closeBox: {
    paddingBottom: '1em',
    paddingLeft: '.7em',
    paddingRight: 0,
  },
  buttonsBox: {
    paddingTop: 0,
  },
  keysItem: {
    backgroundColor: TEXT,
    paddingRight: '3em',
    '&:hover': {
      backgroundColor: TEXT,
    },
  },
}));

interface PreviewProps {
  setMode: Dispatch<SetStateAction<boolean>>;
  uuid: string;
  serverName: string | string[] | undefined;
}

interface VncConnectionProps {
  protocol: string;
  forward_port: number;
}

const extractDomain = (url: string | undefined): string | undefined => {
  const regEx = /[:]\/\/([^/]+)\//;
  let domain;
  let results;

  if (url) {
    results = regEx.exec(url);
    if (results) [, domain] = results;
  }
  return domain;
};

const FullSize = ({ setMode, uuid, serverName }: PreviewProps): JSX.Element => {
  const [anchorEl, setAnchorEl] = useState<null | HTMLElement>(null);
  const rfb = useRef<typeof RFB>(undefined);
  const hostname = useRef<string | undefined>(undefined);
  const [vncConnection, setVncConnection] = useState<
    VncConnectionProps | undefined
  >(undefined);
  const [displaySize] = useState<{
    width: string;
    height: string;
  }>({ width: '75%', height: '80%' });
  const classes = useStyles();

  useEffect(() => {
    if (typeof window !== 'undefined') {
      hostname.current = window.location.hostname;
    }

    if (!vncConnection)
      (async () => {
        const res = await putJSON(
          `${process.env.NEXT_PUBLIC_API_URL}/manage_vnc_pipes`,
          {
            server_uuid: uuid,
            is_open: true,
          },
        );
        setVncConnection(await res.json());
      })();
  }, [uuid, vncConnection]);

  const handleClick = (event: React.MouseEvent<HTMLButtonElement>): void => {
    setAnchorEl(event.currentTarget);
  };

  const handleClickClose = async () => {
    await putJSON(`${process.env.NEXT_PUBLIC_API_URL}/manage_vnc_pipes`, {
      server_uuid: uuid,
      is_open: false,
    });
  };

  const handleSendKeys = (scans: string[]) => {
    if (rfb.current) {
      if (!scans.length) rfb.current.sendCtrlAltDel();
      else {
        // Send pressing keys
        scans.forEach((scan) => {
          rfb.current.sendKey(scan, 1);
        });

        // Send releasing keys in reverse order
        for (let i = scans.length - 1; i >= 0; i -= 1) {
          rfb.current.sendKey(scans[i], 0);
        }
      }
      setAnchorEl(null);
    }
  };
  const extractedDomain = extractDomain(process.env.NEXT_PUBLIC_API_URL);

  return (
    <Panel>
      <Box flexGrow={1}>
        <HeaderText text={`Server: ${serverName}`} />
      </Box>

      <Box display="flex" className={classes.displayBox}>
        {vncConnection && extractedDomain ? (
          <Box className={classes.vncBox}>
            <VncDisplay
              rfb={rfb}
              url={`${vncConnection.protocol}://${hostname.current}:${vncConnection.forward_port}`}
              style={displaySize}
            />
          </Box>
        ) : (
          <Spinner />
        )}
        <Box>
          <Box className={classes.closeBox}>
            <IconButton
              className={classes.closeButton}
              style={{ color: TEXT }}
              component="span"
              onClick={() => {
                handleClickClose();
                setMode(true);
              }}
            >
              <CloseIcon />
            </IconButton>
          </Box>
          <Box className={classes.closeBox}>
            <IconButton
              className={classes.keyboardButton}
              style={{ color: BLACK }}
              aria-label="upload picture"
              component="span"
              onClick={handleClick}
            >
              <KeyboardIcon />
            </IconButton>
            <Menu
              anchorEl={anchorEl}
              keepMounted
              open={Boolean(anchorEl)}
              onClose={() => setAnchorEl(null)}
            >
              {keyCombinations.map(({ keys, scans }) => {
                return (
                  <MenuItem
                    onClick={() => handleSendKeys(scans)}
                    className={classes.keysItem}
                    key={keys}
                  >
                    <Typography variant="subtitle1">{keys}</Typography>
                  </MenuItem>
                );
              })}
            </Menu>
          </Box>
        </Box>
      </Box>
    </Panel>
  );
};

export default FullSize;