diff --git a/packages/components/src/simple-table/simple-table-header.tsx b/packages/components/src/simple-table/simple-table-header.tsx index 3ebd835cb6..13571bd666 100644 --- a/packages/components/src/simple-table/simple-table-header.tsx +++ b/packages/components/src/simple-table/simple-table-header.tsx @@ -1,15 +1,20 @@ +import type { Ref } from 'react' import React from 'react' import { TableHeader, TableHeaderCell, TableRow } from '../table' import type { TableHeaderProps } from '../table' import { flexRender } from '@tanstack/react-table' import type { HeaderGroup } from '@tanstack/react-table' import { IconArrowDown, IconArrowUp } from '@vtex/shoreline-icons' +import { forwardRef } from '@vtex/shoreline-utils' -export function SimpleTableHeader(props: SimpleTableRowProps) { +export const SimpleTableHeader = forwardRef(function SimpleTableHeader( + props: SimpleTableRowProps, + ref: Ref +) { const { headers, children, ...otherProps } = props return ( - + {headers.map((headerGroup) => ( {headerGroup.headers.map((header) => ( @@ -35,7 +40,7 @@ export function SimpleTableHeader(props: SimpleTableRowProps) { ))} ) -} +}) export interface SimpleTableRowProps extends TableHeaderProps { headers: Array> diff --git a/packages/components/src/simple-table/simple-table-row.tsx b/packages/components/src/simple-table/simple-table-row.tsx index 78b7a72161..06dc0d54b1 100644 --- a/packages/components/src/simple-table/simple-table-row.tsx +++ b/packages/components/src/simple-table/simple-table-row.tsx @@ -1,17 +1,22 @@ -import type { ReactNode } from 'react' +import type { ReactNode, Ref } from 'react' import React, { Fragment } from 'react' import { TableCell, TableRow, type TableRowProps } from '../table' import { flexRender, type Row } from '@tanstack/react-table' import type { NavigationTarget } from '../link-box/link-box-utils' import { Clickable } from '../clickable' import { LinkBox } from '../link-box' +import { forwardRef } from '@vtex/shoreline-utils' -export function SimpleTableRow(props: SimpleTableRowProps) { +export const SimpleTableRow = forwardRef(function SimpleTableRow( + props: SimpleTableRowProps, + ref: Ref +) { const { row, id, rowClick, renderDetail, children, ...otherProps } = props return ( (props: SimpleTableRowProps) { )} ) -} +}) export interface SimpleTableRowProps extends TableRowProps { row: Row