import { FlashList, FlashListProps } from "@shopify/flash-list";
import React, { forwardRef, useImperativeHandle, useRef } from "react";
import { View, ViewStyle } from "react-native";
import { Text } from "./Text";

type PartialExcept<T, K extends keyof T> = Partial<T> & Pick<T, K>;

export interface HorizontalScrollRef {
  scrollToIndex: (index: number, viewOffset: number) => void;
}

interface HorizontalScrollProps<T>
  extends PartialExcept<
    Omit<FlashListProps<T>, "renderItem">,
    "estimatedItemSize"
  > {
  data?: T[] | null;
  renderItem: (item: T, index: number) => React.ReactNode;
  keyExtractor?: (item: T, index: number) => string;
  containerStyle?: ViewStyle;
  contentContainerStyle?: ViewStyle;
  loadingContainerStyle?: ViewStyle;
  height?: number;
  loading?: boolean;
  extraData?: any;
  noItemsText?: string;
}

export const HorizontalScroll = forwardRef<
  HorizontalScrollRef,
  HorizontalScrollProps<any>
>(
  <T,>(
    {
      data = [],
      keyExtractor,
      renderItem,
      containerStyle,
      contentContainerStyle,
      loadingContainerStyle,
      loading = false,
      height = 164,
      extraData,
      noItemsText,
      ...props
    }: HorizontalScrollProps<T>,
    ref: React.ForwardedRef<HorizontalScrollRef>
  ) => {
    const flashListRef = useRef<FlashList<T>>(null);

    useImperativeHandle(ref!, () => ({
      scrollToIndex: (index: number, viewOffset: number) => {
        flashListRef.current?.scrollToIndex({
          index,
          animated: true,
          viewPosition: 0,
          viewOffset,
        });
      },
    }));

    const renderFlashListItem = ({
      item,
      index,
    }: {
      item: T;
      index: number;
    }) => (
      <View className="mr-2">
        <React.Fragment>{renderItem(item, index)}</React.Fragment>
      </View>
    );

    if (!data || loading) {
      return (
        <View className="px-4 mb-2">
          <View className="bg-neutral-950 h-24 w-full rounded-md mb-2"></View>
          <View className="bg-neutral-950 h-10 w-full rounded-md mb-1"></View>
        </View>
      );
    }

    return (
      <FlashList<T>
        ref={flashListRef}
        data={data}
        extraData={extraData}
        renderItem={renderFlashListItem}
        horizontal
        estimatedItemSize={200}
        showsHorizontalScrollIndicator={false}
        contentContainerStyle={{
          paddingHorizontal: 16,
          ...contentContainerStyle,
        }}
        keyExtractor={keyExtractor}
        ListEmptyComponent={() => (
          <View className="flex-1 justify-center items-center">
            <Text className="text-center text-gray-500">
              {noItemsText || "No data available"}
            </Text>
          </View>
        )}
        {...props}
      />
    );
  }
);
