route.js 945 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  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 cursor;
  16. for (let i = 0, len = routes.length; i < len; i++) {
  17. const item = routes[i].split('/');
  18. cursor = obj;
  19. for (let j = 0, lenj = item.length; j < lenj; j++) {
  20. const seg = item[j];
  21. cursor[seg] = cursor[seg] || {};
  22. cursor = cursor[seg];
  23. }
  24. }
  25. return obj;
  26. }
  27. function buildNodes(tree) {
  28. const nodes = [];
  29. for (const [key, item] of Object.entries(tree)) {
  30. if (Object.keys(item).length) {
  31. nodes.push({
  32. label: key,
  33. nodes: buildNodes(item)
  34. });
  35. } else {
  36. nodes.push(key);
  37. }
  38. }
  39. return nodes;
  40. }
  41. module.exports = listRoute;