fix(components): tree list item js error (#2071)

Teradata · Nov 6, 2023 · 2ee44df · 2ee44df
1 parent d81f657
commit 2ee44df
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/libs/components/src/tree-list/tree-list-item.ts b/libs/components/src/tree-list/tree-list-item.ts
@@ -3,6 +3,8 @@ import { customElement, property } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
 import styles from './tree-list-item.scss?inline';
 
+import '../icon/icon';
+
 declare global {
   interface HTMLElementTagNameMap {
     'cv-tree-list-item': CovalentTreeListItem;
@@ -82,4 +84,4 @@ export class CovalentTreeListItem extends LitElement {
   }
 }
 
-export default CovalentTreeListItem;
+export default CovalentTreeListItem;
diff --git a/libs/components/src/tree-list/tree-list.ts b/libs/components/src/tree-list/tree-list.ts
@@ -32,22 +32,22 @@ document.addEventListener('select', (e: Event) => {
 const handleSelect = (e: Event): void => {
   // All cv-tree-list-item components.
   const items: any[] = Array.from(
-    document.querySelectorAll('cv-tree-list-item')
+    document?.querySelectorAll('cv-tree-list-item')
   );
 
   // Currently selected item.
   const target: HTMLElement = e.target as HTMLElement;
-  const current = target.shadowRoot!.querySelector('div.itemContent');
+  const current = target.shadowRoot?.querySelector('div.itemContent');
 
   // Find previously selected element and remove the styling.
   items.forEach((item) => {
     item
-      .shadowRoot!.querySelector('div.itemContent')
+      .shadowRoot?.querySelector('div.itemContent')
       ?.classList.remove('selected');
   });
 
   // Add the selected styling to the currently selected item.
-  current!.classList.add('selected');
+  current?.classList.add('selected');
 };
 
 export default CovalentTreeList;