From ed83c2cad515a8c933071a86110735ec54a5844e Mon Sep 17 00:00:00 2001
From: Kevin Wang <kevmo314@gmail.com>
Date: Mon, 23 Oct 2023 07:25:49 -0400
Subject: [PATCH] Add kwargs support to vite_react_refresh

---
 django_vite/templatetags/django_vite.py        | 18 ++++++++++++++----
 .../templatetags/test_vite_react_refresh.py    | 15 +++++++++++++++
 2 files changed, 29 insertions(+), 4 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index fdb3304..1289a11 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -540,12 +540,16 @@ def _generate_vite_server_url(path: str) -> str:
         )
 
     @classmethod
-    def generate_vite_react_refresh_url(cls) -> str:
+    def generate_vite_react_refresh_url(cls, **kwargs) -> str:
         """
         Generates the script for the Vite React Refresh for HMR.
         Only used in development, in production this method returns
         an empty string.
 
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+
         Returns:
             str -- The script or an empty string.
         """
@@ -553,7 +557,9 @@ def generate_vite_react_refresh_url(cls) -> str:
         if not DJANGO_VITE_DEV_MODE:
             return ""
 
-        return f"""<script type="module">
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in kwargs.items()])
+
+        return f"""<script type="module" {attrs_str}>
             import RefreshRuntime from \
             '{cls._generate_vite_server_url(DJANGO_VITE_REACT_REFRESH_URL)}'
             RefreshRuntime.injectIntoGlobalHook(window)
@@ -750,13 +756,17 @@ def vite_legacy_asset(
 
 @register.simple_tag
 @mark_safe
-def vite_react_refresh() -> str:
+def vite_react_refresh(**kwargs: Dict[str, str]) -> str:
     """
     Generates the script for the Vite React Refresh for HMR.
     Only used in development, in production this method returns
     an empty string.
 
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
     Returns:
         str -- The script or an empty string.
     """
-    return DjangoViteAssetLoader.generate_vite_react_refresh_url()
+    return DjangoViteAssetLoader.generate_vite_react_refresh_url(**kwargs)
diff --git a/tests/tests/templatetags/test_vite_react_refresh.py b/tests/tests/templatetags/test_vite_react_refresh.py
index e2c31e7..7c9aa65 100644
--- a/tests/tests/templatetags/test_vite_react_refresh.py
+++ b/tests/tests/templatetags/test_vite_react_refresh.py
@@ -72,3 +72,18 @@ def test_vite_react_refresh_url_setting(patch_settings):
     soup = BeautifulSoup(html, "html.parser")
     script_tag = soup.script
     assert "http://localhost:3000/static/foobar" in script_tag.text
+
+
+def test_vite_react_refresh_uses_kwargs(patch_settings):
+    patch_settings({"DJANGO_VITE_REACT_REFRESH_URL": "foobar"})
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh nonce="woo-nonce" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert script_tag.has_attr("nonce")
+    assert script_tag["nonce"] == "woo-nonce"