67 lines
1.7 KiB
TypeScript
67 lines
1.7 KiB
TypeScript
import { Map as ImmutableMap, List as ImmutableList, Record as ImmutableRecord, fromJS } from 'immutable';
|
|
|
|
import {
|
|
MFA_FETCH_SUCCESS,
|
|
MFA_CONFIRM_SUCCESS,
|
|
MFA_DISABLE_SUCCESS,
|
|
} from '../actions/mfa';
|
|
import {
|
|
FETCH_TOKENS_SUCCESS,
|
|
REVOKE_TOKEN_SUCCESS,
|
|
} from '../actions/security';
|
|
|
|
import type { AnyAction } from 'redux';
|
|
|
|
const TokenRecord = ImmutableRecord({
|
|
id: 0,
|
|
app_name: '',
|
|
valid_until: '',
|
|
});
|
|
|
|
const ReducerRecord = ImmutableRecord({
|
|
tokens: ImmutableList<Token>(),
|
|
mfa: ImmutableMap({
|
|
settings: ImmutableMap({
|
|
totp: false,
|
|
}),
|
|
}),
|
|
});
|
|
|
|
type State = ReturnType<typeof ReducerRecord>;
|
|
|
|
export type Token = ReturnType<typeof TokenRecord>;
|
|
|
|
const deleteToken = (state: State, tokenId: number) => {
|
|
return state.update('tokens', tokens => {
|
|
return tokens.filterNot(token => token.id === tokenId);
|
|
});
|
|
};
|
|
|
|
const importMfa = (state: State, data: any) => {
|
|
return state.set('mfa', data);
|
|
};
|
|
|
|
const enableMfa = (state: State, method: string) => {
|
|
return state.setIn(['mfa', 'settings', method], true);
|
|
};
|
|
|
|
const disableMfa = (state: State, method: string) => {
|
|
return state.setIn(['mfa', 'settings', method], false);
|
|
};
|
|
|
|
export default function security(state = ReducerRecord(), action: AnyAction) {
|
|
switch (action.type) {
|
|
case FETCH_TOKENS_SUCCESS:
|
|
return state.set('tokens', ImmutableList(action.tokens.map(TokenRecord)));
|
|
case REVOKE_TOKEN_SUCCESS:
|
|
return deleteToken(state, action.id);
|
|
case MFA_FETCH_SUCCESS:
|
|
return importMfa(state, fromJS(action.data));
|
|
case MFA_CONFIRM_SUCCESS:
|
|
return enableMfa(state, action.method);
|
|
case MFA_DISABLE_SUCCESS:
|
|
return disableMfa(state, action.method);
|
|
default:
|
|
return state;
|
|
}
|
|
}
|