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 { tc } from "@/utils/textTools";
import { Text } from "./common/Text";
import { type OptionGroup, PlatformDropdown } from "./PlatformDropdown";

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

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

  const subtitleStreams = useMemo(() => {
    return source?.MediaStreams?.filter((x) => x.Type === "Subtitle");
  }, [source]);

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

  const optionGroups: OptionGroup[] = useMemo(() => {
    const options = [
      {
        id: "none",
        type: "radio" as const,
        groupId: "subtitle-streams",
        label: t("item_card.none"),
        selected: selected === -1,
      },
      ...(subtitleStreams?.map((subtitle, idx) => ({
        id: `${subtitle.Index || idx}`,
        type: "radio" as const,
        groupId: "subtitle-streams",
        label: subtitle.DisplayTitle || `Subtitle Stream ${idx + 1}`,
        selected: subtitle.Index === selected,
      })) || []),
    ];

    return [
      {
        id: "subtitle-streams",
        title: "Subtitle tracks",
        options,
      },
    ];
  }, [subtitleStreams, selected, t]);

  const handleOptionSelect = (optionId: string) => {
    if (optionId === "none") {
      onChange(-1);
    } else {
      const selectedStream = subtitleStreams?.find(
        (subtitle, idx) => `${subtitle.Index || idx}` === optionId,
      );
      if (
        selectedStream &&
        selectedStream.Index !== undefined &&
        selectedStream.Index !== null
      ) {
        onChange(selectedStream.Index);
      }
    }
    setOpen(false);
  };

  const trigger = (
    <View className='flex flex-col' {...props}>
      <Text numberOfLines={1} className='opacity-50 mb-1 text-xs'>
        {t("item_card.subtitles")}
      </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>
          {selectedSubtitleSteam
            ? tc(selectedSubtitleSteam?.DisplayTitle, 7)
            : t("item_card.none")}
        </Text>
      </TouchableOpacity>
    </View>
  );

  if (Platform.isTV || subtitleStreams?.length === 0) return null;

  return (
    <View
      className='flex col shrink justify-start place-self-start items-start'
      style={{
        minWidth: 60,
        maxWidth: 200,
      }}
    >
      <PlatformDropdown
        groups={optionGroups}
        trigger={trigger}
        title={t("item_card.subtitles")}
        open={open}
        onOpenChange={setOpen}
        onOptionSelect={handleOptionSelect}
        expoUIConfig={{
          hostStyle: { flex: 1 },
        }}
        bottomSheetConfig={{
          enablePanDownToClose: true,
        }}
      />
    </View>
  );
};