Implementing Undo History
Building an Undo and Redo functionality into an app has traditionally required conscious effort from the developer. It is not an easy problem with classical MVC frameworks because you need to keep track of every past state by cloning all relevant models. In addition, you need to be mindful of the undo stack because the user-initiated changes should be undoable.
This means that implementing Undo and Redo in an MVC application usually forces you to rewrite parts of your application to use a specific data mutation pattern like Command.
With Redux, however, implementing undo history is a breeze. There are three reasons for this:
- There are no multiple models—just a state subtree that you want to keep track of.
- The state is already immutable, and mutations are already described as discrete actions, which is close to the undo stack mental model.
- The reducer
(state, action) => statesignature makes it natural to implement generic “reducer enhancers” or “higher order reducers”. They are functions that take your reducer and enhance it with some additional functionality while preserving its signature. Undo history is exactly such a case.
Before proceeding, make sure you have worked through the basics tutorial and understand reducer composition well. This recipe will build on top of the example described in the basics tutorial.
In the first part of this recipe, we will explain the underlying concepts that make Undo and Redo possible to implement in a generic way.
In the second part of this recipe, we will show how to use Redux Undo package that provides this functionality out of the box.
Understanding Undo History
Designing the State Shape
Undo history is also part of your app’s state, and there is no reason why we should approach it differently. Regardless of the type of the state changing over time, when you implement Undo and Redo, you want to keep track of the history of this state at different points in time.
For example, the state shape of a counter app might look like this:
{
counter: 10
}
If we wanted to implement Undo and Redo in such an app, we’d need to store more state so we can answer the following questions:
- Is there anything left to undo or redo?
- What is the current state?
- What are the past (and future) states in the undo stack?
It is reasonable to suggest that our state shape should change to answer these questions:
{
counter: {
present: 10,
history: {
past: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
future: []
}
}
}
Now, if user presses “Undo”, we want it to change to move into the past:
{
counter: {
present: 9,
history: {
past: [0, 1, 2, 3, 4, 5, 6, 7, 8],
future: [10]
}
}
}
And further yet:
{
counter: {
present: 8,
history: {
past: [0, 1, 2, 3, 4, 5, 6, 7],
future: [9, 10]
}
}
}
When the user presses “Redo”, we want to move one step back into the future:
{
counter: {
present: 9,
history: {
past: [0, 1, 2, 3, 4, 5, 6, 7, 8],
future: [10]
}
}
}
Finally, if the user performs an action (e.g. decrement the counter) while we’re in the middle of the undo stack, we’re going to discard the existing future:
{
counter: {
present: 8,
history: {
past: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
future: []
}
}
}
The interesting part here is that it does not matter whether we want to keep an undo stack of numbers, strings, arrays, or objects. The structure will always be the same:
{
counter: {
present: 3,
history: {
past: [0, 1, 2],
future: [4]
}
}
}
{
todos: {
present: [{ text: 'Use Redux', complete: true }, { text: 'Implement Undo' }],
history: {
past: [
[],
[{ text: 'Use Redux' }],
[{ text: 'Use Redux', complete: true }]
],
future: [
[{ text: 'Use Redux', complete: true }, { text: 'Implement Undo', complete: true }]
]
}
}
}
In general, it looks like this:
{
present: T,
history: {
past: Array<T>,
future: Array<T>
}
}
It is also up to us whether to keep a single top-level history:
{
present: { counterA: 2, counterB: 1 },
history: {
past: [
{ counterA: 1, counterB: 1 },
{ counterA: 1, counterB: 0 },
{ counterA: 0, counterB: 0 }
],
future: []
}
}
Or many granular histories so user can undo and redo actions in them independently:
{
counterA: {
present: 2,
history: {
past: [1, 0],
future: []
}
},
counterB: {
present: 1,
history: {
past: [0],
future: []
}
}
}
We will see later how the approach we take lets us choose how granular Undo and Redo need to be.
Designing the Algorithm
Regardless of the specific data type, the shape of the undo history state is the same:
{
present: T,
history: {
past: Array<T>,
future: Array<T>
}
}
Let’s talk through the algorithm to manipulate the state shape described above. We can define two actions to operate on this state: UNDO and REDO. In our reducer, we will do the following steps to handle these actions:
Handling Undo
- Remove the last element from the
past. - Set the
presentto the element we removed in the previous step. - Insert the old
presentstate at the beginning of thefuture.
Handling Redo
- Remove the first element from the
future. - Set the
presentto the element we removed in the previous step. - Insert the old
presentstate at the end of thepast.
Handling Other Actions
- Insert the
presentat the end of thepast. - Set the
presentto the new state after handling the action. - Clear the
future.
First Attempt: Writing a Reducer
const initialState = {
present: null, // (?) How do we initialize the present?
history: {
past: [],
future: []
}
};
function undoable(state = initialState, action) {
const { present, history } = state;
const { past, future } = history;
switch (action.type) {
case 'UNDO':
const previous = past[past.length - 1];
const newPast = past.slice(0, past.length - 1);
return {
present: previous,
history: {
past: newPast,
future: [present, ...future]
}
};
case 'REDO':
const next = future[0];
const newFuture = future.slice(1);
return {
present: next,
history: {
past: [...past, present],
future: newFuture
}
};
default:
// (?) How do we handle other actions?
return state;
}
}
`
This implementation isn’t usable because it leaves out three important questions:
- Where do we get the initial
presentstate from? We don’t seem to know it beforehand. - Where do we react to the external actions to save the
presentto thepast? - How do we actually delegate the control over the
presentstate to a custom reducer?
It seems that reducer isn’t the right abstraction, but we’re very close.
Meet Reducer Enhancers
You might be familiar with higher order functions. If you use React, you might be familiar with higher order components. Here is a variation on the same pattern, applied to reducers.
A reducer enhancer (or a higher order reducer) is a function that takes a reducer, and returns a new reducer that is able to handle new actions, or to hold more state, delegating control to the inner reducer for the actions it doesn’t understand. This isn’t a new pattern—technically, combineReducers() is also a reducer enhancer because it takes reducers and returns a new reducer.
A reducer enhancer that doesn’t do anything looks like this:
function doNothingWith(reducer) {
return function (state, action) {
// Just call the passed reducer
return reducer(state, action);
};
}
A reducer enhancer that combines other reducers might look like this:
function combineReducers(reducers) {
return function (state = {}, action) {
return Object.keys(reducers).reduce((nextState, key) => {
// Call every reducer with the part of the state it manages
nextState[key] = reducers[key](state[key], action);
return nextState;
}, {});
};
}
Second Attempt: Writing a Reducer Enhancer
Now that we have a better understanding of reducer enhancers, we can see that this is exactly what undoable should have been:
function undoable(reducer) {
// Call the reducer with empty action to populate the initial state
const initialState = {
present: reducer(undefined, {}),
history: {
past: [],
future: []
}
};
// Return a reducer that handles undo and redo
return function (state = initialState, action) {
const { present, history } = state;
const { past, future } = history;
switch (action.type) {
case 'UNDO':
const previous = past[past.length - 1];
const newPast = past.slice(0, past.length - 1);
return {
present: previous,
history: {
past: newPast,
future: [present, ...future]
}
};
case 'REDO':
const next = future[0];
const newFuture = future.slice(1);
return {
present: next,
history: {
past: [...past, present],
future: newFuture
}
};
default:
// Delegate handling the action to the passed reducer
const newPresent = reducer(present, action);
if (present === newPresent) {
return state;
}
return {
present: newPresent,
history: {
past: [...past, present],
future: []
}
};
}
};
}
We can now wrap any reducer into undoable reducer enhancer to teach it to react to UNDO and REDO actions.
// This is a reducer
function todos(state = [], action) {
/* ... */
}
// This is also a reducer!
const undoableTodos = undoable(todos);
import { createStore } from 'redux';
const store = createStore(undoableTodos);
store.dispatch({
type: 'ADD_TODO',
text: 'Use Redux'
});
store.dispatch({
type: 'ADD_TODO',
text: 'Implement Undo'
});
store.dispatch({
type: 'UNDO'
});
There is an important gotcha: you need to remember to append .present to the current state when you retrieve it. You may also check .history.past.length and .history.future.length to determine whether to enable or to disable the Undo and Redo buttons, respectively.
You might have heard that Redux was influenced by Elm Architecture. It shouldn’t come as a surprise that this example is very similar to elm-undo-redo package.
Using Redux Undo
This was all very informative, but can’t we just drop a library and use it instead of implementing undoable ourselves? Sure, we can! Meet Redux Undo, a library that provides simple Undo and Redo functionality for any part of your Redux tree.
In this part of the recipe, you will learn how to make the Todo List example undoable. You can find the full source of this recipe in the todos-with-undo example that comes with Redux.
Installation
First of all, you need to run
npm install --save redux-undo
This installs the package that provides the undoable reducer enhancer.
Wrapping the Reducer
You will need to the reducer you wish to enhance with undoable function. For example, if you use combineReducers(), your code might look like this:
reducers.js
import undoable, { distinctState } from 'redux-undo';
/* ... */
const todoApp = combineReducers({
visibilityFilter,
todos: undoable(todos, { filter: distinctState() })
});
The distinctState() filter serves to ignore the actions that didn’t result in a state change. There are many other options to configure your undoable reducer, like setting the action type for Undo and Redo actions.
You may wrap one or more reducers in undoable at any level of the reducer composition hierarchy. We choose to wrap todos instead of the top-level combined reducer so that changes to visibilityFilter are not reflected in the undo history.
Updating the Selectors
Now the todos part of the state looks like this:
{
visibilityFilter: 'SHOW_ALL',
todos: {
present: [{ text: 'Use Redux', complete: true }, { text: 'Implement Undo' }],
history: {
past: [
[],
[{ text: 'Use Redux' }],
[{ text: 'Use Redux', complete: true }]
],
future: [
[{ text: 'Use Redux', complete: true }, { text: 'Implement Undo', complete: true }]
]
}
}
}
This means you need to access your state with state.todos.present instead of
just state.todos:
containers/App.js
function select(state) {
const presentTodos = state.todos.present;
return {
visibleTodos: selectTodos(presentTodos, state.visibilityFilter),
visibilityFilter: state.visibilityFilter
};
}
In order to disable the Undo and Redo buttons when there is nothing to undo or redo, you need to check whether the past and future in the history are empty:
containers/App.js
function select(state) {
const presentTodos = state.todos.present;
const undoHistory = state.todos.history;
return {
undoDisabled: undoHistory.past.length === 0,
redoDisabled: undoHistory.future.length === 0,
visibleTodos: selectTodos(presentTodos, state.visibilityFilter),
visibilityFilter: state.visibilityFilter
};
}
Adding the Buttons
Now all you need to do is add the buttons for the Undo and Redo actions.
First of all, you need to import ActionCreators from redux-undo and pass their bound versions to the Footer component:
containers/App.js
import { ActionCreators } from 'redux-undo';
/* ... */
class App extends Component {
render() {
const { dispatch, visibleTodos, visibilityFilter } = this.props;
return (
<div>
{/* ... */}
<Footer
filter={visibilityFilter}
onFilterChange={nextFilter => dispatch(setVisibilityFilter(nextFilter))}
onUndo={() => dispatch(ActionCreators.undo())}
onRedo={() => dispatch(ActionCreators.redo())}
undoDisabled={this.props.undoDisabled}
redoDisabled={this.props.redoDisabled} />
</div>
);
}
}
Now you can render the buttons in the footer:
components/Footer.js
export default class Footer extends Component {
/* ... */
renderUndo() {
return (
<p>
<button onClick={this.props.onUndo} disabled={this.props.undoDisabled}>Undo</button>
<button onClick={this.props.onRedo} disabled={this.props.redoDisabled}>Redo</button>
</p>
);
}
render() {
return (
<div>
{this.renderFilters()}
{this.renderUndo()}
</div>
);
}
}
This is it! Run npm install and npm start in the example folder and try it out!