fix: click

packages/core/client/src/flow/models/base/ActionModel.tsx

@@ -18,15 +18,7 @@ export class ActionModel extends FlowModel {
     children: 'Action',
   };
   render() {
-    return (
+    return <Button {...this.defaultProps} {...this.props} />;
-      <Button
-        {...this.defaultProps}
-        {...this.props}
-        onClick={(event) => {
-          this.dispatchEvent('click', { event });
-        }}
-      />
-    );
   }
 }
 
@@ -57,6 +49,12 @@ ActionModel.registerFlow({
       },
       handler(ctx, params) {
         ctx.model.setProps('children', params.children);
+        ctx.model.setProps('onClick', (event) => {
+          ctx.model.dispatchEvent('click', {
+            ...ctx.extra,
+            event,
+          });
+        });
       },
     },
   },

packages/core/client/src/flow/models/data-blocks/table/TableModel.tsx

@@ -65,6 +65,7 @@ export class TableModel extends DataBlockModel<S> {
             },
           ]}
           onModelAdded={async (model: TableColumnModel) => {
+            model.setSharedContext({ currentBlockModel: this });
             await model.applyAutoFlows();
           }}
         />