125 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			125 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| "use strict";
 | |
| 
 | |
| Object.defineProperty(exports, "__esModule", {
 | |
|   value: true
 | |
| });
 | |
| exports.default = void 0;
 | |
| 
 | |
| var _Node = _interopRequireDefault(require("./Node"));
 | |
| 
 | |
| function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 | |
| 
 | |
| class BaseFolder extends _Node.default {
 | |
|   constructor(name, parent) {
 | |
|     super(name, parent);
 | |
|     this.children = Object.create(null);
 | |
|   }
 | |
| 
 | |
|   get src() {
 | |
|     if (!Object.prototype.hasOwnProperty.call(this, '_src')) {
 | |
|       this._src = this.walk((node, src) => src += node.src || '', '', false);
 | |
|     }
 | |
| 
 | |
|     return this._src;
 | |
|   }
 | |
| 
 | |
|   get size() {
 | |
|     if (!Object.prototype.hasOwnProperty.call(this, '_size')) {
 | |
|       this._size = this.walk((node, size) => size + node.size, 0, false);
 | |
|     }
 | |
| 
 | |
|     return this._size;
 | |
|   }
 | |
| 
 | |
|   getChild(name) {
 | |
|     return this.children[name];
 | |
|   }
 | |
| 
 | |
|   addChildModule(module) {
 | |
|     const {
 | |
|       name
 | |
|     } = module;
 | |
|     const currentChild = this.children[name]; // For some reason we already have this node in children and it's a folder.
 | |
| 
 | |
|     if (currentChild && currentChild instanceof BaseFolder) return;
 | |
| 
 | |
|     if (currentChild) {
 | |
|       // We already have this node in children and it's a module.
 | |
|       // Merging it's data.
 | |
|       currentChild.mergeData(module.data);
 | |
|     } else {
 | |
|       // Pushing new module
 | |
|       module.parent = this;
 | |
|       this.children[name] = module;
 | |
|     }
 | |
| 
 | |
|     delete this._size;
 | |
|     delete this._src;
 | |
|   }
 | |
| 
 | |
|   addChildFolder(folder) {
 | |
|     folder.parent = this;
 | |
|     this.children[folder.name] = folder;
 | |
|     delete this._size;
 | |
|     delete this._src;
 | |
|     return folder;
 | |
|   }
 | |
| 
 | |
|   walk(walker, state = {}, deep = true) {
 | |
|     let stopped = false;
 | |
|     Object.values(this.children).forEach(child => {
 | |
|       if (deep && child.walk) {
 | |
|         state = child.walk(walker, state, stop);
 | |
|       } else {
 | |
|         state = walker(child, state, stop);
 | |
|       }
 | |
| 
 | |
|       if (stopped) return false;
 | |
|     });
 | |
|     return state;
 | |
| 
 | |
|     function stop(finalState) {
 | |
|       stopped = true;
 | |
|       return finalState;
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   mergeNestedFolders() {
 | |
|     if (!this.isRoot) {
 | |
|       let childNames;
 | |
| 
 | |
|       while ((childNames = Object.keys(this.children)).length === 1) {
 | |
|         const childName = childNames[0];
 | |
|         const onlyChild = this.children[childName];
 | |
| 
 | |
|         if (onlyChild instanceof this.constructor) {
 | |
|           this.name += `/${onlyChild.name}`;
 | |
|           this.children = onlyChild.children;
 | |
|         } else {
 | |
|           break;
 | |
|         }
 | |
|       }
 | |
|     }
 | |
| 
 | |
|     this.walk(child => {
 | |
|       child.parent = this;
 | |
| 
 | |
|       if (child.mergeNestedFolders) {
 | |
|         child.mergeNestedFolders();
 | |
|       }
 | |
|     }, null, false);
 | |
|   }
 | |
| 
 | |
|   toChartData() {
 | |
|     return {
 | |
|       label: this.name,
 | |
|       path: this.path,
 | |
|       statSize: this.size,
 | |
|       groups: Object.values(this.children).map(child => child.toChartData())
 | |
|     };
 | |
|   }
 | |
| 
 | |
| }
 | |
| 
 | |
| exports.default = BaseFolder;
 | |
| ; |