AdGuardHome/client/src/reducers/stats.js

import { handleActions } from 'redux-actions';
import { normalizeTopClients } from '../helpers/helpers';
import { DAY, HOUR, STATS_INTERVALS_DAYS } from '../helpers/constants';

import * as actions from '../actions/stats';

const defaultStats = {
    dnsQueries: [],
    blockedFiltering: [],
    replacedParental: [],
    replacedSafebrowsing: [],
    topBlockedDomains: [],
    topClients: [],
    topQueriedDomains: [],
    numBlockedFiltering: 0,
    numDnsQueries: 0,
    numReplacedParental: 0,
    numReplacedSafebrowsing: 0,
    numReplacedSafesearch: 0,
    avgProcessingTime: 0,
};

const stats = handleActions(
    {
        [actions.getStatsConfigRequest]: (state) => ({ ...state, processingGetConfig: true }),
        [actions.getStatsConfigFailure]: (state) => ({ ...state, processingGetConfig: false }),
        [actions.getStatsConfigSuccess]: (state, { payload }) => ({
            ...state,
            ...payload,
            customInterval: !STATS_INTERVALS_DAYS.includes(payload.interval)
                ? payload.interval / HOUR
                : null,
            processingGetConfig: false,
        }),

        [actions.setStatsConfigRequest]: (state) => ({ ...state, processingSetConfig: true }),
        [actions.setStatsConfigFailure]: (state) => ({ ...state, processingSetConfig: false }),
        [actions.setStatsConfigSuccess]: (state, { payload }) => ({
            ...state,
            ...payload,
            processingSetConfig: false,
        }),

        [actions.getStatsRequest]: (state) => ({ ...state, processingStats: true }),
        [actions.getStatsFailure]: (state) => ({ ...state, processingStats: false }),
        [actions.getStatsSuccess]: (state, { payload }) => {
            const {
                dns_queries: dnsQueries,
                blocked_filtering: blockedFiltering,
                replaced_parental: replacedParental,
                replaced_safebrowsing: replacedSafebrowsing,
                top_blocked_domains: topBlockedDomains,
                top_clients: topClients,
                top_queried_domains: topQueriedDomains,
                num_blocked_filtering: numBlockedFiltering,
                num_dns_queries: numDnsQueries,
                num_replaced_parental: numReplacedParental,
                num_replaced_safebrowsing: numReplacedSafebrowsing,
                num_replaced_safesearch: numReplacedSafesearch,
                avg_processing_time: avgProcessingTime,
                top_upstreams_responses: topUpstreamsResponses,
                top_upstrems_avg_time: topUpstreamsAvgTime,
            } = payload;

            const newState = {
                ...state,
                processingStats: false,
                dnsQueries,
                blockedFiltering,
                replacedParental,
                replacedSafebrowsing,
                topBlockedDomains,
                topClients,
                normalizedTopClients: normalizeTopClients(topClients),
                topQueriedDomains,
                numBlockedFiltering,
                numDnsQueries,
                numReplacedParental,
                numReplacedSafebrowsing,
                numReplacedSafesearch,
                avgProcessingTime,
                topUpstreamsResponses,
                topUpstreamsAvgTime,
            };

            return newState;
        },

        [actions.resetStatsRequest]: (state) => ({ ...state, processingReset: true }),
        [actions.resetStatsFailure]: (state) => ({ ...state, processingReset: false }),
        [actions.resetStatsSuccess]: (state) => ({
            ...state,
            ...defaultStats,
            processingReset: false,
        }),
    },
    {
        processingGetConfig: false,
        processingSetConfig: false,
        processingStats: true,
        processingReset: false,
        interval: DAY,
        customInterval: null,
        ...defaultStats,
    },
);

export default stats;