zustand middleware for actions

zustand_actions.js

// Extends API with actions
const actionsMiddleware = fn => (set, get, api) => {
  const _actions = {};

  api.actions = () => _actions;

  const state = fn(set, get, api);

  Object.keys(state).forEach(key => {
    if (typeof state[key] === 'function') {
      _actions[key] = state[key];
    }
  });

  return state;
};

// usage
store.actions();