// Copyright 2022 The Parca Authors
// 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 {useEffect, useMemo} from 'react';

import {Provider} from 'react-redux';

import {QueryServiceClient} from '@parca/client';
import {KeyDownProvider, useParcaContext} from '@parca/components';
import {createStore} from '@parca/store';
import {capitalizeOnlyFirstLetter} from '@parca/utilities';

import {useCompareModeMeta} from '../hooks/useCompareModeMeta';
import {useHasProfileData} from '../useHasProfileData';
import ProfileExplorerCompare from './ProfileExplorerCompare';
import ProfileExplorerSingle from './ProfileExplorerSingle';

interface ProfileExplorerProps {
  queryClient: QueryServiceClient;
}

const ErrorContent = ({errorMessage}: {errorMessage: string}): JSX.Element => {
  return (
    <div
      className="relative rounded border border-red-400 bg-red-100 px-4 py-3 text-red-700"
      role="alert"
    >
      <span className="block sm:inline">{errorMessage}</span>
    </div>
  );
};

const ProfileExplorerApp = ({queryClient}: ProfileExplorerProps): JSX.Element => {
  const {
    loading: hasProfileDataLoading,
    data: hasProfileData,
    error: hasProfileDataError,
  } = useHasProfileData(queryClient);

  const {loader, noDataPrompt, onError, authenticationErrorMessage} = useParcaContext();
  const {isCompareMode} = useCompareModeMeta();

  useEffect(() => {
    if (hasProfileDataError !== undefined && hasProfileDataError !== null) {
      onError?.(hasProfileDataError);
    }
  }, [hasProfileDataError, onError]);

  if (hasProfileDataLoading) {
    return <>{loader}</>;
  }

  if (!hasProfileData) {
    return <>{noDataPrompt}</>;
  }

  if (hasProfileDataError !== undefined && hasProfileDataError !== null) {
    if (
      authenticationErrorMessage !== undefined &&
      hasProfileDataError.code === 'UNAUTHENTICATED'
    ) {
      return <ErrorContent errorMessage={authenticationErrorMessage} />;
    }

    return <ErrorContent errorMessage={capitalizeOnlyFirstLetter(hasProfileDataError.message)} />;
  }

  if (isCompareMode) {
    return <ProfileExplorerCompare queryClient={queryClient} />;
  }

  return <ProfileExplorerSingle queryClient={queryClient} />;
};

const ProfileExplorer = ({queryClient}: ProfileExplorerProps): JSX.Element => {
  const {additionalFlamegraphColorProfiles} = useParcaContext();

  const {store: reduxStore} = useMemo(() => {
    return createStore(additionalFlamegraphColorProfiles);
  }, [additionalFlamegraphColorProfiles]);

  return (
    <Provider store={reduxStore}>
      <KeyDownProvider>
        <ProfileExplorerApp queryClient={queryClient} />
      </KeyDownProvider>
    </Provider>
  );
};

export default ProfileExplorer;
