badguardhome/client/src/reducers/index.js

import { combineReducers } from 'redux';
import { handleActions } from 'redux-actions';
import { loadingBarReducer } from 'react-redux-loading-bar';
import { reducer as formReducer } from 'redux-form';
import versionCompare from '../helpers/versionCompare';

import * as actions from '../actions';
import toasts from './toasts';
import encryption from './encryption';
import clients from './clients';
import access from './access';
import rewrites from './rewrites';
import services from './services';
import stats from './stats';

const settings = handleActions({
    [actions.initSettingsRequest]: state => ({ ...state, processing: true }),
    [actions.initSettingsFailure]: state => ({ ...state, processing: false }),
    [actions.initSettingsSuccess]: (state, { payload }) => {
        const { settingsList } = payload;
        const newState = { ...state, settingsList, processing: false };
        return newState;
    },
    [actions.toggleSettingStatus]: (state, { payload }) => {
        const { settingsList } = state;
        const { settingKey } = payload;

        const setting = settingsList[settingKey];

        const newSetting = { ...setting, enabled: !setting.enabled };
        const newSettingsList = { ...settingsList, [settingKey]: newSetting };
        return { ...state, settingsList: newSettingsList };
    },
    [actions.setUpstreamRequest]: state => ({ ...state, processingUpstream: true }),
    [actions.setUpstreamFailure]: state => ({ ...state, processingUpstream: false }),
    [actions.setUpstreamSuccess]: state => ({ ...state, processingUpstream: false }),

    [actions.testUpstreamRequest]: state => ({ ...state, processingTestUpstream: true }),
    [actions.testUpstreamFailure]: state => ({ ...state, processingTestUpstream: false }),
    [actions.testUpstreamSuccess]: state => ({ ...state, processingTestUpstream: false }),
}, {
    processing: true,
    processingTestUpstream: false,
    processingSetUpstream: false,
    processingDhcpStatus: false,
});

const dashboard = handleActions({
    [actions.dnsStatusRequest]: state => ({ ...state, processing: true }),
    [actions.dnsStatusFailure]: state => ({ ...state, processing: false }),
    [actions.dnsStatusSuccess]: (state, { payload }) => {
        const {
            version,
            running,
            dns_port: dnsPort,
            dns_addresses: dnsAddresses,
            querylog_enabled: queryLogEnabled,
            upstream_dns: upstreamDns,
            bootstrap_dns: bootstrapDns,
            all_servers: allServers,
            protection_enabled: protectionEnabled,
            language,
            http_port: httpPort,
        } = payload;
        const newState = {
            ...state,
            isCoreRunning: running,
            processing: false,
            dnsVersion: version,
            dnsPort,
            dnsAddresses,
            queryLogEnabled,
            upstreamDns: upstreamDns.join('\n'),
            bootstrapDns: bootstrapDns.join('\n'),
            allServers,
            protectionEnabled,
            language,
            httpPort,
        };
        return newState;
    },

    [actions.enableDnsRequest]: state => ({ ...state, processing: true }),
    [actions.enableDnsFailure]: state => ({ ...state, processing: false }),
    [actions.enableDnsSuccess]: (state) => {
        const newState = { ...state, isCoreRunning: !state.isCoreRunning, processing: false };
        return newState;
    },

    [actions.disableDnsRequest]: state => ({ ...state, processing: true }),
    [actions.disableDnsFailure]: state => ({ ...state, processing: false }),
    [actions.disableDnsSuccess]: (state) => {
        const newState = { ...state, isCoreRunning: !state.isCoreRunning, processing: false };
        return newState;
    },

    [actions.toggleLogStatusRequest]: state => ({ ...state, logStatusProcessing: true }),
    [actions.toggleLogStatusFailure]: state => ({ ...state, logStatusProcessing: false }),
    [actions.toggleLogStatusSuccess]: (state) => {
        const { queryLogEnabled } = state;
        return ({ ...state, queryLogEnabled: !queryLogEnabled, logStatusProcessing: false });
    },

    [actions.getVersionRequest]: state => ({ ...state, processingVersion: true }),
    [actions.getVersionFailure]: state => ({ ...state, processingVersion: false }),
    [actions.getVersionSuccess]: (state, { payload }) => {
        const currentVersion = state.dnsVersion === 'undefined' ? 0 : state.dnsVersion;

        if (payload && versionCompare(currentVersion, payload.new_version) === -1) {
            const {
                announcement_url: announcementUrl,
                new_version: newVersion,
                can_autoupdate: canAutoUpdate,
            } = payload;

            const newState = {
                ...state,
                announcementUrl,
                newVersion,
                canAutoUpdate,
                isUpdateAvailable: true,
                processingVersion: false,
            };
            return newState;
        }

        return {
            ...state,
            processingVersion: false,
        };
    },

    [actions.getUpdateRequest]: state => ({ ...state, processingUpdate: true }),
    [actions.getUpdateFailure]: state => ({ ...state, processingUpdate: false }),
    [actions.getUpdateSuccess]: (state) => {
        const newState = { ...state, processingUpdate: false };
        return newState;
    },

    [actions.getFilteringRequest]: state => ({ ...state, processingFiltering: true }),
    [actions.getFilteringFailure]: state => ({ ...state, processingFiltering: false }),
    [actions.getFilteringSuccess]: (state, { payload }) => {
        const newState = { ...state, isFilteringEnabled: payload, processingFiltering: false };
        return newState;
    },

    [actions.toggleProtectionRequest]: state => ({ ...state, processingProtection: true }),
    [actions.toggleProtectionFailure]: state => ({ ...state, processingProtection: false }),
    [actions.toggleProtectionSuccess]: (state) => {
        const newState = {
            ...state,
            protectionEnabled: !state.protectionEnabled,
            processingProtection: false,
        };
        return newState;
    },

    [actions.handleUpstreamChange]: (state, { payload }) => {
        const { upstreamDns } = payload;
        return { ...state, upstreamDns };
    },

    [actions.getLanguageSuccess]: (state, { payload }) => {
        const newState = { ...state, language: payload };
        return newState;
    },

    [actions.getClientsRequest]: state => ({ ...state, processingClients: true }),
    [actions.getClientsFailure]: state => ({ ...state, processingClients: false }),
    [actions.getClientsSuccess]: (state, { payload }) => {
        const newState = {
            ...state,
            clients: payload.clients,
            autoClients: payload.autoClients,
            processingClients: false,
        };
        return newState;
    },
}, {
    processing: true,
    isCoreRunning: false,
    logStatusProcessing: false,
    processingVersion: true,
    processingFiltering: true,
    processingClients: true,
    processingUpdate: false,
    upstreamDns: '',
    bootstrapDns: '',
    allServers: false,
    protectionEnabled: false,
    processingProtection: false,
    httpPort: 80,
    dnsPort: 53,
    dnsAddresses: [],
    dnsVersion: '',
    clients: [],
    autoClients: [],
});

