route.js 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. 'use strict';
  2. const archy = require('archy');
  3. function listRoute() {
  4. const routes = this.route.list().sort();
  5. const tree = buildTree(routes);
  6. const nodes = buildNodes(tree);
  7. const s = archy({
  8. label: `Total: ${routes.length}`,
  9. nodes
  10. });
  11. console.log(s);
  12. }
  13. function buildTree(routes) {
  14. const obj = {};
  15. let item, j, lenj, seg, cursor;
  16. for (let i = 0, len = routes.length; i < len; i++) {
  17. item = routes[i].split('/');
  18. cursor = obj;
  19. for (j = 0, lenj = item.length; j < lenj; j++) {
  20. seg = item[j];
  21. cursor = cursor[seg] = cursor[seg] || {};
  22. }
  23. }
  24. return obj;
  25. }
  26. function buildNodes(tree) {
  27. const keys = Object.keys(tree);
  28. const nodes = [];
  29. let key, item;
  30. for (let i = 0, len = keys.length; i < len; i++) {
  31. key = keys[i];
  32. item = tree[key];
  33. if (Object.keys(item).length) {
  34. nodes.push({
  35. label: key,
  36. nodes: buildNodes(item)
  37. });
  38. } else {
  39. nodes.push(key);
  40. }
  41. }
  42. return nodes;
  43. }
  44. module.exports = listRoute;