Adjusted use of BaseComponent and always forwarding ref to it

Author: rubenthoms

frontend/src/framework/components/RealizationPicker/realizationPicker.tsx

@@ -5,17 +5,23 @@ import { resolveClassNames } from "@lib/utils/resolveClassNames";
 export type BaseComponentProps = {
     disabled?: boolean;
     children?: React.ReactNode;
+    className?: string;
+    style?: React.CSSProperties;
 };
 
 export const BaseComponent = React.forwardRef((props: BaseComponentProps, ref: React.ForwardedRef<HTMLDivElement>) => {
     return (
         <div
             ref={ref}
-            className={resolveClassNames({
-                "opacity-50": props.disabled,
-                "pointer-events-none": props.disabled,
-                "cursor-default": props.disabled,
-            })}
+            className={resolveClassNames(
+                {
+                    "opacity-50": props.disabled,
+                    "pointer-events-none": props.disabled,
+                    "cursor-default": props.disabled,
+                },
+                props.className ?? ""
+            )}
+            style={props.style}
         >
             {props.children}
         </div>

frontend/src/lib/components/BaseComponent/baseComponent.tsx

@@ -11,10 +11,15 @@ export type ButtonProps = {
     endIcon?: React.ReactNode;
     color?: "primary" | "danger" | "success" | "secondary";
     size?: "small" | "medium" | "large";
+    buttonRef?: React.Ref<HTMLButtonElement>;
 } & ButtonUnstyledProps;
 
-export const Button = React.forwardRef((props: ButtonProps, ref: React.ForwardedRef<HTMLButtonElement>) => {
-    const { disabled, variant, children, startIcon, endIcon, color, ...rest } = props;
+function ButtonComponent(props: ButtonProps, ref: React.ForwardedRef<HTMLDivElement>) {
+    const { disabled, variant, children, startIcon, endIcon, color, buttonRef, ...rest } = props;
+
+    const internalRef = React.useRef<HTMLButtonElement>(null);
+    React.useImperativeHandle<HTMLButtonElement | null, HTMLButtonElement | null>(buttonRef, () => internalRef.current);
+
     const classNames = [
         "inline-flex",
         "items-center",
@@ -73,10 +78,10 @@ export const Button = React.forwardRef((props: ButtonProps, ref: React.Forwarded
     );
 
     return (
-        <BaseComponent disabled={disabled}>
+        <BaseComponent disabled={disabled} ref={ref}>
             <ButtonUnstyled
                 {...rest}
-                ref={ref}
+                ref={buttonRef}
                 slotProps={{
                     root: {
                         className: resolveClassNames(...classNames),
@@ -87,6 +92,6 @@ export const Button = React.forwardRef((props: ButtonProps, ref: React.Forwarded
             </ButtonUnstyled>
         </BaseComponent>
     );
-});
+}
 
-Button.displayName = "Button";
+export const Button = React.forwardRef(ButtonComponent);

frontend/src/lib/components/Button/button.tsx

@@ -15,7 +15,7 @@ export type CheckboxProps = {
     onChange?: (event: React.ChangeEvent<HTMLInputElement>, checked: boolean) => void;
 } & BaseComponentProps;
 
-export const Checkbox: React.FC<CheckboxProps> = (props) => {
+function CheckboxComponent(props: CheckboxProps, ref: React.ForwardedRef<HTMLDivElement>) {
     const { onChange } = props;
 
     const [checked, setChecked] = React.useState<boolean>(props.checked ?? false);
@@ -36,36 +36,34 @@ export const Checkbox: React.FC<CheckboxProps> = (props) => {
     );
 
     return (
-        <BaseComponent disabled={props.disabled}>
-            <div className="flex gap-2 items-center">
-                <input
-                    id={props.id ?? id.current}
-                    name={props.name}
-                    ref={(el) => el && (el.indeterminate = props.indeterminate ?? false)}
-                    type="checkbox"
-                    checked={checked}
-                    onChange={handleChange}
-                    className={resolveClassNames(
-                        "w-4",
-                        "h-4",
-                        "text-blue-600",
-                        "border-gray-300",
-                        "rounded",
-                        "focus:ring-blue-500",
-                        "cursor-pointer"
-                    )}
-                />
-                {props.label && (
-                    <label
-                        htmlFor={props.id ?? id.current}
-                        className={resolveClassNames("block", "text-gray-900", "cursor-pointer")}
-                    >
-                        {props.label}
-                    </label>
+        <BaseComponent ref={ref} disabled={props.disabled} className="flex gap-2 items-center">
+            <input
+                id={props.id ?? id.current}
+                name={props.name}
+                ref={(el) => el && (el.indeterminate = props.indeterminate ?? false)}
+                type="checkbox"
+                checked={checked}
+                onChange={handleChange}
+                className={resolveClassNames(
+                    "w-4",
+                    "h-4",
+                    "text-blue-600",
+                    "border-gray-300",
+                    "rounded",
+                    "focus:ring-blue-500",
+                    "cursor-pointer"
                 )}
-            </div>
+            />
+            {props.label && (
+                <label
+                    htmlFor={props.id ?? id.current}
+                    className={resolveClassNames("block", "text-gray-900", "cursor-pointer")}
+                >
+                    {props.label}
+                </label>
+            )}
         </BaseComponent>
     );
-};
+}
 
-Checkbox.displayName = "Checkbox";
+export const Checkbox = React.forwardRef(CheckboxComponent);

frontend/src/lib/components/Checkbox/checkbox.tsx

@@ -13,7 +13,7 @@ export type CollapsibleGroupProps = {
     onChange?: (expanded: boolean) => void;
 } & BaseComponentProps;
 
-export const CollapsibleGroup: React.FC<CollapsibleGroupProps> = (props) => {
+function CollapsibleGroupComponent(props: CollapsibleGroupProps, ref: React.ForwardedRef<HTMLDivElement>) {
     const [expanded, setExpanded] = React.useState(props.expanded ?? false);
     const [prevExpanded, setPrevExpanded] = React.useState<boolean | undefined>(props.expanded);
 
@@ -28,30 +28,28 @@ export const CollapsibleGroup: React.FC<CollapsibleGroupProps> = (props) => {
     };
 
     return (
-        <BaseComponent disabled={props.disabled}>
-            <div className="shadow">
-                <div
-                    className={resolveClassNames(
-                        "flex flex-row justify-between items-center bg-slate-100 cursor-pointer p-2 select-none gap-2",
-                        { "border-b": expanded }
-                    )}
-                    onClick={handleClick}
-                    title={expanded ? "Collapse" : "Expand"}
-                >
-                    {props.icon && React.cloneElement(props.icon, { className: "w-4 h-4" })}
-                    <h3 className="text-sm font-semibold flex-grow leading-none">{props.title}</h3>
-                    {expanded ? <ExpandLess fontSize="small" /> : <ExpandMore fontSize="small" />}
-                </div>
-                <div
-                    className={resolveClassNames("p-2", {
-                        hidden: !expanded,
-                    })}
-                >
-                    {props.children}
-                </div>
+        <BaseComponent ref={ref} disabled={props.disabled} className="shadow">
+            <div
+                className={resolveClassNames(
+                    "flex flex-row justify-between items-center bg-slate-100 cursor-pointer p-2 select-none gap-2",
+                    { "border-b": expanded }
+                )}
+                onClick={handleClick}
+                title={expanded ? "Collapse" : "Expand"}
+            >
+                {props.icon && React.cloneElement(props.icon, { className: "w-4 h-4" })}
+                <h3 className="text-sm font-semibold flex-grow leading-none">{props.title}</h3>
+                {expanded ? <ExpandLess fontSize="small" /> : <ExpandMore fontSize="small" />}
+            </div>
+            <div
+                className={resolveClassNames("p-2", {
+                    hidden: !expanded,
+                })}
+            >
+                {props.children}
             </div>
         </BaseComponent>
     );
-};
+}
 
-CollapsibleGroup.displayName = "CollapsibleGroup";
+export const CollapsibleGroup = React.forwardRef(CollapsibleGroupComponent);

frontend/src/lib/components/CollapsibleGroup/collapsibleGroup.tsx

@@ -78,7 +78,7 @@ function makeOptionListItems<TValue>(options: DropdownOption<TValue>[]): OptionL
     return optionsWithSeperators;
 }
 
-export function Dropdown<TValue = string>(props: DropdownProps<TValue>) {
+function DropdownComponent<TValue = string>(props: DropdownProps<TValue>, ref: React.ForwardedRef<HTMLDivElement>) {
     const { onChange } = props;
 
     const valueWithDefault = props.value ?? null;
@@ -103,7 +103,7 @@ export function Dropdown<TValue = string>(props: DropdownProps<TValue>) {
     const [startIndex, setStartIndex] = React.useState<number>(0);
     const [keyboardFocus, setKeyboardFocus] = React.useState<boolean>(false);
 
-    const inputRef = React.useRef<HTMLInputElement>(null);
+    const inputRef = React.useRef<HTMLDivElement>(null);
     const dropdownRef = React.useRef<HTMLDivElement>(null);
     const debounceTimerRef = React.useRef<ReturnType<typeof setTimeout> | null>(null);
 
@@ -482,7 +482,7 @@ export function Dropdown<TValue = string>(props: DropdownProps<TValue>) {
     }
 
     return (
-        <BaseComponent disabled={props.disabled}>
+        <BaseComponent ref={ref} disabled={props.disabled}>
             <div style={{ width: props.width }} id={props.wrapperId} className="flex hover input-comp rounded">
                 <div className="flex-grow">
                     <Input
@@ -569,6 +569,10 @@ export function Dropdown<TValue = string>(props: DropdownProps<TValue>) {
     );
 }
 
+export const Dropdown = React.forwardRef(DropdownComponent) as <TValue = string>(
+    props: DropdownProps<TValue> & { ref?: React.Ref<HTMLDivElement> }
+) => React.ReactElement;
+
 type OptionProps<TValue> = DropdownOption<TValue> & {
     isSelected: boolean;
     isFocused: boolean;
@@ -613,5 +617,3 @@ function SeparatorItem(props: { text: string }): React.ReactNode {
         </div>
     );
 }
-
-Dropdown.displayName = "Dropdown";

frontend/src/lib/components/Dropdown/dropdown.tsx

@@ -2,11 +2,14 @@ import React from "react";
 
 import { Button, ButtonProps } from "../Button/button";
 
-export type HoldPressedIntervalCallbackButtonProps = ButtonProps & {
+export type HoldPressedIntervalCallbackButtonProps = Omit<ButtonProps, "ref"> & {
     onHoldPressedIntervalCallback: () => void;
 };
 
-export function HoldPressedIntervalCallbackButton(props: HoldPressedIntervalCallbackButtonProps): React.ReactNode {
+function HoldPressedIntervalCallbackButtonComponent(
+    props: HoldPressedIntervalCallbackButtonProps,
+    ref: React.ForwardedRef<HTMLDivElement>
+): React.ReactNode {
     const { onHoldPressedIntervalCallback, ...other } = props;
 
     const timeoutRef = React.useRef<ReturnType<typeof setTimeout> | null>(null);
@@ -39,10 +42,13 @@ export function HoldPressedIntervalCallbackButton(props: HoldPressedIntervalCall
     return (
         <Button
             {...other}
+            ref={ref}
             onClick={handleClick}
             onPointerDown={handlePointerDown}
             onPointerUp={handlePointerUp}
             onPointerLeave={handlePointerUp}
         />
     );
 }
+
+export const HoldPressedIntervalCallbackButton = React.forwardRef(HoldPressedIntervalCallbackButtonComponent);