import React, { useCallback, useEffect, useRef, useState } from "react";
import classNames from "classnames";
import { useDrag } from "react-use-gesture";
import { useSprings, useTransition, animated } from "@react-spring/web";
import styles from "./GridDemo.module.css";
import useMeasure from "react-use-measure";

export function GridDemo() {
  const tileKey = useRef(0);
  const [stream, setStream] = useState();
  const [tiles, setTiles] = useState([]);
  const tilePositionsRef = useRef([]);
  const draggingTileRef = useRef(null);

  const startWebcam = useCallback(async () => {
    const stream = await navigator.mediaDevices.getUserMedia({ video: true });
    setStream(stream);
    setTiles([{ stream, key: tileKey.current++ }]);
  }, []);

  const addTile = useCallback(() => {
    const newStream = stream.clone();
    setTiles((tiles) => [
      ...tiles,
      { stream: newStream, key: tileKey.current++ },
    ]);
  }, [stream]);

  const removeTile = useCallback(() => {
    setTiles((tiles) => {
      const newArr = [...tiles];
      newArr.pop();
      return newArr;
    });
  }, []);

  const [gridRef, gridBounds] = useMeasure();

  // useEffect(() => {
  //   const newTilePositions = [];
  //   const tileCount = tiles.length;
  //   const { width: gridWidth, height: gridHeight } = gridBounds;
  //   const gap = 8;

  //   if (tileCount > 0) {
  //     const aspectRatio = gridWidth / gridHeight;

  //     let columnCount, rowCount;

  //     if (aspectRatio < 1) {
  //       if (tileCount <= 4) {
  //         columnCount = 1;
  //         rowCount = tileCount;
  //       } else if (tileCount <= 12) {
  //         columnCount = 2;
  //         rowCount = Math.ceil(tileCount / 2);
  //       }
  //     } else {
  //       if (tileCount === 1) {
  //         columnCount = 1;
  //         rowCount = 1;
  //       } else if (tileCount === 2) {
  //         columnCount = 2;
  //         rowCount = 1;
  //       } else if (tileCount <= 4) {
  //         columnCount = 2;
  //         rowCount = 2;
  //       } else if (tileCount <= 6) {
  //         columnCount = 3;
  //         rowCount = 2;
  //       } else if (tileCount <= 8) {
  //         columnCount = 4;
  //         rowCount = 2;
  //       } else if (tileCount <= 10) {
  //         columnCount = 5;
  //         rowCount = 2;
  //       } else if (tileCount <= 12) {
  //         columnCount = 4;
  //         rowCount = 3;
  //       }
  //     }

  //     let tileHeight = Math.round(
  //       (gridHeight - gap * (rowCount + 1)) / rowCount
  //     );
  //     let tileWidth = Math.round(
  //       (gridWidth - gap * (columnCount + 1)) / columnCount
  //     );

  //     const tileAspectRatio = tileWidth / tileHeight;

  //     if (tileAspectRatio > 16 / 9) {
  //       tileWidth = (16 * tileHeight) / 9;
  //     }

  //     for (let i = 0; i < tiles.length; i++) {
  //       const verticalIndex = Math.floor(i / columnCount);
  //       const top = verticalIndex * tileHeight + (verticalIndex + 1) * gap;

  //       let rowItemCount;

  //       if (verticalIndex + 1 === rowCount && tileCount % rowCount !== 0) {
  //         rowItemCount = Math.floor(tileCount / rowCount);
  //       } else {
  //         rowItemCount = Math.ceil(tileCount / rowCount);
  //       }

  //       const horizontalIndex = i % columnCount;
  //       const totalRowGapWidth = (rowItemCount + 1) * gap;
  //       const totalRowTileWidth = rowItemCount * tileWidth;
  //       const rowLeftMargin = Math.round(
  //         (gridWidth - (totalRowTileWidth + totalRowGapWidth)) / 2
  //       );
  //       const left =
  //         tileWidth * horizontalIndex +
  //         rowLeftMargin +
  //         (horizontalIndex + 1) * gap;

  //       newTilePositions.push({
  //         width: tileWidth,
  //         height: tileHeight,
  //         x: left,
  //         y: top,
  //       });
  //     }
  //   }

  //   tilePositionsRef.current = newTilePositions;

  //   console.log("update tile positions", newTilePositions);
  // }, [gridBounds, tiles]);

  const animate = useCallback(
    (index) => {
      const newTilePositions = [];
      const tileCount = tiles.length;
      const { width: gridWidth, height: gridHeight } = gridBounds;
      const gap = 8;

      if (tileCount > 0) {
        const aspectRatio = gridWidth / gridHeight;

        let columnCount, rowCount;

        if (aspectRatio < 1) {
          if (tileCount <= 4) {
            columnCount = 1;
            rowCount = tileCount;
          } else if (tileCount <= 12) {
            columnCount = 2;
            rowCount = Math.ceil(tileCount / 2);
          }
        } else {
          if (tileCount === 1) {
            columnCount = 1;
            rowCount = 1;
          } else if (tileCount === 2) {
            columnCount = 2;
            rowCount = 1;
          } else if (tileCount <= 4) {
            columnCount = 2;
            rowCount = 2;
          } else if (tileCount <= 6) {
            columnCount = 3;
            rowCount = 2;
          } else if (tileCount <= 8) {
            columnCount = 4;
            rowCount = 2;
          } else if (tileCount <= 10) {
            columnCount = 5;
            rowCount = 2;
          } else if (tileCount <= 12) {
            columnCount = 4;
            rowCount = 3;
          }
        }

        let tileHeight = Math.round(
          (gridHeight - gap * (rowCount + 1)) / rowCount
        );
        let tileWidth = Math.round(
          (gridWidth - gap * (columnCount + 1)) / columnCount
        );

        const tileAspectRatio = tileWidth / tileHeight;

        if (tileAspectRatio > 16 / 9) {
          tileWidth = (16 * tileHeight) / 9;
        }

        for (let i = 0; i < tiles.length; i++) {
          const verticalIndex = Math.floor(i / columnCount);
          const top = verticalIndex * tileHeight + (verticalIndex + 1) * gap;

          let rowItemCount;

          if (verticalIndex + 1 === rowCount && tileCount % rowCount !== 0) {
            rowItemCount = Math.floor(tileCount / rowCount);
          } else {
            rowItemCount = Math.ceil(tileCount / rowCount);
          }

          const horizontalIndex = i % columnCount;
          const totalRowGapWidth = (rowItemCount + 1) * gap;
          const totalRowTileWidth = rowItemCount * tileWidth;
          const rowLeftMargin = Math.round(
            (gridWidth - (totalRowTileWidth + totalRowGapWidth)) / 2
          );
          const left =
            tileWidth * horizontalIndex +
            rowLeftMargin +
            (horizontalIndex + 1) * gap;

          newTilePositions.push({
            width: tileWidth,
            height: tileHeight,
            x: left,
            y: top,
          });
        }
      }

      tilePositionsRef.current = newTilePositions;
      const tilePosition = tilePositionsRef.current[index];
      const draggingTile = draggingTileRef.current;
      const dragging =
        draggingTileRef.current && index === draggingTileRef.current.index;

      if (dragging) {
        return {
          width: tilePosition.width,
          height: tilePosition.height,
          x: tilePosition.x + draggingTile.x,
          y: tilePosition.y + draggingTile.y,
          scale: 1.1,
          zIndex: 1,
          shadow: 15,
          immediate: (key) => key === "zIndex" || key === "x" || key === "y",
        };
      } else {
        return {
          ...tilePosition,
          scale: 1,
          zIndex: 0,
          shadow: 1,
          immediate: false,
        };
      }
    },
    [tiles, gridBounds]
  );

  const [springs, api] = useSprings(tiles.length, animate, [tiles, gridBounds]);

  const bind = useDrag(({ args: [index], active, movement: [x, y] }) => {
    if (active) {
      draggingTileRef.current = {
        index,
        x,
        y,
      };
    } else {
      draggingTileRef.current = null;
    }

    api.start(animate);
  });

  return (
    <div className={styles.gridDemo}>
      <div className={styles.buttons}>
        {!stream && <button onClick={startWebcam}>Start Webcam</button>}
        {stream && <button onClick={addTile}>Add Tile</button>}
        {stream && <button onClick={removeTile}>Remove Tile</button>}
      </div>
      <div className={styles.grid} ref={gridRef}>
        {springs.map((style, i) => {
          const tile = tiles[i];

          return (
            <ParticipantTile
              {...bind(i)}
              key={tile.key}
              style={style}
              {...tile}
            />
          );
        })}
      </div>
    </div>
  );
}

function ParticipantTile({ style, stream, ...rest }) {
  const videoRef = useRef();

  useEffect(() => {
    if (stream) {
      videoRef.current.srcObject = stream;
      videoRef.current.play();
    } else {
      videoRef.current.srcObject = null;
    }
  }, [stream]);

  return (
    <animated.div className={styles.participantTile} style={style} {...rest}>
      <video ref={videoRef} playsInline />
    </animated.div>
  );
}