From 8fd3295ba2d0bcb78d08c14d6df74e6aaf4930dd Mon Sep 17 00:00:00 2001
From: * <8>
Date: Thu, 5 May 2022 16:05:10 +0800
Subject: [PATCH] Match-id-210cbd61873caacc32857203dac50d925672c6a5

---
 libs/extension/src/injector/index.ts    | 6 +++++-
 libs/extension/src/parser/parseVNode.ts | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/libs/extension/src/injector/index.ts b/libs/extension/src/injector/index.ts
index 17725af9..58575844 100644
--- a/libs/extension/src/injector/index.ts
+++ b/libs/extension/src/injector/index.ts
@@ -45,7 +45,11 @@ function postMessage(type: string, data) {
 }
 
 function parseCompAttrs(id: number) {
-  const vNode: VNode = queryVNode(id);
+  const vNode = queryVNode(id);
+  if (!vNode) {
+    console.error('Do not find match vNode, this is a bug, please report us');
+    return;
+  }
   const parsedAttrs = parseVNodeAttrs(vNode);
   postMessage(ComponentAttrs, parsedAttrs);
 }
diff --git a/libs/extension/src/parser/parseVNode.ts b/libs/extension/src/parser/parseVNode.ts
index b62eacc6..314e80e4 100644
--- a/libs/extension/src/parser/parseVNode.ts
+++ b/libs/extension/src/parser/parseVNode.ts
@@ -61,7 +61,7 @@ function parseTreeRoot(treeRoot: VNode) {
   return result;
 }
 
-export function queryVNode(id: number) {
+export function queryVNode(id: number): VNode|undefined {
   return IdToVNodeMap.get(id);
 }