fbredius/storybook

import { action } from './action';
import { ActionsFunction, ActionOptions, ActionsMap } from '../models';
import { config } from './configureActions';

export const actions: ActionsFunction = (...args: any[]) => {
  let options: ActionOptions = config;
  let names = args;
  // args argument can be a single argument as an array
  if (names.length === 1 && Array.isArray(names[0])) {
    [names] = names;
  }
  // last argument can be options
  if (names.length !== 1 && typeof names[names.length - 1] !== 'string') {
    options = {
      ...config,
      ...names.pop(),
    };
  }

  let namesObject = names[0];
  if (names.length !== 1 || typeof namesObject === 'string') {
    namesObject = {};
    names.forEach((name) => {
      namesObject[name] = name;
    });
  }

  const actionsObject: ActionsMap = {};
  Object.keys(namesObject).forEach((name) => {
    actionsObject[name] = action(namesObject[name], options);
  });
  return actionsObject;
};