|
1 | | -import {Component} from 'angular2/core'; |
| 1 | +import {Input, Component, OnInit} from 'angular2/core'; |
| 2 | +import {CORE_DIRECTIVES} from 'angular2/common'; |
| 3 | +import {Ng2TreeService} from './ng2-tree.service'; |
| 4 | +import {EditableNodeDirective} from './editable-node.directive.ts'; |
2 | 5 |
|
3 | 6 | @Component({ |
4 | 7 | selector: 'ng2-tree', |
5 | 8 | template: ` |
6 | | - <pre> |
7 | | - [-] A |
8 | | - | |
9 | | - +-----B |
10 | | - | |
11 | | - +-----C |
12 | | - | |
13 | | - +-----[-] D |
14 | | - | | |
15 | | - | +-----X |
16 | | - | | |
17 | | - | +-----Y |
18 | | - | |
19 | | - [+] |
20 | | - </pre> |
21 | | - ` |
| 9 | + <ul class="tree" (keyup)="cancelActions($event)"> |
| 10 | + <li> |
| 11 | + <div (mouseup)="showMenu($event)"> |
| 12 | + <span class="folding" (click)="switchFolding($event, tree)" [ngClass]="foldingType(tree)"></span> |
| 13 | + <span class="node-value" *ngIf="!edit">{{tree.value}}</span> |
| 14 | + <input type="text" class="node-value" editable [nodeValue]="tree.value" (valueChanged)="applyNewValue($event, tree)" *ngIf="edit"/> |
| 15 | + </div> |
| 16 | + <div class="node-menu" *ngIf="isMenuVisible"> |
| 17 | + <ul class="node-menu-content"> |
| 18 | + <li (click)="rename($event, tree)">Rename node</li> |
| 19 | + <li>Add node</li> |
| 20 | + <li>Remove node</li> |
| 21 | + </ul> |
| 22 | + </div> |
| 23 | + <ng2-tree *ngFor="#child of tree.children" [tree]="child"></ng2-tree> |
| 24 | + </li> |
| 25 | + </ul> |
| 26 | + `, |
| 27 | + directives: [EditableNodeDirective, Ng2Tree, CORE_DIRECTIVES] |
22 | 28 | }) |
23 | | -export class Ng2Tree { |
| 29 | +export class Ng2Tree implements OnInit { |
| 30 | + private static COMPONENT_TAG_NAME: string = 'NG2-TREE'; |
| 31 | + private static FOLDING_NODE_EXPANDED: string = 'node-expanded'; |
| 32 | + private static FOLDING_NODE_COLLAPSED: string = 'node-collapsed'; |
| 33 | + private static FOLDING_NODE_LEAF: string = 'node-leaf'; |
24 | 34 |
|
| 35 | + @Input() |
| 36 | + private tree: any; |
| 37 | + |
| 38 | + private treeService: Ng2TreeService; |
| 39 | + private isMenuVisible: boolean = false; |
| 40 | + private edit: boolean = false; |
| 41 | + |
| 42 | + constructor(treeService: Ng2TreeService) { |
| 43 | + this.treeService = treeService; |
| 44 | + } |
| 45 | + |
| 46 | + private switchFolding($event: any, tree: any): void { |
| 47 | + console.log('fold'); |
| 48 | + this.handleFoldingType($event.target.parentNode.parentNode, tree); |
| 49 | + } |
| 50 | + |
| 51 | + private foldingType(node: any): any { |
| 52 | + if (node.foldingType) { |
| 53 | + return node.foldingType; |
| 54 | + } |
| 55 | + |
| 56 | + if (node.children) { |
| 57 | + node.foldingType = Ng2Tree.FOLDING_NODE_EXPANDED; |
| 58 | + } else { |
| 59 | + node.foldingType = Ng2Tree.FOLDING_NODE_LEAF; |
| 60 | + } |
| 61 | + |
| 62 | + return node.foldingType; |
| 63 | + } |
| 64 | + |
| 65 | + private nextFoldingType(node: any): string { |
| 66 | + if (node.foldingType === Ng2Tree.FOLDING_NODE_EXPANDED) { |
| 67 | + return Ng2Tree.FOLDING_NODE_COLLAPSED; |
| 68 | + } |
| 69 | + |
| 70 | + return Ng2Tree.FOLDING_NODE_EXPANDED; |
| 71 | + } |
| 72 | + |
| 73 | + private handleFoldingType(parent: any, node: any) { |
| 74 | + if (node.foldingType === Ng2Tree.FOLDING_NODE_LEAF) { |
| 75 | + return; |
| 76 | + } |
| 77 | + |
| 78 | + let display = 'block'; |
| 79 | + if (node.foldingType === Ng2Tree.FOLDING_NODE_EXPANDED) { |
| 80 | + display = 'none'; |
| 81 | + } |
| 82 | + |
| 83 | + node.foldingType = this.nextFoldingType(node); |
| 84 | + for (let element of parent.childNodes) { |
| 85 | + if (element.nodeName === Ng2Tree.COMPONENT_TAG_NAME) { |
| 86 | + element.style.display = display; |
| 87 | + } |
| 88 | + } |
| 89 | + } |
| 90 | + |
| 91 | + private rename($event: any, node: any) { |
| 92 | + if ($event.which === 1) { |
| 93 | + this.edit = true; |
| 94 | + this.isMenuVisible = false; |
| 95 | + } |
| 96 | + } |
| 97 | + |
| 98 | + private showMenu($event: any): void { |
| 99 | + if ($event.which === 3) { |
| 100 | + this.isMenuVisible = !this.isMenuVisible; |
| 101 | + this.treeService.emitMenuEvent({sender: this, action: 'close'}) |
| 102 | + } |
| 103 | + $event.preventDefault(); |
| 104 | + } |
| 105 | + |
| 106 | + private cancelActions($event: any): void { |
| 107 | + |
| 108 | + } |
| 109 | + |
| 110 | + private applyNewValue($event: any, node: any): void { |
| 111 | + node.value = $event.value; |
| 112 | + this.edit = false; |
| 113 | + } |
| 114 | + |
| 115 | + ngOnInit(): void { |
| 116 | + this.treeService.menuEventStream() |
| 117 | + .subscribe(menuEvent => { |
| 118 | + if (menuEvent.sender !== this && menuEvent.action === 'close') { |
| 119 | + this.isMenuVisible = false; |
| 120 | + } |
| 121 | + }) |
| 122 | + } |
25 | 123 | } |
0 commit comments