Spaces:
Running
Running
File size: 2,039 Bytes
5621ad9 |
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 |
import { useSearchParams, useLocation } from "react-router-dom";
import { useCallback, useEffect } from "react";
/**
* Hook personnalisé pour gérer l'état de l'URL
* @returns {Object} Méthodes et données pour gérer l'état de l'URL
*/
export const useUrlState = () => {
const [searchParams, setSearchParams] = useSearchParams();
const location = useLocation();
// Convertit les searchParams en objet
const getParams = useCallback(() => {
const params = {};
for (const [key, value] of searchParams.entries()) {
if (key === "categories") {
params[key] = value.split(",").filter(Boolean);
} else if (key === "arena") {
params[key] = value === "true";
} else {
params[key] = value;
}
}
return params;
}, [searchParams]);
// Met à jour les paramètres d'URL
const updateParams = useCallback(
(newParams) => {
const updatedParams = new URLSearchParams(searchParams);
Object.entries(newParams).forEach(([key, value]) => {
if (value && (Array.isArray(value) ? value.length > 0 : true)) {
if (Array.isArray(value)) {
updatedParams.set(key, value.join(","));
} else {
updatedParams.set(key, value.toString());
}
} else {
updatedParams.delete(key);
}
});
setSearchParams(updatedParams);
},
[searchParams, setSearchParams]
);
// Réinitialise tous les paramètres d'URL
const resetParams = useCallback(() => {
setSearchParams(new URLSearchParams());
}, [setSearchParams]);
// Gestion de l'intégration avec Hugging Face
useEffect(() => {
const isHFSpace = window.location !== window.parent.location;
if (!isHFSpace) return;
window.parent.postMessage(
{
queryString: location.search,
hash: location.hash,
},
"https://huggingface.co"
);
}, [location]);
return {
params: getParams(),
updateParams,
resetParams,
searchParams,
location,
};
};
|