const queryLogs = handleActions({
    [actions.getLogsRequest]: state => ({ ...state, getLogsProcessing: true }),
    [actions.getLogsFailure]: state => ({ ...state, getLogsProcessing: false }),
    [actions.getLogsSuccess]: (state, { payload }) => {
        const newState = { ...state, logs: payload, getLogsProcessing: false };
        return newState;
    },
    [actions.downloadQueryLogRequest]: state => ({ ...state, logsDownloading: true }),
    [actions.downloadQueryLogFailure]: state => ({ ...state, logsDownloading: false }),
    [actions.downloadQueryLogSuccess]: state => ({ ...state, logsDownloading: false }),
}, {
    getLogsProcessing: false,
    logsDownloading: false,
    logs: [],
});

const filtering = handleActions({
    [actions.setRulesRequest]: state => ({ ...state, processingRules: true }),
    [actions.setRulesFailure]: state => ({ ...state, processingRules: false }),
    [actions.setRulesSuccess]: state => ({ ...state, processingRules: false }),

    [actions.handleRulesChange]: (state, { payload }) => {
        const { userRules } = payload;
        return { ...state, userRules };
    },

    [actions.getFilteringStatusRequest]: state => ({ ...state, processingFilters: true }),
    [actions.getFilteringStatusFailure]: state => ({ ...state, processingFilters: false }),
    [actions.getFilteringStatusSuccess]: (state, { payload }) => {
        const { status } = payload;
        const { filters, userRules } = status;
        const newState = {
            ...state, filters, userRules, processingFilters: false,
        };
        return newState;
    },

    [actions.addFilterRequest]: state =>
        ({ ...state, processingAddFilter: true, isFilterAdded: false }),
    [actions.addFilterFailure]: (state) => {
        const newState = { ...state, processingAddFilter: false, isFilterAdded: false };
        return newState;
    },
    [actions.addFilterSuccess]: state =>
        ({ ...state, processingAddFilter: false, isFilterAdded: true }),

    [actions.toggleFilteringModal]: (state) => {
        const newState = {
            ...state,
            isFilteringModalOpen: !state.isFilteringModalOpen,
            isFilterAdded: false,
        };
        return newState;
    },

    [actions.toggleFilterRequest]: state => ({ ...state, processingFilters: true }),
    [actions.toggleFilterFailure]: state => ({ ...state, processingFilters: false }),
    [actions.toggleFilterSuccess]: state => ({ ...state, processingFilters: false }),

    [actions.refreshFiltersRequest]: state => ({ ...state, processingRefreshFilters: true }),
    [actions.refreshFiltersFailure]: state => ({ ...state, processingRefreshFilters: false }),
    [actions.refreshFiltersSuccess]: state => ({ ...state, processingRefreshFilters: false }),
}, {
    isFilteringModalOpen: false,
    processingFilters: false,
    processingRules: false,
    processingAddFilter: false,
    processingRefreshFilters: false,
    filters: [],
    userRules: '',
});

