fix: forwardref 组件不能选中的问题

alibaba · Oct 17, 2023 · d4822af · d4822af
1 parent a96961f
commit d4822af
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/packages/renderer-core/src/hoc/index.tsx b/packages/renderer-core/src/hoc/index.tsx
@@ -2,7 +2,25 @@ import { cloneEnumerableProperty } from '@alilc/lowcode-utils';
 import adapter from '../adapter';
 import { IBaseRendererInstance, IRendererProps } from '../types';
 
-function patchDidCatch(Comp: any, { baseRenderer }: { baseRenderer: IBaseRendererInstance }) {
+interface Options {
+  baseRenderer: IBaseRendererInstance;
+  schema: any;
+}
+
+function patchDidMount(Comp: any, { baseRenderer, schema }: Options) {
+  if (Comp.patchedMount) {
+    return;
+  }
+  Comp.patchedMount = true;
+  const { engine } = baseRenderer.context;
+  const originalDidMount = Comp.prototype.componentDidMount;
+  Comp.prototype.componentDidMount = function () {
+    engine.props?.onCompGetRef(schema, this);
+    return originalDidMount && originalDidMount.call(this);
+  };
+}
+
+function patchDidCatch(Comp: any, { baseRenderer }: Options) {
   if (Comp.patchedCatch) {
     return;
   }
@@ -44,7 +62,7 @@ function patchDidCatch(Comp: any, { baseRenderer }: { baseRenderer: IBaseRendere
   }
 }
 
-export function compWrapper(Comp: any, options: { baseRenderer: IBaseRendererInstance }) {
+export function compWrapper(Comp: any, options: Options) {
   const { createElement, Component, forwardRef } = adapter.getRuntime();
   if (
     Comp?.prototype?.isReactComponent || // react
@@ -62,6 +80,7 @@ export function compWrapper(Comp: any, options: { baseRenderer: IBaseRendererIns
   (Wrapper as any).displayName = Comp.displayName;
 
   patchDidCatch(Wrapper, options);
+  patchDidMount(Wrapper, options);
 
   return cloneEnumerableProperty(
     forwardRef((props: any, ref: any) => {

diff --git a/packages/renderer-core/src/renderer/base.tsx b/packages/renderer-core/src/renderer/base.tsx
@@ -615,7 +615,7 @@ export default function baseRendererFactory(): IBaseRenderComponent {
           });
         });
 
-        Comp = compWrapper(Comp, { baseRenderer: this });
+        Comp = compWrapper(Comp, { baseRenderer: this, schema });
         components[schema.componentName] = Comp;
 
         otherProps.ref = (ref: any) => {