From 3c7af9fad4622e76cb0e868ca78fc916c6bdc884 Mon Sep 17 00:00:00 2001
From: Masen Furer <m_github@0x26.net>
Date: Thu, 30 Nov 2023 09:59:09 -0800
Subject: [PATCH] Support hooks in custom components (state, addEvents) (#2234)

---
 reflex/.templates/jinja/web/pages/custom_component.js.jinja2 | 3 +++
 reflex/compiler/utils.py                                     | 1 +
 2 files changed, 4 insertions(+)

diff --git a/reflex/.templates/jinja/web/pages/custom_component.js.jinja2 b/reflex/.templates/jinja/web/pages/custom_component.js.jinja2
index 6b1438c62..210246992 100644
--- a/reflex/.templates/jinja/web/pages/custom_component.js.jinja2
+++ b/reflex/.templates/jinja/web/pages/custom_component.js.jinja2
@@ -22,6 +22,9 @@ export const {{component.name}} = memo(({ {{-component.props|join(", ")-}} }) =>
 
     }
 {% endif %}
+    {% for hook in component.hooks %}
+    {{ hook }}
+    {% endfor %}
     return(
         {{utils.render(component.render)}}
       )
diff --git a/reflex/compiler/utils.py b/reflex/compiler/utils.py
index 0a4df7a4e..dd3a09453 100644
--- a/reflex/compiler/utils.py
+++ b/reflex/compiler/utils.py
@@ -254,6 +254,7 @@ def compile_custom_component(
             "name": component.tag,
             "props": props,
             "render": render.render(),
+            "hooks": render.get_hooks(),
             "custom_code": render.get_custom_code(),
         },
         imports,