/*
Copyright 2026 Element Creations Ltd.

SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial
Please see LICENSE in the repository root for full details.
*/

import {
  useCallback,
  useLayoutEffect,
  useRef,
  useState,
  type FC,
  type Ref,
} from "react";
import { Text } from "@vector-im/compound-web";
import { MicOnIcon } from "@vector-im/compound-design-tokens/assets/web/icons";
import classNames from "classnames";
import { useTranslation } from "react-i18next";

import { distinctUntilChanged, map } from "rxjs";

import styles from "./MicrophoneLevelMeter.module.css";
import { LEVEL_SCALE, type MicrophoneState } from "../state/MicrophoneLevel";
import { observeElementSize$ } from "../utils/elementSize";
import { useMicrophoneLevel } from "./useMicrophoneLevel";

export interface LiveMicrophoneLevelMeterProps {
  /** The microphone to listen to. */
  deviceId: string | undefined;
  /** Whether to hold a capture at all. */
  active: boolean;
  className?: string;
  ref?: Ref<HTMLDivElement>;
}

/**
 * The meter, wired to a live capture. Its own component so a changing level
 * re-renders the meter, not its parent.
 */
export const LiveMicrophoneLevelMeter: FC<LiveMicrophoneLevelMeterProps> = ({
  deviceId,
  active,
  className,
  ref,
}) => {
  const state = useMicrophoneLevel(deviceId, active);
  return <MicrophoneLevelMeter state={state} className={className} ref={ref} />;
};

export interface MicrophoneLevelMeterProps {
  state: MicrophoneState;
  className?: string;
  ref?: Ref<HTMLDivElement>;
}

/** The input level of a microphone, or why there is none. */
export const MicrophoneLevelMeter: FC<MicrophoneLevelMeterProps> = ({
  state,
  className,
  ref,
}) => {
  const { t } = useTranslation();
  // Bars keep one size, so their count follows the width. Starts full so the
  // first paint, and jsdom, draw a whole meter.
  const [barCount, setBarCount] = useState(LEVEL_SCALE);
  const segments = useRef<HTMLDivElement | null>(null);
  const track = useCallback(
    (element: HTMLDivElement | null): (() => void) | undefined => {
      if (element === null) return;
      segments.current = element;
      const subscription = observeElementSize$(element)
        .pipe(
          map(({ width }) => barsThatFit(element, width)),
          distinctUntilChanged(),
        )
        .subscribe(setBarCount);
      return (): void => {
        subscription.unsubscribe();
        segments.current = null;
      };
    },
    [],
  );

  // Drawn straight into the DOM: the level changes many times a second, and
  // re-rendering for each change is what this avoids.
  useLayoutEffect(() => {
    const element = segments.current;
    if (state.type !== "level" || element === null) return;
    const subscription = state.level$.subscribe((level) => {
      element.setAttribute("aria-valuenow", String(level));
      element.setAttribute(
        "aria-valuetext",
        t("microphone_level.value", { level, max: LEVEL_SCALE }),
      );
      // The level is a share of the scale, not a bar count.
      const lit = Math.round((level / LEVEL_SCALE) * barCount);
      Array.from(element.children).forEach((bar, i) =>
        bar.classList.toggle(styles.segmentLit, i < lit),
      );
    });
    return (): void => subscription.unsubscribe();
  }, [state, barCount, t]);

  return (
    <div ref={ref} className={classNames(styles.meter, className)}>
      <MicOnIcon className={styles.icon} aria-hidden />
      {state.type !== "level" ? (
        <Text size="sm" className={styles.message}>
          {state.type === "permission-denied"
            ? t("microphone_level.permission_denied")
            : t("microphone_level.no_device")}
        </Text>
      ) : (
        <div
          ref={track}
          className={styles.segments}
          role="meter"
          aria-label={t("microphone_level.label")}
          aria-valuemin={0}
          aria-valuemax={LEVEL_SCALE}
          // The first paint's value; the effect above keeps it current.
          aria-valuenow={state.level$.value}
          aria-valuetext={t("microphone_level.value", {
            level: state.level$.value,
            max: LEVEL_SCALE,
          })}
        >
          {Array.from({ length: barCount }, (_, i) => (
            <span key={i} aria-hidden className={styles.segment} />
          ))}
        </div>
      )}
    </div>
  );
};

/** How many bars fit across `width`, measured so the stylesheet alone sets their size. */
function barsThatFit(track: HTMLElement, width: number): number {
  const gap = Number.parseFloat(getComputedStyle(track).columnGap);
  const bar = track.firstElementChild?.getBoundingClientRect().width ?? 0;
  // No layout (jsdom): keep the full count.
  if (!(bar > 0) || !(gap >= 0)) return LEVEL_SCALE;
  return Math.max(1, Math.floor((width + gap) / (bar + gap)));
}