// Copyright (C) 2025 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import './split_panel.scss';
import m from 'mithril';
import {classNames} from '../base/classnames';
import {ResizeHandle} from './resize_handle';

type SplitSizePixels = {
  readonly pixels: number;
};

type SplitSizePercent = {
  readonly percent: number;
};

// Split configuration - either percentage or fixed size in pixels.
type SplitSize = SplitSizePixels | SplitSizePercent;

export interface SplitPanelAttrs {
  // Layout direction. Default is 'horizontal'.
  readonly direction?: 'horizontal' | 'vertical';

  // Controls which panel has the controlled size. Default is 'first'.
  readonly controlledPanel?: 'first' | 'second';

  // Initial split configuration for uncontrolled mode. Only read once in
  // oninit. Ignored if `split` is provided.
  readonly initialSplit?: SplitSize;

  // Controlled split configuration. When provided, the component will use this
  // value directly and the parent must update it via onResize to see changes.
  readonly split?: SplitSize;

  // Minimum size in pixels for each panel
  readonly minSize?: number;

  // Additional CSS class for the root element.
  readonly className?: string;

  // Content for the first panel
  readonly firstPanel: m.Children;

  // Content for the second panel
  readonly secondPanel: m.Children;

  // Callback invoked when the user resizes a panel in both controlled and
  // uncontrolled modes.
  readonly onResize?: (size: number) => void;
}

// Factory function to create SplitPanel instances with their own state
export function SplitPanel(
  vnode: m.Vnode<SplitPanelAttrs>,
): m.Component<SplitPanelAttrs> {
  // Internal state for uncontrolled mode
  let internalPercent = 50;
  let internalFixedPx = 150;
  let containerElement: HTMLElement | undefined;

  const initial = vnode.attrs.initialSplit ?? {percent: 50};
  if ('pixels' in initial) {
    internalFixedPx = initial.pixels;
  } else if ('percent' in initial) {
    internalPercent = initial.percent;
  }

  return {
    view(vnode) {
      const {
        direction = 'horizontal',
        minSize = 50,
        split,
        initialSplit,
        firstPanel,
        secondPanel,
        controlledPanel: panel,
      } = vnode.attrs;

      // Determine if we're in controlled or uncontrolled mode
      const isControlled = split !== undefined;
      const effectiveSplit = split ?? initialSplit ?? {percent: 50};
      const isPixelMode = 'pixels' in effectiveSplit;
      const controlledPanel = panel ?? 'first';

      const containerClasses = classNames(
        'pf-split-panel',
        `pf-split-${direction}`,
        vnode.attrs.className,
      );

      // Get current size - from controlled prop or internal state
      let currentPercent: number;
      let currentPixels: number;
      if (isControlled && isPixelMode) {
        currentPixels = effectiveSplit.pixels;
        currentPercent = 50; // unused in pixel mode
      } else if (isControlled && 'percent' in effectiveSplit) {
        currentPercent = effectiveSplit.percent;
        currentPixels = 150; // unused in percent mode
      } else {
        currentPercent = internalPercent;
        currentPixels = internalFixedPx;
      }

      let firstStyle: Record<string, string>;
      let secondStyle: Record<string, string>;

      // Use CSS min/max width/height to enforce size constraints
      const minProp = direction === 'horizontal' ? 'minWidth' : 'minHeight';
      const maxProp = direction === 'horizontal' ? 'maxWidth' : 'maxHeight';
      const maxSize = `calc(100% - ${minSize}px)`;

      if (isPixelMode) {
        // Pixel mode - one panel fixed, one flexible
        if (controlledPanel === 'first') {
          firstStyle = {
            flex: `0 0 ${currentPixels}px`,
            [minProp]: `${minSize}px`,
            [maxProp]: maxSize,
          };
          secondStyle = {flex: '1 1 0', [minProp]: `${minSize}px`};
        } else {
          firstStyle = {flex: '1 1 0', [minProp]: `${minSize}px`};
          secondStyle = {
            flex: `0 0 ${currentPixels}px`,
            [minProp]: `${minSize}px`,
            [maxProp]: maxSize,
          };
        }
      } else {
        // Percentage mode
        const firstPercent =
          controlledPanel === 'first' ? currentPercent : 100 - currentPercent;
        firstStyle = {
          flex: `0 0 ${firstPercent}%`,
          [minProp]: `${minSize}px`,
          [maxProp]: maxSize,
        };
        secondStyle = {
          flex: `0 0 ${100 - firstPercent}%`,
          [minProp]: `${minSize}px`,
          [maxProp]: maxSize,
        };
      }

      const onResizeAbsolute = (pos: number) => {
        if (!containerElement) return;
        m.redraw();

        const rect = containerElement.getBoundingClientRect();
        const containerSize =
          direction === 'horizontal' ? rect.width : rect.height;

        if (isPixelMode) {
          // Pixel mode
          let newSize: number;
          if (controlledPanel === 'first') {
            newSize = pos;
          } else {
            newSize = containerSize - pos;
          }

          // Clamp to min/max
          newSize = Math.max(
            minSize,
            Math.min(containerSize - minSize, newSize),
          );

          // Update internal state only in uncontrolled mode
          if (!isControlled) {
            internalFixedPx = newSize;
          }

          if (vnode.attrs.onResize) {
            vnode.attrs.onResize(newSize);
          }
        } else {
          // Percentage mode
          let newPercent = (pos / containerSize) * 100;

          // If controlling second panel, invert the percentage
          if (controlledPanel === 'second') {
            newPercent = 100 - newPercent;
          }

          const minPercent = (minSize / containerSize) * 100;
          const maxPercent = 100 - minPercent;
          newPercent = Math.max(minPercent, Math.min(maxPercent, newPercent));

          // Update internal state only in uncontrolled mode
          if (!isControlled) {
            internalPercent = newPercent;
          }

          if (vnode.attrs.onResize) {
            vnode.attrs.onResize(newPercent);
          }
        }
      };

      return m(
        'div',
        {
          class: containerClasses,
          oncreate: (v: m.VnodeDOM) => {
            containerElement = v.dom as HTMLElement;
          },
        },
        [
          m('.pf-split-panel__first', {style: firstStyle}, firstPanel),
          m(ResizeHandle, {
            direction: direction === 'horizontal' ? 'horizontal' : 'vertical',
            onResizeAbsolute,
          }),
          m('.pf-split-panel__second', {style: secondStyle}, secondPanel),
        ],
      );
    },
  };
}