// 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 {Table} from '@uwdata/flechette';
import {useQueryState} from 'nuqs';

import {QueryRequest_ReportType} from '@parca/client';
import {useParcaContext} from '@parca/components';

import {
  FIELD_FUNCTION_FILE_NAME,
  FIELD_FUNCTION_START_LINE,
  FIELD_FUNCTION_SYSTEM_NAME,
  FIELD_INLINED,
  FIELD_LOCATION_ADDRESS,
  FIELD_LOCATION_LINE,
  FIELD_MAPPING_BUILD_ID,
  FIELD_MAPPING_FILE,
  FIELD_TIMESTAMP,
} from '../../ProfileFlameGraph/FlameGraphArrow';
import {arrowToString, getLabelPairs} from '../../ProfileFlameGraph/FlameGraphArrow/utils';
import {ProfileSource} from '../../ProfileSource';
import {useProfileViewContext} from '../../ProfileView/context/ProfileViewContext';
import {stringParam} from '../../hooks/urlParsers';
import {useDashboardItems} from '../../hooks/useDashboardItems';
import {useQuery} from '../../useQuery';

interface Props {
  table: Table;
  row: number;
}

interface GraphTooltipMetaInfoData {
  labelPairs: Array<[string, string]>;
  functionFilename: string;
  functionSystemName: string;
  file: string;
  openFile: () => void;
  isSourceAvailable: boolean;
  locationAddress: bigint;
  mappingFile: string | null;
  mappingBuildID: string | null;
  inlined: boolean | null;
  timestamp: bigint | null;
}

export const useGraphTooltipMetaInfo = ({table, row}: Props): GraphTooltipMetaInfoData => {
  const mappingFile: string | null = arrowToString(table.getChild(FIELD_MAPPING_FILE)?.get(row));
  const mappingBuildID: string | null = arrowToString(
    table.getChild(FIELD_MAPPING_BUILD_ID)?.get(row)
  );
  const locationAddress: bigint = table.getChild(FIELD_LOCATION_ADDRESS)?.get(row) ?? 0n;
  const inlined: boolean | null = table.getChild(FIELD_INLINED)?.get(row);
  const locationLine: bigint = table.getChild(FIELD_LOCATION_LINE)?.get(row) ?? 0n;
  const functionFilename: string =
    arrowToString(table.getChild(FIELD_FUNCTION_FILE_NAME)?.get(row)) ?? '';
  const functionSystemName: string =
    arrowToString(table.getChild(FIELD_FUNCTION_SYSTEM_NAME)?.get(row)) ?? '';
  const functionStartLine: bigint = table.getChild(FIELD_FUNCTION_START_LINE)?.get(row) ?? 0n;
  const lineNumber =
    locationLine !== 0n ? locationLine : functionStartLine !== 0n ? functionStartLine : undefined;
  const timestamp = table.getChild(FIELD_TIMESTAMP)?.get(row);

  const {queryServiceClient, enableSourcesView} = useParcaContext();
  const {profileSource} = useProfileViewContext();

  const {isLoading: sourceLoading, response: sourceResponse} = useQuery(
    queryServiceClient,
    profileSource as ProfileSource,
    QueryRequest_ReportType.SOURCE,
    {
      skip:
        enableSourcesView === false ||
        profileSource === undefined ||
        // eslint-disable-next-line no-extra-boolean-cast
        !Boolean(mappingBuildID) ||
        // eslint-disable-next-line no-extra-boolean-cast
        !Boolean(functionFilename),
      sourceBuildID: mappingBuildID !== null ? mappingBuildID : undefined,
      sourceFilename: functionFilename,
      sourceOnly: true,
    }
  );

  const isSourceAvailable = !sourceLoading && sourceResponse?.report != null;

  const getTextForFile = (): string => {
    if (functionFilename === '') return '<unknown>';

    return `${functionFilename} ${lineNumber !== undefined ? ` +${lineNumber.toString()}` : ''}`;
  };
  const file = getTextForFile();

  const labelPairs: Array<[string, string]> = getLabelPairs(table, row);

  const {dashboardItems, setDashboardItems} = useDashboardItems();

  const [_unusedBuildId, setSourceBuildId] = useQueryState('source_buildid', stringParam);

  const [_unusedFilename, setSourceFilename] = useQueryState('source_filename', stringParam);

  const [_unusedLine, setSourceLine] = useQueryState('source_line', stringParam);

  const openFile = (): void => {
    setDashboardItems([dashboardItems[0], 'source']);
    if (mappingBuildID != null) {
      void setSourceBuildId(mappingBuildID);
    }

    void setSourceFilename(functionFilename);
    if (lineNumber !== undefined) {
      void setSourceLine(lineNumber.toString());
    }
  };

  return {
    labelPairs,
    functionFilename,
    functionSystemName,
    file,
    openFile,
    isSourceAvailable,
    locationAddress,
    mappingBuildID,
    mappingFile,
    inlined,
    timestamp,
  };
};