|
| 1 | +import {Injectable} from '@angular/core'; |
| 2 | +import {fromEvent, merge, NEVER, Observable, Subject} from 'rxjs'; |
| 3 | +import {filter, mapTo, switchMap} from 'rxjs/operators'; |
| 4 | + |
| 5 | +enum ChangeAction { |
| 6 | + ADD, |
| 7 | + REMOVE |
| 8 | +} |
| 9 | + |
| 10 | +interface SelectionChange { |
| 11 | + item: Node; |
| 12 | + action: ChangeAction; |
| 13 | +} |
| 14 | + |
| 15 | +@Injectable({ |
| 16 | + providedIn: 'root' |
| 17 | +}) |
| 18 | +export class SelectionService { |
| 19 | + |
| 20 | + private COMMAND_KEY = 'Meta'; |
| 21 | + private CONTROL_KEY = 'Control'; |
| 22 | + |
| 23 | + private selectionChange$ = new Subject<SelectionChange>(); |
| 24 | + private selectedElements: Node[] = []; |
| 25 | + |
| 26 | + constructor() { |
| 27 | + const selectionKeyPressed$ = this.selectionKeyPressed(); |
| 28 | + selectionKeyPressed$.pipe( |
| 29 | + switchMap((pressed) => pressed ? this.selectionChange$ : NEVER) |
| 30 | + ).subscribe((selectionChange: SelectionChange) => this.handleSelectionChange(selectionChange)); |
| 31 | + } |
| 32 | + |
| 33 | + private handleSelectionChange(selectionChange: SelectionChange): void { |
| 34 | + if (selectionChange.action === ChangeAction.ADD) { |
| 35 | + this.selectedElements.push(selectionChange.item); |
| 36 | + } |
| 37 | + if (selectionChange.action === ChangeAction.REMOVE) { |
| 38 | + this.selectedElements = this.selectedElements.filter( |
| 39 | + (element: Node) => element !== selectionChange.item |
| 40 | + ); |
| 41 | + } |
| 42 | + } |
| 43 | + |
| 44 | + private selectionKeyPressed(): Observable<boolean> { |
| 45 | + const selectionKeyPressed = fromEvent(window, 'keydown').pipe( |
| 46 | + filter((keyboardEvent: KeyboardEvent) => keyboardEvent.key === this.COMMAND_KEY || keyboardEvent.key === this.CONTROL_KEY), |
| 47 | + mapTo(true) |
| 48 | + ); |
| 49 | + const keyup = fromEvent(window, 'keyup').pipe(mapTo(false)); |
| 50 | + return merge(selectionKeyPressed, keyup); |
| 51 | + } |
| 52 | + |
| 53 | + public updateSelectedDragItem(item: Node, selected: boolean): void { |
| 54 | + this.selectionChange$.next({ |
| 55 | + item, action: selected ? ChangeAction.ADD : ChangeAction.REMOVE |
| 56 | + }); |
| 57 | + } |
| 58 | + |
| 59 | + public getSelectedElements(): Node[] { |
| 60 | + return this.selectedElements; |
| 61 | + } |
| 62 | +} |
0 commit comments