import {
  View,
  Text,
  ScrollView,
  StyleSheet,
  Dimensions,
  TouchableOpacity,
} from "react-native";
import React, { useEffect, useMemo, useState } from "react";
import Animated, {
  runOnJS,
  useAnimatedGestureHandler,
  useAnimatedStyle,
  useSharedValue,
  withSpring,
} from "react-native-reanimated";
import { PanGestureHandler } from "react-native-gesture-handler";

const { width } = Dimensions.get("window");

// Child component to respect Rules of Hooks (no hooks in loops)
const DraggableItem = ({ item, gestureHandler, draggingItem, translateX, translateY, onActivate }) => {
  const style = useAnimatedStyle(() => ({
    transform: [
      { translateX: draggingItem.value === item.id ? translateX.value : 0 },
      { translateY: draggingItem.value === item.id ? translateY.value : 0 },
    ],
    zIndex: draggingItem.value === item.id ? 10 : 0,
  }));

  return (
    <PanGestureHandler
      onGestureEvent={gestureHandler}
      onHandlerStateChange={() => onActivate(item.id)}
    >
      <Animated.View style={[styles.item, style]}>
        <Text style={styles.text}>{item.label ?? String(item)}</Text>
      </Animated.View>
    </PanGestureHandler>
  );
};

// Props:
// - sourceItems: array of { id, label, value } or strings
// - initialSelected: array of ids to prefill right side (optional)
// - onChange: callback with array of values (or strings) in right order
const SongStructureDragDrop = ({ sourceItems, initialSelected, onChange }) => {
  const normalize = (arr = []) =>
    arr.map((item, idx) =>
      typeof item === "string"
        ? { id: `${item}-${idx}`, label: item, value: item }
        : item,
    );

  const normalizedSource = useMemo(
    () => normalize(sourceItems || []),
    [sourceItems],
  );

  const [rightItems, setRightItems] = useState([]);

  // Compare arrays by item id and order to avoid unnecessary state churn
  const sameById = (a = [], b = []) => {
    if (a === b) return true;
    if (!a || !b) return false;
    if (a.length !== b.length) return false;
    for (let idx = 0; idx < a.length; idx++) {
      if (a[idx]?.id !== b[idx]?.id) return false;
    }
    return true;
  };

  useEffect(() => {
    const selected = Array.isArray(initialSelected) ? initialSelected : [];
    const selectedSet = new Set(selected);
    const nextRight = normalizedSource.filter((i) => selectedSet.has(i.id));
    setRightItems((prev) => (sameById(prev, nextRight) ? prev : nextRight));
  }, [normalizedSource, initialSelected]);

  const leftItems = useMemo(() => {
    if (!rightItems || rightItems.length === 0) return normalizedSource;
    const rightIds = new Set(rightItems.map((item) => item.id));
    return normalizedSource.filter((item) => !rightIds.has(item.id));
  }, [normalizedSource, rightItems]);

  const draggingItem = useSharedValue(null);
  const translateX = useSharedValue(0);
  const translateY = useSharedValue(0);

  const onActivate = (id) => {
    draggingItem.value = id;
  };

  const moveItem = (itemId) => {
    const item = normalizedSource.find((i) => i.id === itemId);
    if (!item) return;
    setRightItems((current) => {
      if (current.some((i) => i.id === itemId)) return current;
      const next = [...current, item];
      onChange?.(next.map((i) => i.value ?? i.label));
      return next;
    });
  };

  const removeItem = (itemId) => {
    setRightItems((current) => {
      if (!current.some((i) => i.id === itemId)) return current;
      const next = current.filter((i) => i.id !== itemId);
      onChange?.(next.map((i) => i.value ?? i.label));
      return next;
    });
  };

  const gestureHandler = useAnimatedGestureHandler({
    onStart: (_, ctx) => {
      ctx.startX = translateX.value;
      ctx.startY = translateY.value;
    },
    onActive: (event, ctx) => {
      translateX.value = ctx.startX + event.translationX;
      translateY.value = ctx.startY + event.translationY;
    },
    onEnd: () => {
      if (translateX.value > width / 2) {
        runOnJS(moveItem)(draggingItem.value);
      }
      translateX.value = withSpring(0);
      translateY.value = withSpring(0);
      draggingItem.value = null;
    },
  });

  return (
    <View style={styles.scrollWrapper}>
      <ScrollView style={{ ...styles.scroll, zIndex: 2 }}>
        <Text style={styles.title}>Left</Text>
        {leftItems.map((item) => (
          <DraggableItem
            key={item.id}
            item={item}
            gestureHandler={gestureHandler}
            draggingItem={draggingItem}
            translateX={translateX}
            translateY={translateY}
            onActivate={onActivate}
          />
        ))}
      </ScrollView>

      <ScrollView style={{ ...styles.scroll, zIndex: -1 }}>
        <Text style={styles.title}>Right</Text>
        <Text style={styles.helpText}>Appuyez pour retirer un élément</Text>
        {rightItems.map((item) => (
          <TouchableOpacity
            key={item.id}
            style={styles.selectedItem}
            onPress={() => removeItem(item.id)}
            activeOpacity={0.7}
            accessibilityRole="button"
            accessibilityLabel={`Retirer ${item.label ?? String(item)}`}
          >
            <Text style={styles.text}>{item.label ?? String(item)}</Text>
            <Text style={styles.removeHint}>Retirer</Text>
          </TouchableOpacity>
        ))}
      </ScrollView>
    </View>
  );
};

export default SongStructureDragDrop;

const styles = StyleSheet.create({
  scrollWrapper: {
    flexDirection: "row",
    justifyContent: "space-around",
  },
  scroll: {
    width: width / 2.2,
    height: "90%",
    backgroundColor: "#f0f0f0",
    borderRadius: 10,
    margin: 5,
    padding: 10,
  },
  title: {
    fontWeight: "bold",
    fontSize: 16,
    marginBottom: 10,
  },
  item: {
    backgroundColor: "#aaf",
    padding: 15,
    marginVertical: 5,
    borderRadius: 10,
  },
  selectedItem: {
    backgroundColor: "#bdf",
    padding: 15,
    marginVertical: 5,
    borderRadius: 10,
    alignItems: "center",
  },
  helpText: {
    fontSize: 12,
    color: "#666",
    marginBottom: 8,
    textAlign: "center",
  },
  removeHint: {
    fontSize: 12,
    color: "#555",
    marginTop: 4,
  },
  text: {
    color: "#333",
    textAlign: "center",
  },
});