From e8ec2816e2f4690f97cc570dd4ef3d0b49eb4122 Mon Sep 17 00:00:00 2001
From: mehmet-erim <bilgi@mehmeterim.com>
Date: Fri, 17 Jul 2020 15:47:14 +0300
Subject: [PATCH] feat(components/tree): add an input named beforeDrop

---
 .../components/tree/src/lib/components/tree.component.html  | 1 +
 .../components/tree/src/lib/components/tree.component.ts    | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.html b/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.html
index 019c2bd0fa..b7c777a449 100644
--- a/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.html
+++ b/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.html
@@ -1,4 +1,5 @@
 <nz-tree
+  [nzBeforeDrop]="beforeDrop"
   [nzDraggable]="draggable"
   [nzCheckStrictly]="checkStrictly"
   [nzCheckable]="checkable"
diff --git a/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.ts b/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.ts
index 90a99d1546..5c39c47057 100644
--- a/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.ts
+++ b/npm/ng-packs/packages/components/tree/src/lib/components/tree.component.ts
@@ -8,6 +8,7 @@ import {
   ViewEncapsulation,
 } from '@angular/core';
 import { NzFormatEmitEvent } from 'ng-zorro-antd/tree';
+import { of } from 'rxjs';
 
 @Component({
   selector: 'abp-tree',
@@ -23,7 +24,7 @@ export class TreeComponent {
   @Output() readonly checkedKeysChange = new EventEmitter();
   @Output() readonly expandedKeysChange = new EventEmitter<string[]>();
   @Output() readonly selectedNodeChange = new EventEmitter();
-  @Output() readonly drop = new EventEmitter<NzFormatEmitEvent>();
+  @Output() readonly dropOver = new EventEmitter<NzFormatEmitEvent>();
   @Input() draggable: boolean;
   @Input() checkable: boolean;
   @Input() checkStrictly: boolean;
@@ -32,6 +33,7 @@ export class TreeComponent {
   @Input() expandedKeys: string[] = [];
   @Input() selectedNode: any;
   @Input() isNodeSelected = node => this.selectedNode?.id === node.key;
+  @Input() beforeDrop = () => of(false);
 
   onSelectedNodeChange(node) {
     this.selectedNode = node.origin.entity;
@@ -52,6 +54,6 @@ export class TreeComponent {
     event.event.stopPropagation();
     event.event.preventDefault();
 
-    this.drop.emit(event);
+    this.dropOver.emit(event);
   }
 }