refactor: reuse method to destroy view (#9949)

taiga-family · Dec 12, 2024 · 16c4f3a · 16c4f3a
1 parent 1144fae
commit 16c4f3a
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/projects/cdk/classes/portals.ts b/projects/cdk/classes/portals.ts
@@ -1,5 +1,11 @@
 /// <reference types="@taiga-ui/tsconfig/ng-dev-mode" />
-import type {ComponentRef, EmbeddedViewRef, Provider, TemplateRef} from '@angular/core';
+import type {
+    ComponentRef,
+    EmbeddedViewRef,
+    Provider,
+    TemplateRef,
+    ViewRef,
+} from '@angular/core';
 import {
     Directive,
     inject,
@@ -56,16 +62,14 @@ export abstract class TuiPortalService {
     }
 
     public remove<C>({hostView}: ComponentRef<C>): void {
-        if (!hostView.destroyed) {
-            hostView.destroy();
-        }
+        this.removeTemplate(hostView);
     }
 
     public addTemplate<C>(templateRef: TemplateRef<C>, context?: C): EmbeddedViewRef<C> {
         return this.safeHost.addTemplateChild(templateRef, context);
     }
 
-    public removeTemplate<C>(viewRef: EmbeddedViewRef<C>): void {
+    public removeTemplate<C>(viewRef: EmbeddedViewRef<C> | ViewRef): void {
         if (!viewRef.destroyed) {
             viewRef.destroy();
         }