shlink-web-client/src/shlink-web-component/visits/reducers/visitsOverview.ts

import type { PayloadAction } from '@reduxjs/toolkit';
import { createSlice } from '@reduxjs/toolkit';
import type { ShlinkApiClient } from '../../../api/services/ShlinkApiClient';
import type { ShlinkVisitsOverview } from '../../../api/types';
import { createAsyncThunk } from '../../../utils/helpers/redux';
import type { CreateVisit } from '../types';
import { groupNewVisitsByType } from '../types/helpers';
import { createNewVisits } from './visitCreation';

const REDUCER_PREFIX = 'shlink/visitsOverview';

export type PartialVisitsSummary = {
  total: number;
  nonBots?: number;
  bots?: number;
};

export type ParsedVisitsOverview = {
  nonOrphanVisits: PartialVisitsSummary;
  orphanVisits: PartialVisitsSummary;
};

export interface VisitsOverview extends ParsedVisitsOverview {
  loading: boolean;
  error: boolean;
}

export type GetVisitsOverviewAction = PayloadAction<ShlinkVisitsOverview>;

const initialState: VisitsOverview = {
  nonOrphanVisits: {
    total: 0,
  },
  orphanVisits: {
    total: 0,
  },
  loading: false,
  error: false,
};

const countBots = (visits: CreateVisit[]) => visits.filter(({ visit }) => visit.potentialBot).length;

export const loadVisitsOverview = (apiClient: ShlinkApiClient) => createAsyncThunk(
  `${REDUCER_PREFIX}/loadVisitsOverview`,
  (): Promise<ParsedVisitsOverview> => apiClient.getVisitsOverview().then(
    ({ nonOrphanVisits, visitsCount, orphanVisits, orphanVisitsCount }) => ({
      nonOrphanVisits: {
        total: nonOrphanVisits?.total ?? visitsCount,
        nonBots: nonOrphanVisits?.nonBots,
        bots: nonOrphanVisits?.bots,
      },
      orphanVisits: {
        total: orphanVisits?.total ?? orphanVisitsCount,
        nonBots: orphanVisits?.nonBots,
        bots: orphanVisits?.bots,
      },
    }),
  ),
);

export const visitsOverviewReducerCreator = (
  loadVisitsOverviewThunk: ReturnType<typeof loadVisitsOverview>,
) => createSlice({
  name: REDUCER_PREFIX,
  initialState,
  reducers: {},
  extraReducers: (builder) => {
    builder.addCase(loadVisitsOverviewThunk.pending, () => ({ ...initialState, loading: true }));
    builder.addCase(loadVisitsOverviewThunk.rejected, () => ({ ...initialState, error: true }));
    builder.addCase(loadVisitsOverviewThunk.fulfilled, (_, { payload }) => ({ ...initialState, ...payload }));

    builder.addCase(createNewVisits, ({ nonOrphanVisits, orphanVisits, ...rest }, { payload }) => {
      const { nonOrphanVisits: newNonOrphanVisits, orphanVisits: newOrphanVisits } = groupNewVisitsByType(
        payload.createdVisits,
      );

      const newNonOrphanTotalVisits = newNonOrphanVisits.length;
      const newNonOrphanBotVisits = countBots(newNonOrphanVisits);
      const newNonOrphanNonBotVisits = newNonOrphanTotalVisits - newNonOrphanBotVisits;

      const newOrphanTotalVisits = newOrphanVisits.length;
      const newOrphanBotVisits = countBots(newOrphanVisits);
      const newOrphanNonBotVisits = newOrphanTotalVisits - newOrphanBotVisits;

      return {
        ...rest,
        nonOrphanVisits: {
          total: nonOrphanVisits.total + newNonOrphanTotalVisits,
          bots: nonOrphanVisits.bots && nonOrphanVisits.bots + newNonOrphanBotVisits,
          nonBots: nonOrphanVisits.nonBots && nonOrphanVisits.nonBots + newNonOrphanNonBotVisits,
        },
        orphanVisits: {
          total: orphanVisits.total + newOrphanTotalVisits,
          bots: orphanVisits.bots && orphanVisits.bots + newOrphanBotVisits,
          nonBots: orphanVisits.nonBots && orphanVisits.nonBots + newOrphanNonBotVisits,
        },
      };
    });
  },
});