@parca/profile 0.19.142 → 0.19.143

package/src/ProfileView/components/Toolbars/TableColumnsDropdown.tsx

@@ -14,14 +14,15 @@
 import {useEffect, useMemo, useState} from 'react';
 
 import {createColumnHelper, type ColumnDef} from '@tanstack/table-core';
+import {useQueryState} from 'nuqs';
 
-import {useURLState} from '@parca/components';
 import {ProfileType} from '@parca/parser';
 import {valueFormatter} from '@parca/utilities';
 
 import {Row} from '../../../Table';
 import ColumnsVisibility from '../../../Table/ColumnsVisibility';
 import {ColumnName, addPlusSign, getRatioString} from '../../../Table/utils/functions';
+import {tableColumnsParser} from '../../../hooks/urlParsers';
 import {useProfileViewContext} from '../../context/ProfileViewContext';
 
 interface Props {
@@ -32,9 +33,7 @@ interface Props {
 
 const TableColumnsDropdown = ({profileType, total, filtered}: Props): JSX.Element => {
   const {compareMode} = useProfileViewContext();
-  const [tableColumns, setTableColumns] = useURLState<string[]>('table_columns', {
-    alwaysReturnArray: true,
-  });
+  const [tableColumns, setTableColumns] = useQueryState('table_columns', tableColumnsParser);
 
   const columnHelper = createColumnHelper<Row>();
 
@@ -190,7 +189,7 @@ const TableColumnsDropdown = ({profileType, total, filtered}: Props): JSX.Elemen
     const newTableColumns = (Object.keys(updatedColumns) as ColumnName[]).filter(
       col => updatedColumns[col]
     );
-    setTableColumns(newTableColumns);
+    void setTableColumns(newTableColumns);
   };
 
   return (

package/src/ProfileView/components/Toolbars/index.tsx

@@ -50,7 +50,7 @@ export interface VisualisationToolbarProps {
   flamechartDimension: string[];
   setFlamechartDimension: (labels: string[]) => void;
   showVisualizationSelector?: boolean;
-  sandwichFunctionName?: string;
+  sandwichFunctionName: string | null;
   alignFunctionName: string;
   setAlignFunctionName: (align: string) => void;
   colorBy: string;
@@ -75,7 +75,7 @@ export interface FlameGraphToolbarProps {
 
 export interface SandwichFlameGraphToolbarProps {
   resetSandwichFunctionName: () => void;
-  sandwichFunctionName?: string;
+  sandwichFunctionName: string | null;
 }
 
 export const TableToolbar: FC<TableToolbarProps> = ({profileType, total, filtered}) => {
@@ -120,7 +120,7 @@ export const SandwichFlameGraphToolbar: FC<SandwichFlameGraphToolbarProps> = ({
           onClick={() => resetSandwichFunctionName()}
           className="w-auto"
           variant="neutral"
-          disabled={sandwichFunctionName === undefined || sandwichFunctionName.length === 0}
+          disabled={sandwichFunctionName == null || sandwichFunctionName.length === 0}
         >
           Reset view
         </Button>

package/src/ProfileView/components/ViewSelector/index.tsx

@@ -13,9 +13,13 @@
 
 import {ReactNode} from 'react';
 
-import {useParcaContext, useURLState, useURLStateBatch} from '@parca/components';
+import {useQueryState} from 'nuqs';
+
+import {useParcaContext} from '@parca/components';
 
 import {ProfileSource} from '../../../ProfileSource';
+import {stringParam} from '../../../hooks/urlParsers';
+import {useDashboardItems} from '../../../hooks/useDashboardItems';
 import Dropdown, {DropdownElement, InnerAction} from './Dropdown';
 
 interface Props {
@@ -23,15 +27,9 @@ interface Props {
 }
 
 const ViewSelector = ({profileSource}: Props): JSX.Element => {
-  const [dashboardItems = ['flamegraph'], setDashboardItems] = useURLState<string[]>(
-    'dashboard_items',
-    {
-      alwaysReturnArray: true,
-    }
-  );
-  const [, setSandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
+  const {dashboardItems, setDashboardItems} = useDashboardItems();
+  const [, setSandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
   const {enableSourcesView, enableSandwichView} = useParcaContext();
-  const batchUpdates = useURLStateBatch();
 
   const allItems: Array<{
     key: string;
@@ -129,14 +127,9 @@ const ViewSelector = ({profileSource}: Props): JSX.Element => {
         } else {
           const newDashboardItems = dashboardItems.filter(v => v !== item.key);
 
-          // Batch updates when removing sandwich panel to combine both URL changes
+          setDashboardItems(newDashboardItems);
           if (item.key === 'sandwich') {
-            batchUpdates(() => {
-              setDashboardItems(newDashboardItems);
-              setSandwichFunctionName(undefined);
-            });
-          } else {
-            setDashboardItems(newDashboardItems);
+            void setSandwichFunctionName(null);
           }
         }
       },

package/src/ProfileView/context/DashboardContext.tsx

@@ -13,8 +13,10 @@
 
 import {FC, PropsWithChildren, createContext, useContext} from 'react';
 
-import {useURLState} from '@parca/components';
+import {useQueryState} from 'nuqs';
 
+import {stringParam} from '../../hooks/urlParsers';
+import {useDashboardItems} from '../../hooks/useDashboardItems';
 import {VisualizationType} from '../types/visualization';
 
 interface DashboardContextType {
@@ -27,10 +29,8 @@ interface DashboardContextType {
 const DashboardContext = createContext<DashboardContextType | undefined>(undefined);
 
 export const DashboardProvider: FC<PropsWithChildren> = ({children}) => {
-  const [dashboardItems, setDashboardItems] = useURLState<string[]>('dashboard_items', {
-    alwaysReturnArray: true,
-  });
-  const [, setSandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
+  const {dashboardItems, setDashboardItems} = useDashboardItems();
+  const [, setSandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
 
   const handleClosePanel = (visualizationType: VisualizationType): void => {
     const newDashboardItems = dashboardItems.filter(item => item !== visualizationType);
@@ -38,7 +38,7 @@ export const DashboardProvider: FC<PropsWithChildren> = ({children}) => {
 
     // Reset sandwich function name when closing sandwich panel
     if (visualizationType === 'sandwich') {
-      setSandwichFunctionName(undefined);
+      void setSandwichFunctionName(null);
     }
   };
 

package/src/ProfileView/hooks/useResetFlameGraphState.ts

@@ -11,17 +11,19 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import {useURLState} from '@parca/components';
+import {useQueryState} from 'nuqs';
+
+import {stringParam} from '../../hooks/urlParsers';
 
 export const useResetFlameGraphState = (): (() => void) => {
-  const [val, setCurPath] = useURLState('cur_path');
+  const [val, setCurPath] = useQueryState('cur_path', stringParam);
 
   return () => {
     setTimeout(() => {
-      if (val === undefined) {
+      if (val === null) {
         return;
       }
-      setCurPath(undefined);
+      void setCurPath(null);
     });
   };
 };

package/src/ProfileView/hooks/useResetStateOnProfileTypeChange.ts

@@ -11,39 +11,37 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import {useURLState, useURLStateBatch} from '@parca/components';
+import {useQueryStates} from 'nuqs';
 
+import {stringParam} from '../../hooks/urlParsers';
 import {useProfileFilters} from '../components/ProfileFilters/useProfileFilters';
 
 export const useResetStateOnProfileTypeChange = (): (() => void) => {
-  const [groupBy, setGroupBy] = useURLState('group_by');
-  const [curPath, setCurPath] = useURLState('cur_path');
-  const [sumByA, setSumByA] = useURLState('sum_by_a');
-  const [sumByB, setSumByB] = useURLState('sum_by_b');
+  const [state, setState] = useQueryStates(
+    {
+      group_by: stringParam,
+      cur_path: stringParam,
+      sum_by_a: stringParam,
+      sum_by_b: stringParam,
+      sandwich_function_name: stringParam,
+    },
+    {history: 'replace'}
+  );
   const {resetFilters} = useProfileFilters();
-  const [sandwichFunctionName, setSandwichFunctionName] = useURLState('sandwich_function_name');
-  const batchUpdates = useURLStateBatch();
 
   return () => {
-    // Batch all URL state resets into a single navigation
-    batchUpdates(() => {
-      if (groupBy !== undefined) {
-        setGroupBy(undefined);
-      }
-      if (curPath !== undefined) {
-        setCurPath(undefined);
-      }
-      if (sandwichFunctionName !== undefined) {
-        setSandwichFunctionName(undefined);
-      }
-      if (sumByA !== undefined) {
-        setSumByA(undefined);
-      }
-      if (sumByB !== undefined) {
-        setSumByB(undefined);
-      }
+    // Atomic reset: clear all params in single URL update
+    const updates: Record<string, null> = {};
+    if (state.group_by !== null) updates.group_by = null;
+    if (state.cur_path !== null) updates.cur_path = null;
+    if (state.sandwich_function_name !== null) updates.sandwich_function_name = null;
+    if (state.sum_by_a !== null) updates.sum_by_a = null;
+    if (state.sum_by_b !== null) updates.sum_by_b = null;
 
-      resetFilters();
-    });
+    if (Object.keys(updates).length > 0) {
+      void setState(updates);
+    }
+
+    resetFilters();
   };
 };

package/src/ProfileView/hooks/useResetStateOnSeriesChange.ts

@@ -11,19 +11,27 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-import {useURLState} from '@parca/components';
+import {useQueryStates} from 'nuqs';
+
+import {stringParam} from '../../hooks/urlParsers';
 
 export const useResetStateOnSeriesChange = (): (() => void) => {
-  const [curPath, setCurPath] = useURLState('cur_path');
-  const [sandwichFunctionName, setSandwichFunctionName] = useURLState('sandwich_function_name');
+  const [state, setState] = useQueryStates(
+    {
+      cur_path: stringParam,
+      sandwich_function_name: stringParam,
+    },
+    {history: 'replace'}
+  );
 
   return () => {
     setTimeout(() => {
-      if (curPath !== undefined) {
-        setCurPath(undefined);
-      }
-      if (sandwichFunctionName !== undefined) {
-        setSandwichFunctionName(undefined);
+      const updates: Record<string, null> = {};
+      if (state.cur_path !== null) updates.cur_path = null;
+      if (state.sandwich_function_name !== null) updates.sandwich_function_name = null;
+
+      if (Object.keys(updates).length > 0) {
+        void setState(updates);
       }
     });
   };

package/src/ProfileView/hooks/useVisualizationState.ts

@@ -13,13 +13,8 @@
 
 import {useCallback, useMemo} from 'react';
 
-import {
-  JSONParser,
-  JSONSerializer,
-  useURLState,
-  useURLStateBatch,
-  useURLStateCustom,
-} from '@parca/components';
+import {useQueryState} from 'nuqs';
+
 import {USER_PREFERENCES, useUserPreference} from '@parca/hooks';
 
 import {
@@ -30,12 +25,19 @@ import {
   FIELD_MAPPING_FILE,
 } from '../../ProfileFlameGraph/FlameGraphArrow';
 import {CurrentPathFrame} from '../../ProfileFlameGraph/FlameGraphArrow/utils';
+import {
+  flamechartDimensionParser,
+  groupByParser,
+  jsonParser,
+  stringParam,
+} from '../../hooks/urlParsers';
+import {useColorBy} from '../../hooks/useColorBy';
 import {useResetFlameGraphState} from './useResetFlameGraphState';
 
 export const useVisualizationState = (): {
   curPathArrow: CurrentPathFrame[];
   setCurPathArrow: (path: CurrentPathFrame[]) => void;
-  colorStackLegend: string | undefined;
+  colorStackLegend: string | null;
   colorBy: string;
   setColorBy: (colorBy: string) => void;
   groupBy: string[];
@@ -44,46 +46,52 @@ export const useVisualizationState = (): {
   setGroupByLabels: (labels: string[]) => void;
   flamechartDimension: string[];
   setFlamechartDimension: (labels: string[]) => void;
-  sandwichFunctionName: string | undefined;
-  setSandwichFunctionName: (sandwichFunctionName: string | undefined) => void;
+  sandwichFunctionName: string | null;
+  setSandwichFunctionName: (sandwichFunctionName: string | null) => void;
   resetSandwichFunctionName: () => void;
   alignFunctionName: string;
   setAlignFunctionName: (align: string) => void;
 } => {
-  const [colorByPreference, setColorByPreference] = useUserPreference<string>(
-    USER_PREFERENCES.COLOR_BY.key
-  );
   const [alignFunctionNamePreference, setAlignFunctionNamePreference] = useUserPreference<string>(
     USER_PREFERENCES.ALIGN_FUNCTION_NAME.key
   );
 
-  const [curPathArrow, setCurPathArrow] = useURLStateCustom<CurrentPathFrame[]>('cur_path', {
-    parse: JSONParser<CurrentPathFrame[]>,
-    stringify: JSONSerializer,
-    defaultValue: '[]',
-  });
-  const [colorStackLegend] = useURLState<string | undefined>('color_stack_legend');
-  const [colorBy, setStoreColorBy] = useURLState('color_by', {
-    defaultValue: colorByPreference,
-  });
-  const [alignFunctionName, setStoreAlignFunctionName] = useURLState('align_function_name', {
-    defaultValue: alignFunctionNamePreference,
-  });
-  const [groupBy, setStoreGroupBy] = useURLState<string[]>('group_by', {
-    defaultValue: [FIELD_FUNCTION_NAME],
-    alwaysReturnArray: true,
-  });
-  const [sandwichFunctionName, setSandwichFunctionName] = useURLState<string | undefined>(
-    'sandwich_function_name'
+  const [curPathArrow, setRawCurPathArrow] = useQueryState(
+    'cur_path',
+    jsonParser<CurrentPathFrame[]>().withDefault([])
+  );
+  const setCurPathArrow = useCallback(
+    (path: CurrentPathFrame[]) => {
+      void setRawCurPathArrow(path);
+    },
+    [setRawCurPathArrow]
+  );
+  const [colorStackLegend] = useQueryState('color_stack_legend', stringParam);
+  const {colorBy, setColorBy} = useColorBy();
+  const [alignFunctionNameRaw, setStoreAlignFunctionName] = useQueryState(
+    'align_function_name',
+    stringParam
+  );
+  const alignFunctionName = alignFunctionNameRaw ?? alignFunctionNamePreference ?? 'left';
+  const [groupBy, setStoreGroupBy] = useQueryState(
+    'group_by',
+    groupByParser.withDefault([FIELD_FUNCTION_NAME])
+  );
+  const [sandwichFunctionName, setRawSandwichFunctionName] = useQueryState(
+    'sandwich_function_name',
+    stringParam
   );
-  const [flamechartDimension, setStoreFlamechartDimension] = useURLState<string[]>(
+  const setSandwichFunctionName = useCallback(
+    (name: string | null) => {
+      void setRawSandwichFunctionName(name);
+    },
+    [setRawSandwichFunctionName]
+  );
+  const [flamechartDimension, setStoreFlamechartDimension] = useQueryState(
     'flamechart_dimension',
-    {
-      alwaysReturnArray: true,
-    }
+    flamechartDimensionParser.withDefault([])
   );
   const resetFlameGraphState = useResetFlameGraphState();
-  const batchUpdates = useURLStateBatch();
 
   const levelsOfProfiling = useMemo(
     () => [
@@ -97,62 +105,46 @@ export const useVisualizationState = (): {
 
   const setGroupBy = useCallback(
     (keys: string[]): void => {
-      setStoreGroupBy(keys);
+      void setStoreGroupBy(keys);
     },
     [setStoreGroupBy]
   );
 
   const toggleGroupBy = useCallback(
     (key: string): void => {
-      // Batch updates to combine setGroupBy + resetFlameGraphState into single URL navigation
-      batchUpdates(() => {
-        if (groupBy.includes(key)) {
-          setGroupBy(groupBy.filter(v => v !== key)); // remove
-        } else {
-          const filteredGroupBy = groupBy.filter(item => !levelsOfProfiling.includes(item));
-          setGroupBy([...filteredGroupBy, key]); // add
-        }
-
-        resetFlameGraphState();
-      });
+      if (groupBy.includes(key)) {
+        setGroupBy(groupBy.filter(v => v !== key));
+      } else {
+        const filteredGroupBy = groupBy.filter(item => !levelsOfProfiling.includes(item));
+        setGroupBy([...filteredGroupBy, key]);
+      }
+      resetFlameGraphState();
     },
-    [groupBy, setGroupBy, levelsOfProfiling, resetFlameGraphState, batchUpdates]
+    [groupBy, setGroupBy, levelsOfProfiling, resetFlameGraphState]
   );
 
   const setGroupByLabels = useCallback(
     (labels: string[]): void => {
-      // Batch updates to combine setGroupBy + resetFlameGraphState into single URL navigation
-      batchUpdates(() => {
-        setGroupBy(groupBy.filter(l => !l.startsWith(`${FIELD_LABELS}.`)).concat(labels));
-
-        resetFlameGraphState();
-      });
+      setGroupBy(groupBy.filter(l => !l.startsWith(`${FIELD_LABELS}.`)).concat(labels));
+      resetFlameGraphState();
     },
-    [groupBy, setGroupBy, resetFlameGraphState, batchUpdates]
+    [groupBy, setGroupBy, resetFlameGraphState]
   );
 
   const setFlamechartDimension = useCallback(
     (labels: string[]): void => {
-      setStoreFlamechartDimension(labels.filter(l => l.startsWith(`${FIELD_LABELS}.`)));
+      void setStoreFlamechartDimension(labels.filter(l => l.startsWith(`${FIELD_LABELS}.`)));
     },
     [setStoreFlamechartDimension]
   );
 
   const resetSandwichFunctionName = useCallback((): void => {
-    setSandwichFunctionName(undefined);
+    setSandwichFunctionName(null);
   }, [setSandwichFunctionName]);
 
-  const setColorBy = useCallback(
-    (value: string): void => {
-      setStoreColorBy(value);
-      setColorByPreference(value);
-    },
-    [setStoreColorBy, setColorByPreference]
-  );
-
   const setAlignFunctionName = useCallback(
     (value: string): void => {
-      setStoreAlignFunctionName(value);
+      void setStoreAlignFunctionName(value);
       setAlignFunctionNamePreference(value);
     },
     [setStoreAlignFunctionName, setAlignFunctionNamePreference]
@@ -162,7 +154,7 @@ export const useVisualizationState = (): {
     curPathArrow,
     setCurPathArrow,
     colorStackLegend,
-    colorBy: (colorBy as string) ?? '',
+    colorBy,
     setColorBy,
     groupBy,
     setGroupBy,
@@ -173,7 +165,7 @@ export const useVisualizationState = (): {
     sandwichFunctionName,
     setSandwichFunctionName,
     resetSandwichFunctionName,
-    alignFunctionName: (alignFunctionName as string) ?? 'left',
+    alignFunctionName,
     setAlignFunctionName,
   };
 };

package/src/ProfileViewWithData.tsx

@@ -13,15 +13,10 @@
 
 import {useEffect, useMemo, useState} from 'react';
 
+import {useQueryState} from 'nuqs';
+
 import {QueryRequest_ReportType, QueryServiceClient} from '@parca/client';
-import {
-  NumberParser,
-  NumberSerializer,
-  useGrpcMetadata,
-  useParcaContext,
-  useURLState,
-  useURLStateCustom,
-} from '@parca/components';
+import {useGrpcMetadata, useParcaContext} from '@parca/components';
 import {saveAsBlob} from '@parca/utilities';
 
 import {validateFlameChartQuery} from './ProfileFlameGraph';
@@ -35,6 +30,14 @@ import {MergedProfileSource, ProfileSource} from './ProfileSource';
 import {ProfileView} from './ProfileView';
 import {useProfileFilters} from './ProfileView/components/ProfileFilters/useProfileFilters';
 import type {SamplesSeries} from './ProfileView/types/visualization';
+import {
+  flamechartDimensionParser,
+  groupByParser,
+  intParam,
+  invertCallStackParser,
+  stringParam,
+} from './hooks/urlParsers';
+import {useDashboardItems} from './hooks/useDashboardItems';
 import {useQuery} from './useQuery';
 import {downloadPprof} from './utils';
 
@@ -53,22 +56,14 @@ export const ProfileViewWithData = ({
   onSwitchToFifteenMinutes,
 }: ProfileViewWithDataProps): JSX.Element => {
   const metadata = useGrpcMetadata();
-  const [dashboardItems, setDashboardItems] = useURLState<string[]>('dashboard_items', {
-    alwaysReturnArray: true,
-  });
-  const [sourceBuildID] = useURLState<string>('source_buildid');
-  const [sourceFilename] = useURLState<string>('source_filename');
-  const [groupBy] = useURLState<string[]>('group_by', {
-    defaultValue: [FIELD_FUNCTION_NAME],
-    alwaysReturnArray: true,
-  });
-  const [sandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
-  const [flamechartDimension] = useURLState<string[]>('flamechart_dimension', {
-    alwaysReturnArray: true,
-  });
+  const {dashboardItems, setDashboardItems} = useDashboardItems();
+  const [sourceBuildID] = useQueryState('source_buildid', stringParam);
+  const [sourceFilename] = useQueryState('source_filename', stringParam);
+  const [groupBy] = useQueryState('group_by', groupByParser.withDefault([FIELD_FUNCTION_NAME]));
+  const [sandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
+  const [flamechartDimension] = useQueryState('flamechart_dimension', flamechartDimensionParser);
 
-  const [invertStack] = useURLState('invert_call_stack');
-  const invertCallStack = invertStack === 'true';
+  const [invertCallStack] = useQueryState('invert_call_stack', invertCallStackParser);
 
   const [pprofDownloading, setPprofDownloading] = useState<boolean>(false);
 
@@ -110,7 +105,7 @@ export const ProfileViewWithData = ({
     skip: !dashboardItems.includes('flamegraph'),
     nodeTrimThreshold,
     groupBy,
-    invertCallStack,
+    invertCallStack: invertCallStack ?? false,
     protoFilters,
   });
 
@@ -134,11 +129,10 @@ export const ProfileViewWithData = ({
   );
 
   // Samples step count: 2px per data point for finer granularity in strips
-  const [samplesStepCount] = useURLStateCustom<number>('samples_step_count', {
-    defaultValue: String(getStepCountFromScreenWidth(2)),
-    parse: NumberParser,
-    stringify: NumberSerializer,
-  });
+  const [samplesStepCount] = useQueryState(
+    'samples_step_count',
+    intParam.withDefault(getStepCountFromScreenWidth(2))
+  );
 
   const {
     isLoading: samplesLoading,
@@ -203,8 +197,8 @@ export const ProfileViewWithData = ({
     error: sourceError,
   } = useQuery(queryClient, profileSource, QueryRequest_ReportType.SOURCE, {
     skip: !dashboardItems.includes('source'),
-    sourceBuildID,
-    sourceFilename,
+    sourceBuildID: sourceBuildID ?? undefined,
+    sourceFilename: sourceFilename ?? undefined,
     protoFilters,
   });
 
@@ -216,8 +210,8 @@ export const ProfileViewWithData = ({
     nodeTrimThreshold,
     groupBy: [FIELD_FUNCTION_NAME],
     invertCallStack: true,
-    sandwichByFunction: sandwichFunctionName,
-    skip: sandwichFunctionName === undefined && !dashboardItems.includes('sandwich'),
+    sandwichByFunction: sandwichFunctionName ?? undefined,
+    skip: sandwichFunctionName == null && !dashboardItems.includes('sandwich'),
     protoFilters,
   });
 
@@ -229,8 +223,8 @@ export const ProfileViewWithData = ({
     nodeTrimThreshold,
     groupBy: [FIELD_FUNCTION_NAME],
     invertCallStack: false,
-    sandwichByFunction: sandwichFunctionName,
-    skip: sandwichFunctionName === undefined && !dashboardItems.includes('sandwich'),
+    sandwichByFunction: sandwichFunctionName ?? undefined,
+    skip: sandwichFunctionName == null && !dashboardItems.includes('sandwich'),
     protoFilters,
   });
 

package/src/Sandwich/index.tsx

@@ -14,14 +14,15 @@
 import React, {useRef, useState} from 'react';
 
 import {AnimatePresence, motion} from 'framer-motion';
+import {useQueryState} from 'nuqs';
 
-import {useURLState} from '@parca/components';
 import {TEST_IDS, testId} from '@parca/test-utils';
 
 import {ProfileSource} from '../ProfileSource';
 import {useDashboard} from '../ProfileView/context/DashboardContext';
 import {useVisualizationState} from '../ProfileView/hooks/useVisualizationState';
 import {SandwichData} from '../ProfileView/types/visualization';
+import {stringParam} from '../hooks/urlParsers';
 import {CalleesSection} from './components/CalleesSection';
 import {CallersSection} from './components/CallersSection';
 
@@ -35,7 +36,7 @@ const Sandwich = React.memo(function Sandwich({
   profileSource,
 }: Props): React.JSX.Element {
   const {dashboardItems} = useDashboard();
-  const [sandwichFunctionName] = useURLState<string | undefined>('sandwich_function_name');
+  const [sandwichFunctionName] = useQueryState('sandwich_function_name', stringParam);
 
   const callersRef = React.useRef<HTMLDivElement | null>(null);
   const calleesRef = React.useRef<HTMLDivElement | null>(null);
@@ -57,7 +58,7 @@ const Sandwich = React.memo(function Sandwich({
           transition={{duration: 0.5}}
         >
           <div className="relative flex flex-row">
-            {sandwichFunctionName !== undefined ? (
+            {sandwichFunctionName != null ? (
               <div className="w-full flex flex-col" ref={callersCalleesContainerRef}>
                 <CallersSection
                   callersRef={callersRef}

package/src/SourceView/index.tsx

@@ -15,12 +15,14 @@ import React, {useCallback, useEffect, useMemo} from 'react';
 
 import {tableFromIPC} from '@uwdata/flechette';
 import {AnimatePresence, motion} from 'framer-motion';
+import {useQueryState} from 'nuqs';
 import {Item, Menu, useContextMenu} from 'react-contexify';
 
 import {Source} from '@parca/client';
-import {SourceSkeleton, useParcaContext, useURLState, type ProfileData} from '@parca/components';
+import {SourceSkeleton, useParcaContext, type ProfileData} from '@parca/components';
 
 import {ExpandOnHover} from '../GraphTooltipArrow/ExpandOnHoverValue';
+import {stringParam} from '../hooks/urlParsers';
 import {alignedUint8Array, truncateStringReverse} from '../utils';
 import {Highlighter, profileAwareRenderer, type LineDataLookup} from './Highlighter';
 import useLineRange from './useSelectedLineRange';
@@ -42,7 +44,7 @@ export const SourceView = React.memo(function SourceView({
   filtered,
   setActionButtons,
 }: SourceViewProps): JSX.Element {
-  const [sourceFileName] = useURLState<string | undefined>('source_filename');
+  const [sourceFileName] = useQueryState('source_filename', stringParam);
   const {isDarkMode, sourceViewContextMenuItems = []} = useParcaContext();
 
   const sourceCode = useMemo(() => {