import { DataCallback } from '../components/material-table'; import { Result, HitEntry } from '../models'; import { requestRest } from '../services/restService'; type propType = string | number | null | undefined | (string | number)[]; type dataType = { [prop: string]: propType }; type resultType = { page: number, rowCount: number, rows: TData[] }; export function createSearchDataHandler(uri: string, additionalParameters?: {}): DataCallback<(TResult & { _id: string })>; export function createSearchDataHandler(uri: string, additionalParameters: {} | null | undefined, mapResult: (res: HitEntry, index: number, arr: HitEntry[]) => (TData & { _id: string }), mapRequest?: (name?: string | null) => string): DataCallback<(TData & { _id: string })> export function createSearchDataHandler(uri: string, additionalParameters?: {} | null | undefined, mapResult?: (res: HitEntry, index: number, arr: HitEntry[]) => (TData & { _id: string }), mapRequest?: (name?: string | null) => string): DataCallback<(TData & { _id: string })> { const url = `${ window.location.origin }/database/${uri}/_search`; const fetchData: DataCallback<(TData & { _id: string }) > = async (page, rowsPerPage, orderBy, order, filter) => { const from = rowsPerPage && page != null && !isNaN(+page) ? (+page) * rowsPerPage : null; const filterKeys = filter && Object.keys(filter) || []; const query = { ...filterKeys.length > 0 ? { query: { bool: { must: filterKeys.reduce((acc, cur) => { if (acc && filter && filter[cur]) { acc.push({ [filter[cur].indexOf("*") > -1 || filter[cur].indexOf("?") > -1 ? "wildcard" : "prefix"]: { [mapRequest ? mapRequest(cur) : cur]: filter[cur] } }); } return acc; }, [] as any[]) } } } : { "query": { "match_all": {} } }, ...rowsPerPage ? { "size": rowsPerPage } : {}, ...from ? { "from": from } : {}, ...orderBy && order ? { "sort": [{ [mapRequest ? mapRequest(orderBy) : orderBy]: order }] } : {}, ...additionalParameters ? additionalParameters : {} }; const result = await requestRest>(url, { method: "POST", // *GET, POST, PUT, DELETE, etc. mode: "no-cors", // no-cors, cors, *same-origin cache: "no-cache", // *default, no-cache, reload, force-cache, only-if-cached headers: { "Content-Type": "application/json; charset=utf-8", // "Content-Type": "application/x-www-form-urlencoded", }, body: JSON.stringify(query), // body data type must match "Content-Type" header }); if (result) { let rows: (TData & { _id: string })[] = []; if (result && result.hits && result.hits.hits) { rows = result.hits.hits.map( mapResult ? mapResult : h => ( { ...(h._source as any as TData), _id: h._id } )) || [] } const data = { page: Math.min(page || 0, result.hits.total || 0 / (rowsPerPage || 1)), rowCount: result.hits.total, rows: rows }; return data; } return { page: 0, rowCount: 0, rows: [] }; }; return fetchData; }