@lobehub/charts
Version:
React modern charts components built on recharts
186 lines (185 loc) • 10.3 kB
JavaScript
'use client';
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
var _excluded = ["data", "blockMargin", "blockRadius", "blockSize", "onValueChange", "fontSize", "hideColorLegend", "hideMonthLabels", "hideTotalCount", "labels", "maxLevel", "loading", "showTooltip", "customTooltip", "showWeekdayLabels", "style", "colors", "totalCount", "weekStart", "className", "noDataText"];
function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
function _objectWithoutProperties(source, excluded) { if (source == null) return {}; var target = _objectWithoutPropertiesLoose(source, excluded); var key, i; if (Object.getOwnPropertySymbols) { var sourceSymbolKeys = Object.getOwnPropertySymbols(source); for (i = 0; i < sourceSymbolKeys.length; i++) { key = sourceSymbolKeys[i]; if (excluded.indexOf(key) >= 0) continue; if (!Object.prototype.propertyIsEnumerable.call(source, key)) continue; target[key] = source[key]; } } return target; }
function _objectWithoutPropertiesLoose(source, excluded) { if (source == null) return {}; var target = {}; var sourceKeys = Object.keys(source); var key, i; for (i = 0; i < sourceKeys.length; i++) { key = sourceKeys[i]; if (excluded.indexOf(key) >= 0) continue; target[key] = source[key]; } return target; }
import { useThemeMode } from 'antd-style';
import { getYear, parseISO } from 'date-fns';
import { forwardRef, useMemo } from 'react';
import { Flexbox } from 'react-layout-kit';
import NoData from "../common/NoData";
import { usePrefersReducedMotion } from "../hooks/usePrefersReducedMotion";
import { isOnSeverSide } from "../utils";
import { DEFAULT_LABELS, generateEmptyData, groupByWeeks, maxWeekdayLabelLength } from "../utils/calendar";
import { createTheme } from "../utils/theme";
import Calendar from "./Calendar";
import ChartLabels from "./ChartLabels";
import Footer from "./Footer";
import { useStyles } from "./styles";
import { jsx as _jsx } from "react/jsx-runtime";
import { jsxs as _jsxs } from "react/jsx-runtime";
var LABEL_MARGIN = 8;
var Heatmaps = /*#__PURE__*/forwardRef(function (props, ref) {
var data = props.data,
_props$blockMargin = props.blockMargin,
blockMargin = _props$blockMargin === void 0 ? 4 : _props$blockMargin,
_props$blockRadius = props.blockRadius,
blockRadius = _props$blockRadius === void 0 ? 2 : _props$blockRadius,
_props$blockSize = props.blockSize,
blockSize = _props$blockSize === void 0 ? 12 : _props$blockSize,
onValueChange = props.onValueChange,
_props$fontSize = props.fontSize,
fontSize = _props$fontSize === void 0 ? 12 : _props$fontSize,
_props$hideColorLegen = props.hideColorLegend,
hideColorLegend = _props$hideColorLegen === void 0 ? false : _props$hideColorLegen,
_props$hideMonthLabel = props.hideMonthLabels,
hideMonthLabels = _props$hideMonthLabel === void 0 ? false : _props$hideMonthLabel,
_props$hideTotalCount = props.hideTotalCount,
hideTotalCount = _props$hideTotalCount === void 0 ? false : _props$hideTotalCount,
labelsProp = props.labels,
_props$maxLevel = props.maxLevel,
maxLevel = _props$maxLevel === void 0 ? Math.max(1, props.maxLevel || 4) : _props$maxLevel,
_props$loading = props.loading,
loading = _props$loading === void 0 ? false : _props$loading,
_props$showTooltip = props.showTooltip,
showTooltip = _props$showTooltip === void 0 ? true : _props$showTooltip,
customTooltip = props.customTooltip,
_props$showWeekdayLab = props.showWeekdayLabels,
showWeekdayLabels = _props$showWeekdayLab === void 0 ? false : _props$showWeekdayLab,
_props$style = props.style,
style = _props$style === void 0 ? {} : _props$style,
colors = props.colors,
totalCountProp = props.totalCount,
_props$weekStart = props.weekStart,
weekStart = _props$weekStart === void 0 ? 0 : _props$weekStart,
className = props.className,
noDataText = props.noDataText,
rest = _objectWithoutProperties(props, _excluded);
var activities = useMemo(function () {
if (loading) return generateEmptyData();
return data.map(function (item) {
return _objectSpread(_objectSpread({}, item), {}, {
level: item.level > maxLevel ? maxLevel : item.level
});
});
}, [data, maxLevel, loading]);
var _useThemeMode = useThemeMode(),
isDarkMode = _useThemeMode.isDarkMode;
var _useStyles = useStyles(),
cx = _useStyles.cx,
styles = _useStyles.styles,
theme = _useStyles.theme;
var defaultColors = useMemo(function () {
switch (maxLevel) {
case 1:
{
return [theme.colorFillSecondary, isDarkMode ? theme.lime8 : theme.green8];
}
case 2:
{
return [theme.colorFillSecondary, isDarkMode ? theme.lime4 : theme.green4, isDarkMode ? theme.lime8 : theme.green8];
}
case 3:
{
return [theme.colorFillSecondary, isDarkMode ? theme.lime3 : theme.green3, isDarkMode ? theme.lime6 : theme.green6, isDarkMode ? theme.lime9 : theme.green9];
}
case 4:
{
return [theme.colorFillSecondary, isDarkMode ? theme.lime2 : theme.green2, isDarkMode ? theme.lime4 : theme.green4, isDarkMode ? theme.lime6 : theme.green6, isDarkMode ? theme.lime8 : theme.green8];
}
default:
{
return [theme.colorFillSecondary, theme.colorSuccess];
}
}
}, [theme, isDarkMode, maxLevel]);
var useAnimation = !usePrefersReducedMotion();
if (activities.length === 0) return /*#__PURE__*/_jsx(NoData, {
noDataText: noDataText
});
if (showWeekdayLabels && isOnSeverSide) return null;
var colorScale = createTheme(colors || defaultColors, maxLevel + 1);
var firstActivity = activities[0];
var year = getYear(parseISO(firstActivity.date));
var weeks = groupByWeeks(activities, weekStart);
var firstWeek = weeks[0];
var labels = Object.assign({}, DEFAULT_LABELS, labelsProp);
var labelHeight = hideMonthLabels ? 0 : fontSize + LABEL_MARGIN;
var weekdayLabelOffset = showWeekdayLabels ? maxWeekdayLabelLength(firstWeek, weekStart, labels.weekdays, fontSize) + LABEL_MARGIN : undefined;
var getDimensions = function getDimensions() {
return {
height: labelHeight + (blockSize + blockMargin) * 7 - blockMargin,
width: weeks.length * (blockSize + blockMargin) - blockMargin
};
};
var _getDimensions = getDimensions(),
width = _getDimensions.width,
height = _getDimensions.height;
var zeroColor = colorScale[0];
var containerStyles = _objectSpread({
fontSize: fontSize
}, useAnimation && _defineProperty(_defineProperty({}, "--lobe-heatmaps-loading", zeroColor), "--lobe-heatmaps-loading-active", theme.colorFill));
return /*#__PURE__*/_jsxs(Flexbox, _objectSpread(_objectSpread({
className: cx(styles.container, className),
ref: ref,
style: _objectSpread(_objectSpread({}, style), containerStyles)
}, rest), {}, {
children: [/*#__PURE__*/_jsx("div", {
className: styles.scrollContainer,
children: /*#__PURE__*/_jsxs("svg", {
className: styles.calendar,
height: height,
style: {
marginLeft: weekdayLabelOffset
},
viewBox: "0 0 ".concat(width, " ").concat(height),
width: width,
children: [!loading && !showWeekdayLabels && hideMonthLabels ? undefined : /*#__PURE__*/_jsx(ChartLabels, {
blockMargin: blockMargin,
blockSize: blockSize,
hideMonthLabels: hideMonthLabels,
labelHeight: labelHeight,
labelMargin: LABEL_MARGIN,
labels: labels,
showWeekdayLabels: showWeekdayLabels,
weekStart: weekStart,
weeks: weeks
}), /*#__PURE__*/_jsx(Calendar, {
blockMargin: blockMargin,
blockRadius: blockRadius,
blockSize: blockSize,
colorScale: colorScale,
customTooltip: customTooltip,
enableAnimation: loading && useAnimation,
labelHeight: labelHeight,
labels: labels,
maxLevel: maxLevel,
onValueChange: onValueChange,
showTooltip: showTooltip,
weeks: weeks
})]
})
}), hideTotalCount && hideColorLegend ? undefined : /*#__PURE__*/_jsx(Footer, {
blockRadius: blockRadius,
blockSize: blockSize,
colorScale: colorScale,
hideColorLegend: hideColorLegend,
hideTotalCount: hideTotalCount,
labels: labels,
loading: loading,
maxLevel: maxLevel,
totalCount: typeof totalCountProp === 'number' ? totalCountProp : activities.reduce(function (sum, activity) {
return sum + activity.count;
}, 0),
weekdayLabelOffset: weekdayLabelOffset,
year: year
})]
}));
});
Heatmaps.displayName = 'Heatmaps';
export default Heatmaps;