// Copyright (c) Meta Platforms, Inc. and affiliates.

'use client';

/**
 * @file useTableCellStyles.ts
 * @input React, StyleX, TableContext, theme tokens
 * @output Exports shared cell styling utilities
 * @position Utility layer; consumed by TableCell and TableHeaderCell
 *
 * Consolidates the shared pattern of building divider style arrays
 * and merging consumer xstyle. Eliminates structural duplication
 * between body cells and header cells.
 *
 * SYNC: When modified, update these files to stay in sync:
 * - /packages/core/src/Table/TableCell.tsx
 * - /packages/core/src/Table/TableHeaderCell.tsx
 */
import { use } from 'react';
import type { StyleXStyles } from "../theme/types.js";
import { TableContext, type TableContextValue } from "./TableContext.js";

// =============================================================================
// Shared Style Builders
// =============================================================================

/**
 * Build the divider styles array for a cell based on context.
 * Shared between body cells and header cells — both apply row/column
 * dividers the same way.
 */
export function buildDividerStyles(ctx: TableContextValue, dividerRowStyle: StyleXStyles, dividerColumnStyle: StyleXStyles): StyleXStyles[] {
  const result: StyleXStyles[] = [];
  if (ctx.dividers === 'rows' || ctx.dividers === 'grid') {
    result.push(dividerRowStyle);
  }
  if (ctx.dividers === 'columns' || ctx.dividers === 'grid') {
    result.push(dividerColumnStyle);
  }
  return result;
}

/**
 * Merge consumer xstyle (single or array) into a styles array.
 * Handles the polymorphic xstyle prop that both cell components accept.
 */
export function mergeXStyle(styles: StyleXStyles[], xstyle: StyleXStyles | StyleXStyles[] | undefined): StyleXStyles[] {
  if (!xstyle) {
    return styles;
  }
  if (Array.isArray(xstyle)) {
    return [...styles, ...xstyle];
  }
  return [...styles, xstyle];
}

/**
 * Read the TableContext. Returns null when outside Table,
 * signaling that the component should render unstyled.
 */
export function useTableContext(): TableContextValue | null {
  return use(TableContext);
}