import { useEffect, useState, useRef } from 'react';
import { getScrollViewportIntersectionRoot } from './useScrollViewport';

/**
 * Track which heading section is currently visible in the viewport
 * Uses Intersection Observer to detect when headings enter/leave view
 */
export function useActiveSection(
  containerRef: React.RefObject<HTMLElement | null>,
  headingCount: number,
  /** Optional explicit scroll element. When provided, takes precedence over
   *  containerRef.current and re-runs the effect when it changes — needed
   *  for OverlayScrollArea, whose viewport is only available after mount. */
  scrollElement?: HTMLElement | null,
) {
  const [activeId, setActiveId] = useState<string | null>(null);
  const observerRef = useRef<IntersectionObserver | null>(null);

  useEffect(() => {
    const contentContainer = containerRef.current ?? scrollElement;
    const viewport = scrollElement ?? contentContainer;
    if (!contentContainer || !viewport) return;
    
    // Only document headings: a heading inside a question card's context has
    // no block id and is not a TOC section.
    const headings = contentContainer.querySelectorAll('[data-block-type="heading"][data-block-id]');
    if (headings.length === 0) return;
    
    // Track which headings are currently intersecting
    const intersectingHeadings = new Map<Element, boolean>();
    
    // Create observer with configuration to detect headings near top of viewport
    observerRef.current = new IntersectionObserver(
      (entries) => {
        // Update intersection state
        entries.forEach(entry => {
          intersectingHeadings.set(entry.target, entry.isIntersecting);
        });
        
        // Find the first intersecting heading (topmost in viewport)
        let topHeading: Element | null = null;
        let topPosition = Infinity;
        
        for (const [heading, isIntersecting] of intersectingHeadings) {
          if (isIntersecting) {
            const rect = heading.getBoundingClientRect();
            if (rect.top < topPosition) {
              topPosition = rect.top;
              topHeading = heading;
            }
          }
        }
        
        // Update active ID
        if (topHeading) {
          const blockId = topHeading.getAttribute('data-block-id');
          if (blockId) {
            setActiveId(blockId);
          }
        }
      },
      {
        root: getScrollViewportIntersectionRoot(viewport),
        rootMargin: '-80px 0px -80% 0px', // Activate when heading is near top
        threshold: [0, 0.1, 0.5, 1.0],
      }
    );
    
    // Observe all headings
    headings.forEach(heading => {
      observerRef.current?.observe(heading);
    });
    
    // Cleanup
    return () => {
      observerRef.current?.disconnect();
    };
  }, [containerRef, headingCount, scrollElement]);
  
  return activeId;
}