superset-frontend/src/dashboard/reducers/nativeFilters.ts

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
import {
  AnyFilterAction,
  SAVE_FILTER_SETS,
  SET_FILTER_CONFIG_COMPLETE,
  SET_FILTER_SETS_CONFIG_COMPLETE,
  SET_FILTERS_STATE,
  UPDATE_EXTRA_FORM_DATA,
  UpdateExtraFormData,
} from 'src/dashboard/actions/nativeFilters';
import {
  FiltersSet,
  FiltersState,
  FilterState,
  FilterStateType,
  NativeFiltersState,
} from './types';
import { FilterConfiguration } from '../components/nativeFilters/types';

export function getInitialFilterState(id: string): FilterState {
  return {
    id,
    extraFormData: {},
    currentState: {},
  };
}

export function getInitialState({
  filterSetsConfig,
  filterConfig,
  state: prevState,
}: {
  filterSetsConfig?: FiltersSet[];
  filterConfig?: FilterConfiguration;
  state?: NativeFiltersState;
}): NativeFiltersState {
  const state: Partial<NativeFiltersState> = {};

  const emptyFiltersState = {
    [FilterStateType.NativeFilters]: {},
    [FilterStateType.CrossFilters]: {},
    [FilterStateType.OwnFilters]: {},
  };

  const filters = {};
  const filtersState = { ...emptyFiltersState };
  if (filterConfig) {
    filterConfig.forEach(filter => {
      const { id } = filter;
      filters[id] = filter;
      filtersState.nativeFilters[id] =
        prevState?.filtersState?.nativeFilters[id] || getInitialFilterState(id);
    });
    state.filters = filters;
    state.filtersState = {
      ...emptyFiltersState,
      ...prevState?.filtersState,
      nativeFilters: filtersState.nativeFilters,
    };
  } else {
    state.filters = prevState?.filters ?? {};
    state.filtersState = prevState?.filtersState ?? { ...emptyFiltersState };
  }

  if (filterSetsConfig) {
    const filterSets = {};
    filterSetsConfig.forEach(filtersSet => {
      const { id } = filtersSet;
      filterSets[id] = filtersSet;
    });
    state.filterSets = filterSets;
  } else {
    state.filterSets = prevState?.filterSets ?? {};
  }
  return state as NativeFiltersState;
}

const getUnitState = (
  unitName: FilterStateType,
  action: UpdateExtraFormData,
  filtersState: FiltersState,
) => {
  if (action[unitName])
    return {
      ...filtersState[unitName],
      [action.filterId]: {
        ...filtersState[unitName][action.filterId],
        ...action[unitName],
      },
    };
  return { ...filtersState[unitName] };
};

export default function nativeFilterReducer(
  state: NativeFiltersState = {
    filters: {},
    filterSets: {},
    filtersState: {
      [FilterStateType.NativeFilters]: {},
      [FilterStateType.CrossFilters]: {},
      [FilterStateType.OwnFilters]: {},
    },
  },
  action: AnyFilterAction,
) {
  const { filters, filtersState, filterSets } = state;
  switch (action.type) {
    case UPDATE_EXTRA_FORM_DATA:
      return {
        ...state,
        filters,
        filtersState: {
          ...filtersState,
          [FilterStateType.NativeFilters]: getUnitState(
            FilterStateType.NativeFilters,
            action,
            filtersState,
          ),
          [FilterStateType.CrossFilters]: getUnitState(
            FilterStateType.CrossFilters,
            action,
            filtersState,
          ),
          [FilterStateType.OwnFilters]: getUnitState(
            FilterStateType.OwnFilters,
            action,
            filtersState,
          ),
        },
      };
    case SAVE_FILTER_SETS:
      return {
        ...state,
        filterSets: {
          ...filterSets,
          [action.filtersSetId]: {
            id: action.filtersSetId,
            name: action.name,
            filtersState: action.filtersState,
          },
        },
      };
    case SET_FILTERS_STATE:
      return {
        ...state,
        filtersState: {
          ...filtersState,
          ...action.filtersState,
        },
      };

    case SET_FILTER_CONFIG_COMPLETE:
      return getInitialState({ filterConfig: action.filterConfig, state });

    case SET_FILTER_SETS_CONFIG_COMPLETE:
      return getInitialState({
        filterSetsConfig: action.filterSetsConfig,
        state,
      });

    // TODO handle SET_FILTER_CONFIG_FAIL action
    default:
      return state;
  }
}