sentinel/static/app/utils/profiling/flamegraph/flamegraphStateProvider/reducers/flamegraphSearch.tsx

import type {FlamegraphFrame} from 'sentry/utils/profiling/flamegraphFrame';
import type {SpanChartNode} from 'sentry/utils/profiling/spanChart';

export type FlamegraphSearchResult = {
  frame: FlamegraphFrame;
  match: ReadonlyArray<[number, number]>;
};

export type SpansSearchResult = {
  match: ReadonlyArray<[number, number]>;
  span: SpanChartNode;
};

export type FlamegraphSearch = {
  highlightFrames: {name: string | undefined; package: string | undefined} | null;
  index: number | null;
  query: string;
  results: {
    frames: Map<string, FlamegraphSearchResult>;
    spans: Map<string, SpansSearchResult>;
  };
};

type ClearFlamegraphSearchAction = {
  type: 'clear search';
};

type SetFlamegraphResultsAction = {
  payload: {
    query: string;
    results: FlamegraphSearch['results'];
  };
  type: 'set search results';
};

type FlamegraphSearchArrowNavigationAction = {
  payload: number;
  type: 'set search index position';
};

type SetHighlightAllFrames = {
  payload: {
    name: string;
    package: string;
  } | null;
  type: 'set highlight all frames';
};

type FlamegraphSearchAction =
  | ClearFlamegraphSearchAction
  | FlamegraphSearchArrowNavigationAction
  | SetFlamegraphResultsAction
  | SetHighlightAllFrames;

export function flamegraphSearchReducer(
  state: FlamegraphSearch,
  action: FlamegraphSearchAction
): FlamegraphSearch {
  switch (action.type) {
    case 'clear search': {
      return {
        ...state,
        query: '',
        index: null,
        highlightFrames: null,
        results: {
          frames: new Map(),
          spans: new Map(),
        },
      };
    }
    case 'set highlight all frames': {
      return {
        ...state,
        highlightFrames: action.payload,
      };
    }
    case 'set search results': {
      return {...state, highlightFrames: null, ...action.payload};
    }
    case 'set search index position': {
      return {...state, index: action.payload};
    }
    default: {
      return state;
    }
  }
}