LingvAI/utils/reducer.ts

import { replaceBoth } from "./language";

export const initialState = {
    source: "auto",
    target: "en",
    query: "",
    delayedQuery: "",
    translation: "",
    isLoading: true
}

type State = typeof initialState;

export enum Actions {
    SET_FIELD,
    SET_ALL,
    SWITCH_LANGS
}

type Action = {
    type: Actions.SET_FIELD,
    payload: {
        key: string,
        value: any
    }
} | {
    type: Actions.SET_ALL,
    payload: {
        state: State
    }
} | {
    type: Actions.SWITCH_LANGS
}

export default function reducer(state: State, action: Action) {
    const { source, target } = replaceBoth("exception", {
        source: state.target,
        target: state.source
    });

    switch (action.type) {
        case Actions.SET_FIELD:
            const { key, value } = action.payload;
            if (key === "source" && value === state.target)
                return { ...state, [key]: value, target: target !== value ? target : "eo" };
            if (key === "target" && value === state.source)
                return { ...state, [key]: value, source };
            return { ...state, [key]: value };
        case Actions.SET_ALL:
            return { ...state, ...action.payload.state };
        case Actions.SWITCH_LANGS:
            return {
                ...state,
                source: source !== target
                    ? source
                    : initialState.source,
                target,
                query: state.translation,
                delayedQuery: state.translation,
                translation: ""
            };
        default:
            return state;
    }
}