immich-folio/components/AssetProtection.tsx

'use client';

import { useEffect } from 'react';

interface Props {
  disableRightClick?: boolean;
  disableImageDrag?: boolean;
}

export default function AssetProtection({ disableRightClick, disableImageDrag }: Props) {
  useEffect(() => {
    if (!disableRightClick && !disableImageDrag) return;

    function handleContextMenu(e: MouseEvent) {
      const target = e.target as HTMLElement;
      if (disableRightClick && (target.tagName === 'IMG' || target.closest('img') || target.classList.contains('photo-tile') || target.classList.contains('lightbox-img'))) {
        e.preventDefault();
      }
    }

    function handleDragStart(e: DragEvent) {
      const target = e.target as HTMLElement;
      if (disableImageDrag && (target.tagName === 'IMG' || target.closest('img'))) {
        e.preventDefault();
      }
    }

    if (disableRightClick) {
      document.addEventListener('contextmenu', handleContextMenu);
    }
    if (disableImageDrag) {
      document.addEventListener('dragstart', handleDragStart);
    }

    return () => {
      document.removeEventListener('contextmenu', handleContextMenu);
      document.removeEventListener('dragstart', handleDragStart);
    };
  }, [disableRightClick, disableImageDrag]);

  return null;
}