From c98dc5fca565be57fe46a3fb87ed19e30cc3e8d3 Mon Sep 17 00:00:00 2001
From: AleksanderBodurri <alex.bodurri@mail.utoronto.ca>
Date: Fri, 31 Jan 2020 14:33:56 -0500
Subject: [PATCH] refactor(devtools): create interface for forest builder
 options object; change patch template back

---
 projects/ng-devtools-backend/src/lib/component-tree.ts | 7 ++++++-
 projects/ng-devtools-backend/src/lib/utils.ts          | 4 ++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/projects/ng-devtools-backend/src/lib/component-tree.ts b/projects/ng-devtools-backend/src/lib/component-tree.ts
index 43ab7035727..65df2e79dbd 100644
--- a/projects/ng-devtools-backend/src/lib/component-tree.ts
+++ b/projects/ng-devtools-backend/src/lib/component-tree.ts
@@ -24,6 +24,11 @@ export interface ComponentTreeNode extends Node<DirectiveInstanceType, Component
   children: ComponentTreeNode[];
 }
 
+export interface DirectiveForestBuilderOptions {
+  getDirectives?: boolean;
+  includeNativeElement?: boolean;
+}
+
 export const getLatestComponentState = (query: ComponentExplorerViewQuery): DirectivesProperties | undefined => {
   let result;
   if (query.selectedElement && query.expandedProperties) {
@@ -86,7 +91,7 @@ export const getComponentForest = (root = document.documentElement): ComponentTr
 const buildDirectiveForest = (
   node: Element,
   tree: ComponentTreeNode | undefined,
-  options: { [option: string]: boolean } = {}
+  options: DirectiveForestBuilderOptions = {}
 ): ComponentTreeNode[] => {
   if (!node) {
     return [tree];
diff --git a/projects/ng-devtools-backend/src/lib/utils.ts b/projects/ng-devtools-backend/src/lib/utils.ts
index 8818583e380..9431ae11338 100644
--- a/projects/ng-devtools-backend/src/lib/utils.ts
+++ b/projects/ng-devtools-backend/src/lib/utils.ts
@@ -12,8 +12,8 @@ export const patchTemplate = (instance: any, fn: () => void) => {
   const metadata = componentMetadata(instance);
   const original = metadata.template;
 
-  metadata.tView.template = metadata.template = (...args) => {
-    const result = original(...args);
+  metadata.tView.template = metadata.template = function() {
+    const result = original.apply(this, arguments);
     fn();
     return result;
   };