91 lines
2.8 KiB
TypeScript
91 lines
2.8 KiB
TypeScript
import { QueryClient, QueryFunction } from 'react-query';
|
|
import { getJSON, isNumeric } from '@utils';
|
|
import { matchSorter } from 'match-sorter';
|
|
|
|
const API_EDUCATION_QUERY_LIMIT = 5;
|
|
const API_GEO_QUERY_LIMIT = 5;
|
|
const API_ADRESSE_QUERY_LIMIT = 5;
|
|
|
|
// When searching for short strings like "mer", le exact match shows up quite far in
|
|
// the ordering (~50).
|
|
//
|
|
// That's why we deliberately fetch a lot of results, and then let the local matching
|
|
// (match-sorter) do the work.
|
|
//
|
|
// NB: 60 is arbitrary, we may add more if needed.
|
|
const API_GEO_COMMUNES_QUERY_LIMIT = 60;
|
|
|
|
const { api_geo_url, api_adresse_url, api_education_url } =
|
|
(window as any).gon.autocomplete || {};
|
|
|
|
type QueryKey = readonly [
|
|
scope: string,
|
|
term: string,
|
|
extra: string | undefined
|
|
];
|
|
|
|
function buildURL(scope: string, term: string, extra?: string) {
|
|
term = encodeURIComponent(term.replace(/\(|\)/g, ''));
|
|
if (scope === 'adresse') {
|
|
return `${api_adresse_url}/search?q=${term}&limit=${API_ADRESSE_QUERY_LIMIT}`;
|
|
} else if (scope === 'annuaire-education') {
|
|
return `${api_education_url}/search?dataset=fr-en-annuaire-education&q=${term}&rows=${API_EDUCATION_QUERY_LIMIT}`;
|
|
} else if (scope === 'communes') {
|
|
const limit = `limit=${API_GEO_COMMUNES_QUERY_LIMIT}`;
|
|
const url = extra
|
|
? `${api_geo_url}/communes?codeDepartement=${extra}&${limit}&`
|
|
: `${api_geo_url}/communes?${limit}&`;
|
|
if (isNumeric(term)) {
|
|
return `${url}codePostal=${term}`;
|
|
}
|
|
return `${url}nom=${term}&boost=population`;
|
|
} else if (isNumeric(term)) {
|
|
const code = term.padStart(2, '0');
|
|
return `${api_geo_url}/${scope}?code=${code}&limit=${API_GEO_QUERY_LIMIT}`;
|
|
}
|
|
return `${api_geo_url}/${scope}?nom=${term}&limit=${API_GEO_QUERY_LIMIT}`;
|
|
}
|
|
|
|
function buildOptions(): [RequestInit, AbortController | null] {
|
|
if (window.AbortController) {
|
|
const controller = new AbortController();
|
|
const signal = controller.signal;
|
|
return [{ signal }, controller];
|
|
}
|
|
return [{}, null];
|
|
}
|
|
|
|
const defaultQueryFn: QueryFunction<unknown, QueryKey> = async ({
|
|
queryKey: [scope, term, extra]
|
|
}) => {
|
|
if (scope == 'pays') {
|
|
return matchSorter(await getPays(), term, { keys: ['label'] });
|
|
}
|
|
|
|
const url = buildURL(scope, term, extra);
|
|
const [options, controller] = buildOptions();
|
|
const promise = fetch(url, options).then((response) => {
|
|
if (response.ok) {
|
|
return response.json();
|
|
}
|
|
throw new Error(`Error fetching from "${scope}" API`);
|
|
});
|
|
(promise as any).cancel = () => controller && controller.abort();
|
|
return promise;
|
|
};
|
|
|
|
let paysCache: { label: string }[];
|
|
async function getPays(): Promise<{ label: string }[]> {
|
|
if (!paysCache) {
|
|
paysCache = await getJSON('/api/pays', null);
|
|
}
|
|
return paysCache;
|
|
}
|
|
|
|
export const queryClient = new QueryClient({
|
|
defaultOptions: {
|
|
queries: {
|
|
queryFn: defaultQueryFn as any
|
|
}
|
|
}
|
|
});
|