import type { MediaSourceInfo } from "@jellyfin/sdk/lib/generated-client/models";
import { useMemo, useState } from "react";
import { useTranslation } from "react-i18next";
import { Platform, TouchableOpacity, View } from "react-native";
import { Text } from "./common/Text";
import { type OptionGroup, PlatformOptionsMenu } from "./PlatformOptionsMenu";

interface Props extends React.ComponentProps<typeof View> {
  source?: MediaSourceInfo;
  onChange: (value: number) => void;
  selected?: number | undefined;
}

export const AudioTrackSelector: React.FC<Props> = ({
  source,
  onChange,
  selected,
  ...props
}) => {
  const isTv = Platform.isTV;
  const [open, setOpen] = useState(false);
  const { t } = useTranslation();

  const audioStreams = useMemo(
    () => source?.MediaStreams?.filter((x) => x.Type === "Audio"),
    [source],
  );

  const selectedAudioSteam = useMemo(
    () => audioStreams?.find((x) => x.Index === selected),
    [audioStreams, selected],
  );

  const optionGroups: OptionGroup[] = useMemo(
    () => [
      {
        id: "audio-streams",
        title: "Audio streams",
        options:
          audioStreams?.map((audio, idx) => ({
            id: `${audio.Index || idx}`,
            type: "radio" as const,
            groupId: "audio-streams",
            label: audio.DisplayTitle || `Audio Stream ${idx + 1}`,
            selected: audio.Index === selected,
          })) || [],
      },
    ],
    [audioStreams, selected],
  );

  const handleOptionSelect = (optionId: string) => {
    const selectedStream = audioStreams?.find(
      (audio, idx) => `${audio.Index || idx}` === optionId,
    );
    if (
      selectedStream &&
      selectedStream.Index !== null &&
      selectedStream.Index !== undefined
    ) {
      onChange(selectedStream.Index);
    }
    setOpen(false);
  };

  const trigger = (
    <View className='flex flex-col' {...props}>
      <Text className='opacity-50 mb-1 text-xs'>{t("item_card.audio")}</Text>
      <TouchableOpacity
        className='bg-neutral-900 h-10 rounded-xl border-neutral-800 border px-3 py-2 flex flex-row items-center justify-between'
        onPress={() => setOpen(true)}
      >
        <Text numberOfLines={1}>{selectedAudioSteam?.DisplayTitle}</Text>
      </TouchableOpacity>
    </View>
  );

  if (isTv) return null;

  return (
    <View
      className='flex shrink'
      style={{
        minWidth: 50,
      }}
    >
      <PlatformOptionsMenu
        groups={optionGroups}
        trigger={trigger}
        title={t("item_card.audio")}
        open={open}
        onOpenChange={setOpen}
        onOptionSelect={handleOptionSelect}
        expoUIConfig={{
          hostStyle: { flex: 1 },
        }}
        bottomSheetConfig={{
          enableDynamicSizing: true,
          enablePanDownToClose: true,
        }}
      />
    </View>
  );
};