mirror of
https://github.com/apache/superset.git
synced 2024-09-19 12:09:42 -04:00
50 lines
1.5 KiB
TypeScript
50 lines
1.5 KiB
TypeScript
|
import {
|
||
|
Encodable,
|
||
|
ColorSchemeResolver,
|
||
|
TimeFormatResolver,
|
||
|
CategoricalColorScaleResolver,
|
||
|
defaultColorSchemeResolver,
|
||
|
addPrefix,
|
||
|
} from 'encodable';
|
||
|
import {
|
||
|
CategoricalColorNamespace,
|
||
|
getCategoricalSchemeRegistry,
|
||
|
getSequentialSchemeRegistry,
|
||
|
getNumberFormatter,
|
||
|
getTimeFormatter,
|
||
|
LOCAL_PREFIX,
|
||
|
getTimeFormatterRegistry,
|
||
|
} from '@superset-ui/core';
|
||
|
|
||
|
const timeFormat: TimeFormatResolver = ({ format, formatInLocalTime = false } = {}) => {
|
||
|
const formatString = formatInLocalTime
|
||
|
? addPrefix(LOCAL_PREFIX, format ?? getTimeFormatterRegistry().getDefaultKey()!)
|
||
|
: format;
|
||
|
|
||
|
return getTimeFormatter(formatString);
|
||
|
};
|
||
|
|
||
|
const colorSchemeResolver: ColorSchemeResolver = ({ name, type = 'categorical' } = {}) => {
|
||
|
if (type === 'sequential') {
|
||
|
const scheme = getSequentialSchemeRegistry().get(name);
|
||
|
|
||
|
return typeof scheme === 'undefined' ? scheme : { type: 'sequential', ...scheme };
|
||
|
}
|
||
|
if (type === 'categorical') {
|
||
|
const scheme = getCategoricalSchemeRegistry().get(name);
|
||
|
|
||
|
return typeof scheme === 'undefined' ? scheme : { type: 'categorical', ...scheme };
|
||
|
}
|
||
|
return defaultColorSchemeResolver({ name, type });
|
||
|
};
|
||
|
|
||
|
const colorScaleResolver: CategoricalColorScaleResolver = ({ name, namespace } = {}) =>
|
||
|
CategoricalColorNamespace.getScale(name, namespace);
|
||
|
|
||
|
export default function configureEncodable() {
|
||
|
Encodable.setNumberFormatResolver(getNumberFormatter)
|
||
|
.setTimeFormatResolver(timeFormat)
|
||
|
.setColorSchemeResolver(colorSchemeResolver)
|
||
|
.setCategoricalColorScaleResolver(colorScaleResolver);
|
||
|
}
|