All files / client/src/components/ui unified-scroll-area.tsx

88.23% Statements 75/85
71.42% Branches 45/63
80% Functions 12/15
92% Lines 69/75

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208                                          8x           112x 32x       65x 65x 65x 65x 65x     65x       65x                                                 356x 356x 356x   4x 4x 1x 1x 1x 1x 1x 1x 1x 1x 1x       4x         1x 1x 1x 1x 1x 1x 1x 1x 1x                                             8x                               182x 182x 182x 182x 65x   182x 112x 112x 112x     182x 117x 61x 61x     117x 117x 61x     117x 117x 117x 61x 61x 61x       182x 182x 182x   182x                       2x 2x     1x 1x 1x 1x 1x 1x                        
import * as React from "react";
import { cn } from "@/lib/utils";
 
type ScrollOrientation = "horizontal" | "vertical" | "both";
type ScrollbarVisibility = "hover" | "always";
 
interface ScrollMetrics {
  horizontal: { visible: boolean; size: number; offset: number };
  vertical: { visible: boolean; size: number; offset: number };
}
 
interface UnifiedScrollAreaProps extends React.HTMLAttributes<HTMLDivElement> {
  readonly orientation?: ScrollOrientation;
  readonly scrollbarVisibility?: ScrollbarVisibility;
  readonly viewportClassName?: string;
  readonly viewportProps?: React.HTMLAttributes<HTMLDivElement>;
  readonly viewportRef?: React.Ref<HTMLDivElement>;
  readonly viewportTestId?: string;
  readonly shiftWheelHorizontal?: boolean;
}
 
const EMPTY_METRICS: ScrollMetrics = {
  horizontal: { visible: false, size: 100, offset: 0 },
  vertical: { visible: false, size: 100, offset: 0 },
};
 
function assignRef<T>(ref: React.Ref<T> | undefined, value: T | null): void {
  Iif (typeof ref === "function") ref(value);
  else if (ref) (ref as React.MutableRefObject<T | null>).current = value;
}
 
function readMetrics(viewport: HTMLDivElement): ScrollMetrics {
  const horizontalMax = viewport.scrollWidth - viewport.clientWidth;
  const verticalMax = viewport.scrollHeight - viewport.clientHeight;
  const horizontalVisible = horizontalMax > 1;
  const verticalVisible = verticalMax > 1;
  const horizontalSize = horizontalVisible
    ? Math.max(8, (viewport.clientWidth / viewport.scrollWidth) * 100)
    : 100;
  const verticalSize = verticalVisible
    ? Math.max(8, (viewport.clientHeight / viewport.scrollHeight) * 100)
    : 100;
 
  return {
    horizontal: {
      visible: horizontalVisible,
      size: horizontalSize,
      offset: horizontalVisible
        ? (viewport.scrollLeft / horizontalMax) * (100 - horizontalSize)
        : 0,
    },
    vertical: {
      visible: verticalVisible,
      size: verticalSize,
      offset: verticalVisible
        ? (viewport.scrollTop / verticalMax) * (100 - verticalSize)
        : 0,
    },
  };
}
 
interface UnifiedScrollbarProps {
  readonly orientation: "horizontal" | "vertical";
  readonly viewport: HTMLDivElement | null;
  readonly metric: ScrollMetrics["horizontal"];
}
 
