fix: allow users to pass template for label (#2923)

Signed-off-by: Akshat Patel <[email protected]>

Author: Akshat55

src/treeview/tree-node.component.ts

@@ -80,7 +80,11 @@ import { Node } from "./tree-node.types";
 							size="16">
 						</svg>
 					</ng-container>
-					<ng-template *ngIf="isTemplate(icon)" [ngTemplateOutlet]="icon"></ng-template>
+					<ng-template
+						*ngIf="isTemplate(icon)"
+						[ngTemplateOutlet]="icon"
+						[ngTemplateOutletContext]="{ $implicit: iconContext }">
+					</ng-template>
 					{{label}}
 				</span>
 			</div>
@@ -113,6 +117,7 @@ export class TreeNodeComponent implements AfterContentChecked, OnInit, OnDestroy
 	@Input() selected = false;
 	@Input() value;
 	@Input() icon: string | TemplateRef<any>;
+	@Input() iconContext: any;
 	@Input() children: Node[] = [];
 
 	/**
@@ -138,6 +143,7 @@ export class TreeNodeComponent implements AfterContentChecked, OnInit, OnDestroy
 		this.selected = node.selected ?? this.selected;
 		this.depth = node.depth ?? this.depth;
 		this.children = node.children ?? this.children;
+		this.iconContext = node.iconText ?? this.iconContext;
 	}
 
 	get node() {

src/treeview/tree-node.types.ts

@@ -9,6 +9,7 @@ export interface Node {
 	expanded?: boolean;
 	selected?: boolean;
 	icon?: string | TemplateRef<any>;
+	iconContext?: any;
 	children?: Node[];
 	[key: string]: any;
 }

src/treeview/treeview.component.ts

@@ -32,7 +32,12 @@ import { TreeViewService } from "./treeview.service";
 			*ngIf="label"
 			[id]="id"
 			class="cds--label">
-			{{label}}
+			<ng-container *ngIf="!isTemplate(label)">{{label}}</ng-container>
+			<ng-template
+				*ngIf="isTemplate(label)"
+				[ngTemplateOutlet]="label"
+				[ngTemplateOutletContext]="{ $implicit: labelContext }">
+			</ng-template>
 		</label>
 		<div
 			class="cds--tree"
@@ -80,6 +85,10 @@ export class TreeViewComponent implements AfterViewInit, OnInit, OnDestroy {
 	 * Tree view label
 	 */
 	@Input() label: string | TemplateRef<any>;
+	/**
+	 * Optional context for label if it's a template
+	 */
+	@Input() labelContext: any;
 	/**
 	 * Specify the size of the list items in the tree
 	 */
@@ -154,6 +163,10 @@ export class TreeViewComponent implements AfterViewInit, OnInit, OnDestroy {
 		}
 	}
 
+	public isTemplate(value) {
+		return value instanceof TemplateRef;
+	}
+
 	public isProjected() {
 		return this.treeViewService.contentProjected;
 	}