| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file | |
| 2 // for details. All rights reserved. Use of this source code is governed by a | |
| 3 // BSD-style license that can be found in the LICENSE file. | |
| 4 | |
| 5 /// A simple library for rendering a list of files as a directory tree. | |
| 6 library directory_tree; | |
| 7 | |
| 8 import 'package:pathos/path.dart' as path; | |
| 9 | |
| 10 import 'log.dart' as log; | |
| 11 | |
| 12 /// Draws a directory tree for the given list of files. Given a list of files | |
| 13 /// like: | |
| 14 /// | |
| 15 /// TODO | |
| 16 /// example/console_example.dart | |
| 17 /// example/main.dart | |
| 18 /// example/web copy/web_example.dart | |
| 19 /// test/absolute_test.dart | |
| 20 /// test/basename_test.dart | |
| 21 /// test/dirname_test.dart | |
| 22 /// test/extension_test.dart | |
| 23 /// test/is_absolute_test.dart | |
| 24 /// test/is_relative_test.dart | |
| 25 /// test/join_test.dart | |
| 26 /// test/normalize_test.dart | |
| 27 /// test/relative_test.dart | |
| 28 /// test/split_test.dart | |
| 29 /// .gitignore | |
| 30 /// README.md | |
| 31 /// lib/path.dart | |
| 32 /// pubspec.yaml | |
| 33 /// test/all_test.dart | |
| 34 /// test/path_posix_test.dart | |
| 35 /// test/path_windows_test.dart | |
| 36 /// | |
| 37 /// this will render: | |
| 38 /// | |
| 39 /// |-- .gitignore | |
| 40 /// |-- README.md | |
| 41 /// |-- TODO | |
| 42 /// |-- example | |
| 43 /// | |-- console_example.dart | |
| 44 /// | |-- main.dart | |
| 45 /// | '-- web copy | |
| 46 /// | '-- web_example.dart | |
| 47 /// |-- lib | |
| 48 /// | '-- path.dart | |
| 49 /// |-- pubspec.yaml | |
| 50 /// '-- test | |
| 51 /// |-- absolute_test.dart | |
| 52 /// |-- all_test.dart | |
| 53 /// |-- basename_test.dart | |
| 54 /// | (7 more...) | |
| 55 /// |-- path_windows_test.dart | |
| 56 /// |-- relative_test.dart | |
| 57 /// '-- split_test.dart | |
| 58 /// | |
| 59 String generateTree(List<String> files) { | |
| 60 // Parse out the files into a tree of nested maps. | |
| 61 var root = {}; | |
| 62 for (var file in files) { | |
| 63 var parts = path.split(file); | |
| 64 var directory = root; | |
| 65 for (var part in path.split(file)) { | |
| 66 directory = directory.putIfAbsent(part, () => {}); | |
| 67 } | |
| 68 } | |
| 69 | |
| 70 // Walk the map recursively and render to a string. | |
| 71 var buffer = new StringBuffer(); | |
| 72 _draw(buffer, '', false, null, root); | |
| 73 return buffer.toString(); | |
| 74 } | |
| 75 | |
| 76 void _drawLine(StringBuffer buffer, String prefix, bool isLastChild, | |
| 77 String name) { | |
| 78 // Print lines. | |
| 79 buffer.write(prefix); | |
| 80 if (name != null) { | |
| 81 if (isLastChild) { | |
| 82 buffer.write("'-- "); | |
| 83 } else { | |
| 84 buffer.write("|-- "); | |
| 85 } | |
| 86 } | |
| 87 | |
| 88 // Print name. | |
| 89 buffer.writeln(name); | |
| 90 } | |
| 91 | |
| 92 String _getPrefix(bool isRoot, bool isLast) { | |
| 93 if (isRoot) return ""; | |
| 94 if (isLast) return " "; | |
| 95 return "| "; | |
| 96 } | |
| 97 | |
| 98 void _draw(StringBuffer buffer, String prefix, bool isLast, | |
| 99 String name, Map children) { | |
| 100 // Don't draw a line for the root node. | |
| 101 if (name != null) _drawLine(buffer, prefix, isLast, name); | |
| 102 | |
| 103 // Recurse to the children. | |
| 104 var childNames = new List.from(children.keys); | |
| 105 childNames.sort(); | |
| 106 | |
| 107 _drawChild(bool isLastChild, String child) { | |
| 108 var childPrefix = _getPrefix(name == null, isLast); | |
| 109 _draw(buffer, '$prefix$childPrefix', isLastChild, child, children[child]); | |
| 110 } | |
| 111 | |
| 112 if (name == null || childNames.length <= 10) { | |
| 113 // Not too many, so show all the children. | |
| 114 for (var i = 0; i < childNames.length; i++) { | |
| 115 _drawChild(i == childNames.length - 1, childNames[i]); | |
| 116 } | |
| 117 } else { | |
| 118 // Show the first few. | |
| 119 _drawChild(false, childNames[0]); | |
| 120 _drawChild(false, childNames[1]); | |
| 121 _drawChild(false, childNames[2]); | |
| 122 | |
| 123 // Elide the middle ones. | |
| 124 buffer.write(prefix); | |
| 125 buffer.write(_getPrefix(name == null, isLast)); | |
| 126 buffer.writeln('| (${childNames.length - 6} more...)'); | |
| 127 | |
| 128 // Show the last few. | |
| 129 _drawChild(false, childNames[childNames.length - 3]); | |
| 130 _drawChild(false, childNames[childNames.length - 2]); | |
| 131 _drawChild(true, childNames[childNames.length - 1]); | |
| 132 } | |
| 133 } | |
| OLD | NEW |