import { useEffect, useRef } from "react";
import type { RefObject } from "react";
import type { Object3D } from "three";
import { Octree } from "three/addons/math/Octree.js";
import type { OctreeReadyHandler } from "@/types/three";

export function useOctreeGraphNode(
  graphNodeRef: RefObject<Object3D | null>,
  onOctreeReady: OctreeReadyHandler,
  rebuildKey: string | number = 0,
): void {
  const octreeBuilt = useRef(false);

  useEffect(() => {
    octreeBuilt.current = false;
  }, [rebuildKey]);

  useEffect(() => {
    const graphNode = graphNodeRef.current;
    if (octreeBuilt.current || !graphNode) return;
    octreeBuilt.current = true;

    graphNode.updateMatrixWorld(true);

    const octree = new Octree();
    octree.fromGraphNode(graphNode);
    onOctreeReady(octree);
  }, [graphNodeRef, onOctreeReady, rebuildKey]);
}