TopBar.tsx

/*
Copyright 2019-2020 Gravitational, Inc.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

import React, { Suspense, useState } from 'react';
import styled, { useTheme } from 'styled-components';
import { Flex, Text, TopNav } from 'design';

import { matchPath, useHistory } from 'react-router';

import { BrainIcon, OpenAIIcon } from 'design/SVGIcon';

import { useLocalStorage } from 'shared/hooks/useLocalStorage';

import useTeleport from 'teleport/useTeleport';
import useStickyClusterId from 'teleport/useStickyClusterId';
import { UserMenuNav } from 'teleport/components/UserMenuNav';
import { useFeatures } from 'teleport/FeaturesContext';

import cfg from 'teleport/config';

import { useLayout } from 'teleport/Main/LayoutContext';

import { KeysEnum } from 'teleport/services/localStorage';
import {
  Popup,
  PopupButton,
  PopupFooter,
  PopupLogos,
  PopupLogosSpacer,
  PopupTitle,
  PopupTitleBackground,
  TeleportIcon,
} from 'teleport/Assist/Popup/Popup';

import ClusterSelector from './ClusterSelector';

const Assist = React.lazy(() => import('teleport/Assist'));

const AssistButton = styled.div`
  padding: 0 10px;
  height: 48px;
  display: flex;
  align-items: center;
  justify-content: center;
  border-radius: 5px;
  cursor: pointer;
  user-select: none;
  margin-right: 5px;

  &:hover {
    background: ${props => props.theme.colors.spotBackground[0]};
  }
`;

const AssistButtonContainer = styled.div`
  position: relative;
`;

const Background = styled.div`
  position: fixed;
  top: 0;
  left: 0;
  right: 0;
  bottom: 0;
  z-index: 98;
  background: rgba(0, 0, 0, 0.6);
`;

export function TopBar() {
  const theme = useTheme();

  const ctx = useTeleport();
  const history = useHistory();
  const features = useFeatures();

  const assistEnabled = ctx.getFeatureFlags().assist && ctx.assistEnabled;

  const [showAssistPopup, setShowAssistPopup] = useLocalStorage(
    KeysEnum.SHOW_ASSIST_POPUP,
    assistEnabled
  );

  const [showAssist, setShowAssist] = useState(false);

  const { clusterId, hasClusterUrl } = useStickyClusterId();

  const { hasDockedElement } = useLayout();

  function loadClusters() {
    return ctx.clusterService.fetchClusters();
  }

  function changeCluster(value: string) {
    const newPrefix = cfg.getClusterRoute(value);

    const oldPrefix = cfg.getClusterRoute(clusterId);

    const newPath = history.location.pathname.replace(oldPrefix, newPrefix);
    history.push(newPath);
  }

  // find active feature
  const feature = features
    .filter(feature => Boolean(feature.route))
    .find(f =>
      matchPath(history.location.pathname, {
        path: f.route.path,
        exact: f.route.exact ?? false,
      })
    );

  const title = feature?.route?.title || '';

  // instead of re-creating an expensive react-select component,
  // hide/show it instead
  const styles = {
    display: !hasClusterUrl ? 'none' : 'block',
  };

  return (
    <TopBarContainer>
      {!hasClusterUrl && (
        <Text fontSize="18px" bold data-testid="title">
          {title}
        </Text>
      )}
      <Text fontSize="18px" id="topbar-portal" ml={2}></Text>
      <ClusterSelector
        value={clusterId}
        width="384px"
        maxMenuHeight={200}
        mr="20px"
        onChange={changeCluster}
        onLoad={loadClusters}
        style={styles}
      />
      <Flex ml="auto" height="100%" alignItems="center">
        {!hasDockedElement && assistEnabled && (
          <AssistButtonContainer>
            <AssistButton onClick={() => setShowAssist(true)}>
              <BrainIcon />
            </AssistButton>

            {showAssistPopup && (
              <>
                <Background />
                <Popup>
                  <PopupTitle>
                    <PopupTitleBackground>New!</PopupTitleBackground>
                  </PopupTitle>{' '}
                  Try out Teleport Assist, a GPT-4-powered AI assistant that
                  leverages your infrastructure
                  <PopupFooter>
                    <PopupLogos>
                      <OpenAIIcon size={30} />
                      <PopupLogosSpacer>+</PopupLogosSpacer>
                      <TeleportIcon light={theme.name === 'light'} />
                    </PopupLogos>

                    <PopupButton onClick={() => setShowAssistPopup(false)}>
                      Close
                    </PopupButton>
                  </PopupFooter>
                </Popup>
              </>
            )}
          </AssistButtonContainer>
        )}
        <UserMenuNav username={ctx.storeUser.state.username} />
      </Flex>

      {showAssist && (
        <Suspense fallback={null}>
          <Assist onClose={() => setShowAssist(false)} />
        </Suspense>
      )}
    </TopBarContainer>
  );
}

export const TopBarContainer = styled(TopNav)`
  height: 72px;
  background-color: inherit;
  padding-left: ${({ theme }) => `${theme.space[6]}px`};
  overflow-y: initial;
  flex-shrink: 0;
  border-bottom: 1px solid ${({ theme }) => theme.colors.spotBackground[0]};
`;