const dhcp = handleActions({
    [actions.getDhcpStatusRequest]: state => ({ ...state, processing: true }),
    [actions.getDhcpStatusFailure]: state => ({ ...state, processing: false }),
    [actions.getDhcpStatusSuccess]: (state, { payload }) => {
        const {
            static_leases: staticLeases,
            ...values
        } = payload;

        const newState = {
            ...state,
            staticLeases,
            processing: false,
            ...values,
        };

        return newState;
    },

    [actions.getDhcpInterfacesRequest]: state => ({ ...state, processingInterfaces: true }),
    [actions.getDhcpInterfacesFailure]: state => ({ ...state, processingInterfaces: false }),
    [actions.getDhcpInterfacesSuccess]: (state, { payload }) => {
        const newState = {
            ...state,
            interfaces: payload,
            processingInterfaces: false,
        };
        return newState;
    },

    [actions.findActiveDhcpRequest]: state => ({ ...state, processingStatus: true }),
    [actions.findActiveDhcpFailure]: state => ({ ...state, processingStatus: false }),
    [actions.findActiveDhcpSuccess]: (state, { payload }) => {
        const {
            other_server: otherServer,
            static_ip: staticIP,
        } = payload;

        const newState = {
            ...state,
            check: {
                otherServer,
                staticIP,
            },
            processingStatus: false,
        };
        return newState;
    },

    [actions.toggleDhcpRequest]: state => ({ ...state, processingDhcp: true }),
    [actions.toggleDhcpFailure]: state => ({ ...state, processingDhcp: false }),
    [actions.toggleDhcpSuccess]: (state) => {
        const { config } = state;
        const newConfig = { ...config, enabled: !config.enabled };
        const newState = {
            ...state, config: newConfig, check: null, processingDhcp: false,
        };
        return newState;
    },

    [actions.setDhcpConfigRequest]: state => ({ ...state, processingConfig: true }),
    [actions.setDhcpConfigFailure]: state => ({ ...state, processingConfig: false }),
    [actions.setDhcpConfigSuccess]: (state, { payload }) => {
        const { config } = state;
        const newConfig = { ...config, ...payload };
        const newState = { ...state, config: newConfig, processingConfig: false };
        return newState;
    },

    [actions.toggleLeaseModal]: (state) => {
        const newState = {
            ...state,
            isModalOpen: !state.isModalOpen,
        };
        return newState;
    },

    [actions.addStaticLeaseRequest]: state => ({ ...state, processingAdding: true }),
    [actions.addStaticLeaseFailure]: state => ({ ...state, processingAdding: false }),
    [actions.addStaticLeaseSuccess]: (state, { payload }) => {
        const {
            ip, mac, hostname,
        } = payload;
        const newLease = {
            ip,
            mac,
            hostname: hostname || '',
        };
        const leases = [...state.staticLeases, newLease];
        const newState = {
            ...state,
            staticLeases: leases,
            processingAdding: false,
        };
        return newState;
    },

    [actions.removeStaticLeaseRequest]: state => ({ ...state, processingDeleting: true }),
    [actions.removeStaticLeaseFailure]: state => ({ ...state, processingDeleting: false }),
    [actions.removeStaticLeaseSuccess]: (state, { payload }) => {
        const leaseToRemove = payload.ip;
        const leases = state.staticLeases.filter(item => item.ip !== leaseToRemove);
        const newState = {
            ...state,
            staticLeases: leases,
            processingDeleting: false,
        };
        return newState;
    },
}, {
    processing: true,
    processingStatus: false,
    processingInterfaces: false,
    processingDhcp: false,
    processingConfig: false,
    processingAdding: false,
    processingDeleting: false,
    config: {
        enabled: false,
    },
    check: null,
    leases: [],
    staticLeases: [],
    isModalOpen: false,
});

export default combineReducers({
    settings,
    dashboard,
    queryLogs,
    filtering,
    toasts,
    dhcp,
    encryption,
    clients,
    access,
    rewrites,
    services,
    stats,
    loadingBar: loadingBarReducer,
    form: formReducer,
});