Доработка хуков useQuery, useParams, покрытие тестами (#30)

This commit is contained in:
Nikolay
2020-12-28 00:31:34 +03:00
committed by GitHub
parent 94525edd98
commit c051c23896
12 changed files with 273 additions and 87 deletions

View File

@ -1,70 +1,9 @@
import {parse, ParsedUrlQuery} from 'querystring';
import {head} from 'lodash';
import {useMemo} from 'react';
import {useLocation} from 'react-router-dom';
import {toNumber} from '../utils/parsers';
import {getQueryFromUrl, QueryParsers} from '../utils/getQueryFromUrl';
type QueryParser<T> = (value?: string | string[]) => T;
export type QueryParsers<T> = Partial<{[K in keyof T]: QueryParser<T[K]>}>;
export function stringParser<T extends string>(): QueryParser<Undefinable<T>>;
export function stringParser<T extends string>(defaultValue: T): QueryParser<T>;
export function stringParser<T extends string>(defaultValue?: T) {
return (val?: string | string[]) => {
const value = Array.isArray(val) ? head(val) : val;
return value ?? defaultValue;
};
}
export function numberParser(): QueryParser<Undefinable<number>>;
export function numberParser(defaultValue?: number): QueryParser<number>;
export function numberParser(defaultValue?: number) {
return (val?: string | string[]) => {
const value = Array.isArray(val) ? head(val) : val;
return toNumber(value) ?? defaultValue;
};
}
export function booleanParser(): QueryParser<Undefinable<boolean>>;
export function booleanParser(defaultValue: boolean): QueryParser<boolean>;
export function booleanParser(defaultValue?: boolean) {
return (val?: string | string[]) => {
const value = Array.isArray(val) ? head(val) : val;
if (value === 'true') {
return true;
}
if (value === 'false') {
return false;
}
return defaultValue;
};
}
// Date parser
// Array parser (должен уметь с enum)
export function useQuery<T extends {[name: string]: unknown}>(
queryParsers: QueryParsers<T>
): ParsedUrlQuery | Partial<T> {
export function useQuery<T extends Record<string, unknown>>(queryParsers: QueryParsers<T>): T {
const {search} = useLocation();
return useMemo(() => {
const query = parse(search.slice(1));
return queryParsers ? Object.keys(queryParsers).reduce<T>((memo, key) => {
if (key in queryParsers) {
const parser = queryParsers[key];
return {
...memo,
[key]: parser?.(query[key]),
};
}
return memo;
}, {} as T) : query;
}, [search, queryParsers]);
return useMemo(() => getQueryFromUrl(queryParsers, search), [search, queryParsers]);
}