nits

facebook · May 27, 2024 · beff622 · beff622
1 parent 50c168b
commit beff622
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/packages/lexical-playground/src/plugins/CollapsiblePlugin/CollapsibleContentNode.ts b/packages/lexical-playground/src/plugins/CollapsiblePlugin/CollapsibleContentNode.ts
@@ -16,8 +16,9 @@ import {
   LexicalNode,
   SerializedElementNode,
 } from 'lexical';
+import invariant from 'shared/invariant';
 
-import {CollapsibleContainerNode} from './CollapsibleContainerNode';
+import {$isCollapsibleContainerNode} from './CollapsibleContainerNode';
 
 type SerializedCollapsibleContentNode = SerializedElementNode;
 
@@ -43,7 +44,11 @@ export class CollapsibleContentNode extends ElementNode {
     const dom = document.createElement('div');
     dom.classList.add('Collapsible__content');
     editor.getEditorState().read(() => {
-      const containerNode = this.getParentOrThrow<CollapsibleContainerNode>();
+      const containerNode = this.getParentOrThrow();
+      invariant(
+        $isCollapsibleContainerNode(containerNode),
+        'Expected parent node to be a CollapsibleContainerNode',
+      );
       if (!containerNode.__open) {
         dom.style.display = 'none';
       }

diff --git a/packages/lexical-playground/src/plugins/CollapsiblePlugin/CollapsibleTitleNode.ts b/packages/lexical-playground/src/plugins/CollapsiblePlugin/CollapsibleTitleNode.ts
@@ -51,7 +51,7 @@ export class CollapsibleTitleNode extends ElementNode {
         const collapsibleContainer = this.getLatest().getParentOrThrow();
         invariant(
           $isCollapsibleContainerNode(collapsibleContainer),
-          'something',
+          'Expected parent node to be a CollapsibleContainerNode',
         );
         collapsibleContainer.toggleOpen();
       });