import _ from 'lodash';
import { ComponentType, forwardRef, PropsWithChildren } from 'react';
import { NewBBoxClass } from '../NewBBox';
import {
  Constraints,
  Measure,
  Placeable,
  Layout,
  useBluefishLayout,
  withBluefish,
  LayoutFn,
  withBluefishFn,
  NewPlaceable,
} from '../bluefish';

export type PaddingProps = { left: number; right: number; top: number; bottom: number } | { all: number };

const paddingMeasurePolicy =
  (options: PaddingProps): Measure =>
  (measurables, constraints: Constraints) => {
    if (measurables.length !== 1) {
      throw new Error('Padding must have exactly one child');
    }

    const { left, right, top, bottom } =
      'all' in options ? { left: options.all, right: options.all, top: options.all, bottom: options.all } : options;
    const childConstraints = {
      width: constraints.width ? constraints.width - left - right : undefined,
      height: constraints.height ? constraints.height - top - bottom : undefined,
    };
    const placeables = measurables.map((measurable) => measurable.measure(constraints));

    placeables[0].left = left;
    placeables[0].top = top;

    return {
      left: 0,
      top: 0,
      width: placeables[0].width! + left + right,
      height: placeables[0].height! + top + bottom,
    };
  };

export const Padding = LayoutFn((props: PaddingProps) => paddingMeasurePolicy(props));