fix: emit 'unmount'

retejs · Jul 16, 2023 · 4d17a2e · 4d17a2e
1 parent 4032bfa
commit 4d17a2e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/ref.ts b/src/ref.ts
@@ -1,10 +1,10 @@
-import { Input, ElementRef, OnChanges } from '@angular/core';
+import { Input, ElementRef, OnChanges, OnDestroy } from '@angular/core';
 import { Directive } from '@angular/core';
 
 @Directive({
   selector: '[refComponent]'
 })
-export class RefDirective implements OnChanges {
+export class RefDirective implements OnChanges, OnDestroy {
   @Input() data!: any
   @Input() emit!: any
 
@@ -13,4 +13,8 @@ export class RefDirective implements OnChanges {
   ngOnChanges() {
     this.emit({ type: 'render', data: { ...this.data, element: this.el.nativeElement } })
   }
+
+  ngOnDestroy() {
+    this.emit({ type: 'unmount', data: { element: this.el.nativeElement } })
+  }
 }