reflex/reflex/components/sonner/toast.py
Masen Furer 109b272bc2
Bump frontend and backend deps (#4667)
* relock backend deps

* bump frontend library versions

* go back to react 18.3

* revert to react 18.3.1 (as it was before)

* reflex-web keep reflex from current revision, rather than main

* relock backend deps (again)

* don't check ag-grid version on main repo

* config_path is passed via `extra` to avoid pyright error when None

* bump next again

---------

Co-authored-by: Lendemor <thomas.brandeho@gmail.com>
2025-01-22 15:18:05 -08:00

384 lines
11 KiB
Python

"""Sonner toast component."""
from __future__ import annotations
from typing import Any, ClassVar, Literal, Optional, Union
from reflex.base import Base
from reflex.components.component import Component, ComponentNamespace
from reflex.components.lucide.icon import Icon
from reflex.components.props import NoExtrasAllowedProps, PropsBase
from reflex.event import EventSpec, run_script
from reflex.style import Style, resolved_color_mode
from reflex.utils import format
from reflex.utils.imports import ImportVar
from reflex.utils.serializers import serializer
from reflex.vars import VarData
from reflex.vars.base import LiteralVar, Var
from reflex.vars.function import FunctionVar
from reflex.vars.object import ObjectVar
LiteralPosition = Literal[
"top-left",
"top-center",
"top-right",
"bottom-left",
"bottom-center",
"bottom-right",
]
toast_ref = Var(_js_expr="refs['__toast']")
class ToastAction(Base):
"""A toast action that render a button in the toast."""
label: str
on_click: Any
@serializer
def serialize_action(action: ToastAction) -> dict:
"""Serialize a toast action.
Args:
action: The toast action to serialize.
Returns:
The serialized toast action with on_click formatted to queue the given event.
"""
return {
"label": action.label,
"onClick": format.format_queue_events(action.on_click),
}
def _toast_callback_signature(toast: Var) -> list[Var]:
"""The signature for the toast callback, stripping out unserializable keys.
Args:
toast: The toast variable.
Returns:
A function call stripping non-serializable members of the toast object.
"""
return [
Var(
_js_expr=f"(() => {{let {{action, cancel, onDismiss, onAutoClose, ...rest}} = {toast!s}; return rest}})()"
)
]
class ToastProps(PropsBase, NoExtrasAllowedProps):
"""Props for the toast component."""
# Toast's title, renders above the description.
title: Optional[Union[str, Var]]
# Toast's description, renders underneath the title.
description: Optional[Union[str, Var]]
# Whether to show the close button.
close_button: Optional[bool]
# Dark toast in light mode and vice versa.
invert: Optional[bool]
# Control the sensitivity of the toast for screen readers
important: Optional[bool]
# Time in milliseconds that should elapse before automatically closing the toast.
duration: Optional[int]
# Position of the toast.
position: Optional[LiteralPosition]
# If false, it'll prevent the user from dismissing the toast.
dismissible: Optional[bool]
# TODO: fix serialization of icons for toast? (might not be possible yet)
# Icon displayed in front of toast's text, aligned vertically.
# icon: Optional[Icon] = None # noqa: ERA001
# TODO: fix implementation for action / cancel buttons
# Renders a primary button, clicking it will close the toast.
action: Optional[ToastAction]
# Renders a secondary button, clicking it will close the toast.
cancel: Optional[ToastAction]
# Custom id for the toast.
id: Optional[Union[str, Var]]
# Removes the default styling, which allows for easier customization.
unstyled: Optional[bool]
# Custom style for the toast.
style: Optional[Style]
# Class name for the toast.
class_name: Optional[str]
# XXX: These still do not seem to work
# Custom style for the toast primary button.
action_button_styles: Optional[Style]
# Custom style for the toast secondary button.
cancel_button_styles: Optional[Style]
# The function gets called when either the close button is clicked, or the toast is swiped.
on_dismiss: Optional[Any]
# Function that gets called when the toast disappears automatically after it's timeout (duration` prop).
on_auto_close: Optional[Any]
def dict(self, *args, **kwargs) -> dict[str, Any]:
"""Convert the object to a dictionary.
Args:
*args: The arguments to pass to the base class.
**kwargs: The keyword arguments to pass to the base
Returns:
The object as a dictionary with ToastAction fields intact.
"""
kwargs.setdefault("exclude_none", True) # type: ignore
d = super().dict(*args, **kwargs)
# Keep these fields as ToastAction so they can be serialized specially
if "action" in d:
d["action"] = self.action
if isinstance(self.action, dict):
d["action"] = ToastAction(**self.action)
if "cancel" in d:
d["cancel"] = self.cancel
if isinstance(self.cancel, dict):
d["cancel"] = ToastAction(**self.cancel)
if "onDismiss" in d:
d["onDismiss"] = format.format_queue_events(
self.on_dismiss, _toast_callback_signature
)
if "onAutoClose" in d:
d["onAutoClose"] = format.format_queue_events(
self.on_auto_close, _toast_callback_signature
)
return d
class Toaster(Component):
"""A Toaster Component for displaying toast notifications."""
library: str = "sonner@1.7.2"
tag = "Toaster"
# the theme of the toast
theme: Var[str] = resolved_color_mode
# whether to show rich colors
rich_colors: Var[bool] = LiteralVar.create(True)
# whether to expand the toast
expand: Var[bool] = LiteralVar.create(True)
# the number of toasts that are currently visible
visible_toasts: Var[int]
# the position of the toast
position: Var[LiteralPosition] = LiteralVar.create("bottom-right")
# whether to show the close button
close_button: Var[bool] = LiteralVar.create(False)
# offset of the toast
offset: Var[str]
# directionality of the toast (default: ltr)
dir: Var[str]
# Keyboard shortcut that will move focus to the toaster area.
hotkey: Var[str]
# Dark toasts in light mode and vice versa.
invert: Var[bool]
# These will act as default options for all toasts. See toast() for all available options.
toast_options: Var[ToastProps]
# Gap between toasts when expanded
gap: Var[int]
# Changes the default loading icon
loading_icon: Var[Icon]
# Pauses toast timers when the page is hidden, e.g., when the tab is backgrounded, the browser is minimized, or the OS is locked.
pause_when_page_is_hidden: Var[bool]
# Marked True when any Toast component is created.
is_used: ClassVar[bool] = False
def add_hooks(self) -> list[Var | str]:
"""Add hooks for the toaster component.
Returns:
The hooks for the toaster component.
"""
hook = Var(
_js_expr=f"{toast_ref} = toast",
_var_data=VarData(
imports={
"$/utils/state": [ImportVar(tag="refs")],
self.library: [ImportVar(tag="toast", install=False)],
}
),
)
return [hook]
@staticmethod
def send_toast(
message: str | Var = "", level: str | None = None, **props
) -> EventSpec:
"""Send a toast message.
Args:
message: The message to display.
level: The level of the toast.
**props: The options for the toast.
Raises:
ValueError: If the Toaster component is not created.
Returns:
The toast event.
"""
if not Toaster.is_used:
raise ValueError(
"Toaster component must be created before sending a toast. (use `rx.toast.provider()`)"
)
toast_command = (
ObjectVar.__getattr__(toast_ref.to(dict), level) if level else toast_ref
).to(FunctionVar)
if isinstance(message, Var):
props.setdefault("title", message)
message = ""
elif message == "" and "title" not in props and "description" not in props:
raise ValueError("Toast message or title or description must be provided.")
if props:
args = LiteralVar.create(ToastProps(component_name="rx.toast", **props)) # pyright: ignore [reportCallIssue, reportGeneralTypeIssues]
toast = toast_command.call(message, args)
else:
toast = toast_command.call(message)
return run_script(toast)
@staticmethod
def toast_info(message: str | Var = "", **kwargs):
"""Display an info toast message.
Args:
message: The message to display.
kwargs: Additional toast props.
Returns:
The toast event.
"""
return Toaster.send_toast(message, level="info", **kwargs)
@staticmethod
def toast_warning(message: str | Var = "", **kwargs):
"""Display a warning toast message.
Args:
message: The message to display.
kwargs: Additional toast props.
Returns:
The toast event.
"""
return Toaster.send_toast(message, level="warning", **kwargs)
@staticmethod
def toast_error(message: str | Var = "", **kwargs):
"""Display an error toast message.
Args:
message: The message to display.
kwargs: Additional toast props.
Returns:
The toast event.
"""
return Toaster.send_toast(message, level="error", **kwargs)
@staticmethod
def toast_success(message: str | Var = "", **kwargs):
"""Display a success toast message.
Args:
message: The message to display.
kwargs: Additional toast props.
Returns:
The toast event.
"""
return Toaster.send_toast(message, level="success", **kwargs)
@staticmethod
def toast_dismiss(id: Var | str | None = None):
"""Dismiss a toast.
Args:
id: The id of the toast to dismiss.
Returns:
The toast dismiss event.
"""
dismiss_var_data = None
if isinstance(id, Var):
dismiss = f"{toast_ref}.dismiss({id!s})"
dismiss_var_data = id._get_all_var_data()
elif isinstance(id, str):
dismiss = f"{toast_ref}.dismiss('{id}')"
else:
dismiss = f"{toast_ref}.dismiss()"
dismiss_action = Var(
_js_expr=dismiss, _var_data=VarData.merge(dismiss_var_data)
)
return run_script(dismiss_action)
@classmethod
def create(cls, *children, **props) -> Component:
"""Create a toaster component.
Args:
*children: The children of the toaster.
**props: The properties of the toaster.
Returns:
The toaster component.
"""
cls.is_used = True
return super().create(*children, **props)
# TODO: figure out why loading toast stay open forever when using level="loading" in toast()
class ToastNamespace(ComponentNamespace):
"""Namespace for toast components."""
provider = staticmethod(Toaster.create)
options = staticmethod(ToastProps)
info = staticmethod(Toaster.toast_info)
warning = staticmethod(Toaster.toast_warning)
error = staticmethod(Toaster.toast_error)
success = staticmethod(Toaster.toast_success)
dismiss = staticmethod(Toaster.toast_dismiss)
__call__ = staticmethod(Toaster.send_toast)
toast = ToastNamespace()