Explicit state and event id minification

.gitignore

@@ -27,6 +27,7 @@ node_modules
 package-lock.json
 *.pyi
 .pre-commit-config.yaml
+uploaded_files/*
 .claude/.worktrees
 .claude/settings.local.json
 CLAUDE.local.md

packages/reflex-base/src/reflex_base/.templates/web/utils/state.js

@@ -15,7 +15,9 @@ import {
   initialState,
   onLoadInternalEvent,
   state_name,
-  exception_state_name,
+  handle_frontend_exception,
+  main_state_name,
+  update_vars_internal,
 } from "$/utils/context";
 import debounce from "$/utils/helpers/debounce";
 import throttle from "$/utils/helpers/throttle";
@@ -142,7 +144,7 @@ export const isStateful = () => {
   if (event_queue.length === 0) {
     return false;
   }
-  return event_queue.some((event) => event.name.startsWith("reflex___state"));
+  return event_queue.some((event) => event.name.startsWith(main_state_name));
 };
 
 /**
@@ -1007,7 +1009,7 @@ export const useEventLoop = (
 
     window.onerror = function (msg, url, lineNo, columnNo, error) {
       addEvents([
-        ReflexEvent(`${exception_state_name}.handle_frontend_exception`, {
+        ReflexEvent(handle_frontend_exception, {
           info: error.name + ": " + error.message + "\n" + error.stack,
           component_stack: "",
         }),
@@ -1019,7 +1021,7 @@ export const useEventLoop = (
     //https://github.com/mknichel/javascript-errors?tab=readme-ov-file#promise-rejection-events
     window.onunhandledrejection = function (event) {
       addEvents([
-        ReflexEvent(`${exception_state_name}.handle_frontend_exception`, {
+        ReflexEvent(handle_frontend_exception, {
           info:
             event.reason?.name +
             ": " +
@@ -1084,10 +1086,9 @@ export const useEventLoop = (
       if (storage_to_state_map[e.key]) {
         const vars = {};
         vars[storage_to_state_map[e.key]] = e.newValue;
-        const event = ReflexEvent(
-          `${state_name}.reflex___state____update_vars_internal_state.update_vars_internal`,
-          { vars: vars },
-        );
+        const event = ReflexEvent(update_vars_internal, {
+          vars: vars,
+        });
         addEvents([event], e);
       }
     };
@@ -1122,7 +1123,7 @@ export const useEventLoop = (
     }
 
     // Equivalent to routeChangeStart - runs when navigation begins
-    const main_state_dispatch = dispatch["reflex___state____state"];
+    const main_state_dispatch = dispatch[main_state_name];
     if (main_state_dispatch !== undefined) {
       main_state_dispatch({ is_hydrated_rx_state_: false });
     }

packages/reflex-base/src/reflex_base/compiler/templates.py

@@ -9,7 +9,7 @@
 from reflex_base import constants
 from reflex_base.constants import Hooks
 from reflex_base.utils import memo_paths
-from reflex_base.utils.format import format_state_name, json_dumps
+from reflex_base.utils.format import format_event_handler, format_state_name, json_dumps
 from reflex_base.vars.base import VarData
 
 if TYPE_CHECKING:
@@ -288,6 +288,23 @@ def context_template(
     Returns:
         Rendered context file content as string.
     """
+    # Import state classes to get dynamic names (supports minification)
+    from reflex.state import (
+        FrontendEventExceptionState,
+        OnLoadInternalState,
+        State,
+        UpdateVarsInternalState,
+    )
+
+    main_state_name = State.get_name()
+    on_load_internal = format_event_handler(OnLoadInternalState.on_load_internal)
+    update_vars_internal = format_event_handler(
+        UpdateVarsInternalState.update_vars_internal
+    )
+    handle_frontend_exception = format_event_handler(
+        FrontendEventExceptionState.handle_frontend_exception
+    )
+
     initial_state = initial_state or {}
     state_contexts_str = "".join([
         f"{format_state_name(state_name)}: createContext(null),"
@@ -298,7 +315,11 @@ def context_template(
         rf"""
 export const state_name = "{state_name}"
 
-export const exception_state_name = "{constants.CompileVars.FRONTEND_EXCEPTION_STATE_FULL}"
+export const main_state_name = "{main_state_name}"
+
+export const update_vars_internal = "{update_vars_internal}"
+
+export const handle_frontend_exception = "{handle_frontend_exception}"
 
 // These events are triggered on initial load and each page navigation.
 export const onLoadInternalEvent = () => {{
@@ -310,15 +331,15 @@ def context_template(
     if (client_storage_vars && Object.keys(client_storage_vars).length !== 0) {{
         internal_events.push(
             ReflexEvent(
-                '{state_name}.{constants.CompileVars.UPDATE_VARS_INTERNAL}',
+                '{update_vars_internal}',
                 {{vars: client_storage_vars}},
             ),
         );
     }}
 
     // `on_load_internal` triggers the correct on_load event(s) for the current page.
     // If the page does not define any on_load event, this will just set `is_hydrated = true`.
-    internal_events.push(ReflexEvent('{state_name}.{constants.CompileVars.ON_LOAD_INTERNAL}'));
+    internal_events.push(ReflexEvent('{on_load_internal}'));
 
     return internal_events;
 }}
@@ -333,7 +354,11 @@ def context_template(
         else """
 export const state_name = undefined
 
-export const exception_state_name = undefined
+export const main_state_name = undefined
+
+export const update_vars_internal = undefined
+
+export const handle_frontend_exception = undefined
 
 export const onLoadInternalEvent = () => []
 

packages/reflex-base/src/reflex_base/constants/compiler.py

@@ -67,18 +67,6 @@ class CompileVars(SimpleNamespace):
     CONNECT_ERROR = "connectErrors"
     # The name of the function for converting a dict to an event.
     TO_EVENT = "ReflexEvent"
-    # The name of the internal on_load event.
-    ON_LOAD_INTERNAL = "reflex___state____on_load_internal_state.on_load_internal"
-    # The name of the internal event to update generic state vars.
-    UPDATE_VARS_INTERNAL = (
-        "reflex___state____update_vars_internal_state.update_vars_internal"
-    )
-    # The name of the frontend event exception state
-    FRONTEND_EXCEPTION_STATE = "reflex___state____frontend_event_exception_state"
-    # The full name of the frontend exception state
-    FRONTEND_EXCEPTION_STATE_FULL = (
-        f"reflex___state____state.{FRONTEND_EXCEPTION_STATE}"
-    )
 
 
 class PageNames(SimpleNamespace):

packages/reflex-base/src/reflex_base/constants/event.py

@@ -3,6 +3,9 @@
 from enum import Enum
 from types import SimpleNamespace
 
+# The name of the setvar event handler.
+SETVAR = "setvar"
+
 
 class Endpoint(Enum):
     """Endpoints for the reflex backend API."""

packages/reflex-base/src/reflex_base/environment.py

@@ -549,6 +549,13 @@ class PathExistsFlag:
 ExistingPath = Annotated[Path, PathExistsFlag]
 
 
+class MinifyMode(enum.Enum):
+    """Mode for minification of state/event IDs."""
+
+    ENABLED = "enabled"
+    DISABLED = "disabled"
+
+
 class PerformanceMode(enum.Enum):
     """Performance mode for the app."""
 
@@ -741,6 +748,12 @@ class EnvironmentVariables:
     # How long to opportunistically hold the redis lock in milliseconds (must be less than the token expiration).
     REFLEX_OPLOCK_HOLD_TIME_MS: EnvVar[int] = env_var(0)
 
+    # Whether to enable state ID minification (requires minify.json).
+    REFLEX_MINIFY_STATES: EnvVar[MinifyMode] = env_var(MinifyMode.DISABLED)
+
+    # Whether to enable event ID minification (requires minify.json).
+    REFLEX_MINIFY_EVENTS: EnvVar[MinifyMode] = env_var(MinifyMode.DISABLED)
+
     # Extra plugins to append to the config's plugins list.
     REFLEX_EXTRA_PLUGINS: EnvVar[list[type[Plugin]]] = env_var([])
 

packages/reflex-base/src/reflex_base/event/__init__.py

@@ -545,6 +545,26 @@ def __call__(self, *args: Any, **kwargs: Any) -> "EventSpec":
         )
 
 
+if TYPE_CHECKING:
+
+    def typing_event(fn: Callable[..., Any]) -> EventHandler:
+        """Mark ``fn`` so pyright sees it as the ``EventHandler`` that
+        ``BaseState.__init_subclass__`` will rewrite it into. No-op at runtime.
+
+        Args:
+            fn: The method to mark.
+
+        Returns:
+            ``fn`` typed as :class:`EventHandler`.
+        """
+        ...
+
+else:
+
+    def typing_event(fn: Callable[..., Any]) -> Callable[..., Any]:  # noqa: D103
+        return fn
+
+
 @dataclasses.dataclass(
     init=True,
     frozen=True,
@@ -3004,6 +3024,7 @@ def BaseState(self) -> "type[BaseState]":  # noqa: N802
 
 event = EventNamespace
 event.event = event  # pyright: ignore[reportAttributeAccessIssue]
+event.typing_event = staticmethod(typing_event)  # pyright: ignore[reportAttributeAccessIssue]
 _this = sys.modules[__name__]
 event.__path__ = _this.__path__  # pyright: ignore[reportAttributeAccessIssue]
 event.__spec__ = _this.__spec__  # pyright: ignore[reportAttributeAccessIssue]