...

import { PayloadAction } from '@reduxjs/toolkit';
import { AnyAction } from 'redux';

import { toStateKey } from '../utils';

import { getTemplatingReducers, TemplatingState } from './reducers';
import { variablesInitTransaction } from './transactionReducer';

export interface KeyedVariablesState {
  lastKey?: string;
  keys: Record<string, TemplatingState>;
}

export const initialKeyedVariablesState: KeyedVariablesState = { keys: {} };

export interface KeyedAction {
  key: string;
  action: PayloadAction<any>;
}

const keyedAction = (payload: KeyedAction) => ({
  type: `templating/keyed/${payload.action.type.replace(/^templating\//, '')}`,
  payload,
});

export function toKeyedAction(key: string, action: PayloadAction<any>): PayloadAction<KeyedAction> {
  const keyAsString = toStateKey(key);
  return keyedAction({ key: keyAsString, action });
}

const isKeyedAction = (action: AnyAction): action is PayloadAction<KeyedAction> => {
  return (
    typeof action.type === 'string' &&
    action.type.startsWith('templating/keyed') &&
    'payload' in action &&
    typeof action.payload.key === 'string'
  );
};

export function keyedVariablesReducer(state = initialKeyedVariablesState, outerAction: AnyAction): KeyedVariablesState {
  if (isKeyedAction(outerAction)) {
    const { key, action } = outerAction.payload;
    const stringKey = toStateKey(key);
    const lastKey = variablesInitTransaction.match(action) ? stringKey : state.lastKey;
    const templatingReducers = getTemplatingReducers();
    const prevKeyState = state.keys[stringKey];
    const nextKeyState = templatingReducers(prevKeyState, action);

return {
      ...state,
      lastKey,
      keys: {
        ...state.keys,
        [stringKey]: nextKeyState,
      },
    };
  }

return state;
}

export default {
  templating: keyedVariablesReducer,
};

.	Edit
..	Edit
__tests__	Edit
actions.test.ts	Edit
actions.ts	Edit
helpers.ts	Edit
initVariableTransaction.test.ts	Edit
keyedVariablesReducer.test.ts	Edit
keyedVariablesReducer.ts	Edit
migrateVariablesDatasourceNameToRef.test.ts	Edit
onTimeRangeUpdated.test.ts	Edit
processVariable.test.ts	Edit
reducers.test.ts	Edit
reducers.ts	Edit
selectors.ts	Edit
setOptionFromUrl.test.ts	Edit
sharedReducer.test.ts	Edit
sharedReducer.ts	Edit
templateVarsChangedInUrl.test.ts	Edit
transactionReducer.test.ts	Edit
transactionReducer.ts	Edit
types.ts	Edit
upgradeLegacyQueries.test.ts	Edit
variablesReducer.ts	Edit