Spaces:
Running
Running
File size: 3,466 Bytes
16ab111 6bc7874 16ab111 6bc7874 16ab111 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
import React, {
createContext,
useState,
useEffect,
ReactNode,
useCallback,
} from "react";
import { processDroppedFiles } from "@/lib/UrdfDragAndDrop";
import { useUrdf } from "@/hooks/useUrdf";
export type DragAndDropContextType = {
isDragging: boolean;
setIsDragging: (isDragging: boolean) => void;
handleDrop: (e: DragEvent) => Promise<void>;
};
export const DragAndDropContext = createContext<
DragAndDropContextType | undefined
>(undefined);
interface DragAndDropProviderProps {
children: ReactNode;
}
export const DragAndDropProvider: React.FC<DragAndDropProviderProps> = ({
children,
}) => {
const [isDragging, setIsDragging] = useState(false);
// Get the Urdf context
const { urdfProcessor, processUrdfFiles } = useUrdf();
const handleDragOver = (e: DragEvent) => {
e.preventDefault();
e.stopPropagation();
};
const handleDragEnter = (e: DragEvent) => {
e.preventDefault();
e.stopPropagation();
setIsDragging(true);
};
const handleDragLeave = (e: DragEvent) => {
e.preventDefault();
e.stopPropagation();
// Only set isDragging to false if we're leaving the document
// This prevents flickering when moving between elements
if (!e.relatedTarget || !(e.relatedTarget as Element).closest("html")) {
setIsDragging(false);
}
};
const handleDrop = useCallback(
async (e: DragEvent) => {
e.preventDefault();
e.stopPropagation();
setIsDragging(false);
console.log("🔄 DragAndDropContext: Drop event detected");
if (!e.dataTransfer || !urdfProcessor) {
console.error("❌ No dataTransfer or urdfProcessor available");
return;
}
try {
console.log("🔍 Processing dropped files with urdfProcessor");
// Process files first
const { availableModels, files } = await processDroppedFiles(
e.dataTransfer,
urdfProcessor
);
// Delegate further processing to UrdfContext
await processUrdfFiles(files, availableModels);
} catch (error) {
console.error("❌ Error in handleDrop:", error);
}
},
[urdfProcessor, processUrdfFiles]
);
// Set up global event listeners
useEffect(() => {
document.addEventListener("dragover", handleDragOver);
document.addEventListener("dragenter", handleDragEnter);
document.addEventListener("dragleave", handleDragLeave);
document.addEventListener("drop", handleDrop);
return () => {
document.removeEventListener("dragover", handleDragOver);
document.removeEventListener("dragenter", handleDragEnter);
document.removeEventListener("dragleave", handleDragLeave);
document.removeEventListener("drop", handleDrop);
};
}, [handleDrop]); // Re-register when handleDrop changes
return (
<DragAndDropContext.Provider
value={{
isDragging,
setIsDragging,
handleDrop,
}}
>
{children}
{isDragging && (
<div className="fixed inset-0 bg-primary/10 pointer-events-none z-50 flex items-center justify-center">
<div className="bg-background p-8 rounded-lg shadow-lg text-center">
<div className="text-3xl font-bold mb-4">Drop Urdf Files Here</div>
<p className="text-muted-foreground">
Release to upload your robot model
</p>
</div>
</div>
)}
</DragAndDropContext.Provider>
);
};
|