function UnifiedScrollbar({ orientation, viewport, metric }: UnifiedScrollbarProps) {
  const dragOffsetRef = React.useRef(0);
  const draggingRef = React.useRef(false);
  if (!metric.visible) return null;
 
  const isHorizontal = orientation === "horizontal";
  const moveToPointer = (event: React.PointerEvent<HTMLDivElement>, preserveOffset: boolean) => {
    Iif (!viewport) return;
    const track = event.currentTarget;
    const rect = track.getBoundingClientRect();
    const trackSize = isHorizontal ? rect.width : rect.height;
    const thumbSize = trackSize * metric.size / 100;
    const coordinate = isHorizontal ? event.clientX - rect.left : event.clientY - rect.top;
    const offset = preserveOffset ? dragOffsetRef.current : thumbSize / 2;
    const ratio = Math.max(0, Math.min(1, (coordinate - offset) / Math.max(1, trackSize - thumbSize)));
    if (isHorizontal) viewport.scrollLeft = ratio * (viewport.scrollWidth - viewport.clientWidth);
    else Eviewport.scrollTop = ratio * (viewport.scrollHeight - viewport.clientHeight);
  };
 
  return (
    <div
      aria-hidden="true"
      className={cn("unified-scrollbar", `unified-scrollbar--${orientation}`)}
      onPointerDown={(event) => {
        event.preventDefault();
        const thumb = (event.target as HTMLElement).closest(".unified-scrollbar__thumb");
        const rect = event.currentTarget.getBoundingClientRect();
        const thumbStart = (isHorizontal ? rect.width : rect.height) * metric.offset / 100;
        const coordinate = isHorizontal ? event.clientX - rect.left : event.clientY - rect.top;
        dragOffsetRef.current = thumb ? coordinate - thumbStart : 0;
        draggingRef.current = true;
        event.currentTarget.setPointerCapture?.(event.pointerId);
        moveToPointer(event, Boolean(thumb));
      }}
      onPointerMove={(event) => {
        if (draggingRef.current) moveToPointer(event, true);
      }}
      onPointerUp={(event) => {
        draggingRef.current = false;
        if (event.currentTarget.hasPointerCapture?.(event.pointerId)) {
          event.currentTarget.releasePointerCapture?.(event.pointerId);
        }
      }}
      onPointerCancel={() => { draggingRef.current = false; }}
    >
      <span
        className="unified-scrollbar__thumb"
        style={isHorizontal
          ? { left: `${metric.offset}%`, width: `${metric.size}%` }
          : { height: `${metric.size}%`, top: `${metric.offset}%` }}
      />
    </div>
  );
}
 
export const UnifiedScrollArea = React.forwardRef<HTMLDivElement, UnifiedScrollAreaProps>(
  function UnifiedScrollArea(
    {
      children,
      className,
      orientation = "both",
      scrollbarVisibility = "hover",
      shiftWheelHorizontal = true,
      viewportClassName,
      viewportProps,
      viewportRef,
      viewportTestId,
      ...rootProps
    },
    forwardedRef,
  ) {
    const internalViewportRef = React.useRef<HTMLDivElement | null>(null);
    const [viewport, setViewport] = React.useState<HTMLDivElement | null>(null);
    const [metrics, setMetrics] = React.useState<ScrollMetrics>(EMPTY_METRICS);
    const updateMetrics = React.useCallback(() => {
      Eif (internalViewportRef.current) setMetrics(readMetrics(internalViewportRef.current));
    }, []);
    const setViewportRef = React.useCallback((node: HTMLDivElement | null) => {
      internalViewportRef.current = node;
      setViewport(node);
      assignRef(viewportRef, node);
    }, [viewportRef]);
 
    React.useLayoutEffect(() => {
      if (!viewport) return;
      updateMetrics();
      const resizeObserver = typeof ResizeObserver === "undefined"
        ? null
        : new ResizeObserver(updateMetrics);
      resizeObserver?.observe(viewport);
      if (viewport.firstElementChild) resizeObserver?.observe(viewport.firstElementChild);
      const mutationObserver = typeof MutationObserver === "undefined"
        ? null
        : new MutationObserver(updateMetrics);
      mutationObserver?.observe(viewport, { childList: true, subtree: true, characterData: true });
      globalThis.addEventListener("resize", updateMetrics);
      return () => {
        resizeObserver?.disconnect();
        mutationObserver?.disconnect();
        globalThis.removeEventListener("resize", updateMetrics);
      };
    }, [children, updateMetrics, viewport]);
 
    const allowHorizontal = orientation !== "vertical";
    const allowVertical = orientation !== "horizontal";
    const { onScroll, onWheel, className: viewportPropsClassName, ...restViewportProps } = viewportProps ?? {};
 
    return (
      <div
        ref={forwardedRef}
        className={cn("unified-scroll-area", className)}
        data-scrollbar-visibility={scrollbarVisibility}
        {...rootProps}
      >
        <div
          ref={setViewportRef}
          data-testid={viewportTestId}
          className={cn("unified-scroll-area__viewport", viewportClassName, viewportPropsClassName)}
          onScroll={(event) => {
            updateMetrics();
            onScroll?.(event);
          }}
          onWheel={(event) => {
            onWheel?.(event);
            Iif (event.defaultPrevented || !allowHorizontal || !shiftWheelHorizontal || !event.shiftKey) return;
            event.currentTarget.scrollLeft += event.deltaY || event.deltaX;
            event.preventDefault();
            event.stopPropagation();
            updateMetrics();
          }}
          {...restViewportProps}
        >
          {children}
        </div>
        {allowHorizontal && <UnifiedScrollbar orientation="horizontal" viewport={viewport} metric={metrics.horizontal} />}
        {allowVertical && <UnifiedScrollbar orientation="vertical" viewport={viewport} metric={metrics.vertical} />}
      </div>
    );
  },
);