import { IApi } from 'umi-types';
import assert from 'assert';
import exclude from './exclude';

function optsToArray(item) {
  if (!item) return [];
  if (Array.isArray(item)) {
    return item;
  } else {
    return [item];
  }
}

interface IOpts {
  exclude: string[] | string;
  update: Function;
}

export default function(api: IApi, opts: IOpts) {
  // disable if routes if configured
  if (api.config.routes) return;

  api.onOptionChange(newOpts => {
    opts = newOpts;
    api.rebuildTmpFiles();
  });

  api.modifyRoutes(routes => {
    routes = exclude(routes, optsToArray(opts.exclude));

    if (opts.update) {
      assert(
        typeof opts.update === 'function',
        `opts.update should be function, but got ${opts.update}`,
      );
      routes = opts.update(routes);
    }

    return routes;
  });
}