import { format } from 'date-fns' import { genericObject } from '@/types/generic' import { ISport } from '@/types/sports' import { IStatisticsChartData, IStatisticsChartDataset, IStatisticsDateParams, TDatasetKeys, TStatisticsDatasets, TStatisticsFromApi, } from '@/types/statistics' import { incrementDate, startDate } from '@/utils/dates' // date format from api const dateFormats: genericObject = { week: 'yyyy-MM-dd', month: 'yyyy-MM', year: 'yyyy', } const data: TDatasetKeys[] = ['nb_workouts', 'total_duration', 'total_distance'] export const getDateKeys = ( params: IStatisticsDateParams, weekStartingMonday: boolean ): Date[] => { const days = [] for ( let day = startDate(params.duration, params.start, weekStartingMonday); day <= params.end; day = incrementDate(params.duration, day) ) { days.push(day) } return days } const getStatisticsChartDataset = ( sportLabel: string ): IStatisticsChartDataset => Object.assign( {}, { label: sportLabel, backgroundColor: [], data: [], } ) export const getDatasets = (displayedSports: ISport[]): TStatisticsDatasets => { const datasets: TStatisticsDatasets = { nb_workouts: [], total_distance: [], total_duration: [], } displayedSports.map((sport) => { datasets.nb_workouts.push(getStatisticsChartDataset(sport.label)) datasets.total_distance.push(getStatisticsChartDataset(sport.label)) datasets.total_duration.push(getStatisticsChartDataset(sport.label)) }) return datasets } export const formatStats = ( params: IStatisticsDateParams, weekStartingMonday: boolean, sports: ISport[], displayedSportsId: number[], apiStats: TStatisticsFromApi ): IStatisticsChartData => { const dayKeys = getDateKeys(params, weekStartingMonday) const dateFormat = dateFormats[params.duration] const displayedSports = sports.filter((sport) => displayedSportsId.length == 0 ? true : displayedSportsId.includes(sport.id) ) const labels: string[] = [] const datasets = getDatasets(displayedSports) const sportsId: Record = {} displayedSports.map( (displayedSport) => (sportsId[displayedSport.label] = displayedSport.id) ) dayKeys.map((key) => { const date: string = format(key, dateFormat) labels.push(date) data.map((datasetKey) => { datasets[datasetKey].map((dataset) => dataset.data.push( apiStats !== {} && date in apiStats && sportsId[dataset.label] in apiStats[date] ? apiStats[date][sportsId[dataset.label]][datasetKey] : null ) ) }) }) return { labels, datasets, } }