From f6468bfac4aa606634a0d40c0348308d5e40651f Mon Sep 17 00:00:00 2001
From: MrBin99 <paul.pages12@gmail.com>
Date: Fri, 29 Jul 2022 14:36:04 +0200
Subject: [PATCH 01/29] feat(multi-config): rework how configuration definition
 is handled

---
 django_vite/templatetags/django_vite.py | 456 +++++++++++++++++-------
 1 file changed, 326 insertions(+), 130 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 75b0e98..e4a17b6 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -1,6 +1,6 @@
 import json
 from pathlib import Path
-from typing import Dict, List
+from typing import Dict, List, NamedTuple, Optional, Type, Union
 from urllib.parse import urljoin
 
 from django import template
@@ -10,66 +10,78 @@
 register = template.Library()
 
 
-# If using in development or production mode.
-DJANGO_VITE_DEV_MODE = getattr(settings, "DJANGO_VITE_DEV_MODE", False)
-
-# Default Vite server protocol (http or https)
-DJANGO_VITE_DEV_SERVER_PROTOCOL = getattr(
-    settings, "DJANGO_VITE_DEV_SERVER_PROTOCOL", "http"
-)
-
-# Default vite server hostname.
-DJANGO_VITE_DEV_SERVER_HOST = getattr(
-    settings, "DJANGO_VITE_DEV_SERVER_HOST", "localhost"
-)
-
-# Default Vite server port.
-DJANGO_VITE_DEV_SERVER_PORT = getattr(
-    settings, "DJANGO_VITE_DEV_SERVER_PORT", 3000
-)
-
-# Default Vite server path to HMR script.
-DJANGO_VITE_WS_CLIENT_URL = getattr(
-    settings, "DJANGO_VITE_WS_CLIENT_URL", "@vite/client"
-)
-
-# Location of Vite compiled assets (only used in Vite production mode).
-# Must be included in your "STATICFILES_DIRS".
-# In Django production mode this folder need to be collected as static
-# files using "python manage.py collectstatic".
-DJANGO_VITE_ASSETS_PATH = Path(getattr(settings, "DJANGO_VITE_ASSETS_PATH"))
-
-# Prefix for STATIC_URL
-DJANGO_VITE_STATIC_URL_PREFIX = getattr(
-    settings, "DJANGO_VITE_STATIC_URL_PREFIX", ""
-)
-
-DJANGO_VITE_STATIC_ROOT = (
-    DJANGO_VITE_ASSETS_PATH
-    if DJANGO_VITE_DEV_MODE
-    else Path(settings.STATIC_ROOT) / DJANGO_VITE_STATIC_URL_PREFIX
-)
-
-# Path to your manifest file generated by Vite.
-# Should by in "DJANGO_VITE_ASSETS_PATH".
-DJANGO_VITE_MANIFEST_PATH = getattr(
-    settings,
-    "DJANGO_VITE_MANIFEST_PATH",
-    DJANGO_VITE_STATIC_ROOT / "manifest.json",
-)
-
-# Motif in the 'manifest.json' to find the polyfills generated by Vite.
-DJANGO_VITE_LEGACY_POLYFILLS_MOTIF = getattr(
-    settings, "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF", "legacy-polyfills"
-)
-
-DJANGO_VITE_STATIC_URL = urljoin(
-    settings.STATIC_URL, DJANGO_VITE_STATIC_URL_PREFIX
-)
-
-# Make sure 'DJANGO_VITE_STATIC_URL' finish with a '/'
-if DJANGO_VITE_STATIC_URL[-1] != "/":
-    DJANGO_VITE_STATIC_URL += "/"
+class DjangoViteManifest(NamedTuple):
+    """
+    Represent an entry for a file inside the "manifest.json".
+    """
+
+    file: str
+    src: str
+    isEntry: bool
+    css: Optional[List[str]] = []
+    imports: Optional[List[str]] = []
+
+
+class DjangoViteConfig(NamedTuple):
+    """
+    Represent the Django Vite configuration structure.
+    """
+
+    # Location of Vite compiled assets (only used in Vite production mode).
+    assets_path: Union[Path, str]
+
+    # If using in development or production mode.
+    dev_mode: bool = False
+
+    # Default Vite server protocol (http or https)
+    dev_server_protocol: str = "http"
+
+    # Default vite server hostname.
+    dev_server_host: str = "localhost"
+
+    # Default Vite server port.
+    dev_server_port: int = 3000
+
+    # Default Vite server path to HMR script.
+    ws_client_url: str = "@vite/client"
+
+    # Prefix for STATIC_URL.
+    static_url_prefix: str = ""
+
+    # Motif in the "manifest.json" to find the polyfills generated by Vite.
+    legacy_polyfills_motif: str = "legacy-polyfills"
+
+    # Path to your manifest file generated by Vite.
+    manifest_path: Union[Path, str] = ""
+
+    @property
+    def static_root(self) -> Union[Path, str]:
+        """
+        Compute the static root URL of assets.
+
+        Returns:
+            Union[Path, str] -- Static root URL.
+        """
+
+        return (
+            self.assets_path
+            if self.dev_mode
+            else Path(settings.STATIC_ROOT) / self.static_url_prefix
+        )
+
+    def get_computed_manifest_path(self) -> Union[Path, str]:
+        """
+        Compute the path to the "manifest.json".
+
+        Returns:
+            Union[Path, str] -- Path to the "manifest.json".
+        """
+
+        return (
+            self.manifest_path
+            if self.manifest_path
+            else self.static_root / "manifest.json"
+        )
 
 
 class DjangoViteAssetLoader:
@@ -79,12 +91,17 @@ class DjangoViteAssetLoader:
 
     _instance = None
 
+    _configs = Dict[str, Type[DjangoViteConfig]]
+    _manifests: Dict[str, Type[DjangoViteManifest]]
+    _static_urls: Dict[str, str]
+
     def __init__(self) -> None:
         raise RuntimeError("Use the instance() method instead.")
 
     def generate_vite_asset(
         self,
         path: str,
+        config_key: str,
         **kwargs: Dict[str, str],
     ) -> str:
         """
@@ -95,6 +112,7 @@ def generate_vite_asset(
 
         Arguments:
             path {str} -- Path to a Vite JS/TS asset to include.
+            config_key {str} -- Key of the configuration to use.
 
         Returns:
             str -- All tags to import this file in your HTML page.
@@ -112,29 +130,35 @@ def generate_vite_asset(
                 this asset in your page.
         """
 
-        if DJANGO_VITE_DEV_MODE:
+        config = self._get_config(config_key)
+        static_url = self._get_static_url(config_key)
+
+        if config.dev_mode:
             return DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_vite_server_url(path),
+                DjangoViteAssetLoader._generate_vite_server_url(
+                    path, static_url, config
+                ),
                 {"type": "module"},
             )
 
-        if not self._manifest or path not in self._manifest:
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
             raise RuntimeError(
                 f"Cannot find {path} in Vite manifest "
-                f"at {DJANGO_VITE_MANIFEST_PATH}"
+                f"at {config.get_computed_manifest_path()}"
             )
 
         tags = []
-        manifest_entry = self._manifest[path]
         scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
 
         # Add dependent CSS
-        tags.extend(self._generate_css_files_of_asset(path, []))
+        tags.extend(self._generate_css_files_of_asset(path, config_key, []))
 
         # Add the script by itself
         tags.append(
             DjangoViteAssetLoader._generate_script_tag(
-                urljoin(DJANGO_VITE_STATIC_URL, manifest_entry["file"]),
+                urljoin(static_url, manifest[path].file),
                 attrs=scripts_attrs,
             )
         )
@@ -142,13 +166,17 @@ def generate_vite_asset(
         return "\n".join(tags)
 
     def _generate_css_files_of_asset(
-        self, path: str, already_processed: List[str]
+        self,
+        path: str,
+        config_key: str,
+        already_processed: List[str],
     ) -> List[str]:
         """
         Generates all CSS tags for dependencies of an asset.
 
         Arguments:
             path {str} -- Path to an asset in the 'manifest.json'.
+            config_key {str} -- Key of the configuration to use.
             already_processed {list} -- List of already processed CSS file.
 
         Returns:
@@ -156,36 +184,37 @@ def _generate_css_files_of_asset(
         """
 
         tags = []
-        manifest_entry = self._manifest[path]
-
-        if "imports" in manifest_entry:
-            for import_path in manifest_entry["imports"]:
-                tags.extend(
-                    self._generate_css_files_of_asset(
-                        import_path, already_processed
-                    )
+        static_url = self._get_static_url(config_key)
+        manifest = self._get_manifest(config_key)
+        manifest_entry = manifest[path]
+
+        for import_path in manifest_entry.imports:
+            tags.extend(
+                self._generate_css_files_of_asset(
+                    import_path, already_processed
                 )
+            )
 
-        if "css" in manifest_entry:
-            for css_path in manifest_entry["css"]:
-                if css_path not in already_processed:
-                    tags.append(
-                        DjangoViteAssetLoader._generate_stylesheet_tag(
-                            urljoin(DJANGO_VITE_STATIC_URL, css_path)
-                        )
+        for css_path in manifest_entry.css:
+            if css_path not in already_processed:
+                tags.append(
+                    DjangoViteAssetLoader._generate_stylesheet_tag(
+                        urljoin(static_url, css_path)
                     )
+                )
 
-                already_processed.append(css_path)
+            already_processed.append(css_path)
 
         return tags
 
-    def generate_vite_asset_url(self, path: str) -> str:
+    def generate_vite_asset_url(self, path: str, config_key: str) -> str:
         """
         Generates only the URL of an asset managed by ViteJS.
         Warning, this function does not generate URLs for dependant assets.
 
         Arguments:
             path {str} -- Path to a Vite asset.
+            config_key {str} -- Key of the configuration to use.
 
         Raises:
             RuntimeError: If cannot find the asset path in the
@@ -195,19 +224,27 @@ def generate_vite_asset_url(self, path: str) -> str:
             str -- The URL of this asset.
         """
 
-        if DJANGO_VITE_DEV_MODE:
-            return DjangoViteAssetLoader._generate_vite_server_url(path)
+        config = self._get_config(config_key)
+        static_url = self._get_static_url(config_key)
+
+        if config.dev_mode:
+            return DjangoViteAssetLoader._generate_vite_server_url(
+                path, config
+            )
+
+        manifest = self._get_manifest(config_key)
 
-        if not self._manifest or path not in self._manifest:
+        if path not in manifest:
             raise RuntimeError(
                 f"Cannot find {path} in Vite manifest "
-                f"at {DJANGO_VITE_MANIFEST_PATH}"
+                f"at {config.get_computed_manifest_path()}"
             )
 
-        return urljoin(DJANGO_VITE_STATIC_URL, self._manifest[path]["file"])
+        return urljoin(static_url, manifest[path].file)
 
     def generate_vite_legacy_polyfills(
         self,
+        config_key: str,
         **kwargs: Dict[str, str],
     ) -> str:
         """
@@ -216,6 +253,9 @@ def generate_vite_legacy_polyfills(
         This tag must be included at end of the <body> before
         including other legacy scripts.
 
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
         Keyword Arguments:
             **kwargs {Dict[str, str]} -- Adds new attributes to generated
                 script tags.
@@ -228,26 +268,31 @@ def generate_vite_legacy_polyfills(
             str -- The script tag to the polyfills.
         """
 
-        if DJANGO_VITE_DEV_MODE:
+        config = self._get_config(config_key)
+        manifest = self._get_manifest(config_key)
+        static_url = self._get_static_url(config_key)
+
+        if config.dev_mode:
             return ""
 
         scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
 
-        for path, content in self._manifest.items():
-            if DJANGO_VITE_LEGACY_POLYFILLS_MOTIF in path:
+        for path, content in manifest.items():
+            if config.legacy_polyfills_motif in path:
                 return DjangoViteAssetLoader._generate_script_tag(
-                    urljoin(DJANGO_VITE_STATIC_URL, content["file"]),
+                    urljoin(static_url, content.file),
                     attrs=scripts_attrs,
                 )
 
         raise RuntimeError(
             f"Vite legacy polyfills not found in manifest "
-            f"at {DJANGO_VITE_MANIFEST_PATH}"
+            f"at {config.get_computed_manifest_path()}"
         )
 
     def generate_vite_legacy_asset(
         self,
         path: str,
+        config_key: str,
         **kwargs: Dict[str, str],
     ) -> str:
         """
@@ -258,6 +303,7 @@ def generate_vite_legacy_asset(
         Arguments:
             path {str} -- Path to a Vite asset to include
                 (must contains '-legacy' in its name).
+            config_key {str} -- Key of the configuration to use.
 
         Keyword Arguments:
             **kwargs {Dict[str, str]} -- Adds new attributes to generated
@@ -271,48 +317,127 @@ def generate_vite_legacy_asset(
             str -- The script tag of this legacy asset .
         """
 
-        if DJANGO_VITE_DEV_MODE:
+        config = self._get_config(config_key)
+        static_url = self._get_static_url(config_key)
+
+        if config.dev_mode:
             return ""
 
-        if not self._manifest or path not in self._manifest:
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
             raise RuntimeError(
                 f"Cannot find {path} in Vite manifest "
-                f"at {DJANGO_VITE_MANIFEST_PATH}"
+                f"at {config.get_computed_manifest_path()}"
             )
 
-        manifest_entry = self._manifest[path]
         scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
 
         return DjangoViteAssetLoader._generate_script_tag(
-            urljoin(DJANGO_VITE_STATIC_URL, manifest_entry["file"]),
+            urljoin(static_url, manifest[path].file),
             attrs=scripts_attrs,
         )
 
-    def _parse_manifest(self) -> None:
+    def _get_config(self, config_key: str) -> Type[DjangoViteConfig]:
+        """
+        Get configuration object registered with the key passed in
+        parameter.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to retrieve.
+
+        Raises:
+            RuntimeError: If no configuration exists for this key.
+
+        Returns:
+            Type[DjangoViteConfig] -- The configuration.
+        """
+
+        if config_key not in self._configs:
+            raise RuntimeError(f'Cannot find "{config_key}" configuration')
+
+        return self._configs[config_key]
+
+    def _parse_manifest(
+        self, config_key: str
+    ) -> Dict[str, Type[DjangoViteManifest]]:
         """
         Read and parse the Vite manifest file.
 
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
         Raises:
             RuntimeError: if cannot load the file or JSON in file is malformed.
         """
 
+        config = self._get_config(config_key)
+
         try:
-            manifest_file = open(DJANGO_VITE_MANIFEST_PATH, "r")
-            manifest_content = manifest_file.read()
-            manifest_file.close()
-            self._manifest = json.loads(manifest_content)
+            with open(
+                config.get_computed_manifest_path(), "r"
+            ) as manifest_file:
+                manifest_content = manifest_file.read()
+                manifest_json = json.loads(manifest_content)
+
+                return {
+                    k: DjangoViteManifest(**v)
+                    for k, v in manifest_json.items()
+                }
+
         except Exception as error:
             raise RuntimeError(
                 f"Cannot read Vite manifest file at "
-                f"{DJANGO_VITE_MANIFEST_PATH} : {str(error)}"
+                f"{config.get_computed_manifest_path()} : {str(error)}"
+            )
+
+    def _get_manifest(
+        self, config_key: str
+    ) -> Dict[str, Type[DjangoViteManifest]]:
+        """
+        Load if needed and parse the "manifest.json" of the specified
+        configuration.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            Dict[str, Type[DjangoViteManifest]] -- Parsed content of
+                the "manifest.json"
+        """
+
+        if config_key not in self._manifests:
+            self._manifests[config_key] = self._parse_manifest(config_key)
+
+        return self._manifests[config_key]
+
+    def _get_static_url(self, config_key: str) -> str:
+        """
+        Build the static URL of a specified configuration.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            str -- The static URL.
+        """
+
+        if config_key not in self._static_urls:
+            config = self._get_config(config_key)
+            static_url = urljoin(settings.STATIC_URL, config.static_url_prefix)
+
+            self._static_urls[config_key] = (
+                static_url if static_url[-1] == "/" else static_url + "/"
             )
 
+        return self._static_urls[config_key]
+
     @classmethod
     def instance(cls):
         """
         Singleton.
-        Uses singleton to keep parsed manifest in memory after
-        the first time it's loaded.
+        Uses singleton to keep parsed manifests in memory after
+        the first time they are loaded.
 
         Returns:
             DjangoViteAssetLoader -- only instance of the class.
@@ -320,30 +445,74 @@ def instance(cls):
 
         if cls._instance is None:
             cls._instance = cls.__new__(cls)
-            cls._instance._manifest = None
-
-            # Manifest is only used in production.
-            if not DJANGO_VITE_DEV_MODE:
-                cls._instance._parse_manifest()
+            cls._instance._configs = {}
+            cls._instance._manifests = {}
+            cls._instance._static_urls = {}
+
+            if hasattr(settings, "DJANGO_VITE"):
+                config = getattr(settings, "DJANGO_VITE")
+
+                for config_key, config_values in config.items():
+                    if isinstance(config_values, DjangoViteConfig):
+                        cls._instance._configs[config_key] = config_values
+                    elif isinstance(config_values, dict):
+                        cls._instance._configs[config_key] = DjangoViteConfig(
+                            **config_values
+                        )
+                    else:
+                        raise RuntimeError(
+                            f"Cannot read configuration for key '{config_key}'"
+                        )
+            else:
+                # Warning : This branch will be remove in further
+                # releases. Please use new way of handling configuration.
+
+                _config_keys = {
+                    "DJANGO_VITE_DEV_MODE": "dev_mode",
+                    "DJANGO_VITE_DEV_SERVER_PROTOCOL": "dev_server_protocol",
+                    "DJANGO_VITE_DEV_SERVER_HOST": "dev_server_host",
+                    "DJANGO_VITE_DEV_SERVER_PORT": "dev_server_port",
+                    "DJANGO_VITE_WS_CLIENT_URL": "ws_client_url",
+                    "DJANGO_VITE_ASSETS_PATH": "assets_path",
+                    "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
+                    "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
+                    "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
+                }
+
+                config = {
+                    _config_keys[setting_key]: getattr(settings, setting_key)
+                    for setting_key in dir(settings)
+                    if setting_key in _config_keys.keys()
+                }
+
+                cls._instance._configs["default"] = DjangoViteConfig(**config)
 
         return cls._instance
 
     @classmethod
-    def generate_vite_ws_client(cls) -> str:
+    def generate_vite_ws_client(cls, config_key: str) -> str:
         """
         Generates the script tag for the Vite WS client for HMR.
         Only used in development, in production this method returns
         an empty string.
 
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
         Returns:
             str -- The script tag or an empty string.
         """
 
-        if not DJANGO_VITE_DEV_MODE:
+        config = cls._get_config(config_key)
+        static_url = cls._get_static_url(config_key)
+
+        if not config.dev_mode:
             return ""
 
         return cls._generate_script_tag(
-            cls._generate_vite_server_url(DJANGO_VITE_WS_CLIENT_URL),
+            cls._generate_vite_server_url(
+                config.ws_client_url, static_url, config
+            ),
             {"type": "module"},
         )
 
@@ -384,21 +553,26 @@ def _generate_stylesheet_tag(href: str) -> str:
         return f'<link rel="stylesheet" href="{href}" />'
 
     @staticmethod
-    def _generate_vite_server_url(path: str) -> str:
+    def _generate_vite_server_url(
+        path: str,
+        static_url: str,
+        config: Type[DjangoViteConfig],
+    ) -> str:
         """
         Generates an URL to and asset served by the Vite development server.
 
         Keyword Arguments:
             path {str} -- Path to the asset.
+            config {Type[DjangoViteConfig]} -- Config object to use.
 
         Returns:
             str -- Full URL to the asset.
         """
 
         return urljoin(
-            f"{DJANGO_VITE_DEV_SERVER_PROTOCOL}://"
-            f"{DJANGO_VITE_DEV_SERVER_HOST}:{DJANGO_VITE_DEV_SERVER_PORT}",
-            urljoin(DJANGO_VITE_STATIC_URL, path),
+            f"{config.dev_server_protocol}://"
+            f"{config.dev_server_host}:{config.dev_server_port}",
+            urljoin(static_url, path),
         )
 
 
@@ -408,23 +582,27 @@ def _generate_vite_server_url(path: str) -> str:
 
 @register.simple_tag
 @mark_safe
-def vite_hmr_client() -> str:
+def vite_hmr_client(config: str = "default") -> str:
     """
     Generates the script tag for the Vite WS client for HMR.
     Only used in development, in production this method returns
     an empty string.
 
+    Arguments:
+        config {str} -- Configuration to use.
+
     Returns:
         str -- The script tag or an empty string.
     """
 
-    return DjangoViteAssetLoader.generate_vite_ws_client()
+    return DjangoViteAssetLoader.generate_vite_ws_client(config)
 
 
 @register.simple_tag
 @mark_safe
 def vite_asset(
     path: str,
+    config: str = "default",
     **kwargs: Dict[str, str],
 ) -> str:
     """
@@ -435,6 +613,7 @@ def vite_asset(
 
     Arguments:
         path {str} -- Path to a Vite JS/TS asset to include.
+        config {str} -- Configuration to use.
 
     Returns:
         str -- All tags to import this file in your HTML page.
@@ -453,18 +632,22 @@ def vite_asset(
     """
 
     assert path is not None
+    assert config is not None
 
-    return DjangoViteAssetLoader.instance().generate_vite_asset(path, **kwargs)
+    return DjangoViteAssetLoader.instance().generate_vite_asset(
+        path, config, **kwargs
+    )
 
 
 @register.simple_tag
-def vite_asset_url(path: str) -> str:
+def vite_asset_url(path: str, config: str = "default") -> str:
     """
     Generates only the URL of an asset managed by ViteJS.
     Warning, this function does not generate URLs for dependant assets.
 
     Arguments:
         path {str} -- Path to a Vite asset.
+        config {str} -- Configuration to use.
 
     Raises:
         RuntimeError: If cannot find the asset path in the
@@ -475,19 +658,27 @@ def vite_asset_url(path: str) -> str:
     """
 
     assert path is not None
+    assert config is not None
 
-    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path)
+    return DjangoViteAssetLoader.instance().generate_vite_asset_url(
+        path, config
+    )
 
 
 @register.simple_tag
 @mark_safe
-def vite_legacy_polyfills(**kwargs: Dict[str, str]) -> str:
+def vite_legacy_polyfills(
+    config: str = "default", **kwargs: Dict[str, str]
+) -> str:
     """
     Generates a <script> tag to the polyfills generated
     by '@vitejs/plugin-legacy' if used.
     This tag must be included at end of the <body> before including
     other legacy scripts.
 
+    Arguments:
+        config {str} -- Configuration to use.
+
     Keyword Arguments:
         **kwargs {Dict[str, str]} -- Adds new attributes to generated
             script tags.
@@ -500,8 +691,10 @@ def vite_legacy_polyfills(**kwargs: Dict[str, str]) -> str:
         str -- The script tag to the polyfills.
     """
 
+    assert config is not None
+
     return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
-        **kwargs
+        config, **kwargs
     )
 
 
@@ -509,6 +702,7 @@ def vite_legacy_polyfills(**kwargs: Dict[str, str]) -> str:
 @mark_safe
 def vite_legacy_asset(
     path: str,
+    config: str = "default",
     **kwargs: Dict[str, str],
 ) -> str:
     """
@@ -519,6 +713,7 @@ def vite_legacy_asset(
     Arguments:
         path {str} -- Path to a Vite asset to include
             (must contains '-legacy' in its name).
+        config {str} -- Configuration to use.
 
     Keyword Arguments:
         **kwargs {Dict[str, str]} -- Adds new attributes to generated
@@ -529,11 +724,12 @@ def vite_legacy_asset(
             the manifest (only in production).
 
     Returns:
-        str -- The script tag of this legacy asset .
+        str -- The script tag of this legacy asset.
     """
 
     assert path is not None
+    assert config is not None
 
     return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
-        path, **kwargs
+        path, config, **kwargs
     )

From c5d6f36b53d35b1b67fe17ca5346c85d27e84018 Mon Sep 17 00:00:00 2001
From: MrBin99 <paul.pages12@gmail.com>
Date: Fri, 29 Jul 2022 14:45:50 +0200
Subject: [PATCH 02/29] style: disable flake8 line-length check

---
 .flake8 | 2 ++
 1 file changed, 2 insertions(+)
 create mode 100644 .flake8

diff --git a/.flake8 b/.flake8
new file mode 100644
index 0000000..5eb7392
--- /dev/null
+++ b/.flake8
@@ -0,0 +1,2 @@
+[flake8]
+max-line-length = 800

From 95711598015166862895648efef2528e43c72469 Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijs@fourdigits.nl>
Date: Sun, 14 May 2023 14:58:59 +0200
Subject: [PATCH 03/29] setup tests, linting and codestyle checks

---
 .gitignore     |  4 ++++
 pyproject.toml | 29 +++++++++++++++++++++++++++--
 tox.ini        | 40 ++++++++++++++++++++++++++++++++++++++++
 3 files changed, 71 insertions(+), 2 deletions(-)
 create mode 100644 tox.ini

diff --git a/.gitignore b/.gitignore
index c18dd8d..3659bc9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,5 @@
 __pycache__/
+.tox
+.vscode
+.coverage
+*.egg-info
diff --git a/pyproject.toml b/pyproject.toml
index a8f43fe..0614b47 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,2 +1,27 @@
-[tool.black]
-line-length = 79
+[tool.pytest.ini_options]
+DJANGO_SETTINGS_MODULE = "tests.settings"
+django_find_project = false
+pythonpath = "."
+addopts = '''
+  --cov=django_vite
+  --cov-report html
+  --cov-report term-missing
+  --cov-branch
+'''
+
+[tool.ruff]
+select = [
+  "E",   # pycodestyle
+  "F",   # pyflakes
+  "C",  # pyupgrade
+  "B", # bugbear,
+  "PT",  # pytest,
+  "SIM",  # simplify,
+  "DJ", # django
+]
+
+# do not autofix the following violations due to bad DX:
+unfixable = [
+  "F401",  # Module imported but unused
+  "F841", # Unused variables
+]
diff --git a/tox.ini b/tox.ini
new file mode 100644
index 0000000..5076ea8
--- /dev/null
+++ b/tox.ini
@@ -0,0 +1,40 @@
+[tox]
+envlist =
+    codestyle,
+    lint,
+    {py38,py39}-django{32,40,41,42},
+    {py310,py311}-django{41,42,-latest},
+isolated_build = true
+minversion = 1.9
+
+[testenv]
+description = run unit tests
+deps =
+    pytest>=7
+    pytest-cov
+    pytest-django
+    pytest-sugar
+    django32: Django>=3.2,<4.0
+    django40: Django>=4.0,<4.1
+    django41: Django>=4.1,<4.2
+    django42: Django>=4.2,<4.3
+    django-latest: https://github.com/django/django/archive/main.tar.gz
+commands =
+    pytest {posargs:tests}
+ignore_outcome =
+    django-latest: True
+
+[testenv:codestyle]
+basepython = python3
+commands =
+    black --check --diff .
+deps =
+    black
+skip_install = true
+
+[testenv:lint]
+commands =
+    ruff check django_vite
+deps =
+    ruff
+skip_install = true

From a2dd8ce112b63bd0528a4fb83e6eacc40dd985cc Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijs@fourdigits.nl>
Date: Sun, 14 May 2023 14:59:15 +0200
Subject: [PATCH 04/29] add a simple smoke test with some standard django
 settings

---
 tests/settings.py       | 28 ++++++++++++++++++++++++++++
 tests/test_singleton.py |  8 ++++++++
 2 files changed, 36 insertions(+)
 create mode 100644 tests/settings.py
 create mode 100644 tests/test_singleton.py

diff --git a/tests/settings.py b/tests/settings.py
new file mode 100644
index 0000000..189a195
--- /dev/null
+++ b/tests/settings.py
@@ -0,0 +1,28 @@
+import os
+
+BASE_DIR = os.path.dirname(__file__)
+
+STATIC_URL = "static"
+DJANGO_VITE_DEV_MODE = True
+DJANGO_VITE_ASSETS_PATH = "/"
+USE_TZ = True
+
+
+INSTALLED_APPS = [
+    "django_vite",
+]
+
+TEMPLATE_DEBUG = True
+TEMPLATE_DIRS = (os.path.join(BASE_DIR, "templates"),)
+
+TEMPLATES = [
+    {
+        "BACKEND": "django.template.backends.django.DjangoTemplates",
+        "DIRS": TEMPLATE_DIRS,
+        "APP_DIRS": True,
+        "OPTIONS": {
+            "context_processors": [],
+            "debug": TEMPLATE_DEBUG,
+        },
+    },
+]
diff --git a/tests/test_singleton.py b/tests/test_singleton.py
new file mode 100644
index 0000000..7ee8935
--- /dev/null
+++ b/tests/test_singleton.py
@@ -0,0 +1,8 @@
+import pytest
+
+from django_vite.templatetags.django_vite import DjangoViteAssetLoader
+
+
+def test_django_vite_asset_loader_cannot_be_instantiated():
+    with pytest.raises(RuntimeError):
+        DjangoViteAssetLoader()

From 98a6788ec49bf20da18b51204f6f3c425f91a82c Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijs@fourdigits.nl>
Date: Sun, 14 May 2023 15:05:07 +0200
Subject: [PATCH 05/29] add Github Action to run Tox test matrix

---
 .github/workflows/test.yml | 31 +++++++++++++++++++++++++++++++
 tox.ini                    |  7 +++++++
 2 files changed, 38 insertions(+)
 create mode 100644 .github/workflows/test.yml

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
new file mode 100644
index 0000000..ff9e985
--- /dev/null
+++ b/.github/workflows/test.yml
@@ -0,0 +1,31 @@
+name: Run Test Matrix
+
+on:
+  push:
+    branches: [master]
+  pull_request:
+    branches: [master]
+
+concurrency:
+  group: ${{ github.head_ref }}
+  cancel-in-progress: true
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
+
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install tox tox-gh-actions
+      - name: Test with tox
+        run: tox
diff --git a/tox.ini b/tox.ini
index 5076ea8..c746865 100644
--- a/tox.ini
+++ b/tox.ini
@@ -7,6 +7,13 @@ envlist =
 isolated_build = true
 minversion = 1.9
 
+[gh-actions]
+python =
+    3.8: py38
+    3.9: py39
+    3.10: py310
+    3.11: py311
+
 [testenv]
 description = run unit tests
 deps =

From e0b752e42fe04cf05957569c68edfa9be1ad8fb5 Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijskramer@gmail.com>
Date: Fri, 2 Jun 2023 10:56:58 +0200
Subject: [PATCH 06/29] fix codestyle warning

---
 django_vite/apps.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/django_vite/apps.py b/django_vite/apps.py
index ab781bc..98dae96 100644
--- a/django_vite/apps.py
+++ b/django_vite/apps.py
@@ -1,3 +1,5 @@
+from contextlib import suppress
+
 from django.apps import AppConfig
 from django.core.checks import Warning, register
 
@@ -9,12 +11,10 @@ class DjangoViteAppConfig(AppConfig):
     verbose_name = "Django Vite"
 
     def ready(self) -> None:
-        try:
-            # Create Loader instance at startup to prevent threading problems.
+        with suppress(RuntimeError):
+            # Create Loader instance at startup to prevent threading problems,
+            # but do not crash while doing so.
             DjangoViteAssetLoader.instance()
-        except RuntimeError:
-            # Just continue, the system check below outputs a warning.
-            pass
 
 
 @register

From 7c02345653e223a7eb8f0f12bfc4e411d82def59 Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijskramer@gmail.com>
Date: Fri, 2 Jun 2023 11:08:57 +0200
Subject: [PATCH 07/29] update pre-commit config

---
 .github/workflows/pre-commit.yml | 2 +-
 .pre-commit-config.yaml          | 9 ++-------
 2 files changed, 3 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/pre-commit.yml b/.github/workflows/pre-commit.yml
index e373bbc..0bcc3ed 100644
--- a/.github/workflows/pre-commit.yml
+++ b/.github/workflows/pre-commit.yml
@@ -12,4 +12,4 @@ jobs:
       - uses: "actions/setup-python@v4"
         with:
           python-version: 3.x
-      - uses: "pre-commit/action@v2.0.3"
+      - uses: "pre-commit/action@v3.0.0"
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index e5d7f53..c1391dc 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,6 +1,6 @@
 repos:
   - repo: "https://github.com/pre-commit/pre-commit-hooks"
-    rev: "v4.1.0"
+    rev: "v4.4.0"
     hooks:
       - id: "check-toml"
       - id: "check-yaml"
@@ -10,15 +10,10 @@ repos:
       - id: "trailing-whitespace"
 
   - repo: "https://github.com/psf/black"
-    rev: "22.3.0"
+    rev: "23.3.0"
     hooks:
       - id: "black"
 
-  - repo: "https://github.com/PyCQA/flake8"
-    rev: "4.0.1"
-    hooks:
-      - id: "flake8"
-
   - repo: "https://github.com/commitizen-tools/commitizen"
     rev: master
     hooks:

From adedfde12b640bc4ee3d309f0e5fbe9453a029e1 Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijskramer@gmail.com>
Date: Fri, 2 Jun 2023 11:10:36 +0200
Subject: [PATCH 08/29] re-format code with black

---
 django_vite/templatetags/django_vite.py | 40 +++++++------------------
 1 file changed, 11 insertions(+), 29 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index c923b5b..d8a14f7 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -25,9 +25,7 @@
 )
 
 # Default Vite server port.
-DJANGO_VITE_DEV_SERVER_PORT = getattr(
-    settings, "DJANGO_VITE_DEV_SERVER_PORT", 3000
-)
+DJANGO_VITE_DEV_SERVER_PORT = getattr(settings, "DJANGO_VITE_DEV_SERVER_PORT", 3000)
 
 # Default Vite server path to HMR script.
 DJANGO_VITE_WS_CLIENT_URL = getattr(
@@ -42,12 +40,10 @@
 # Must be included in your "STATICFILES_DIRS".
 # In Django production mode this folder need to be collected as static
 # files using "python manage.py collectstatic".
-DJANGO_VITE_ASSETS_PATH = Path(getattr(settings, "DJANGO_VITE_ASSETS_PATH"))
+DJANGO_VITE_ASSETS_PATH = Path(settings.DJANGO_VITE_ASSETS_PATH)
 
 # Prefix for STATIC_URL
-DJANGO_VITE_STATIC_URL_PREFIX = getattr(
-    settings, "DJANGO_VITE_STATIC_URL_PREFIX", ""
-)
+DJANGO_VITE_STATIC_URL_PREFIX = getattr(settings, "DJANGO_VITE_STATIC_URL_PREFIX", "")
 
 DJANGO_VITE_STATIC_ROOT = (
     DJANGO_VITE_ASSETS_PATH
@@ -68,9 +64,7 @@
     settings, "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF", "legacy-polyfills"
 )
 
-DJANGO_VITE_STATIC_URL = urljoin(
-    settings.STATIC_URL, DJANGO_VITE_STATIC_URL_PREFIX
-)
+DJANGO_VITE_STATIC_URL = urljoin(settings.STATIC_URL, DJANGO_VITE_STATIC_URL_PREFIX)
 
 # Make sure 'DJANGO_VITE_STATIC_URL' finish with a '/'
 if DJANGO_VITE_STATIC_URL[-1] != "/":
@@ -168,22 +162,16 @@ def _generate_css_files_of_asset(
         if "imports" in manifest_entry:
             for import_path in manifest_entry["imports"]:
                 tags.extend(
-                    self._generate_css_files_of_asset(
-                        import_path, already_processed
-                    )
+                    self._generate_css_files_of_asset(import_path, already_processed)
                 )
 
         if "css" in manifest_entry:
             for css_path in manifest_entry["css"]:
                 if css_path not in already_processed:
-                    url = (
-                        DjangoViteAssetLoader._generate_production_server_url(
-                            css_path
-                        )
-                    )
-                    tags.append(
-                        DjangoViteAssetLoader._generate_stylesheet_tag(url)
+                    url = DjangoViteAssetLoader._generate_production_server_url(
+                        css_path
                     )
+                    tags.append(DjangoViteAssetLoader._generate_stylesheet_tag(url))
 
                 already_processed.append(css_path)
 
@@ -383,9 +371,7 @@ def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
             str -- The script tag.
         """
 
-        attrs_str = " ".join(
-            [f'{key}="{value}"' for key, value in attrs.items()]
-        )
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
 
         return f'<script {attrs_str} src="{src}"></script>'
 
@@ -567,9 +553,7 @@ def vite_legacy_polyfills(**kwargs: Dict[str, str]) -> str:
         str -- The script tag to the polyfills.
     """
 
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
-        **kwargs
-    )
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(**kwargs)
 
 
 @register.simple_tag
@@ -601,9 +585,7 @@ def vite_legacy_asset(
 
     assert path is not None
 
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
-        path, **kwargs
-    )
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(path, **kwargs)
 
 
 @register.simple_tag

From 1b68f58a307634532715e7a0c5c5b93619485a8d Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijskramer@gmail.com>
Date: Fri, 2 Jun 2023 11:23:03 +0200
Subject: [PATCH 09/29] add troves to clarify for which django and python
 versions Django-Vite is intended

---
 setup.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index ecbbd11..35e7831 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
 setup(
     name="django-vite",
     version="2.1.2",
-    description="Integration of ViteJS in a Django project.",
+    description="Integration of Vite in a Django project.",
     long_description=README,
     long_description_content_type="text/markdown",
     author="MrBin99",
@@ -20,13 +20,23 @@
     include_package_data=True,
     packages=find_packages(),
     requires=[
-        "Django (>=1.11)",
+        "Django (>=3.2)",
     ],
     install_requires=[
-        "Django>=1.11",
+        "Django>=3.2",
     ],
     classifiers=[
         "License :: OSI Approved :: Apache Software License",
+        "Framework :: Django",
+        "Framework :: Django :: 3.2",
+        "Framework :: Django :: 4.0",
+        "Framework :: Django :: 4.1",
+        "Framework :: Django :: 4.2",
+        "Programming Language :: Python",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
     ],
-    extras_require={"dev": ["black", "flake8"]},
+    extras_require={"dev": ["black"]},
 )

From 9c1ed2cd42af014f76e5a1addcaa17e57a53fdd1 Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijskramer@gmail.com>
Date: Mon, 5 Jun 2023 21:07:52 +0200
Subject: [PATCH 10/29] fix linting errors

---
 django_vite/templatetags/django_vite.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index d8a14f7..4ff55de 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -301,15 +301,14 @@ def _parse_manifest(self) -> None:
         """
 
         try:
-            manifest_file = open(DJANGO_VITE_MANIFEST_PATH, "r")
-            manifest_content = manifest_file.read()
-            manifest_file.close()
+            with open(DJANGO_VITE_MANIFEST_PATH, "r") as manifest_file:
+                manifest_content = manifest_file.read()
             self._manifest = json.loads(manifest_content)
         except Exception as error:
             raise RuntimeError(
                 f"Cannot read Vite manifest file at "
                 f"{DJANGO_VITE_MANIFEST_PATH} : {str(error)}"
-            )
+            ) from error
 
     @classmethod
     def instance(cls):

From 05d21215c847c1857beeb73259e3e5bf89ce8184 Mon Sep 17 00:00:00 2001
From: Thijs Kramer <thijskramer@gmail.com>
Date: Mon, 5 Jun 2023 22:24:38 +0200
Subject: [PATCH 11/29] add tests for templatetags

---
 ...test_singleton.py => test_asset_loader.py} |   0
 tests/test_templatetags.py                    | 150 ++++++++++++++++++
 tox.ini                                       |   1 +
 3 files changed, 151 insertions(+)
 rename tests/{test_singleton.py => test_asset_loader.py} (100%)
 create mode 100644 tests/test_templatetags.py

diff --git a/tests/test_singleton.py b/tests/test_asset_loader.py
similarity index 100%
rename from tests/test_singleton.py
rename to tests/test_asset_loader.py
diff --git a/tests/test_templatetags.py b/tests/test_templatetags.py
new file mode 100644
index 0000000..79bb5c3
--- /dev/null
+++ b/tests/test_templatetags.py
@@ -0,0 +1,150 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template, TemplateSyntaxError
+
+
+@pytest.fixture()
+def override_setting(monkeypatch):
+    def _override_setting(setting, value):
+        monkeypatch.setattr(
+            f"django_vite.templatetags.django_vite.{setting}",
+            value,
+        )
+
+    return _override_setting
+
+
+def test_vite_hmr_client_returns_script_tag():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/@vite/client"
+    assert script_tag["type"] == "module"
+
+
+def test_vite_hmr_client_kwargs():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client blocking="render" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag.has_attr("blocking")
+    assert script_tag["blocking"] == "render"
+
+
+def test_vite_hmr_client_returns_nothing_with_dev_mode_off(settings, monkeypatch):
+    settings.DJANGO_VITE_DEV_MODE = False
+    monkeypatch.setattr(
+        "django_vite.templatetags.django_vite.DJANGO_VITE_DEV_MODE",
+        settings.DJANGO_VITE_DEV_MODE,
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+def test_vite_hmr_client_uses_correct_settings(override_setting):
+    override_setting("DJANGO_VITE_DEV_SERVER_PROTOCOL", "https")
+    override_setting("DJANGO_VITE_DEV_SERVER_HOST", "127.0.0.2")
+    override_setting("DJANGO_VITE_DEV_SERVER_PORT", "5174")
+    override_setting("DJANGO_VITE_STATIC_URL", "static/")
+    override_setting("DJANGO_VITE_WS_CLIENT_URL", "foo/bar")
+
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "https://127.0.0.2:5174/static/foo/bar"
+
+
+def test_vite_asset_returns_script_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.tsx" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.tsx"
+    assert script_tag["type"] == "module"
+
+
+def test_vite_asset_raises_without_path():
+    with pytest.raises(TemplateSyntaxError):
+        Template(
+            """
+            {% load django_vite %}
+            {% vite_asset %}
+        """
+        )
+
+
+def test_vite_react_refresh_happy_flow():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert not script_tag.has_attr("src")
+    assert script_tag.has_attr("type")
+    assert script_tag["type"] == "module"
+    assert "__vite_plugin_react_preamble_installed__" in script_tag.text
+    assert "http://localhost:3000/static/@react-refresh" in script_tag.text
+
+
+def test_vite_react_refresh_returns_nothing_with_dev_mode_off(settings, monkeypatch):
+    settings.DJANGO_VITE_DEV_MODE = False
+    monkeypatch.setattr(
+        "django_vite.templatetags.django_vite.DJANGO_VITE_DEV_MODE",
+        settings.DJANGO_VITE_DEV_MODE,
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+def test_vite_react_refresh_url_setting(override_setting):
+    override_setting("DJANGO_VITE_REACT_REFRESH_URL", "foobar")
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert "http://localhost:3000/static/foobar" in script_tag.text
diff --git a/tox.ini b/tox.ini
index c746865..193fa9c 100644
--- a/tox.ini
+++ b/tox.ini
@@ -17,6 +17,7 @@ python =
 [testenv]
 description = run unit tests
 deps =
+    beautifulsoup4
     pytest>=7
     pytest-cov
     pytest-django

From ebc5f68ce671f4ed82949ec945de962ff0b93d98 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 17:02:48 -0700
Subject: [PATCH 12/29] config-ify DJANGO_VITE_REACT_REFRESH_URL

---
 django_vite/templatetags/django_vite.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 7b74564..89d4992 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -55,6 +55,9 @@ class DjangoViteConfig(NamedTuple):
     # Path to your manifest file generated by Vite.
     manifest_path: Union[Path, str] = ""
 
+    # Default Vite server path to React RefreshRuntime for @vitejs/plugin-react.
+    react_refresh_url: str = "@react-refresh"
+
     @property
     def static_root(self) -> Union[Path, str]:
         """
@@ -763,7 +766,7 @@ def _generate_vite_server_url(
         )
 
     @classmethod
-    def generate_vite_react_refresh_url(cls) -> str:
+    def generate_vite_react_refresh_url(self, config_key: str = "default") -> str:
         """
         Generates the script for the Vite React Refresh for HMR.
         Only used in development, in production this method returns
@@ -772,13 +775,14 @@ def generate_vite_react_refresh_url(cls) -> str:
         Returns:
             str -- The script or an empty string.
         """
+        config = self._get_config(config_key)
 
-        if not settings.get("DJANGO_VITE_DEV_MODE"):
+        if not config.dev_mode:
             return ""
 
         return f"""<script type="module">
             import RefreshRuntime from \
-            '{cls._generate_vite_server_url(settings.get("DJANGO_VITE_REACT_REFRESH_URL"))}'
+            '{self._generate_vite_server_url(config.react_refresh_url)}'
             RefreshRuntime.injectIntoGlobalHook(window)
             window.$RefreshReg$ = () => {{}}
             window.$RefreshSig$ = () => (type) => type
@@ -1000,7 +1004,7 @@ def vite_legacy_asset(
 
 @register.simple_tag
 @mark_safe
-def vite_react_refresh() -> str:
+def vite_react_refresh(config_key: str = "default") -> str:
     """
     Generates the script for the Vite React Refresh for HMR.
     Only used in development, in production this method returns
@@ -1009,4 +1013,4 @@ def vite_react_refresh() -> str:
     Returns:
         str -- The script or an empty string.
     """
-    return DjangoViteAssetLoader.generate_vite_react_refresh_url()
+    return DjangoViteAssetLoader.generate_vite_react_refresh_url(config_key)

From b77eb63e08d14a4abb3b194d54c07488d717ddcd Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 17:15:06 -0700
Subject: [PATCH 13/29] run pre-commit, remove duplicated
 _generate_css_files_of_asset

---
 django_vite/templatetags/django_vite.py | 58 +++++--------------------
 1 file changed, 10 insertions(+), 48 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 89d4992..87f9c97 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -191,48 +191,6 @@ def generate_vite_asset(
 
         return "\n".join(tags)
 
-    def _generate_css_files_of_asset(
-        self,
-        path: str,
-        config_key: str,
-        already_processed: List[str],
-    ) -> List[str]:
-        """
-        Generates all CSS tags for dependencies of an asset.
-
-        Arguments:
-            path {str} -- Path to an asset in the 'manifest.json'.
-            config_key {str} -- Key of the configuration to use.
-            already_processed {list} -- List of already processed CSS file.
-
-        Returns:
-            list -- List of CSS tags.
-        """
-
-        tags = []
-        static_url = self._get_static_url(config_key)
-        manifest = self._get_manifest(config_key)
-        manifest_entry = manifest[path]
-
-        for import_path in manifest_entry.imports:
-            tags.extend(
-                self._generate_css_files_of_asset(
-                    import_path, already_processed
-                )
-            )
-
-        for css_path in manifest_entry.css:
-            if css_path not in already_processed:
-                tags.append(
-                    DjangoViteAssetLoader._generate_stylesheet_tag(
-                        urljoin(static_url, css_path)
-                    )
-                )
-
-            already_processed.append(css_path)
-
-        return tags
-
     def preload_vite_asset(
         self,
         path: str,
@@ -327,7 +285,11 @@ def _load_css_files_of_asset(
         )
 
     def _generate_css_files_of_asset(
-        self, path: str, already_processed: List[str], config_key: str, tag_generator: Callable
+        self,
+        path: str,
+        already_processed: List[str],
+        config_key: str,
+        tag_generator: Callable,
     ) -> List[str]:
         """
         Generates all CSS tags for dependencies of an asset.
@@ -355,10 +317,8 @@ def _generate_css_files_of_asset(
 
         for css_path in manifest_entry.css:
             if css_path not in already_processed:
-                url = (
-                    DjangoViteAssetLoader._generate_production_server_url(
-                        css_path
-                    )
+                url = DjangoViteAssetLoader._generate_production_server_url(
+                    css_path
                 )
                 tags.append(tag_generator(url))
 
@@ -655,7 +615,9 @@ def instance(cls):
         return cls._instance
 
     @classmethod
-    def generate_vite_ws_client(cls, config_key: str, **kwargs: Dict[str, str]) -> str:
+    def generate_vite_ws_client(
+        cls, config_key: str, **kwargs: Dict[str, str]
+    ) -> str:
         """
         Generates the script tag for the Vite WS client for HMR.
         Only used in development, in production this method returns

From 502ed2fc7a6e21dae033418e563c8427778c4d85 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 17:22:31 -0700
Subject: [PATCH 14/29] set rev for pre-commit commitizen

---
 .pre-commit-config.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index e5d7f53..0d5e749 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -20,7 +20,7 @@ repos:
       - id: "flake8"
 
   - repo: "https://github.com/commitizen-tools/commitizen"
-    rev: master
+    rev: "3.10.0"
     hooks:
       - id: "commitizen"
         stages: ["commit-msg"]

From c7c213f927cecf5b1862cb8940d7a33525ff6a90 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 22:50:11 -0700
Subject: [PATCH 15/29] config-ify _generate_production_server_url; set
 defaults for config_key

---
 django_vite/templatetags/django_vite.py | 116 ++++++++++--------------
 1 file changed, 48 insertions(+), 68 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 87f9c97..d8556b0 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -10,6 +10,8 @@
 
 register = template.Library()
 
+DEFAULT_CONFIG_KEY = "default"
+
 
 class DjangoViteManifest(NamedTuple):
     """
@@ -58,6 +60,13 @@ class DjangoViteConfig(NamedTuple):
     # Default Vite server path to React RefreshRuntime for @vitejs/plugin-react.
     react_refresh_url: str = "@react-refresh"
 
+    @property
+    def static_url(self) -> str:
+        url = urljoin(settings.STATIC_URL, self.static_url_prefix)
+        if not url.endswith("/"):
+            url += "/"
+        return url
+
     @property
     def static_root(self) -> Union[Path, str]:
         """
@@ -136,13 +145,11 @@ def generate_vite_asset(
         """
 
         config = self._get_config(config_key)
-        static_url = self._get_static_url(config_key)
+        static_url = config.static_url
 
         if config.dev_mode:
             return DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_vite_server_url(
-                    path, static_url, config
-                ),
+                DjangoViteAssetLoader._generate_vite_server_url(path, config),
                 {"type": "module", **kwargs},
             )
 
@@ -180,7 +187,7 @@ def generate_vite_asset(
             dep_manifest_entry = self._manifest[dep]
             dep_file = dep_manifest_entry["file"]
             url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file
+                dep_file, config.static_url_prefix
             )
             tags.append(
                 DjangoViteAssetLoader._generate_preload_tag(
@@ -304,7 +311,7 @@ def _generate_css_files_of_asset(
         """
 
         tags = []
-        static_url = self._get_static_url(config_key)
+        config = self._get_config(config_key)
         manifest = self._get_manifest(config_key)
         manifest_entry = manifest[path]
 
@@ -318,7 +325,7 @@ def _generate_css_files_of_asset(
         for css_path in manifest_entry.css:
             if css_path not in already_processed:
                 url = DjangoViteAssetLoader._generate_production_server_url(
-                    css_path
+                    css_path, config.static_url_prefix
                 )
                 tags.append(tag_generator(url))
 
@@ -344,7 +351,6 @@ def generate_vite_asset_url(self, path: str, config_key: str) -> str:
         """
 
         config = self._get_config(config_key)
-        static_url = self._get_static_url(config_key)
 
         if config.dev_mode:
             return DjangoViteAssetLoader._generate_vite_server_url(
@@ -360,7 +366,7 @@ def generate_vite_asset_url(self, path: str, config_key: str) -> str:
             )
 
         return DjangoViteAssetLoader._generate_production_server_url(
-            self._manifest[path]["file"]
+            self._manifest[path]["file"], config.static_url_prefix
         )
 
     def generate_vite_legacy_polyfills(
@@ -391,7 +397,6 @@ def generate_vite_legacy_polyfills(
 
         config = self._get_config(config_key)
         manifest = self._get_manifest(config_key)
-        static_url = self._get_static_url(config_key)
 
         if config.dev_mode:
             return ""
@@ -402,7 +407,7 @@ def generate_vite_legacy_polyfills(
             if config.legacy_polyfills_motif in path:
                 return DjangoViteAssetLoader._generate_script_tag(
                     DjangoViteAssetLoader._generate_production_server_url(
-                        content["file"]
+                        content["file"], config.static_url_prefix
                     ),
                     attrs=scripts_attrs,
                 )
@@ -441,7 +446,6 @@ def generate_vite_legacy_asset(
         """
 
         config = self._get_config(config_key)
-        static_url = self._get_static_url(config_key)
 
         if config.dev_mode:
             return ""
@@ -458,7 +462,7 @@ def generate_vite_legacy_asset(
 
         return DjangoViteAssetLoader._generate_script_tag(
             DjangoViteAssetLoader._generate_production_server_url(
-                manifest[path].file
+                manifest[path].file, config.static_url_prefix
             ),
             attrs=scripts_attrs,
         )
@@ -536,27 +540,6 @@ def _get_manifest(
 
         return self._manifests[config_key]
 
-    def _get_static_url(self, config_key: str) -> str:
-        """
-        Build the static URL of a specified configuration.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            str -- The static URL.
-        """
-
-        if config_key not in self._static_urls:
-            config = self._get_config(config_key)
-            static_url = urljoin(settings.STATIC_URL, config.static_url_prefix)
-
-            self._static_urls[config_key] = (
-                static_url if static_url[-1] == "/" else static_url + "/"
-            )
-
-        return self._static_urls[config_key]
-
     @classmethod
     def instance(cls):
         """
@@ -610,13 +593,15 @@ def instance(cls):
                     if setting_key in _config_keys.keys()
                 }
 
-                cls._instance._configs["default"] = DjangoViteConfig(**config)
+                cls._instance._configs[DEFAULT_CONFIG_KEY] = DjangoViteConfig(
+                    **config
+                )
 
         return cls._instance
 
     @classmethod
     def generate_vite_ws_client(
-        cls, config_key: str, **kwargs: Dict[str, str]
+        cls, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
     ) -> str:
         """
         Generates the script tag for the Vite WS client for HMR.
@@ -635,15 +620,12 @@ def generate_vite_ws_client(
         """
 
         config = cls._get_config(config_key)
-        static_url = cls._get_static_url(config_key)
 
         if not config.dev_mode:
             return ""
 
         return cls._generate_script_tag(
-            cls._generate_vite_server_url(
-                config.ws_client_url, static_url, config
-            ),
+            cls._generate_vite_server_url(config.ws_client_url, config),
             {"type": "module", **kwargs},
         )
 
@@ -707,7 +689,6 @@ def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
     @staticmethod
     def _generate_vite_server_url(
         path: str,
-        static_url: str,
         config: Type[DjangoViteConfig],
     ) -> str:
         """
@@ -724,11 +705,13 @@ def _generate_vite_server_url(
         return urljoin(
             f"{config.dev_server_protocol}://"
             f"{config.dev_server_host}:{config.dev_server_port}",
-            urljoin(static_url, path),
+            urljoin(config.static_url, path),
         )
 
     @classmethod
-    def generate_vite_react_refresh_url(self, config_key: str = "default") -> str:
+    def generate_vite_react_refresh_url(
+        self, config_key: str = DEFAULT_CONFIG_KEY
+    ) -> str:
         """
         Generates the script for the Vite React Refresh for HMR.
         Only used in development, in production this method returns
@@ -744,7 +727,7 @@ def generate_vite_react_refresh_url(self, config_key: str = "default") -> str:
 
         return f"""<script type="module">
             import RefreshRuntime from \
-            '{self._generate_vite_server_url(config.react_refresh_url)}'
+            '{self._generate_vite_server_url(config.react_refresh_url, config)}'
             RefreshRuntime.injectIntoGlobalHook(window)
             window.$RefreshReg$ = () => {{}}
             window.$RefreshSig$ = () => (type) => type
@@ -752,7 +735,9 @@ def generate_vite_react_refresh_url(self, config_key: str = "default") -> str:
         </script>"""
 
     @staticmethod
-    def _generate_production_server_url(path: str) -> str:
+    def _generate_production_server_url(
+        path: str, static_url_prefix=""
+    ) -> str:
         """
         Generates an URL to an asset served during production.
 
@@ -764,8 +749,8 @@ def _generate_production_server_url(path: str) -> str:
         """
 
         production_server_url = path
-        if prefix := settings.get("DJANGO_VITE_STATIC_URL_PREFIX", ""):
-            if not settings.get("DJANGO_VITE_STATIC_URL_PREFIX", "").endswith("/"):
+        if prefix := static_url_prefix:
+            if not static_url_prefix.endswith("/"):
                 prefix += "/"
             production_server_url = urljoin(prefix, path)
 
@@ -779,7 +764,9 @@ def _generate_production_server_url(path: str) -> str:
 
 @register.simple_tag
 @mark_safe
-def vite_hmr_client(config: str = "default", **kwargs: Dict[str, str]) -> str:
+def vite_hmr_client(
+    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+) -> str:
     """
     Generates the script tag for the Vite WS client for HMR.
     Only used in development, in production this method returns
@@ -796,14 +783,14 @@ def vite_hmr_client(config: str = "default", **kwargs: Dict[str, str]) -> str:
             script tags.
     """
 
-    return DjangoViteAssetLoader.generate_vite_ws_client(config, **kwargs)
+    return DjangoViteAssetLoader.generate_vite_ws_client(config_key, **kwargs)
 
 
 @register.simple_tag
 @mark_safe
 def vite_asset(
     path: str,
-    config: str = "default",
+    config_key: str = DEFAULT_CONFIG_KEY,
     **kwargs: Dict[str, str],
 ) -> str:
     """
@@ -833,10 +820,10 @@ def vite_asset(
     """
 
     assert path is not None
-    assert config is not None
+    assert config_key is not None
 
     return DjangoViteAssetLoader.instance().generate_vite_asset(
-        path, config, **kwargs
+        path, config_key, **kwargs
     )
 
 
@@ -869,7 +856,7 @@ def vite_preload_asset(
 
 
 @register.simple_tag
-def vite_asset_url(path: str, config: str = "default") -> str:
+def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
     """
     Generates only the URL of an asset managed by ViteJS.
     Warning, this function does not generate URLs for dependant assets.
@@ -885,19 +872,16 @@ def vite_asset_url(path: str, config: str = "default") -> str:
     Returns:
         str -- The URL of this asset.
     """
-
     assert path is not None
-    assert config is not None
-
     return DjangoViteAssetLoader.instance().generate_vite_asset_url(
-        path, config
+        path, config_key
     )
 
 
 @register.simple_tag
 @mark_safe
 def vite_legacy_polyfills(
-    config: str = "default", **kwargs: Dict[str, str]
+    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
 ) -> str:
     """
     Generates a <script> tag to the polyfills generated
@@ -906,7 +890,7 @@ def vite_legacy_polyfills(
     other legacy scripts.
 
     Arguments:
-        config {str} -- Configuration to use.
+        config_key {str} -- Configuration to use.
 
     Keyword Arguments:
         **kwargs {Dict[str, str]} -- Adds new attributes to generated
@@ -919,11 +903,8 @@ def vite_legacy_polyfills(
     Returns:
         str -- The script tag to the polyfills.
     """
-
-    assert config is not None
-
     return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
-        config, **kwargs
+        config_key, **kwargs
     )
 
 
@@ -931,7 +912,7 @@ def vite_legacy_polyfills(
 @mark_safe
 def vite_legacy_asset(
     path: str,
-    config: str = "default",
+    config_key: str = DEFAULT_CONFIG_KEY,
     **kwargs: Dict[str, str],
 ) -> str:
     """
@@ -942,7 +923,7 @@ def vite_legacy_asset(
     Arguments:
         path {str} -- Path to a Vite asset to include
             (must contains '-legacy' in its name).
-        config {str} -- Configuration to use.
+        config_key {str} -- Configuration to use.
 
     Keyword Arguments:
         **kwargs {Dict[str, str]} -- Adds new attributes to generated
@@ -957,16 +938,15 @@ def vite_legacy_asset(
     """
 
     assert path is not None
-    assert config is not None
 
     return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
-        path, config, **kwargs
+        path, config_key, **kwargs
     )
 
 
 @register.simple_tag
 @mark_safe
-def vite_react_refresh(config_key: str = "default") -> str:
+def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
     """
     Generates the script for the Vite React Refresh for HMR.
     Only used in development, in production this method returns

From 47eaaa49bcad0fc0702a5df9cdfbdcac14ece2cd Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 22:53:45 -0700
Subject: [PATCH 16/29] remove flake; increase black line-length

---
 .flake8                                 |  2 --
 .pre-commit-config.yaml                 |  5 ---
 django_vite/templatetags/django_vite.py | 41 ++++++-------------------
 pyproject.toml                          |  2 +-
 4 files changed, 11 insertions(+), 39 deletions(-)
 delete mode 100644 .flake8

diff --git a/.flake8 b/.flake8
deleted file mode 100644
index 5eb7392..0000000
--- a/.flake8
+++ /dev/null
@@ -1,2 +0,0 @@
-[flake8]
-max-line-length = 800
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 0d5e749..04a9a88 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -14,11 +14,6 @@ repos:
     hooks:
       - id: "black"
 
-  - repo: "https://github.com/PyCQA/flake8"
-    rev: "4.0.1"
-    hooks:
-      - id: "flake8"
-
   - repo: "https://github.com/commitizen-tools/commitizen"
     rev: "3.10.0"
     hooks:
diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index d8556b0..d737e82 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -244,9 +244,7 @@ def preload_vite_asset(
         }
 
         manifest_file = manifest_entry["file"]
-        url = DjangoViteAssetLoader._generate_production_server_url(
-            manifest_file
-        )
+        url = DjangoViteAssetLoader._generate_production_server_url(manifest_file)
         tags.append(
             DjangoViteAssetLoader._generate_preload_tag(
                 url,
@@ -261,9 +259,7 @@ def preload_vite_asset(
         for dep in manifest_entry.get("imports", []):
             dep_manifest_entry = self._manifest[dep]
             dep_file = dep_manifest_entry["file"]
-            url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file
-            )
+            url = DjangoViteAssetLoader._generate_production_server_url(dep_file)
             tags.append(
                 DjangoViteAssetLoader._generate_preload_tag(
                     url,
@@ -353,9 +349,7 @@ def generate_vite_asset_url(self, path: str, config_key: str) -> str:
         config = self._get_config(config_key)
 
         if config.dev_mode:
-            return DjangoViteAssetLoader._generate_vite_server_url(
-                path, config
-            )
+            return DjangoViteAssetLoader._generate_vite_server_url(path, config)
 
         manifest = self._get_manifest(config_key)
 
@@ -503,16 +497,11 @@ def _parse_manifest(
         config = self._get_config(config_key)
 
         try:
-            with open(
-                config.get_computed_manifest_path(), "r"
-            ) as manifest_file:
+            with open(config.get_computed_manifest_path(), "r") as manifest_file:
                 manifest_content = manifest_file.read()
                 manifest_json = json.loads(manifest_content)
 
-                return {
-                    k: DjangoViteManifest(**v)
-                    for k, v in manifest_json.items()
-                }
+                return {k: DjangoViteManifest(**v) for k, v in manifest_json.items()}
 
         except Exception as error:
             raise RuntimeError(
@@ -520,9 +509,7 @@ def _parse_manifest(
                 f"{config.get_computed_manifest_path()} : {str(error)}"
             )
 
-    def _get_manifest(
-        self, config_key: str
-    ) -> Dict[str, Type[DjangoViteManifest]]:
+    def _get_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
         """
         Load if needed and parse the "manifest.json" of the specified
         configuration.
@@ -645,9 +632,7 @@ def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
             str -- The script tag.
         """
 
-        attrs_str = " ".join(
-            [f'{key}="{value}"' for key, value in attrs.items()]
-        )
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
 
         return f'<script {attrs_str} src="{src}"></script>'
 
@@ -680,9 +665,7 @@ def _generate_stylesheet_preload_tag(href: str) -> str:
 
     @staticmethod
     def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
-        attrs_str = " ".join(
-            [f'{key}="{value}"' for key, value in attrs.items()]
-        )
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
 
         return f'<link href="{href}" {attrs_str} />'
 
@@ -735,9 +718,7 @@ def generate_vite_react_refresh_url(
         </script>"""
 
     @staticmethod
-    def _generate_production_server_url(
-        path: str, static_url_prefix=""
-    ) -> str:
+    def _generate_production_server_url(path: str, static_url_prefix="") -> str:
         """
         Generates an URL to an asset served during production.
 
@@ -873,9 +854,7 @@ def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
         str -- The URL of this asset.
     """
     assert path is not None
-    return DjangoViteAssetLoader.instance().generate_vite_asset_url(
-        path, config_key
-    )
+    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, config_key)
 
 
 @register.simple_tag
diff --git a/pyproject.toml b/pyproject.toml
index a8f43fe..a674447 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,2 +1,2 @@
 [tool.black]
-line-length = 79
+line-length = 85

From b72a032ee3c347222c98c16964793828da117d4c Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 23:31:36 -0700
Subject: [PATCH 17/29] config-ify preload_vite_asset

---
 django_vite/templatetags/django_vite.py | 65 +++++++++++++++----------
 1 file changed, 40 insertions(+), 25 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index d737e82..7cd6b9d 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -165,7 +165,7 @@ def generate_vite_asset(
         scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
 
         # Add dependent CSS
-        tags.extend(self._generate_css_files_of_asset(path, config_key, []))
+        tags.extend(self._load_css_files_of_asset(path, [], config_key))
 
         # Add the script by itself
         tags.append(
@@ -201,6 +201,7 @@ def generate_vite_asset(
     def preload_vite_asset(
         self,
         path: str,
+        config_key: str = DEFAULT_CONFIG_KEY,
     ) -> str:
         """
         Generates a <link modulepreload> tag for this JS/TS asset, a
@@ -211,6 +212,7 @@ def preload_vite_asset(
 
         Arguments:
             path {str} -- Path to a Vite JS/TS asset to preload.
+            config_key {str} -- Key of the configuration to use.
 
         Returns:
             str -- All tags to preload this file in your HTML page.
@@ -223,18 +225,20 @@ def preload_vite_asset(
             str -- all <link> tags to preload
                 this asset.
         """
-        if settings.DJANGO_VITE_DEV_MODE:
+        tags = []
+        config = self._get_config(config_key)
+        manifest = self._get_manifest(config_key)
+        manifest_entry = manifest[path]
+
+        if not config.dev_mode:
             return ""
 
-        if not self._manifest or path not in self._manifest:
+        if path not in manifest:
             raise RuntimeError(
                 f"Cannot find {path} in Vite manifest "
-                f"at {settings.DJANGO_VITE_MANIFEST_PATH}"
+                f"at {config.get_computed_manifest_path()}"
             )
 
-        tags = []
-        manifest_entry = self._manifest[path]
-
         # Add the script by itself
         script_attrs = {
             "type": "text/javascript",
@@ -243,8 +247,10 @@ def preload_vite_asset(
             "as": "script",
         }
 
-        manifest_file = manifest_entry["file"]
-        url = DjangoViteAssetLoader._generate_production_server_url(manifest_file)
+        manifest_file = manifest_entry.file
+        url = DjangoViteAssetLoader._generate_production_server_url(
+            manifest_file, config.static_url_prefix
+        )
         tags.append(
             DjangoViteAssetLoader._generate_preload_tag(
                 url,
@@ -253,13 +259,14 @@ def preload_vite_asset(
         )
 
         # Add dependent CSS
-        tags.extend(self._preload_css_files_of_asset(path, []))
+        tags.extend(self._preload_css_files_of_asset(path, [], config_key))
 
         # Preload imports
-        for dep in manifest_entry.get("imports", []):
-            dep_manifest_entry = self._manifest[dep]
-            dep_file = dep_manifest_entry["file"]
-            url = DjangoViteAssetLoader._generate_production_server_url(dep_file)
+        for dependency_path in manifest_entry.imports:
+            dependency_file = manifest[dependency_path].file
+            url = DjangoViteAssetLoader._generate_production_server_url(
+                dependency_file, config.static_url_prefix
+            )
             tags.append(
                 DjangoViteAssetLoader._generate_preload_tag(
                     url,
@@ -270,29 +277,37 @@ def preload_vite_asset(
         return "\n".join(tags)
 
     def _preload_css_files_of_asset(
-        self, path: str, already_processed: List[str]
+        self,
+        path: str,
+        already_processed: List[str],
+        config_key: str = DEFAULT_CONFIG_KEY,
     ) -> List[str]:
         return self._generate_css_files_of_asset(
             path,
             already_processed,
             DjangoViteAssetLoader._generate_stylesheet_preload_tag,
+            config_key,
         )
 
     def _load_css_files_of_asset(
-        self, path: str, already_processed: List[str]
+        self,
+        path: str,
+        already_processed: List[str],
+        config_key: str = DEFAULT_CONFIG_KEY,
     ) -> List[str]:
         return self._generate_css_files_of_asset(
             path,
             already_processed,
             DjangoViteAssetLoader._generate_stylesheet_tag,
+            config_key,
         )
 
     def _generate_css_files_of_asset(
         self,
         path: str,
         already_processed: List[str],
-        config_key: str,
         tag_generator: Callable,
+        config_key: str = DEFAULT_CONFIG_KEY,
     ) -> List[str]:
         """
         Generates all CSS tags for dependencies of an asset.
@@ -314,7 +329,7 @@ def _generate_css_files_of_asset(
         for import_path in manifest_entry.imports:
             tags.extend(
                 self._generate_css_files_of_asset(
-                    import_path, already_processed, tag_generator
+                    import_path, already_processed, tag_generator, config_key
                 )
             )
 
@@ -454,10 +469,11 @@ def generate_vite_legacy_asset(
 
         scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
 
+        url = DjangoViteAssetLoader._generate_production_server_url(
+            manifest[path].file, config.static_url_prefix
+        )
         return DjangoViteAssetLoader._generate_script_tag(
-            DjangoViteAssetLoader._generate_production_server_url(
-                manifest[path].file, config.static_url_prefix
-            ),
+            url,
             attrs=scripts_attrs,
         )
 
@@ -702,6 +718,7 @@ def generate_vite_react_refresh_url(
 
         Returns:
             str -- The script or an empty string.
+            config_key {str} -- Key of the configuration to use.
         """
         config = self._get_config(config_key)
 
@@ -810,9 +827,7 @@ def vite_asset(
 
 @register.simple_tag
 @mark_safe
-def vite_preload_asset(
-    path: str,
-) -> str:
+def vite_preload_asset(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
     """
     Generates preloadmodule tag for this JS/TS asset and preloads
     all of its CSS and JS dependencies by reading the manifest
@@ -833,7 +848,7 @@ def vite_preload_asset(
 
     assert path is not None
 
-    return DjangoViteAssetLoader.instance().preload_vite_asset(path)
+    return DjangoViteAssetLoader.instance().preload_vite_asset(path, config_key)
 
 
 @register.simple_tag

From 1ff26b83ab2e5c1f9ffc794920576a29ef75f27d Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 23:34:34 -0700
Subject: [PATCH 18/29] remove unused 'already_processed' arg from
 _preload_css_files_of_asset and _load_css_files_of_asset

---
 django_vite/templatetags/django_vite.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 7cd6b9d..fe25c65 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -165,7 +165,7 @@ def generate_vite_asset(
         scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
 
         # Add dependent CSS
-        tags.extend(self._load_css_files_of_asset(path, [], config_key))
+        tags.extend(self._load_css_files_of_asset(path, config_key))
 
         # Add the script by itself
         tags.append(
@@ -259,7 +259,7 @@ def preload_vite_asset(
         )
 
         # Add dependent CSS
-        tags.extend(self._preload_css_files_of_asset(path, [], config_key))
+        tags.extend(self._preload_css_files_of_asset(path, config_key))
 
         # Preload imports
         for dependency_path in manifest_entry.imports:
@@ -279,12 +279,11 @@ def preload_vite_asset(
     def _preload_css_files_of_asset(
         self,
         path: str,
-        already_processed: List[str],
         config_key: str = DEFAULT_CONFIG_KEY,
     ) -> List[str]:
         return self._generate_css_files_of_asset(
             path,
-            already_processed,
+            [],
             DjangoViteAssetLoader._generate_stylesheet_preload_tag,
             config_key,
         )
@@ -292,12 +291,11 @@ def _preload_css_files_of_asset(
     def _load_css_files_of_asset(
         self,
         path: str,
-        already_processed: List[str],
         config_key: str = DEFAULT_CONFIG_KEY,
     ) -> List[str]:
         return self._generate_css_files_of_asset(
             path,
-            already_processed,
+            [],
             DjangoViteAssetLoader._generate_stylesheet_tag,
             config_key,
         )

From a71580ded2b8c7921e7889d3570585ed593ffbbc Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Fri, 13 Oct 2023 23:37:22 -0700
Subject: [PATCH 19/29] remove _static_urls since static_url is now a property
 of DjangoViteConfig

---
 django_vite/templatetags/django_vite.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index fe25c65..e23b717 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -106,7 +106,6 @@ class DjangoViteAssetLoader:
 
     _configs = Dict[str, Type[DjangoViteConfig]]
     _manifests: Dict[str, Type[DjangoViteManifest]]
-    _static_urls: Dict[str, str]
 
     def __init__(self) -> None:
         raise RuntimeError("Use the instance() method instead.")
@@ -556,7 +555,6 @@ def instance(cls):
             cls._instance = cls.__new__(cls)
             cls._instance._configs = {}
             cls._instance._manifests = {}
-            cls._instance._static_urls = {}
 
             if hasattr(settings, "DJANGO_VITE"):
                 config = getattr(settings, "DJANGO_VITE")

From 8e7f9e3493f79108504f70a9380a942dbe88f5eb Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Sat, 14 Oct 2023 08:47:57 -0700
Subject: [PATCH 20/29] fix missed merge conflict issues

---
 django_vite/templatetags/django_vite.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index e23b717..91af64d 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -144,7 +144,6 @@ def generate_vite_asset(
         """
 
         config = self._get_config(config_key)
-        static_url = config.static_url
 
         if config.dev_mode:
             return DjangoViteAssetLoader._generate_script_tag(
@@ -169,7 +168,9 @@ def generate_vite_asset(
         # Add the script by itself
         tags.append(
             DjangoViteAssetLoader._generate_script_tag(
-                urljoin(static_url, manifest[path].file),
+                DjangoViteAssetLoader._generate_production_server_url(
+                    manifest[path].file, config.static_url_prefix
+                ),
                 attrs=scripts_attrs,
             )
         )
@@ -183,7 +184,7 @@ def generate_vite_asset(
         }
 
         for dep in manifest.imports:
-            dep_manifest_entry = self._manifest[dep]
+            dep_manifest_entry = manifest[dep]
             dep_file = dep_manifest_entry["file"]
             url = DjangoViteAssetLoader._generate_production_server_url(
                 dep_file, config.static_url_prefix
@@ -261,10 +262,11 @@ def preload_vite_asset(
         tags.extend(self._preload_css_files_of_asset(path, config_key))
 
         # Preload imports
-        for dependency_path in manifest_entry.imports:
-            dependency_file = manifest[dependency_path].file
+        for dep in manifest_entry.imports:
+            dep_manifest_entry = manifest[dep]
+            dep_file = dep_manifest_entry.file
             url = DjangoViteAssetLoader._generate_production_server_url(
-                dependency_file, config.static_url_prefix
+                dep_file, config.static_url_prefix
             )
             tags.append(
                 DjangoViteAssetLoader._generate_preload_tag(

From 94a6fcb9b6f8868665d71ca5e11db15c893550e6 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Sat, 14 Oct 2023 18:16:36 -0700
Subject: [PATCH 21/29] create patch_settings fixture

---
 tests/conftest.py                             |  38 +++++
 tests/data/staticfiles/manifest.json          |   3 +
 tests/settings.py                             |  11 +-
 tests/test_templatetags.py                    | 150 ------------------
 tests/tests/templatetags/test_vite_asset.py   |  27 ++++
 .../tests/templatetags/test_vite_asset_url.py |   0
 .../templatetags/test_vite_hmr_client.py      |  70 ++++++++
 .../templatetags/test_vite_legacy_asset.py    |   0
 .../test_vite_legacy_polyfills.py             |   0
 .../templatetags/test_vite_react_refresh.py   |  50 ++++++
 tests/{ => tests}/test_asset_loader.py        |   0
 tox.ini                                       |  21 +++
 12 files changed, 216 insertions(+), 154 deletions(-)
 create mode 100644 tests/conftest.py
 create mode 100644 tests/data/staticfiles/manifest.json
 delete mode 100644 tests/test_templatetags.py
 create mode 100644 tests/tests/templatetags/test_vite_asset.py
 create mode 100644 tests/tests/templatetags/test_vite_asset_url.py
 create mode 100644 tests/tests/templatetags/test_vite_hmr_client.py
 create mode 100644 tests/tests/templatetags/test_vite_legacy_asset.py
 create mode 100644 tests/tests/templatetags/test_vite_legacy_polyfills.py
 create mode 100644 tests/tests/templatetags/test_vite_react_refresh.py
 rename tests/{ => tests}/test_asset_loader.py (100%)

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..469c143
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,38 @@
+from typing import Dict, Any
+import pytest
+from importlib import reload
+from django_vite.templatetags import django_vite
+
+
+@pytest.fixture()
+def patch_settings(settings):
+    """
+    1. Patch new settings into django.conf.settings.
+    2. Reload django_vite module so that variables on the module level that use settings
+    get recalculated.
+    3. Restore the original settings once the test is over.
+
+    TODO: refactor django_vite so that we don't set variables on the module level using
+    settings.
+    """
+    __PYTEST_EMPTY__ = "__PYTEST_EMPTY__"
+    original_settings_cache = {}
+
+    def _patch_settings(new_settings: Dict[str, Any]):
+        for key, value in new_settings.items():
+            original_settings_cache[key] = getattr(settings, key, __PYTEST_EMPTY__)
+            setattr(settings, key, value)
+
+        reload(django_vite)
+        django_vite.DjangoViteAssetLoader.instance()
+
+    yield _patch_settings
+
+    for key, value in original_settings_cache.items():
+        if value == __PYTEST_EMPTY__:
+            delattr(settings, key)
+        else:
+            setattr(settings, key, value)
+
+    reload(django_vite)
+    django_vite.DjangoViteAssetLoader.instance()
diff --git a/tests/data/staticfiles/manifest.json b/tests/data/staticfiles/manifest.json
new file mode 100644
index 0000000..2a40f18
--- /dev/null
+++ b/tests/data/staticfiles/manifest.json
@@ -0,0 +1,3 @@
+{
+    "placeholder": "placeholder"
+}
diff --git a/tests/settings.py b/tests/settings.py
index 189a195..0801bd6 100644
--- a/tests/settings.py
+++ b/tests/settings.py
@@ -1,13 +1,11 @@
 import os
+from pathlib import Path
 
-BASE_DIR = os.path.dirname(__file__)
+BASE_DIR = Path(__file__).resolve().parent
 
 STATIC_URL = "static"
-DJANGO_VITE_DEV_MODE = True
-DJANGO_VITE_ASSETS_PATH = "/"
 USE_TZ = True
 
-
 INSTALLED_APPS = [
     "django_vite",
 ]
@@ -26,3 +24,8 @@
         },
     },
 ]
+
+# django-vite defaults
+DJANGO_VITE_DEV_MODE = True
+DJANGO_VITE_ASSETS_PATH = "/"
+STATIC_ROOT = BASE_DIR / "data" / "staticfiles"
diff --git a/tests/test_templatetags.py b/tests/test_templatetags.py
deleted file mode 100644
index 79bb5c3..0000000
--- a/tests/test_templatetags.py
+++ /dev/null
@@ -1,150 +0,0 @@
-import pytest
-from bs4 import BeautifulSoup
-from django.template import Context, Template, TemplateSyntaxError
-
-
-@pytest.fixture()
-def override_setting(monkeypatch):
-    def _override_setting(setting, value):
-        monkeypatch.setattr(
-            f"django_vite.templatetags.django_vite.{setting}",
-            value,
-        )
-
-    return _override_setting
-
-
-def test_vite_hmr_client_returns_script_tag():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag["src"] == "http://localhost:3000/static/@vite/client"
-    assert script_tag["type"] == "module"
-
-
-def test_vite_hmr_client_kwargs():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client blocking="render" %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag.has_attr("blocking")
-    assert script_tag["blocking"] == "render"
-
-
-def test_vite_hmr_client_returns_nothing_with_dev_mode_off(settings, monkeypatch):
-    settings.DJANGO_VITE_DEV_MODE = False
-    monkeypatch.setattr(
-        "django_vite.templatetags.django_vite.DJANGO_VITE_DEV_MODE",
-        settings.DJANGO_VITE_DEV_MODE,
-    )
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    assert str(soup).strip() == ""
-
-
-def test_vite_hmr_client_uses_correct_settings(override_setting):
-    override_setting("DJANGO_VITE_DEV_SERVER_PROTOCOL", "https")
-    override_setting("DJANGO_VITE_DEV_SERVER_HOST", "127.0.0.2")
-    override_setting("DJANGO_VITE_DEV_SERVER_PORT", "5174")
-    override_setting("DJANGO_VITE_STATIC_URL", "static/")
-    override_setting("DJANGO_VITE_WS_CLIENT_URL", "foo/bar")
-
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_hmr_client %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag["src"] == "https://127.0.0.2:5174/static/foo/bar"
-
-
-def test_vite_asset_returns_script_tags():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_asset "src/entry.tsx" %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.find("script")
-    assert script_tag["src"] == "http://localhost:3000/static/src/entry.tsx"
-    assert script_tag["type"] == "module"
-
-
-def test_vite_asset_raises_without_path():
-    with pytest.raises(TemplateSyntaxError):
-        Template(
-            """
-            {% load django_vite %}
-            {% vite_asset %}
-        """
-        )
-
-
-def test_vite_react_refresh_happy_flow():
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_react_refresh %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.script
-    assert not script_tag.has_attr("src")
-    assert script_tag.has_attr("type")
-    assert script_tag["type"] == "module"
-    assert "__vite_plugin_react_preamble_installed__" in script_tag.text
-    assert "http://localhost:3000/static/@react-refresh" in script_tag.text
-
-
-def test_vite_react_refresh_returns_nothing_with_dev_mode_off(settings, monkeypatch):
-    settings.DJANGO_VITE_DEV_MODE = False
-    monkeypatch.setattr(
-        "django_vite.templatetags.django_vite.DJANGO_VITE_DEV_MODE",
-        settings.DJANGO_VITE_DEV_MODE,
-    )
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_react_refresh %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    assert str(soup).strip() == ""
-
-
-def test_vite_react_refresh_url_setting(override_setting):
-    override_setting("DJANGO_VITE_REACT_REFRESH_URL", "foobar")
-    template = Template(
-        """
-        {% load django_vite %}
-        {% vite_react_refresh %}
-    """
-    )
-    html = template.render(Context({}))
-    soup = BeautifulSoup(html, "html.parser")
-    script_tag = soup.script
-    assert "http://localhost:3000/static/foobar" in script_tag.text
diff --git a/tests/tests/templatetags/test_vite_asset.py b/tests/tests/templatetags/test_vite_asset.py
new file mode 100644
index 0000000..c09f5d2
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_asset.py
@@ -0,0 +1,27 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template, TemplateSyntaxError
+
+
+def test_vite_asset_returns_script_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.tsx" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.tsx"
+    assert script_tag["type"] == "module"
+
+
+def test_vite_asset_raises_without_path():
+    with pytest.raises(TemplateSyntaxError):
+        Template(
+            """
+            {% load django_vite %}
+            {% vite_asset %}
+        """
+        )
diff --git a/tests/tests/templatetags/test_vite_asset_url.py b/tests/tests/templatetags/test_vite_asset_url.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/tests/templatetags/test_vite_hmr_client.py b/tests/tests/templatetags/test_vite_hmr_client.py
new file mode 100644
index 0000000..5f35364
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_hmr_client.py
@@ -0,0 +1,70 @@
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+
+
+def test_vite_hmr_client_returns_script_tag():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/@vite/client"
+    assert script_tag["type"] == "module"
+
+
+def test_vite_hmr_client_kwargs():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client blocking="render" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag.has_attr("blocking")
+    assert script_tag["blocking"] == "render"
+
+
+def test_vite_hmr_client_returns_nothing_with_dev_mode_off(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+def test_vite_hmr_client_uses_correct_settings(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_SERVER_PROTOCOL": "https",
+            "DJANGO_VITE_DEV_SERVER_HOST": "127.0.0.2",
+            "DJANGO_VITE_DEV_SERVER_PORT": "5174",
+            "DJANGO_VITE_STATIC_URL": "static/",
+            "DJANGO_VITE_WS_CLIENT_URL": "foo/bar",
+        }
+    )
+
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_hmr_client %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "https://127.0.0.2:5174/static/foo/bar"
diff --git a/tests/tests/templatetags/test_vite_legacy_asset.py b/tests/tests/templatetags/test_vite_legacy_asset.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/tests/templatetags/test_vite_legacy_polyfills.py b/tests/tests/templatetags/test_vite_legacy_polyfills.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/tests/templatetags/test_vite_react_refresh.py b/tests/tests/templatetags/test_vite_react_refresh.py
new file mode 100644
index 0000000..4787112
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_react_refresh.py
@@ -0,0 +1,50 @@
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+
+
+def test_vite_react_refresh_happy_flow():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert not script_tag.has_attr("src")
+    assert script_tag.has_attr("type")
+    assert script_tag["type"] == "module"
+    assert "__vite_plugin_react_preamble_installed__" in script_tag.text
+    assert "http://localhost:3000/static/@react-refresh" in script_tag.text
+
+
+def test_vite_react_refresh_returns_nothing_with_dev_mode_off(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+def test_vite_react_refresh_url_setting(patch_settings):
+    patch_settings({"DJANGO_VITE_REACT_REFRESH_URL": "foobar"})
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.script
+    assert "http://localhost:3000/static/foobar" in script_tag.text
diff --git a/tests/test_asset_loader.py b/tests/tests/test_asset_loader.py
similarity index 100%
rename from tests/test_asset_loader.py
rename to tests/tests/test_asset_loader.py
diff --git a/tox.ini b/tox.ini
index 193fa9c..fb651ac 100644
--- a/tox.ini
+++ b/tox.ini
@@ -32,6 +32,27 @@ commands =
 ignore_outcome =
     django-latest: True
 
+[testenv:debugpy]
+description = run unit tests with debugpy
+basepython = python3.8
+deps =
+    beautifulsoup4
+    pytest>=7
+    debugpy
+    pytest-django
+    pytest-sugar
+    Django>=3.2,<4.0
+setenv =
+    DEBUGPY_PORT_PYTEST = 56789
+commands =
+    python \
+    -m debugpy --listen localhost:{env:DEBUGPY_PORT_PYTEST} --wait-for-client \
+    -m pytest \
+    -o addopts="" \
+    {posargs:tests}
+ignore_outcome =
+    django-latest: True
+
 [testenv:codestyle]
 basepython = python3
 commands =

From da1dae6ef3f57eb750a5ab41e2ab8563b4967310 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Sun, 15 Oct 2023 09:54:05 -0700
Subject: [PATCH 22/29] 100% test coverage

---
 django_vite/apps.py                           |   5 +-
 django_vite/exceptions.py                     |  10 ++
 django_vite/templatetags/django_vite.py       |  29 ++---
 pyproject.toml                                |   4 +
 tests/conftest.py                             |  18 ++-
 .../custom-motif-polyfills-manifest.json      |  24 ++++
 .../staticfiles/custom/prefix/manifest.json   |  72 ++++++++++++
 tests/data/staticfiles/manifest.json          |  71 +++++++++++-
 .../data/staticfiles/polyfills-manifest.json  |  24 ++++
 tests/settings.py                             |   6 +-
 tests/tests/templatetags/test_vite_asset.py   | 103 +++++++++++++++++-
 .../tests/templatetags/test_vite_asset_url.py |  43 ++++++++
 .../templatetags/test_vite_hmr_client.py      |   9 +-
 .../templatetags/test_vite_legacy_asset.py    |  49 +++++++++
 .../test_vite_legacy_polyfills.py             |  78 +++++++++++++
 .../templatetags/test_vite_preload_asset.py   |  45 ++++++++
 .../templatetags/test_vite_react_refresh.py   |  30 ++++-
 tests/tests/test_asset_loader.py              |  25 +++++
 18 files changed, 608 insertions(+), 37 deletions(-)
 create mode 100644 django_vite/exceptions.py
 create mode 100644 tests/data/staticfiles/custom-motif-polyfills-manifest.json
 create mode 100644 tests/data/staticfiles/custom/prefix/manifest.json
 create mode 100644 tests/data/staticfiles/polyfills-manifest.json
 create mode 100644 tests/tests/templatetags/test_vite_preload_asset.py

diff --git a/django_vite/apps.py b/django_vite/apps.py
index 98dae96..29280ad 100644
--- a/django_vite/apps.py
+++ b/django_vite/apps.py
@@ -4,6 +4,7 @@
 from django.core.checks import Warning, register
 
 from .templatetags.django_vite import DjangoViteAssetLoader
+from .exceptions import DjangoViteManifestError
 
 
 class DjangoViteAppConfig(AppConfig):
@@ -11,7 +12,7 @@ class DjangoViteAppConfig(AppConfig):
     verbose_name = "Django Vite"
 
     def ready(self) -> None:
-        with suppress(RuntimeError):
+        with suppress(DjangoViteManifestError):
             # Create Loader instance at startup to prevent threading problems,
             # but do not crash while doing so.
             DjangoViteAssetLoader.instance()
@@ -25,7 +26,7 @@ def check_loader_instance(**kwargs):
         # Make Loader instance at startup to prevent threading problems
         DjangoViteAssetLoader.instance()
         return []
-    except RuntimeError as exception:
+    except DjangoViteManifestError as exception:
         return [
             Warning(
                 exception,
diff --git a/django_vite/exceptions.py b/django_vite/exceptions.py
new file mode 100644
index 0000000..1030ca8
--- /dev/null
+++ b/django_vite/exceptions.py
@@ -0,0 +1,10 @@
+class DjangoViteManifestError(RuntimeError):
+    """Manifest parsing failed."""
+
+    pass
+
+
+class DjangoViteAssetNotFoundError(RuntimeError):
+    """Vite Asset could not be found."""
+
+    pass
diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 56601eb..52deacf 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -8,8 +8,9 @@
 from django.conf import settings
 from django.utils.safestring import mark_safe
 
-register = template.Library()
+from django_vite.exceptions import DjangoViteManifestError, DjangoViteAssetNotFoundError
 
+register = template.Library()
 
 # If using in development or production mode.
 DJANGO_VITE_DEV_MODE = getattr(settings, "DJANGO_VITE_DEV_MODE", False)
@@ -104,7 +105,7 @@ def generate_vite_asset(
                 script tags.
 
         Raises:
-            RuntimeError: If cannot find the file path in the
+            DjangoViteAssetNotFoundError: If cannot find the file path in the
                 manifest (only in production).
 
         Returns:
@@ -119,7 +120,7 @@ def generate_vite_asset(
             )
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -180,7 +181,7 @@ def preload_vite_asset(
             str -- All tags to preload this file in your HTML page.
 
         Raises:
-            RuntimeError: If cannot find the file path in the
+            DjangoViteAssetNotFoundError: if cannot find the file path in the
                 manifest.
 
         Returns:
@@ -191,7 +192,7 @@ def preload_vite_asset(
             return ""
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -297,7 +298,7 @@ def generate_vite_asset_url(self, path: str) -> str:
             path {str} -- Path to a Vite asset.
 
         Raises:
-            RuntimeError: If cannot find the asset path in the
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
                 manifest (only in production).
 
         Returns:
@@ -308,7 +309,7 @@ def generate_vite_asset_url(self, path: str) -> str:
             return DjangoViteAssetLoader._generate_vite_server_url(path)
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -332,7 +333,7 @@ def generate_vite_legacy_polyfills(
                 script tags.
 
         Raises:
-            RuntimeError: If polyfills path not found inside
+            DjangoViteAssetNotFoundError: If polyfills path not found inside
                 the 'manifest.json' (only in production).
 
         Returns:
@@ -353,7 +354,7 @@ def generate_vite_legacy_polyfills(
                     attrs=scripts_attrs,
                 )
 
-        raise RuntimeError(
+        raise DjangoViteAssetNotFoundError(
             f"Vite legacy polyfills not found in manifest "
             f"at {DJANGO_VITE_MANIFEST_PATH}"
         )
@@ -377,7 +378,7 @@ def generate_vite_legacy_asset(
                 script tags.
 
         Raises:
-            RuntimeError: If cannot find the asset path in the
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
                 manifest (only in production).
 
         Returns:
@@ -388,7 +389,7 @@ def generate_vite_legacy_asset(
             return ""
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -408,7 +409,8 @@ def _parse_manifest(self) -> None:
         Read and parse the Vite manifest file.
 
         Raises:
-            RuntimeError: if cannot load the file or JSON in file is malformed.
+            DjangoViteManifestError: if cannot load the file or JSON in file is
+            malformed.
         """
 
         try:
@@ -416,7 +418,7 @@ def _parse_manifest(self) -> None:
                 manifest_content = manifest_file.read()
             self._manifest = json.loads(manifest_content)
         except Exception as error:
-            raise RuntimeError(
+            raise DjangoViteManifestError(
                 f"Cannot read Vite manifest file at "
                 f"{DJANGO_VITE_MANIFEST_PATH} : {str(error)}"
             ) from error
@@ -663,7 +665,6 @@ def vite_preload_asset(
             manifest (only in production).
 
     """
-
     assert path is not None
 
     return DjangoViteAssetLoader.instance().preload_vite_asset(path)
diff --git a/pyproject.toml b/pyproject.toml
index 0614b47..c332ae0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,8 +7,12 @@ addopts = '''
   --cov-report html
   --cov-report term-missing
   --cov-branch
+  --cov-fail-under=100
 '''
 
+[tool.black]
+line-length = 88
+
 [tool.ruff]
 select = [
   "E",   # pycodestyle
diff --git a/tests/conftest.py b/tests/conftest.py
index 469c143..2eedec1 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,9 +1,16 @@
 from typing import Dict, Any
 import pytest
 from importlib import reload
+from django.apps import apps
 from django_vite.templatetags import django_vite
 
 
+def reload_django_vite():
+    reload(django_vite)
+    django_vite_app_config = apps.get_app_config("django_vite")
+    django_vite_app_config.ready()
+
+
 @pytest.fixture()
 def patch_settings(settings):
     """
@@ -22,9 +29,7 @@ def _patch_settings(new_settings: Dict[str, Any]):
         for key, value in new_settings.items():
             original_settings_cache[key] = getattr(settings, key, __PYTEST_EMPTY__)
             setattr(settings, key, value)
-
-        reload(django_vite)
-        django_vite.DjangoViteAssetLoader.instance()
+        reload_django_vite()
 
     yield _patch_settings
 
@@ -33,6 +38,9 @@ def _patch_settings(new_settings: Dict[str, Any]):
             delattr(settings, key)
         else:
             setattr(settings, key, value)
+    reload_django_vite()
 
-    reload(django_vite)
-    django_vite.DjangoViteAssetLoader.instance()
+
+@pytest.fixture()
+def dev_mode_off(patch_settings):
+    patch_settings({"DJANGO_VITE_DEV_MODE": False})
diff --git a/tests/data/staticfiles/custom-motif-polyfills-manifest.json b/tests/data/staticfiles/custom-motif-polyfills-manifest.json
new file mode 100644
index 0000000..c954ade
--- /dev/null
+++ b/tests/data/staticfiles/custom-motif-polyfills-manifest.json
@@ -0,0 +1,24 @@
+{
+    "../../vite/custom-motif-legacy": {
+      "file": "assets/polyfills-legacy-6e7a4b9c.js",
+      "isEntry": true,
+      "src": "../../vite/custom-motif-legacy"
+    },
+    "src/entry-legacy.ts": {
+      "file": "assets/entry-legacy-4c50596f.js",
+      "isEntry": true,
+      "src": "src/entry-legacy.ts"
+    },
+    "src/entry.css": {
+      "file": "assets/entry-5e7d9c21.css",
+      "src": "src/entry.css"
+    },
+    "src/entry.ts": {
+      "css": [
+        "assets/entry-5e7d9c21.css"
+      ],
+      "file": "assets/entry-8a2f6b3d.js",
+      "isEntry": true,
+      "src": "src/entry.ts"
+    }
+}
diff --git a/tests/data/staticfiles/custom/prefix/manifest.json b/tests/data/staticfiles/custom/prefix/manifest.json
new file mode 100644
index 0000000..fbe4a74
--- /dev/null
+++ b/tests/data/staticfiles/custom/prefix/manifest.json
@@ -0,0 +1,72 @@
+{
+  "src/entry.ts": {
+    "css": ["assets/entry-74d0d5dd.css"],
+    "file": "assets/entry-29e38a60.js",
+    "imports": [
+      "_vue.esm-bundler-96356fb1.js",
+      "_index-62f37ad0.js",
+      "_vue.esm-bundler-49e6b475.js",
+      "__plugin-vue_export-helper-c27b6911.js",
+      "_messages-a0a9e13b.js",
+      "_use-outside-click-224980bf.js",
+      "_use-resolve-button-type-c5656cba.js",
+      "_use-event-listener-153dc639.js",
+      "_hidden-84ddb9a5.js",
+      "_apiClient-01d20438.js",
+      "_pinia-5d7892fd.js"
+    ],
+    "isEntry": true,
+    "src": "entry.ts"
+  },
+  "src/entry.css": {
+    "file": "assets/entry-74d0d5dd.css",
+    "src": "entry.css"
+  },
+  "src/extra.css": {
+    "file": "assets/extra-a9f3b2c1.css",
+    "src": "extra.css"
+  },
+  "_vue.esm-bundler-96356fb1.js": {
+    "file": "vue.esm-bundler-96356fb1.js"
+  },
+  "_index-62f37ad0.js": {
+    "file": "index-62f37ad0.js",
+    "imports": ["_vue.esm-bundler-49e6b475.js"]
+  },
+  "_vue.esm-bundler-49e6b475.js": {
+    "file": "vue.esm-bundler-49e6b475.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "__plugin-vue_export-helper-c27b6911.js": {
+    "file": "_plugin-vue_export-helper-c27b6911.js"
+  },
+  "_messages-a0a9e13b.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "messages-a0a9e13b.js",
+    "imports": ["_pinia-5d7892fd.js", "_vue.esm-bundler-96356fb1.js"]
+  },
+  "_pinia-5d7892fd.js": {
+    "file": "pinia-5d7892fd.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-outside-click-224980bf.js": {
+    "file": "use-outside-click-224980bf.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-resolve-button-type-c5656cba.js": {
+    "file": "use-resolve-button-type-c5656cba.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_use-event-listener-153dc639.js": {
+    "file": "use-event-listener-153dc639.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_hidden-84ddb9a5.js": {
+    "file": "hidden-84ddb9a5.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_apiClient-01d20438.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "apiClient-01d20438.js"
+  }
+}
diff --git a/tests/data/staticfiles/manifest.json b/tests/data/staticfiles/manifest.json
index 2a40f18..fbe4a74 100644
--- a/tests/data/staticfiles/manifest.json
+++ b/tests/data/staticfiles/manifest.json
@@ -1,3 +1,72 @@
 {
-    "placeholder": "placeholder"
+  "src/entry.ts": {
+    "css": ["assets/entry-74d0d5dd.css"],
+    "file": "assets/entry-29e38a60.js",
+    "imports": [
+      "_vue.esm-bundler-96356fb1.js",
+      "_index-62f37ad0.js",
+      "_vue.esm-bundler-49e6b475.js",
+      "__plugin-vue_export-helper-c27b6911.js",
+      "_messages-a0a9e13b.js",
+      "_use-outside-click-224980bf.js",
+      "_use-resolve-button-type-c5656cba.js",
+      "_use-event-listener-153dc639.js",
+      "_hidden-84ddb9a5.js",
+      "_apiClient-01d20438.js",
+      "_pinia-5d7892fd.js"
+    ],
+    "isEntry": true,
+    "src": "entry.ts"
+  },
+  "src/entry.css": {
+    "file": "assets/entry-74d0d5dd.css",
+    "src": "entry.css"
+  },
+  "src/extra.css": {
+    "file": "assets/extra-a9f3b2c1.css",
+    "src": "extra.css"
+  },
+  "_vue.esm-bundler-96356fb1.js": {
+    "file": "vue.esm-bundler-96356fb1.js"
+  },
+  "_index-62f37ad0.js": {
+    "file": "index-62f37ad0.js",
+    "imports": ["_vue.esm-bundler-49e6b475.js"]
+  },
+  "_vue.esm-bundler-49e6b475.js": {
+    "file": "vue.esm-bundler-49e6b475.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "__plugin-vue_export-helper-c27b6911.js": {
+    "file": "_plugin-vue_export-helper-c27b6911.js"
+  },
+  "_messages-a0a9e13b.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "messages-a0a9e13b.js",
+    "imports": ["_pinia-5d7892fd.js", "_vue.esm-bundler-96356fb1.js"]
+  },
+  "_pinia-5d7892fd.js": {
+    "file": "pinia-5d7892fd.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-outside-click-224980bf.js": {
+    "file": "use-outside-click-224980bf.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-resolve-button-type-c5656cba.js": {
+    "file": "use-resolve-button-type-c5656cba.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_use-event-listener-153dc639.js": {
+    "file": "use-event-listener-153dc639.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_hidden-84ddb9a5.js": {
+    "file": "hidden-84ddb9a5.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_apiClient-01d20438.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "apiClient-01d20438.js"
+  }
 }
diff --git a/tests/data/staticfiles/polyfills-manifest.json b/tests/data/staticfiles/polyfills-manifest.json
new file mode 100644
index 0000000..ddebd6a
--- /dev/null
+++ b/tests/data/staticfiles/polyfills-manifest.json
@@ -0,0 +1,24 @@
+{
+    "../../vite/legacy-polyfills-legacy": {
+      "file": "assets/polyfills-legacy-f4c2b91e.js",
+      "isEntry": true,
+      "src": "../../vite/legacy-polyfills-legacy"
+    },
+    "src/entry-legacy.ts": {
+      "file": "assets/entry-legacy-32083566.js",
+      "isEntry": true,
+      "src": "src/entry-legacy.ts"
+    },
+    "src/entry.css": {
+      "file": "assets/entry-74d0d5dd.css",
+      "src": "src/entry.css"
+    },
+    "src/entry.ts": {
+      "css": [
+        "assets/entry-74d0d5dd.css"
+      ],
+      "file": "assets/entry-2e8a3a7a.js",
+      "isEntry": true,
+      "src": "src/entry.ts"
+    }
+}
diff --git a/tests/settings.py b/tests/settings.py
index 0801bd6..541df85 100644
--- a/tests/settings.py
+++ b/tests/settings.py
@@ -3,7 +3,7 @@
 
 BASE_DIR = Path(__file__).resolve().parent
 
-STATIC_URL = "static"
+STATIC_URL = "/static/"
 USE_TZ = True
 
 INSTALLED_APPS = [
@@ -25,7 +25,7 @@
     },
 ]
 
-# django-vite defaults
+STATIC_ROOT = BASE_DIR / "data" / "staticfiles"
+
 DJANGO_VITE_DEV_MODE = True
 DJANGO_VITE_ASSETS_PATH = "/"
-STATIC_ROOT = BASE_DIR / "data" / "staticfiles"
diff --git a/tests/tests/templatetags/test_vite_asset.py b/tests/tests/templatetags/test_vite_asset.py
index c09f5d2..7923954 100644
--- a/tests/tests/templatetags/test_vite_asset.py
+++ b/tests/tests/templatetags/test_vite_asset.py
@@ -1,22 +1,40 @@
 import pytest
 from bs4 import BeautifulSoup
 from django.template import Context, Template, TemplateSyntaxError
+from django_vite.exceptions import DjangoViteAssetNotFoundError
 
 
-def test_vite_asset_returns_script_tags():
+def test_vite_asset_returns_dev_tags():
     template = Template(
         """
         {% load django_vite %}
-        {% vite_asset "src/entry.tsx" %}
+        {% vite_asset "src/entry.ts" %}
     """
     )
     html = template.render(Context({}))
     soup = BeautifulSoup(html, "html.parser")
     script_tag = soup.find("script")
-    assert script_tag["src"] == "http://localhost:3000/static/src/entry.tsx"
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.ts"
     assert script_tag["type"] == "module"
 
 
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_returns_production_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
+
+
 def test_vite_asset_raises_without_path():
     with pytest.raises(TemplateSyntaxError):
         Template(
@@ -25,3 +43,82 @@ def test_vite_asset_raises_without_path():
             {% vite_asset %}
         """
         )
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
+
+
+@pytest.mark.parametrize("prefix", ["custom/prefix", "custom/prefix/"])
+def test_vite_asset_dev_prefix(prefix, patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_STATIC_URL_PREFIX": prefix,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert (
+        script_tag["src"] == "http://localhost:3000/static/custom/prefix/src/entry.ts"
+    )
+    assert script_tag["type"] == "module"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+@pytest.mark.parametrize("prefix", ["custom/prefix", "custom/prefix/"])
+def test_vite_asset_production_prefix(prefix, patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_STATIC_URL_PREFIX": prefix,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "custom/prefix/assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_production_staticfiles_storage(patch_settings):
+    patch_settings(
+        {
+            "INSTALLED_APPS": ["django_vite", "django.contrib.staticfiles"],
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "/static/assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
diff --git a/tests/tests/templatetags/test_vite_asset_url.py b/tests/tests/templatetags/test_vite_asset_url.py
index e69de29..e0eef07 100644
--- a/tests/tests/templatetags/test_vite_asset_url.py
+++ b/tests/tests/templatetags/test_vite_asset_url.py
@@ -0,0 +1,43 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_asset_url_returns_dev_url():
+    template = Template(
+        """
+        {% load django_vite %}
+        <script src="{% vite_asset_url "src/entry.ts" %}">
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.ts"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_url_returns_production_url():
+    template = Template(
+        """
+        {% load django_vite %}
+        <script src="{% vite_asset_url "src/entry.ts" %}">
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-29e38a60.js"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_url_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            <script src="{% vite_asset_url "src/fake.ts" %}">
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_hmr_client.py b/tests/tests/templatetags/test_vite_hmr_client.py
index 5f35364..2fee91b 100644
--- a/tests/tests/templatetags/test_vite_hmr_client.py
+++ b/tests/tests/templatetags/test_vite_hmr_client.py
@@ -1,3 +1,4 @@
+import pytest
 from bs4 import BeautifulSoup
 from django.template import Context, Template
 
@@ -30,12 +31,8 @@ def test_vite_hmr_client_kwargs():
     assert script_tag["blocking"] == "render"
 
 
-def test_vite_hmr_client_returns_nothing_with_dev_mode_off(patch_settings):
-    patch_settings(
-        {
-            "DJANGO_VITE_DEV_MODE": False,
-        }
-    )
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_hmr_client_returns_nothing_with_dev_mode_off():
     template = Template(
         """
         {% load django_vite %}
diff --git a/tests/tests/templatetags/test_vite_legacy_asset.py b/tests/tests/templatetags/test_vite_legacy_asset.py
index e69de29..86c99ff 100644
--- a/tests/tests/templatetags/test_vite_legacy_asset.py
+++ b/tests/tests/templatetags/test_vite_legacy_asset.py
@@ -0,0 +1,49 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_legacy_asset_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_legacy_returns_production_tags(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-2e8a3a7a.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_legacy_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_legacy_polyfills.py b/tests/tests/templatetags/test_vite_legacy_polyfills.py
index e69de29..409c032 100644
--- a/tests/tests/templatetags/test_vite_legacy_polyfills.py
+++ b/tests/tests/templatetags/test_vite_legacy_polyfills.py
@@ -0,0 +1,78 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_legacy_polyfills_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_production(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-f4c2b91e.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_custom_motif(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "custom-motif",
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "custom-motif-polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-6e7a4b9c.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_nonexistent_motif(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+            "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "fake-legacy-polyfills",
+        }
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_legacy_polyfills %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_preload_asset.py b/tests/tests/templatetags/test_vite_preload_asset.py
new file mode 100644
index 0000000..bf14523
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_preload_asset.py
@@ -0,0 +1,45 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_hmr_client_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_preload_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_preload_vite_asset_returns_production_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_preload_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    links = soup.find_all("link")
+    assert len(links) == 14
+    first_link = links[0]
+    assert first_link["href"] == "assets/entry-29e38a60.js"
+    assert first_link["rel"] == ["modulepreload"]
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_preload_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_react_refresh.py b/tests/tests/templatetags/test_vite_react_refresh.py
index 4787112..e2c31e7 100644
--- a/tests/tests/templatetags/test_vite_react_refresh.py
+++ b/tests/tests/templatetags/test_vite_react_refresh.py
@@ -1,8 +1,9 @@
+import pytest
 from bs4 import BeautifulSoup
 from django.template import Context, Template
 
 
-def test_vite_react_refresh_happy_flow():
+def test_vite_react_refresh_returns_script_tag():
     template = Template(
         """
         {% load django_vite %}
@@ -19,12 +20,35 @@ def test_vite_react_refresh_happy_flow():
     assert "http://localhost:3000/static/@react-refresh" in script_tag.text
 
 
-def test_vite_react_refresh_returns_nothing_with_dev_mode_off(patch_settings):
+def test_vite_react_refresh_uses_correct_settings(patch_settings):
     patch_settings(
         {
-            "DJANGO_VITE_DEV_MODE": False,
+            "DJANGO_VITE_DEV_SERVER_PROTOCOL": "https",
+            "DJANGO_VITE_DEV_SERVER_HOST": "127.0.0.2",
+            "DJANGO_VITE_DEV_SERVER_PORT": "5174",
+            "DJANGO_VITE_STATIC_URL": "static/",
+            "DJANGO_VITE_REACT_REFRESH_URL": "foo/bar",
         }
     )
+
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert not script_tag.has_attr("src")
+    assert script_tag.has_attr("type")
+    assert script_tag["type"] == "module"
+    assert "__vite_plugin_react_preamble_installed__" in script_tag.text
+    assert "https://127.0.0.2:5174/static/foo/bar" in script_tag.text
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_react_refresh_returns_nothing_with_dev_mode_off():
     template = Template(
         """
         {% load django_vite %}
diff --git a/tests/tests/test_asset_loader.py b/tests/tests/test_asset_loader.py
index 7ee8935..470867c 100644
--- a/tests/tests/test_asset_loader.py
+++ b/tests/tests/test_asset_loader.py
@@ -1,8 +1,33 @@
 import pytest
 
 from django_vite.templatetags.django_vite import DjangoViteAssetLoader
+from django_vite.apps import check_loader_instance
 
 
 def test_django_vite_asset_loader_cannot_be_instantiated():
     with pytest.raises(RuntimeError):
         DjangoViteAssetLoader()
+
+
+def test_check_loader_instance_happy(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+        }
+    )
+    DjangoViteAssetLoader._instance = None
+    warnings = check_loader_instance()
+    assert len(warnings) == 0
+
+
+def test_check_loader_instance_warnings(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+            "DJANGO_VITE_MANIFEST_PATH": "fake.json",
+        }
+    )
+    DjangoViteAssetLoader._instance = None
+    warnings = check_loader_instance()
+    assert len(warnings) == 1
+    assert "Make sure you have generated a manifest file" in warnings[0].hint

From 6aa177ce80290f2cfc1a50d17e12d13a7362dafd Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Sun, 15 Oct 2023 09:54:05 -0700
Subject: [PATCH 23/29] 100% test coverage

---
 django_vite/apps.py                           |   5 +-
 django_vite/exceptions.py                     |  10 ++
 django_vite/templatetags/django_vite.py       |  29 ++---
 pyproject.toml                                |   4 +
 tests/conftest.py                             |  18 ++-
 .../custom-motif-polyfills-manifest.json      |  24 ++++
 .../staticfiles/custom/prefix/manifest.json   |  72 ++++++++++++
 tests/data/staticfiles/manifest.json          |  71 +++++++++++-
 .../data/staticfiles/polyfills-manifest.json  |  24 ++++
 tests/settings.py                             |   6 +-
 tests/tests/templatetags/test_vite_asset.py   | 103 +++++++++++++++++-
 .../tests/templatetags/test_vite_asset_url.py |  43 ++++++++
 .../templatetags/test_vite_hmr_client.py      |   9 +-
 .../templatetags/test_vite_legacy_asset.py    |  49 +++++++++
 .../test_vite_legacy_polyfills.py             |  78 +++++++++++++
 .../templatetags/test_vite_preload_asset.py   |  45 ++++++++
 .../templatetags/test_vite_react_refresh.py   |  30 ++++-
 tests/tests/test_asset_loader.py              |  25 +++++
 18 files changed, 608 insertions(+), 37 deletions(-)
 create mode 100644 django_vite/exceptions.py
 create mode 100644 tests/data/staticfiles/custom-motif-polyfills-manifest.json
 create mode 100644 tests/data/staticfiles/custom/prefix/manifest.json
 create mode 100644 tests/data/staticfiles/polyfills-manifest.json
 create mode 100644 tests/tests/templatetags/test_vite_preload_asset.py

diff --git a/django_vite/apps.py b/django_vite/apps.py
index 98dae96..29280ad 100644
--- a/django_vite/apps.py
+++ b/django_vite/apps.py
@@ -4,6 +4,7 @@
 from django.core.checks import Warning, register
 
 from .templatetags.django_vite import DjangoViteAssetLoader
+from .exceptions import DjangoViteManifestError
 
 
 class DjangoViteAppConfig(AppConfig):
@@ -11,7 +12,7 @@ class DjangoViteAppConfig(AppConfig):
     verbose_name = "Django Vite"
 
     def ready(self) -> None:
-        with suppress(RuntimeError):
+        with suppress(DjangoViteManifestError):
             # Create Loader instance at startup to prevent threading problems,
             # but do not crash while doing so.
             DjangoViteAssetLoader.instance()
@@ -25,7 +26,7 @@ def check_loader_instance(**kwargs):
         # Make Loader instance at startup to prevent threading problems
         DjangoViteAssetLoader.instance()
         return []
-    except RuntimeError as exception:
+    except DjangoViteManifestError as exception:
         return [
             Warning(
                 exception,
diff --git a/django_vite/exceptions.py b/django_vite/exceptions.py
new file mode 100644
index 0000000..1030ca8
--- /dev/null
+++ b/django_vite/exceptions.py
@@ -0,0 +1,10 @@
+class DjangoViteManifestError(RuntimeError):
+    """Manifest parsing failed."""
+
+    pass
+
+
+class DjangoViteAssetNotFoundError(RuntimeError):
+    """Vite Asset could not be found."""
+
+    pass
diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 56601eb..52deacf 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -8,8 +8,9 @@
 from django.conf import settings
 from django.utils.safestring import mark_safe
 
-register = template.Library()
+from django_vite.exceptions import DjangoViteManifestError, DjangoViteAssetNotFoundError
 
+register = template.Library()
 
 # If using in development or production mode.
 DJANGO_VITE_DEV_MODE = getattr(settings, "DJANGO_VITE_DEV_MODE", False)
@@ -104,7 +105,7 @@ def generate_vite_asset(
                 script tags.
 
         Raises:
-            RuntimeError: If cannot find the file path in the
+            DjangoViteAssetNotFoundError: If cannot find the file path in the
                 manifest (only in production).
 
         Returns:
@@ -119,7 +120,7 @@ def generate_vite_asset(
             )
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -180,7 +181,7 @@ def preload_vite_asset(
             str -- All tags to preload this file in your HTML page.
 
         Raises:
-            RuntimeError: If cannot find the file path in the
+            DjangoViteAssetNotFoundError: if cannot find the file path in the
                 manifest.
 
         Returns:
@@ -191,7 +192,7 @@ def preload_vite_asset(
             return ""
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -297,7 +298,7 @@ def generate_vite_asset_url(self, path: str) -> str:
             path {str} -- Path to a Vite asset.
 
         Raises:
-            RuntimeError: If cannot find the asset path in the
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
                 manifest (only in production).
 
         Returns:
@@ -308,7 +309,7 @@ def generate_vite_asset_url(self, path: str) -> str:
             return DjangoViteAssetLoader._generate_vite_server_url(path)
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -332,7 +333,7 @@ def generate_vite_legacy_polyfills(
                 script tags.
 
         Raises:
-            RuntimeError: If polyfills path not found inside
+            DjangoViteAssetNotFoundError: If polyfills path not found inside
                 the 'manifest.json' (only in production).
 
         Returns:
@@ -353,7 +354,7 @@ def generate_vite_legacy_polyfills(
                     attrs=scripts_attrs,
                 )
 
-        raise RuntimeError(
+        raise DjangoViteAssetNotFoundError(
             f"Vite legacy polyfills not found in manifest "
             f"at {DJANGO_VITE_MANIFEST_PATH}"
         )
@@ -377,7 +378,7 @@ def generate_vite_legacy_asset(
                 script tags.
 
         Raises:
-            RuntimeError: If cannot find the asset path in the
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
                 manifest (only in production).
 
         Returns:
@@ -388,7 +389,7 @@ def generate_vite_legacy_asset(
             return ""
 
         if not self._manifest or path not in self._manifest:
-            raise RuntimeError(
+            raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {DJANGO_VITE_MANIFEST_PATH}"
             )
@@ -408,7 +409,8 @@ def _parse_manifest(self) -> None:
         Read and parse the Vite manifest file.
 
         Raises:
-            RuntimeError: if cannot load the file or JSON in file is malformed.
+            DjangoViteManifestError: if cannot load the file or JSON in file is
+            malformed.
         """
 
         try:
@@ -416,7 +418,7 @@ def _parse_manifest(self) -> None:
                 manifest_content = manifest_file.read()
             self._manifest = json.loads(manifest_content)
         except Exception as error:
-            raise RuntimeError(
+            raise DjangoViteManifestError(
                 f"Cannot read Vite manifest file at "
                 f"{DJANGO_VITE_MANIFEST_PATH} : {str(error)}"
             ) from error
@@ -663,7 +665,6 @@ def vite_preload_asset(
             manifest (only in production).
 
     """
-
     assert path is not None
 
     return DjangoViteAssetLoader.instance().preload_vite_asset(path)
diff --git a/pyproject.toml b/pyproject.toml
index 0614b47..c332ae0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,8 +7,12 @@ addopts = '''
   --cov-report html
   --cov-report term-missing
   --cov-branch
+  --cov-fail-under=100
 '''
 
+[tool.black]
+line-length = 88
+
 [tool.ruff]
 select = [
   "E",   # pycodestyle
diff --git a/tests/conftest.py b/tests/conftest.py
index 469c143..2eedec1 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,9 +1,16 @@
 from typing import Dict, Any
 import pytest
 from importlib import reload
+from django.apps import apps
 from django_vite.templatetags import django_vite
 
 
+def reload_django_vite():
+    reload(django_vite)
+    django_vite_app_config = apps.get_app_config("django_vite")
+    django_vite_app_config.ready()
+
+
 @pytest.fixture()
 def patch_settings(settings):
     """
@@ -22,9 +29,7 @@ def _patch_settings(new_settings: Dict[str, Any]):
         for key, value in new_settings.items():
             original_settings_cache[key] = getattr(settings, key, __PYTEST_EMPTY__)
             setattr(settings, key, value)
-
-        reload(django_vite)
-        django_vite.DjangoViteAssetLoader.instance()
+        reload_django_vite()
 
     yield _patch_settings
 
@@ -33,6 +38,9 @@ def _patch_settings(new_settings: Dict[str, Any]):
             delattr(settings, key)
         else:
             setattr(settings, key, value)
+    reload_django_vite()
 
-    reload(django_vite)
-    django_vite.DjangoViteAssetLoader.instance()
+
+@pytest.fixture()
+def dev_mode_off(patch_settings):
+    patch_settings({"DJANGO_VITE_DEV_MODE": False})
diff --git a/tests/data/staticfiles/custom-motif-polyfills-manifest.json b/tests/data/staticfiles/custom-motif-polyfills-manifest.json
new file mode 100644
index 0000000..c954ade
--- /dev/null
+++ b/tests/data/staticfiles/custom-motif-polyfills-manifest.json
@@ -0,0 +1,24 @@
+{
+    "../../vite/custom-motif-legacy": {
+      "file": "assets/polyfills-legacy-6e7a4b9c.js",
+      "isEntry": true,
+      "src": "../../vite/custom-motif-legacy"
+    },
+    "src/entry-legacy.ts": {
+      "file": "assets/entry-legacy-4c50596f.js",
+      "isEntry": true,
+      "src": "src/entry-legacy.ts"
+    },
+    "src/entry.css": {
+      "file": "assets/entry-5e7d9c21.css",
+      "src": "src/entry.css"
+    },
+    "src/entry.ts": {
+      "css": [
+        "assets/entry-5e7d9c21.css"
+      ],
+      "file": "assets/entry-8a2f6b3d.js",
+      "isEntry": true,
+      "src": "src/entry.ts"
+    }
+}
diff --git a/tests/data/staticfiles/custom/prefix/manifest.json b/tests/data/staticfiles/custom/prefix/manifest.json
new file mode 100644
index 0000000..fbe4a74
--- /dev/null
+++ b/tests/data/staticfiles/custom/prefix/manifest.json
@@ -0,0 +1,72 @@
+{
+  "src/entry.ts": {
+    "css": ["assets/entry-74d0d5dd.css"],
+    "file": "assets/entry-29e38a60.js",
+    "imports": [
+      "_vue.esm-bundler-96356fb1.js",
+      "_index-62f37ad0.js",
+      "_vue.esm-bundler-49e6b475.js",
+      "__plugin-vue_export-helper-c27b6911.js",
+      "_messages-a0a9e13b.js",
+      "_use-outside-click-224980bf.js",
+      "_use-resolve-button-type-c5656cba.js",
+      "_use-event-listener-153dc639.js",
+      "_hidden-84ddb9a5.js",
+      "_apiClient-01d20438.js",
+      "_pinia-5d7892fd.js"
+    ],
+    "isEntry": true,
+    "src": "entry.ts"
+  },
+  "src/entry.css": {
+    "file": "assets/entry-74d0d5dd.css",
+    "src": "entry.css"
+  },
+  "src/extra.css": {
+    "file": "assets/extra-a9f3b2c1.css",
+    "src": "extra.css"
+  },
+  "_vue.esm-bundler-96356fb1.js": {
+    "file": "vue.esm-bundler-96356fb1.js"
+  },
+  "_index-62f37ad0.js": {
+    "file": "index-62f37ad0.js",
+    "imports": ["_vue.esm-bundler-49e6b475.js"]
+  },
+  "_vue.esm-bundler-49e6b475.js": {
+    "file": "vue.esm-bundler-49e6b475.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "__plugin-vue_export-helper-c27b6911.js": {
+    "file": "_plugin-vue_export-helper-c27b6911.js"
+  },
+  "_messages-a0a9e13b.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "messages-a0a9e13b.js",
+    "imports": ["_pinia-5d7892fd.js", "_vue.esm-bundler-96356fb1.js"]
+  },
+  "_pinia-5d7892fd.js": {
+    "file": "pinia-5d7892fd.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-outside-click-224980bf.js": {
+    "file": "use-outside-click-224980bf.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-resolve-button-type-c5656cba.js": {
+    "file": "use-resolve-button-type-c5656cba.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_use-event-listener-153dc639.js": {
+    "file": "use-event-listener-153dc639.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_hidden-84ddb9a5.js": {
+    "file": "hidden-84ddb9a5.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_apiClient-01d20438.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "apiClient-01d20438.js"
+  }
+}
diff --git a/tests/data/staticfiles/manifest.json b/tests/data/staticfiles/manifest.json
index 2a40f18..fbe4a74 100644
--- a/tests/data/staticfiles/manifest.json
+++ b/tests/data/staticfiles/manifest.json
@@ -1,3 +1,72 @@
 {
-    "placeholder": "placeholder"
+  "src/entry.ts": {
+    "css": ["assets/entry-74d0d5dd.css"],
+    "file": "assets/entry-29e38a60.js",
+    "imports": [
+      "_vue.esm-bundler-96356fb1.js",
+      "_index-62f37ad0.js",
+      "_vue.esm-bundler-49e6b475.js",
+      "__plugin-vue_export-helper-c27b6911.js",
+      "_messages-a0a9e13b.js",
+      "_use-outside-click-224980bf.js",
+      "_use-resolve-button-type-c5656cba.js",
+      "_use-event-listener-153dc639.js",
+      "_hidden-84ddb9a5.js",
+      "_apiClient-01d20438.js",
+      "_pinia-5d7892fd.js"
+    ],
+    "isEntry": true,
+    "src": "entry.ts"
+  },
+  "src/entry.css": {
+    "file": "assets/entry-74d0d5dd.css",
+    "src": "entry.css"
+  },
+  "src/extra.css": {
+    "file": "assets/extra-a9f3b2c1.css",
+    "src": "extra.css"
+  },
+  "_vue.esm-bundler-96356fb1.js": {
+    "file": "vue.esm-bundler-96356fb1.js"
+  },
+  "_index-62f37ad0.js": {
+    "file": "index-62f37ad0.js",
+    "imports": ["_vue.esm-bundler-49e6b475.js"]
+  },
+  "_vue.esm-bundler-49e6b475.js": {
+    "file": "vue.esm-bundler-49e6b475.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "__plugin-vue_export-helper-c27b6911.js": {
+    "file": "_plugin-vue_export-helper-c27b6911.js"
+  },
+  "_messages-a0a9e13b.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "messages-a0a9e13b.js",
+    "imports": ["_pinia-5d7892fd.js", "_vue.esm-bundler-96356fb1.js"]
+  },
+  "_pinia-5d7892fd.js": {
+    "file": "pinia-5d7892fd.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-outside-click-224980bf.js": {
+    "file": "use-outside-click-224980bf.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js"]
+  },
+  "_use-resolve-button-type-c5656cba.js": {
+    "file": "use-resolve-button-type-c5656cba.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_use-event-listener-153dc639.js": {
+    "file": "use-event-listener-153dc639.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_hidden-84ddb9a5.js": {
+    "file": "hidden-84ddb9a5.js",
+    "imports": ["_vue.esm-bundler-96356fb1.js", "_use-outside-click-224980bf.js"]
+  },
+  "_apiClient-01d20438.js": {
+    "css": ["assets/extra-a9f3b2c1.css"],
+    "file": "apiClient-01d20438.js"
+  }
 }
diff --git a/tests/data/staticfiles/polyfills-manifest.json b/tests/data/staticfiles/polyfills-manifest.json
new file mode 100644
index 0000000..ddebd6a
--- /dev/null
+++ b/tests/data/staticfiles/polyfills-manifest.json
@@ -0,0 +1,24 @@
+{
+    "../../vite/legacy-polyfills-legacy": {
+      "file": "assets/polyfills-legacy-f4c2b91e.js",
+      "isEntry": true,
+      "src": "../../vite/legacy-polyfills-legacy"
+    },
+    "src/entry-legacy.ts": {
+      "file": "assets/entry-legacy-32083566.js",
+      "isEntry": true,
+      "src": "src/entry-legacy.ts"
+    },
+    "src/entry.css": {
+      "file": "assets/entry-74d0d5dd.css",
+      "src": "src/entry.css"
+    },
+    "src/entry.ts": {
+      "css": [
+        "assets/entry-74d0d5dd.css"
+      ],
+      "file": "assets/entry-2e8a3a7a.js",
+      "isEntry": true,
+      "src": "src/entry.ts"
+    }
+}
diff --git a/tests/settings.py b/tests/settings.py
index 0801bd6..541df85 100644
--- a/tests/settings.py
+++ b/tests/settings.py
@@ -3,7 +3,7 @@
 
 BASE_DIR = Path(__file__).resolve().parent
 
-STATIC_URL = "static"
+STATIC_URL = "/static/"
 USE_TZ = True
 
 INSTALLED_APPS = [
@@ -25,7 +25,7 @@
     },
 ]
 
-# django-vite defaults
+STATIC_ROOT = BASE_DIR / "data" / "staticfiles"
+
 DJANGO_VITE_DEV_MODE = True
 DJANGO_VITE_ASSETS_PATH = "/"
-STATIC_ROOT = BASE_DIR / "data" / "staticfiles"
diff --git a/tests/tests/templatetags/test_vite_asset.py b/tests/tests/templatetags/test_vite_asset.py
index c09f5d2..7923954 100644
--- a/tests/tests/templatetags/test_vite_asset.py
+++ b/tests/tests/templatetags/test_vite_asset.py
@@ -1,22 +1,40 @@
 import pytest
 from bs4 import BeautifulSoup
 from django.template import Context, Template, TemplateSyntaxError
+from django_vite.exceptions import DjangoViteAssetNotFoundError
 
 
-def test_vite_asset_returns_script_tags():
+def test_vite_asset_returns_dev_tags():
     template = Template(
         """
         {% load django_vite %}
-        {% vite_asset "src/entry.tsx" %}
+        {% vite_asset "src/entry.ts" %}
     """
     )
     html = template.render(Context({}))
     soup = BeautifulSoup(html, "html.parser")
     script_tag = soup.find("script")
-    assert script_tag["src"] == "http://localhost:3000/static/src/entry.tsx"
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.ts"
     assert script_tag["type"] == "module"
 
 
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_returns_production_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
+
+
 def test_vite_asset_raises_without_path():
     with pytest.raises(TemplateSyntaxError):
         Template(
@@ -25,3 +43,82 @@ def test_vite_asset_raises_without_path():
             {% vite_asset %}
         """
         )
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
+
+
+@pytest.mark.parametrize("prefix", ["custom/prefix", "custom/prefix/"])
+def test_vite_asset_dev_prefix(prefix, patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_STATIC_URL_PREFIX": prefix,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert (
+        script_tag["src"] == "http://localhost:3000/static/custom/prefix/src/entry.ts"
+    )
+    assert script_tag["type"] == "module"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+@pytest.mark.parametrize("prefix", ["custom/prefix", "custom/prefix/"])
+def test_vite_asset_production_prefix(prefix, patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_STATIC_URL_PREFIX": prefix,
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "custom/prefix/assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_production_staticfiles_storage(patch_settings):
+    patch_settings(
+        {
+            "INSTALLED_APPS": ["django_vite", "django.contrib.staticfiles"],
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "/static/assets/entry-29e38a60.js"
+    assert script_tag["type"] == "module"
+    links = soup.find_all("link")
+    assert len(links) == 13
diff --git a/tests/tests/templatetags/test_vite_asset_url.py b/tests/tests/templatetags/test_vite_asset_url.py
index e69de29..e0eef07 100644
--- a/tests/tests/templatetags/test_vite_asset_url.py
+++ b/tests/tests/templatetags/test_vite_asset_url.py
@@ -0,0 +1,43 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_asset_url_returns_dev_url():
+    template = Template(
+        """
+        {% load django_vite %}
+        <script src="{% vite_asset_url "src/entry.ts" %}">
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "http://localhost:3000/static/src/entry.ts"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_url_returns_production_url():
+    template = Template(
+        """
+        {% load django_vite %}
+        <script src="{% vite_asset_url "src/entry.ts" %}">
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-29e38a60.js"
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_asset_url_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            <script src="{% vite_asset_url "src/fake.ts" %}">
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_hmr_client.py b/tests/tests/templatetags/test_vite_hmr_client.py
index 5f35364..2fee91b 100644
--- a/tests/tests/templatetags/test_vite_hmr_client.py
+++ b/tests/tests/templatetags/test_vite_hmr_client.py
@@ -1,3 +1,4 @@
+import pytest
 from bs4 import BeautifulSoup
 from django.template import Context, Template
 
@@ -30,12 +31,8 @@ def test_vite_hmr_client_kwargs():
     assert script_tag["blocking"] == "render"
 
 
-def test_vite_hmr_client_returns_nothing_with_dev_mode_off(patch_settings):
-    patch_settings(
-        {
-            "DJANGO_VITE_DEV_MODE": False,
-        }
-    )
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_hmr_client_returns_nothing_with_dev_mode_off():
     template = Template(
         """
         {% load django_vite %}
diff --git a/tests/tests/templatetags/test_vite_legacy_asset.py b/tests/tests/templatetags/test_vite_legacy_asset.py
index e69de29..4c2111a 100644
--- a/tests/tests/templatetags/test_vite_legacy_asset.py
+++ b/tests/tests/templatetags/test_vite_legacy_asset.py
@@ -0,0 +1,49 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_legacy_asset_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_asset_returns_production_tags(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/entry-2e8a3a7a.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_legacy_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_legacy_polyfills.py b/tests/tests/templatetags/test_vite_legacy_polyfills.py
index e69de29..409c032 100644
--- a/tests/tests/templatetags/test_vite_legacy_polyfills.py
+++ b/tests/tests/templatetags/test_vite_legacy_polyfills.py
@@ -0,0 +1,78 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_vite_legacy_polyfills_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_production(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-f4c2b91e.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_custom_motif(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "custom-motif",
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "custom-motif-polyfills-manifest.json",
+        }
+    )
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_legacy_polyfills %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert script_tag["src"] == "assets/polyfills-legacy-6e7a4b9c.js"
+    assert script_tag["nomodule"] == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_legacy_polyfills_nonexistent_motif(patch_settings, settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_MANIFEST_PATH": settings.STATIC_ROOT
+            / "polyfills-manifest.json",
+            "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "fake-legacy-polyfills",
+        }
+    )
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_legacy_polyfills %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_preload_asset.py b/tests/tests/templatetags/test_vite_preload_asset.py
new file mode 100644
index 0000000..e8ec44a
--- /dev/null
+++ b/tests/tests/templatetags/test_vite_preload_asset.py
@@ -0,0 +1,45 @@
+import pytest
+from bs4 import BeautifulSoup
+from django.template import Context, Template
+from django_vite.exceptions import DjangoViteAssetNotFoundError
+
+
+def test_preload_vite_asset_returns_nothing_with_dev_mode_on():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_preload_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    assert str(soup).strip() == ""
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_preload_vite_asset_returns_production_tags():
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_preload_asset "src/entry.ts" %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    links = soup.find_all("link")
+    assert len(links) == 14
+    first_link = links[0]
+    assert first_link["href"] == "assets/entry-29e38a60.js"
+    assert first_link["rel"] == ["modulepreload"]
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_preload_asset_raises_nonexistent_entry():
+    with pytest.raises(DjangoViteAssetNotFoundError):
+        template = Template(
+            """
+            {% load django_vite %}
+            {% vite_preload_asset "src/fake.ts" %}
+        """
+        )
+        template.render(Context({}))
diff --git a/tests/tests/templatetags/test_vite_react_refresh.py b/tests/tests/templatetags/test_vite_react_refresh.py
index 4787112..e2c31e7 100644
--- a/tests/tests/templatetags/test_vite_react_refresh.py
+++ b/tests/tests/templatetags/test_vite_react_refresh.py
@@ -1,8 +1,9 @@
+import pytest
 from bs4 import BeautifulSoup
 from django.template import Context, Template
 
 
-def test_vite_react_refresh_happy_flow():
+def test_vite_react_refresh_returns_script_tag():
     template = Template(
         """
         {% load django_vite %}
@@ -19,12 +20,35 @@ def test_vite_react_refresh_happy_flow():
     assert "http://localhost:3000/static/@react-refresh" in script_tag.text
 
 
-def test_vite_react_refresh_returns_nothing_with_dev_mode_off(patch_settings):
+def test_vite_react_refresh_uses_correct_settings(patch_settings):
     patch_settings(
         {
-            "DJANGO_VITE_DEV_MODE": False,
+            "DJANGO_VITE_DEV_SERVER_PROTOCOL": "https",
+            "DJANGO_VITE_DEV_SERVER_HOST": "127.0.0.2",
+            "DJANGO_VITE_DEV_SERVER_PORT": "5174",
+            "DJANGO_VITE_STATIC_URL": "static/",
+            "DJANGO_VITE_REACT_REFRESH_URL": "foo/bar",
         }
     )
+
+    template = Template(
+        """
+        {% load django_vite %}
+        {% vite_react_refresh %}
+    """
+    )
+    html = template.render(Context({}))
+    soup = BeautifulSoup(html, "html.parser")
+    script_tag = soup.find("script")
+    assert not script_tag.has_attr("src")
+    assert script_tag.has_attr("type")
+    assert script_tag["type"] == "module"
+    assert "__vite_plugin_react_preamble_installed__" in script_tag.text
+    assert "https://127.0.0.2:5174/static/foo/bar" in script_tag.text
+
+
+@pytest.mark.usefixtures("dev_mode_off")
+def test_vite_react_refresh_returns_nothing_with_dev_mode_off():
     template = Template(
         """
         {% load django_vite %}
diff --git a/tests/tests/test_asset_loader.py b/tests/tests/test_asset_loader.py
index 7ee8935..470867c 100644
--- a/tests/tests/test_asset_loader.py
+++ b/tests/tests/test_asset_loader.py
@@ -1,8 +1,33 @@
 import pytest
 
 from django_vite.templatetags.django_vite import DjangoViteAssetLoader
+from django_vite.apps import check_loader_instance
 
 
 def test_django_vite_asset_loader_cannot_be_instantiated():
     with pytest.raises(RuntimeError):
         DjangoViteAssetLoader()
+
+
+def test_check_loader_instance_happy(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+        }
+    )
+    DjangoViteAssetLoader._instance = None
+    warnings = check_loader_instance()
+    assert len(warnings) == 0
+
+
+def test_check_loader_instance_warnings(patch_settings):
+    patch_settings(
+        {
+            "DJANGO_VITE_DEV_MODE": False,
+            "DJANGO_VITE_MANIFEST_PATH": "fake.json",
+        }
+    )
+    DjangoViteAssetLoader._instance = None
+    warnings = check_loader_instance()
+    assert len(warnings) == 1
+    assert "Make sure you have generated a manifest file" in warnings[0].hint

From 6f8901cb75b339a4da218466d710a0195e0f685c Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Sun, 15 Oct 2023 16:06:07 -0700
Subject: [PATCH 24/29] pass tests after merge

---
 django_vite/templatetags/django_vite.py | 54 +++++++++++++++----------
 1 file changed, 32 insertions(+), 22 deletions(-)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index 17e965d..0196a00 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -21,8 +21,8 @@ class DjangoViteManifest(NamedTuple):
     """
 
     file: str
-    src: str
-    isEntry: bool
+    src: Optional[str] = None
+    isEntry: Optional[bool] = False
     css: Optional[List[str]] = []
     imports: Optional[List[str]] = []
 
@@ -70,7 +70,7 @@ def static_url(self) -> str:
         return url
 
     @property
-    def static_root(self) -> Union[Path, str]:
+    def static_root(self) -> Path:
         """
         Compute the static root URL of assets.
 
@@ -79,7 +79,7 @@ def static_root(self) -> Union[Path, str]:
         """
 
         return (
-            self.assets_path
+            Path(self.assets_path)
             if self.dev_mode
             else Path(settings.STATIC_ROOT) / self.static_url_prefix
         )
@@ -161,6 +161,7 @@ def generate_vite_asset(
                 f"at {config.get_computed_manifest_path()}"
             )
 
+        manifest_entry = manifest[path]
         tags = []
         scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
 
@@ -171,7 +172,7 @@ def generate_vite_asset(
         tags.append(
             DjangoViteAssetLoader._generate_script_tag(
                 DjangoViteAssetLoader._generate_production_server_url(
-                    manifest[path].file, config.static_url_prefix
+                    manifest_entry.file, config.static_url_prefix
                 ),
                 attrs=scripts_attrs,
             )
@@ -185,9 +186,9 @@ def generate_vite_asset(
             "as": "script",
         }
 
-        for dep in manifest.imports:
+        for dep in manifest_entry.imports:
             dep_manifest_entry = manifest[dep]
-            dep_file = dep_manifest_entry["file"]
+            dep_file = dep_manifest_entry.file
             url = DjangoViteAssetLoader._generate_production_server_url(
                 dep_file, config.static_url_prefix
             )
@@ -229,18 +230,20 @@ def preload_vite_asset(
         """
         tags = []
         config = self._get_config(config_key)
-        manifest = self._get_manifest(config_key)
-        manifest_entry = manifest[path]
 
-        if not config.dev_mode:
+        if config.dev_mode:
             return ""
 
+        manifest = self._get_manifest(config_key)
+
         if path not in manifest:
             raise DjangoViteAssetNotFoundError(
                 f"Cannot find {path} in Vite manifest "
                 f"at {config.get_computed_manifest_path()}"
             )
 
+        manifest_entry = manifest[path]
+
         # Add the script by itself
         script_attrs = {
             "type": "text/javascript",
@@ -375,8 +378,10 @@ def generate_vite_asset_url(self, path: str, config_key: str) -> str:
                 f"at {config.get_computed_manifest_path()}"
             )
 
+        manifest_entry = manifest[path]
+
         return DjangoViteAssetLoader._generate_production_server_url(
-            self._manifest[path]["file"], config.static_url_prefix
+            manifest_entry.file, config.static_url_prefix
         )
 
     def generate_vite_legacy_polyfills(
@@ -406,18 +411,18 @@ def generate_vite_legacy_polyfills(
         """
 
         config = self._get_config(config_key)
-        manifest = self._get_manifest(config_key)
 
         if config.dev_mode:
             return ""
 
+        manifest = self._get_manifest(config_key)
         scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
 
-        for path, content in manifest.items():
+        for path, manifest_entry in manifest.items():
             if config.legacy_polyfills_motif in path:
                 return DjangoViteAssetLoader._generate_script_tag(
                     DjangoViteAssetLoader._generate_production_server_url(
-                        content["file"], config.static_url_prefix
+                        manifest_entry.file, config.static_url_prefix
                     ),
                     attrs=scripts_attrs,
                 )
@@ -587,6 +592,7 @@ def instance(cls):
                     "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
                     "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
                     "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
+                    "DJANGO_VITE_REACT_REFRESH_URL": "react_refresh_url",
                 }
 
                 config = {
@@ -597,11 +603,14 @@ def instance(cls):
 
                 cls._instance._configs[DEFAULT_CONFIG_KEY] = DjangoViteConfig(**config)
 
+                # Manifest is only used in production.
+                if not cls._instance._configs[DEFAULT_CONFIG_KEY].dev_mode:
+                    cls._instance._parse_manifest(DEFAULT_CONFIG_KEY)
+
         return cls._instance
 
-    @classmethod
     def generate_vite_ws_client(
-        cls, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+        self, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
     ) -> str:
         """
         Generates the script tag for the Vite WS client for HMR.
@@ -619,13 +628,13 @@ def generate_vite_ws_client(
                 script tags.
         """
 
-        config = cls._get_config(config_key)
+        config = self._get_config(config_key)
 
         if not config.dev_mode:
             return ""
 
-        return cls._generate_script_tag(
-            cls._generate_vite_server_url(config.ws_client_url, config),
+        return self._generate_script_tag(
+            self._generate_vite_server_url(config.ws_client_url, config),
             {"type": "module", **kwargs},
         )
 
@@ -704,7 +713,6 @@ def _generate_vite_server_url(
             urljoin(config.static_url, path),
         )
 
-    @classmethod
     def generate_vite_react_refresh_url(
         self, config_key: str = DEFAULT_CONFIG_KEY
     ) -> str:
@@ -778,7 +786,9 @@ def vite_hmr_client(
             script tags.
     """
 
-    return DjangoViteAssetLoader.generate_vite_ws_client(config_key, **kwargs)
+    return DjangoViteAssetLoader.instance().generate_vite_ws_client(
+        config_key, **kwargs
+    )
 
 
 @register.simple_tag
@@ -945,4 +955,4 @@ def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
     Returns:
         str -- The script or an empty string.
     """
-    return DjangoViteAssetLoader.generate_vite_react_refresh_url(config_key)
+    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(config_key)

From e6100b05f3b58c566d46cc185d6f1023b7744a53 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Mon, 16 Oct 2023 10:28:20 -0700
Subject: [PATCH 25/29] duplicate django_vite/templatetags/django_vite.py to
 django_vite/loader.py

---
 django_vite/{templatetags/django_vite.py => loader.py} | 0
 1 file changed, 0 insertions(+), 0 deletions(-)
 rename django_vite/{templatetags/django_vite.py => loader.py} (100%)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/loader.py
similarity index 100%
rename from django_vite/templatetags/django_vite.py
rename to django_vite/loader.py

From bd97cf0badc03261b7ab5e29b9ff58efce69cc8c Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Mon, 16 Oct 2023 10:29:21 -0700
Subject: [PATCH 26/29] restore duplicated
 django_vite/templatetags/django_vite.py

---
 django_vite/templatetags/django_vite.py | 958 ++++++++++++++++++++++++
 1 file changed, 958 insertions(+)
 create mode 100644 django_vite/templatetags/django_vite.py

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
new file mode 100644
index 0000000..c1e9956
--- /dev/null
+++ b/django_vite/templatetags/django_vite.py
@@ -0,0 +1,958 @@
+import json
+from pathlib import Path
+from typing import Dict, List, Callable, NamedTuple, Optional, Type, Union
+from urllib.parse import urljoin
+
+from django import template
+from django.apps import apps
+from django.conf import settings
+from django.utils.safestring import mark_safe
+
+from django_vite.exceptions import DjangoViteAssetNotFoundError, DjangoViteManifestError
+
+DEFAULT_CONFIG_KEY = "default"
+
+register = template.Library()
+
+
+class DjangoViteManifest(NamedTuple):
+    """
+    Represent an entry for a file inside the "manifest.json".
+    """
+
+    file: str
+    src: Optional[str] = None
+    isEntry: Optional[bool] = False
+    css: Optional[List[str]] = []
+    imports: Optional[List[str]] = []
+
+
+class DjangoViteConfig(NamedTuple):
+    """
+    Represent the Django Vite configuration structure.
+    """
+
+    # Location of Vite compiled assets (only used in Vite production mode).
+    assets_path: Union[Path, str]
+
+    # If using in development or production mode.
+    dev_mode: bool = False
+
+    # Default Vite server protocol (http or https)
+    dev_server_protocol: str = "http"
+
+    # Default vite server hostname.
+    dev_server_host: str = "localhost"
+
+    # Default Vite server port.
+    dev_server_port: int = 3000
+
+    # Default Vite server path to HMR script.
+    ws_client_url: str = "@vite/client"
+
+    # Prefix for STATIC_URL.
+    static_url_prefix: str = ""
+
+    # Motif in the "manifest.json" to find the polyfills generated by Vite.
+    legacy_polyfills_motif: str = "legacy-polyfills"
+
+    # Path to your manifest file generated by Vite.
+    manifest_path: Union[Path, str] = ""
+
+    # Default Vite server path to React RefreshRuntime for @vitejs/plugin-react.
+    react_refresh_url: str = "@react-refresh"
+
+    @property
+    def static_url(self) -> str:
+        url = urljoin(settings.STATIC_URL, self.static_url_prefix)
+        if not url.endswith("/"):
+            url += "/"
+        return url
+
+    @property
+    def static_root(self) -> Path:
+        """
+        Compute the static root URL of assets.
+
+        Returns:
+            Union[Path, str] -- Static root URL.
+        """
+
+        return (
+            Path(self.assets_path)
+            if self.dev_mode
+            else Path(settings.STATIC_ROOT) / self.static_url_prefix
+        )
+
+    def get_computed_manifest_path(self) -> Union[Path, str]:
+        """
+        Compute the path to the "manifest.json".
+
+        Returns:
+            Union[Path, str] -- Path to the "manifest.json".
+        """
+
+        return (
+            self.manifest_path
+            if self.manifest_path
+            else self.static_root / "manifest.json"
+        )
+
+
+class DjangoViteAssetLoader:
+    """
+    Class handling Vite asset loading.
+    """
+
+    _instance = None
+
+    _configs = Dict[str, Type[DjangoViteConfig]]
+    _manifests: Dict[str, Type[DjangoViteManifest]]
+
+    def __init__(self) -> None:
+        raise RuntimeError("Use the instance() method instead.")
+
+    def generate_vite_asset(
+        self,
+        path: str,
+        config_key: str,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        """
+        Generates a <script> tag for this JS/TS asset, a <link> tag for
+        all of its CSS dependencies, and a <link modulepreload>
+        for the js dependencies, as listed in the manifest file
+        (for production only).
+        In development Vite loads all by itself.
+
+        Arguments:
+            path {str} -- Path to a Vite JS/TS asset to include.
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            str -- All tags to import this file in your HTML page.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If cannot find the file path in the
+                manifest (only in production).
+
+        Returns:
+            str -- The <script> tag and all <link> tags to import
+                this asset in your page.
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return DjangoViteAssetLoader._generate_script_tag(
+                DjangoViteAssetLoader._generate_vite_server_url(path, config),
+                {"type": "module", **kwargs},
+            )
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        manifest_entry = manifest[path]
+        tags = []
+        scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
+
+        # Add dependent CSS
+        tags.extend(self._load_css_files_of_asset(path, config_key))
+
+        # Add the script by itself
+        tags.append(
+            DjangoViteAssetLoader._generate_script_tag(
+                DjangoViteAssetLoader._generate_production_server_url(
+                    manifest_entry.file, config.static_url_prefix
+                ),
+                attrs=scripts_attrs,
+            )
+        )
+
+        # Preload imports
+        preload_attrs = {
+            "type": "text/javascript",
+            "crossorigin": "anonymous",
+            "rel": "modulepreload",
+            "as": "script",
+        }
+
+        for dep in manifest_entry.imports:
+            dep_manifest_entry = manifest[dep]
+            dep_file = dep_manifest_entry.file
+            url = DjangoViteAssetLoader._generate_production_server_url(
+                dep_file, config.static_url_prefix
+            )
+            tags.append(
+                DjangoViteAssetLoader._generate_preload_tag(
+                    url,
+                    attrs=preload_attrs,
+                )
+            )
+
+        return "\n".join(tags)
+
+    def preload_vite_asset(
+        self,
+        path: str,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> str:
+        """
+        Generates a <link modulepreload> tag for this JS/TS asset, a
+        <link preload> tag for all of its CSS dependencies,
+        and a <link modulepreload> for the js dependencies.
+        In development this template tag renders nothing,
+        since files aren't compiled yet"
+
+        Arguments:
+            path {str} -- Path to a Vite JS/TS asset to preload.
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            str -- All tags to preload this file in your HTML page.
+
+        Raises:
+            DjangoViteAssetNotFoundError: if cannot find the file path in the
+                manifest.
+
+        Returns:
+            str -- all <link> tags to preload
+                this asset.
+        """
+        tags = []
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return ""
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        manifest_entry = manifest[path]
+
+        # Add the script by itself
+        script_attrs = {
+            "type": "text/javascript",
+            "crossorigin": "anonymous",
+            "rel": "modulepreload",
+            "as": "script",
+        }
+
+        manifest_file = manifest_entry.file
+        url = DjangoViteAssetLoader._generate_production_server_url(
+            manifest_file, config.static_url_prefix
+        )
+        tags.append(
+            DjangoViteAssetLoader._generate_preload_tag(
+                url,
+                attrs=script_attrs,
+            )
+        )
+
+        # Add dependent CSS
+        tags.extend(self._preload_css_files_of_asset(path, config_key))
+
+        # Preload imports
+        for dep in manifest_entry.imports:
+            dep_manifest_entry = manifest[dep]
+            dep_file = dep_manifest_entry.file
+            url = DjangoViteAssetLoader._generate_production_server_url(
+                dep_file, config.static_url_prefix
+            )
+            tags.append(
+                DjangoViteAssetLoader._generate_preload_tag(
+                    url,
+                    attrs=script_attrs,
+                )
+            )
+
+        return "\n".join(tags)
+
+    def _preload_css_files_of_asset(
+        self,
+        path: str,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> List[str]:
+        return self._generate_css_files_of_asset(
+            path,
+            [],
+            DjangoViteAssetLoader._generate_stylesheet_preload_tag,
+            config_key,
+        )
+
+    def _load_css_files_of_asset(
+        self,
+        path: str,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> List[str]:
+        return self._generate_css_files_of_asset(
+            path,
+            [],
+            DjangoViteAssetLoader._generate_stylesheet_tag,
+            config_key,
+        )
+
+    def _generate_css_files_of_asset(
+        self,
+        path: str,
+        already_processed: List[str],
+        tag_generator: Callable,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> List[str]:
+        """
+        Generates all CSS tags for dependencies of an asset.
+
+        Arguments:
+            path {str} -- Path to an asset in the 'manifest.json'.
+            config_key {str} -- Key of the configuration to use.
+            already_processed {list} -- List of already processed CSS file.
+
+        Returns:
+            list -- List of CSS tags.
+        """
+
+        tags = []
+        config = self._get_config(config_key)
+        manifest = self._get_manifest(config_key)
+        manifest_entry = manifest[path]
+
+        for import_path in manifest_entry.imports:
+            tags.extend(
+                self._generate_css_files_of_asset(
+                    import_path, already_processed, tag_generator, config_key
+                )
+            )
+
+        for css_path in manifest_entry.css:
+            if css_path not in already_processed:
+                url = DjangoViteAssetLoader._generate_production_server_url(
+                    css_path, config.static_url_prefix
+                )
+                tags.append(tag_generator(url))
+
+            already_processed.append(css_path)
+
+        return tags
+
+    def generate_vite_asset_url(self, path: str, config_key: str) -> str:
+        """
+        Generates only the URL of an asset managed by ViteJS.
+        Warning, this function does not generate URLs for dependant assets.
+
+        Arguments:
+            path {str} -- Path to a Vite asset.
+            config_key {str} -- Key of the configuration to use.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
+                manifest (only in production).
+
+        Returns:
+            str -- The URL of this asset.
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return DjangoViteAssetLoader._generate_vite_server_url(path, config)
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        manifest_entry = manifest[path]
+
+        return DjangoViteAssetLoader._generate_production_server_url(
+            manifest_entry.file, config.static_url_prefix
+        )
+
+    def generate_vite_legacy_polyfills(
+        self,
+        config_key: str,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        """
+        Generates a <script> tag to the polyfills
+        generated by '@vitejs/plugin-legacy' if used.
+        This tag must be included at end of the <body> before
+        including other legacy scripts.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If polyfills path not found inside
+                the 'manifest.json' (only in production).
+
+        Returns:
+            str -- The script tag to the polyfills.
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return ""
+
+        manifest = self._get_manifest(config_key)
+        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
+
+        for path, manifest_entry in manifest.items():
+            if config.legacy_polyfills_motif in path:
+                return DjangoViteAssetLoader._generate_script_tag(
+                    DjangoViteAssetLoader._generate_production_server_url(
+                        manifest_entry.file, config.static_url_prefix
+                    ),
+                    attrs=scripts_attrs,
+                )
+
+        raise DjangoViteAssetNotFoundError(
+            f"Vite legacy polyfills not found in manifest "
+            f"at {config.get_computed_manifest_path()}"
+        )
+
+    def generate_vite_legacy_asset(
+        self,
+        path: str,
+        config_key: str,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        """
+        Generates a <script> tag for legacy assets JS/TS
+        generated by '@vitejs/plugin-legacy'
+        (in production only, in development do nothing).
+
+        Arguments:
+            path {str} -- Path to a Vite asset to include
+                (must contains '-legacy' in its name).
+            config_key {str} -- Key of the configuration to use.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
+                manifest (only in production).
+
+        Returns:
+            str -- The script tag of this legacy asset .
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return ""
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
+
+        url = DjangoViteAssetLoader._generate_production_server_url(
+            manifest[path].file, config.static_url_prefix
+        )
+        return DjangoViteAssetLoader._generate_script_tag(
+            url,
+            attrs=scripts_attrs,
+        )
+
+    def _get_config(self, config_key: str) -> Type[DjangoViteConfig]:
+        """
+        Get configuration object registered with the key passed in
+        parameter.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to retrieve.
+
+        Raises:
+            RuntimeError: If no configuration exists for this key.
+
+        Returns:
+            Type[DjangoViteConfig] -- The configuration.
+        """
+
+        if config_key not in self._configs:
+            raise RuntimeError(f'Cannot find "{config_key}" configuration')
+
+        return self._configs[config_key]
+
+    def _parse_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
+        """
+        Read and parse the Vite manifest file.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Raises:
+            DjangoViteManifestError: if cannot load the file or JSON in file is
+            malformed.
+        """
+
+        config = self._get_config(config_key)
+
+        try:
+            with open(config.get_computed_manifest_path(), "r") as manifest_file:
+                manifest_content = manifest_file.read()
+                manifest_json = json.loads(manifest_content)
+
+                return {k: DjangoViteManifest(**v) for k, v in manifest_json.items()}
+
+        except Exception as error:
+            raise DjangoViteManifestError(
+                f"Cannot read Vite manifest file at "
+                f"{config.get_computed_manifest_path()} : {str(error)}"
+            ) from error
+
+    def _get_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
+        """
+        Load if needed and parse the "manifest.json" of the specified
+        configuration.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            Dict[str, Type[DjangoViteManifest]] -- Parsed content of
+                the "manifest.json"
+        """
+
+        if config_key not in self._manifests:
+            self._manifests[config_key] = self._parse_manifest(config_key)
+
+        return self._manifests[config_key]
+
+    @classmethod
+    def instance(cls):
+        """
+        Singleton.
+        Uses singleton to keep parsed manifests in memory after
+        the first time they are loaded.
+
+        Returns:
+            DjangoViteAssetLoader -- only instance of the class.
+        """
+
+        if cls._instance is None:
+            cls._instance = cls.__new__(cls)
+            cls._instance._configs = {}
+            cls._instance._manifests = {}
+
+            if hasattr(settings, "DJANGO_VITE"):
+                config = getattr(settings, "DJANGO_VITE")
+
+                for config_key, config_values in config.items():
+                    if isinstance(config_values, DjangoViteConfig):
+                        cls._instance._configs[config_key] = config_values
+                    elif isinstance(config_values, dict):
+                        cls._instance._configs[config_key] = DjangoViteConfig(
+                            **config_values
+                        )
+                    else:
+                        raise RuntimeError(
+                            f"Cannot read configuration for key '{config_key}'"
+                        )
+            else:
+                # Warning : This branch will be remove in further
+                # releases. Please use new way of handling configuration.
+
+                _config_keys = {
+                    "DJANGO_VITE_DEV_MODE": "dev_mode",
+                    "DJANGO_VITE_DEV_SERVER_PROTOCOL": "dev_server_protocol",
+                    "DJANGO_VITE_DEV_SERVER_HOST": "dev_server_host",
+                    "DJANGO_VITE_DEV_SERVER_PORT": "dev_server_port",
+                    "DJANGO_VITE_WS_CLIENT_URL": "ws_client_url",
+                    "DJANGO_VITE_ASSETS_PATH": "assets_path",
+                    "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
+                    "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
+                    "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
+                    "DJANGO_VITE_REACT_REFRESH_URL": "react_refresh_url",
+                }
+
+                config = {
+                    _config_keys[setting_key]: getattr(settings, setting_key)
+                    for setting_key in dir(settings)
+                    if setting_key in _config_keys.keys()
+                }
+
+                cls._instance._configs[DEFAULT_CONFIG_KEY] = DjangoViteConfig(**config)
+
+                # Manifest is only used in production.
+                if not cls._instance._configs[DEFAULT_CONFIG_KEY].dev_mode:
+                    cls._instance._parse_manifest(DEFAULT_CONFIG_KEY)
+
+        return cls._instance
+
+    def generate_vite_ws_client(
+        self, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+    ) -> str:
+        """
+        Generates the script tag for the Vite WS client for HMR.
+        Only used in development, in production this method returns
+        an empty string.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            str -- The script tag or an empty string.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+        """
+
+        config = self._get_config(config_key)
+
+        if not config.dev_mode:
+            return ""
+
+        return self._generate_script_tag(
+            self._generate_vite_server_url(config.ws_client_url, config),
+            {"type": "module", **kwargs},
+        )
+
+    @staticmethod
+    def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
+        """
+        Generates an HTML script tag.
+
+        Arguments:
+            src {str} -- Source of the script.
+
+        Keyword Arguments:
+            attrs {Dict[str, str]} -- List of custom attributes
+                for the tag.
+
+        Returns:
+            str -- The script tag.
+        """
+
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
+
+        return f'<script {attrs_str} src="{src}"></script>'
+
+    @staticmethod
+    def _generate_stylesheet_tag(href: str) -> str:
+        """
+        Generates an HTML <link> stylesheet tag for CSS.
+
+        Arguments:
+            href {str} -- CSS file URL.
+
+        Returns:
+            str -- CSS link tag.
+        """
+
+        return f'<link rel="stylesheet" href="{href}" />'
+
+    def _generate_stylesheet_preload_tag(href: str) -> str:
+        """
+        Generates an HTML <link> preload tag for CSS.
+
+        Arguments:
+            href {str} -- CSS file URL.
+
+        Returns:
+            str -- CSS link tag.
+        """
+
+        return f'<link rel="preload" href="{href}" as="style" />'
+
+    @staticmethod
+    def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
+
+        return f'<link href="{href}" {attrs_str} />'
+
+    @staticmethod
+    def _generate_vite_server_url(
+        path: str,
+        config: Type[DjangoViteConfig],
+    ) -> str:
+        """
+        Generates an URL to and asset served by the Vite development server.
+
+        Keyword Arguments:
+            path {str} -- Path to the asset.
+            config {Type[DjangoViteConfig]} -- Config object to use.
+
+        Returns:
+            str -- Full URL to the asset.
+        """
+
+        return urljoin(
+            f"{config.dev_server_protocol}://"
+            f"{config.dev_server_host}:{config.dev_server_port}",
+            urljoin(config.static_url, path),
+        )
+
+    def generate_vite_react_refresh_url(
+        self, config_key: str = DEFAULT_CONFIG_KEY
+    ) -> str:
+        """
+        Generates the script for the Vite React Refresh for HMR.
+        Only used in development, in production this method returns
+        an empty string.
+
+        Returns:
+            str -- The script or an empty string.
+            config_key {str} -- Key of the configuration to use.
+        """
+        config = self._get_config(config_key)
+
+        if not config.dev_mode:
+            return ""
+
+        return f"""<script type="module">
+            import RefreshRuntime from \
+            '{self._generate_vite_server_url(config.react_refresh_url, config)}'
+            RefreshRuntime.injectIntoGlobalHook(window)
+            window.$RefreshReg$ = () => {{}}
+            window.$RefreshSig$ = () => (type) => type
+            window.__vite_plugin_react_preamble_installed__ = true
+        </script>"""
+
+    @staticmethod
+    def _generate_production_server_url(path: str, static_url_prefix="") -> str:
+        """
+        Generates an URL to an asset served during production.
+
+        Keyword Arguments:
+            path {str} -- Path to the asset.
+
+        Returns:
+            str -- Full URL to the asset.
+        """
+
+        production_server_url = path
+        if prefix := static_url_prefix:
+            if not static_url_prefix.endswith("/"):
+                prefix += "/"
+            production_server_url = urljoin(prefix, path)
+
+        if apps.is_installed("django.contrib.staticfiles"):
+            from django.contrib.staticfiles.storage import staticfiles_storage
+
+            return staticfiles_storage.url(production_server_url)
+
+        return production_server_url
+
+
+@register.simple_tag
+@mark_safe
+def vite_hmr_client(
+    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+) -> str:
+    """
+    Generates the script tag for the Vite WS client for HMR.
+    Only used in development, in production this method returns
+    an empty string.
+
+    Arguments:
+        config {str} -- Configuration to use.
+
+    Returns:
+        str -- The script tag or an empty string.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+    """
+
+    return DjangoViteAssetLoader.instance().generate_vite_ws_client(
+        config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_asset(
+    path: str,
+    config_key: str = DEFAULT_CONFIG_KEY,
+    **kwargs: Dict[str, str],
+) -> str:
+    """
+    Generates a <script> tag for this JS/TS asset, a <link> tag for
+    all of its CSS dependencies, and a <link rel="modulepreload">
+    for all js dependencies, as listed in the manifest file
+    In development Vite loads all by itself.
+
+    Arguments:
+        path {str} -- Path to a Vite JS/TS asset to include.
+        config {str} -- Configuration to use.
+
+    Returns:
+        str -- All tags to import this file in your HTML page.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If cannot find the file path in the
+            manifest (only in production).
+
+    Returns:
+        str -- The <script> tag and all <link> tags to import this
+            asset in your page.
+    """
+
+    assert path is not None
+    assert config_key is not None
+
+    return DjangoViteAssetLoader.instance().generate_vite_asset(
+        path, config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_preload_asset(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
+    """
+    Generates preloadmodule tag for this JS/TS asset and preloads
+    all of its CSS and JS dependencies by reading the manifest
+    file (for production only).
+    In development does nothing.
+
+    Arguments:
+        path {str} -- Path to a Vite JS/TS asset to include.
+
+    Returns:
+        str -- All tags to import this file in your HTML page.
+
+    Raises:
+        RuntimeError: If cannot find the file path in the
+            manifest (only in production).
+
+    """
+    assert path is not None
+
+    return DjangoViteAssetLoader.instance().preload_vite_asset(path, config_key)
+
+
+@register.simple_tag
+def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
+    """
+    Generates only the URL of an asset managed by ViteJS.
+    Warning, this function does not generate URLs for dependant assets.
+
+    Arguments:
+        path {str} -- Path to a Vite asset.
+        config {str} -- Configuration to use.
+
+    Raises:
+        RuntimeError: If cannot find the asset path in the
+            manifest (only in production).
+
+    Returns:
+        str -- The URL of this asset.
+    """
+    assert path is not None
+    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, config_key)
+
+
+@register.simple_tag
+@mark_safe
+def vite_legacy_polyfills(
+    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+) -> str:
+    """
+    Generates a <script> tag to the polyfills generated
+    by '@vitejs/plugin-legacy' if used.
+    This tag must be included at end of the <body> before including
+    other legacy scripts.
+
+    Arguments:
+        config_key {str} -- Configuration to use.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If polyfills path not found inside
+            the 'manifest.json' (only in production).
+
+    Returns:
+        str -- The script tag to the polyfills.
+    """
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
+        config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_legacy_asset(
+    path: str,
+    config_key: str = DEFAULT_CONFIG_KEY,
+    **kwargs: Dict[str, str],
+) -> str:
+    """
+    Generates a <script> tag for legacy assets JS/TS
+    generated by '@vitejs/plugin-legacy'
+    (in production only, in development do nothing).
+
+    Arguments:
+        path {str} -- Path to a Vite asset to include
+            (must contains '-legacy' in its name).
+        config_key {str} -- Configuration to use.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If cannot find the asset path in
+            the manifest (only in production).
+
+    Returns:
+        str -- The script tag of this legacy asset.
+    """
+
+    assert path is not None
+
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
+        path, config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
+    """
+    Generates the script for the Vite React Refresh for HMR.
+    Only used in development, in production this method returns
+    an empty string.
+
+    Returns:
+        str -- The script or an empty string.
+    """
+    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(config_key)

From e099d4cea5096556c7278cb66d7ce5e56bf0901b Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Mon, 16 Oct 2023 10:31:49 -0700
Subject: [PATCH 27/29] duplicate django_vite/templatetags/django_vite.py to
 django_vite/tag_generator.py

---
 django_vite/{templatetags/django_vite.py => tag_generator.py} | 0
 1 file changed, 0 insertions(+), 0 deletions(-)
 rename django_vite/{templatetags/django_vite.py => tag_generator.py} (100%)

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/tag_generator.py
similarity index 100%
rename from django_vite/templatetags/django_vite.py
rename to django_vite/tag_generator.py

From 2a024e8e67876b3b3ee0fce4c34639fc006923f3 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Mon, 16 Oct 2023 10:32:12 -0700
Subject: [PATCH 28/29] restore duplicated
 django_vite/templatetags/django_vite.py

---
 django_vite/templatetags/django_vite.py | 958 ++++++++++++++++++++++++
 1 file changed, 958 insertions(+)
 create mode 100644 django_vite/templatetags/django_vite.py

diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
new file mode 100644
index 0000000..c1e9956
--- /dev/null
+++ b/django_vite/templatetags/django_vite.py
@@ -0,0 +1,958 @@
+import json
+from pathlib import Path
+from typing import Dict, List, Callable, NamedTuple, Optional, Type, Union
+from urllib.parse import urljoin
+
+from django import template
+from django.apps import apps
+from django.conf import settings
+from django.utils.safestring import mark_safe
+
+from django_vite.exceptions import DjangoViteAssetNotFoundError, DjangoViteManifestError
+
+DEFAULT_CONFIG_KEY = "default"
+
+register = template.Library()
+
+
+class DjangoViteManifest(NamedTuple):
+    """
+    Represent an entry for a file inside the "manifest.json".
+    """
+
+    file: str
+    src: Optional[str] = None
+    isEntry: Optional[bool] = False
+    css: Optional[List[str]] = []
+    imports: Optional[List[str]] = []
+
+
+class DjangoViteConfig(NamedTuple):
+    """
+    Represent the Django Vite configuration structure.
+    """
+
+    # Location of Vite compiled assets (only used in Vite production mode).
+    assets_path: Union[Path, str]
+
+    # If using in development or production mode.
+    dev_mode: bool = False
+
+    # Default Vite server protocol (http or https)
+    dev_server_protocol: str = "http"
+
+    # Default vite server hostname.
+    dev_server_host: str = "localhost"
+
+    # Default Vite server port.
+    dev_server_port: int = 3000
+
+    # Default Vite server path to HMR script.
+    ws_client_url: str = "@vite/client"
+
+    # Prefix for STATIC_URL.
+    static_url_prefix: str = ""
+
+    # Motif in the "manifest.json" to find the polyfills generated by Vite.
+    legacy_polyfills_motif: str = "legacy-polyfills"
+
+    # Path to your manifest file generated by Vite.
+    manifest_path: Union[Path, str] = ""
+
+    # Default Vite server path to React RefreshRuntime for @vitejs/plugin-react.
+    react_refresh_url: str = "@react-refresh"
+
+    @property
+    def static_url(self) -> str:
+        url = urljoin(settings.STATIC_URL, self.static_url_prefix)
+        if not url.endswith("/"):
+            url += "/"
+        return url
+
+    @property
+    def static_root(self) -> Path:
+        """
+        Compute the static root URL of assets.
+
+        Returns:
+            Union[Path, str] -- Static root URL.
+        """
+
+        return (
+            Path(self.assets_path)
+            if self.dev_mode
+            else Path(settings.STATIC_ROOT) / self.static_url_prefix
+        )
+
+    def get_computed_manifest_path(self) -> Union[Path, str]:
+        """
+        Compute the path to the "manifest.json".
+
+        Returns:
+            Union[Path, str] -- Path to the "manifest.json".
+        """
+
+        return (
+            self.manifest_path
+            if self.manifest_path
+            else self.static_root / "manifest.json"
+        )
+
+
+class DjangoViteAssetLoader:
+    """
+    Class handling Vite asset loading.
+    """
+
+    _instance = None
+
+    _configs = Dict[str, Type[DjangoViteConfig]]
+    _manifests: Dict[str, Type[DjangoViteManifest]]
+
+    def __init__(self) -> None:
+        raise RuntimeError("Use the instance() method instead.")
+
+    def generate_vite_asset(
+        self,
+        path: str,
+        config_key: str,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        """
+        Generates a <script> tag for this JS/TS asset, a <link> tag for
+        all of its CSS dependencies, and a <link modulepreload>
+        for the js dependencies, as listed in the manifest file
+        (for production only).
+        In development Vite loads all by itself.
+
+        Arguments:
+            path {str} -- Path to a Vite JS/TS asset to include.
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            str -- All tags to import this file in your HTML page.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If cannot find the file path in the
+                manifest (only in production).
+
+        Returns:
+            str -- The <script> tag and all <link> tags to import
+                this asset in your page.
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return DjangoViteAssetLoader._generate_script_tag(
+                DjangoViteAssetLoader._generate_vite_server_url(path, config),
+                {"type": "module", **kwargs},
+            )
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        manifest_entry = manifest[path]
+        tags = []
+        scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
+
+        # Add dependent CSS
+        tags.extend(self._load_css_files_of_asset(path, config_key))
+
+        # Add the script by itself
+        tags.append(
+            DjangoViteAssetLoader._generate_script_tag(
+                DjangoViteAssetLoader._generate_production_server_url(
+                    manifest_entry.file, config.static_url_prefix
+                ),
+                attrs=scripts_attrs,
+            )
+        )
+
+        # Preload imports
+        preload_attrs = {
+            "type": "text/javascript",
+            "crossorigin": "anonymous",
+            "rel": "modulepreload",
+            "as": "script",
+        }
+
+        for dep in manifest_entry.imports:
+            dep_manifest_entry = manifest[dep]
+            dep_file = dep_manifest_entry.file
+            url = DjangoViteAssetLoader._generate_production_server_url(
+                dep_file, config.static_url_prefix
+            )
+            tags.append(
+                DjangoViteAssetLoader._generate_preload_tag(
+                    url,
+                    attrs=preload_attrs,
+                )
+            )
+
+        return "\n".join(tags)
+
+    def preload_vite_asset(
+        self,
+        path: str,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> str:
+        """
+        Generates a <link modulepreload> tag for this JS/TS asset, a
+        <link preload> tag for all of its CSS dependencies,
+        and a <link modulepreload> for the js dependencies.
+        In development this template tag renders nothing,
+        since files aren't compiled yet"
+
+        Arguments:
+            path {str} -- Path to a Vite JS/TS asset to preload.
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            str -- All tags to preload this file in your HTML page.
+
+        Raises:
+            DjangoViteAssetNotFoundError: if cannot find the file path in the
+                manifest.
+
+        Returns:
+            str -- all <link> tags to preload
+                this asset.
+        """
+        tags = []
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return ""
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        manifest_entry = manifest[path]
+
+        # Add the script by itself
+        script_attrs = {
+            "type": "text/javascript",
+            "crossorigin": "anonymous",
+            "rel": "modulepreload",
+            "as": "script",
+        }
+
+        manifest_file = manifest_entry.file
+        url = DjangoViteAssetLoader._generate_production_server_url(
+            manifest_file, config.static_url_prefix
+        )
+        tags.append(
+            DjangoViteAssetLoader._generate_preload_tag(
+                url,
+                attrs=script_attrs,
+            )
+        )
+
+        # Add dependent CSS
+        tags.extend(self._preload_css_files_of_asset(path, config_key))
+
+        # Preload imports
+        for dep in manifest_entry.imports:
+            dep_manifest_entry = manifest[dep]
+            dep_file = dep_manifest_entry.file
+            url = DjangoViteAssetLoader._generate_production_server_url(
+                dep_file, config.static_url_prefix
+            )
+            tags.append(
+                DjangoViteAssetLoader._generate_preload_tag(
+                    url,
+                    attrs=script_attrs,
+                )
+            )
+
+        return "\n".join(tags)
+
+    def _preload_css_files_of_asset(
+        self,
+        path: str,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> List[str]:
+        return self._generate_css_files_of_asset(
+            path,
+            [],
+            DjangoViteAssetLoader._generate_stylesheet_preload_tag,
+            config_key,
+        )
+
+    def _load_css_files_of_asset(
+        self,
+        path: str,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> List[str]:
+        return self._generate_css_files_of_asset(
+            path,
+            [],
+            DjangoViteAssetLoader._generate_stylesheet_tag,
+            config_key,
+        )
+
+    def _generate_css_files_of_asset(
+        self,
+        path: str,
+        already_processed: List[str],
+        tag_generator: Callable,
+        config_key: str = DEFAULT_CONFIG_KEY,
+    ) -> List[str]:
+        """
+        Generates all CSS tags for dependencies of an asset.
+
+        Arguments:
+            path {str} -- Path to an asset in the 'manifest.json'.
+            config_key {str} -- Key of the configuration to use.
+            already_processed {list} -- List of already processed CSS file.
+
+        Returns:
+            list -- List of CSS tags.
+        """
+
+        tags = []
+        config = self._get_config(config_key)
+        manifest = self._get_manifest(config_key)
+        manifest_entry = manifest[path]
+
+        for import_path in manifest_entry.imports:
+            tags.extend(
+                self._generate_css_files_of_asset(
+                    import_path, already_processed, tag_generator, config_key
+                )
+            )
+
+        for css_path in manifest_entry.css:
+            if css_path not in already_processed:
+                url = DjangoViteAssetLoader._generate_production_server_url(
+                    css_path, config.static_url_prefix
+                )
+                tags.append(tag_generator(url))
+
+            already_processed.append(css_path)
+
+        return tags
+
+    def generate_vite_asset_url(self, path: str, config_key: str) -> str:
+        """
+        Generates only the URL of an asset managed by ViteJS.
+        Warning, this function does not generate URLs for dependant assets.
+
+        Arguments:
+            path {str} -- Path to a Vite asset.
+            config_key {str} -- Key of the configuration to use.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
+                manifest (only in production).
+
+        Returns:
+            str -- The URL of this asset.
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return DjangoViteAssetLoader._generate_vite_server_url(path, config)
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        manifest_entry = manifest[path]
+
+        return DjangoViteAssetLoader._generate_production_server_url(
+            manifest_entry.file, config.static_url_prefix
+        )
+
+    def generate_vite_legacy_polyfills(
+        self,
+        config_key: str,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        """
+        Generates a <script> tag to the polyfills
+        generated by '@vitejs/plugin-legacy' if used.
+        This tag must be included at end of the <body> before
+        including other legacy scripts.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If polyfills path not found inside
+                the 'manifest.json' (only in production).
+
+        Returns:
+            str -- The script tag to the polyfills.
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return ""
+
+        manifest = self._get_manifest(config_key)
+        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
+
+        for path, manifest_entry in manifest.items():
+            if config.legacy_polyfills_motif in path:
+                return DjangoViteAssetLoader._generate_script_tag(
+                    DjangoViteAssetLoader._generate_production_server_url(
+                        manifest_entry.file, config.static_url_prefix
+                    ),
+                    attrs=scripts_attrs,
+                )
+
+        raise DjangoViteAssetNotFoundError(
+            f"Vite legacy polyfills not found in manifest "
+            f"at {config.get_computed_manifest_path()}"
+        )
+
+    def generate_vite_legacy_asset(
+        self,
+        path: str,
+        config_key: str,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        """
+        Generates a <script> tag for legacy assets JS/TS
+        generated by '@vitejs/plugin-legacy'
+        (in production only, in development do nothing).
+
+        Arguments:
+            path {str} -- Path to a Vite asset to include
+                (must contains '-legacy' in its name).
+            config_key {str} -- Key of the configuration to use.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+
+        Raises:
+            DjangoViteAssetNotFoundError: If cannot find the asset path in the
+                manifest (only in production).
+
+        Returns:
+            str -- The script tag of this legacy asset .
+        """
+
+        config = self._get_config(config_key)
+
+        if config.dev_mode:
+            return ""
+
+        manifest = self._get_manifest(config_key)
+
+        if path not in manifest:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest "
+                f"at {config.get_computed_manifest_path()}"
+            )
+
+        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
+
+        url = DjangoViteAssetLoader._generate_production_server_url(
+            manifest[path].file, config.static_url_prefix
+        )
+        return DjangoViteAssetLoader._generate_script_tag(
+            url,
+            attrs=scripts_attrs,
+        )
+
+    def _get_config(self, config_key: str) -> Type[DjangoViteConfig]:
+        """
+        Get configuration object registered with the key passed in
+        parameter.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to retrieve.
+
+        Raises:
+            RuntimeError: If no configuration exists for this key.
+
+        Returns:
+            Type[DjangoViteConfig] -- The configuration.
+        """
+
+        if config_key not in self._configs:
+            raise RuntimeError(f'Cannot find "{config_key}" configuration')
+
+        return self._configs[config_key]
+
+    def _parse_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
+        """
+        Read and parse the Vite manifest file.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Raises:
+            DjangoViteManifestError: if cannot load the file or JSON in file is
+            malformed.
+        """
+
+        config = self._get_config(config_key)
+
+        try:
+            with open(config.get_computed_manifest_path(), "r") as manifest_file:
+                manifest_content = manifest_file.read()
+                manifest_json = json.loads(manifest_content)
+
+                return {k: DjangoViteManifest(**v) for k, v in manifest_json.items()}
+
+        except Exception as error:
+            raise DjangoViteManifestError(
+                f"Cannot read Vite manifest file at "
+                f"{config.get_computed_manifest_path()} : {str(error)}"
+            ) from error
+
+    def _get_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
+        """
+        Load if needed and parse the "manifest.json" of the specified
+        configuration.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            Dict[str, Type[DjangoViteManifest]] -- Parsed content of
+                the "manifest.json"
+        """
+
+        if config_key not in self._manifests:
+            self._manifests[config_key] = self._parse_manifest(config_key)
+
+        return self._manifests[config_key]
+
+    @classmethod
+    def instance(cls):
+        """
+        Singleton.
+        Uses singleton to keep parsed manifests in memory after
+        the first time they are loaded.
+
+        Returns:
+            DjangoViteAssetLoader -- only instance of the class.
+        """
+
+        if cls._instance is None:
+            cls._instance = cls.__new__(cls)
+            cls._instance._configs = {}
+            cls._instance._manifests = {}
+
+            if hasattr(settings, "DJANGO_VITE"):
+                config = getattr(settings, "DJANGO_VITE")
+
+                for config_key, config_values in config.items():
+                    if isinstance(config_values, DjangoViteConfig):
+                        cls._instance._configs[config_key] = config_values
+                    elif isinstance(config_values, dict):
+                        cls._instance._configs[config_key] = DjangoViteConfig(
+                            **config_values
+                        )
+                    else:
+                        raise RuntimeError(
+                            f"Cannot read configuration for key '{config_key}'"
+                        )
+            else:
+                # Warning : This branch will be remove in further
+                # releases. Please use new way of handling configuration.
+
+                _config_keys = {
+                    "DJANGO_VITE_DEV_MODE": "dev_mode",
+                    "DJANGO_VITE_DEV_SERVER_PROTOCOL": "dev_server_protocol",
+                    "DJANGO_VITE_DEV_SERVER_HOST": "dev_server_host",
+                    "DJANGO_VITE_DEV_SERVER_PORT": "dev_server_port",
+                    "DJANGO_VITE_WS_CLIENT_URL": "ws_client_url",
+                    "DJANGO_VITE_ASSETS_PATH": "assets_path",
+                    "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
+                    "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
+                    "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
+                    "DJANGO_VITE_REACT_REFRESH_URL": "react_refresh_url",
+                }
+
+                config = {
+                    _config_keys[setting_key]: getattr(settings, setting_key)
+                    for setting_key in dir(settings)
+                    if setting_key in _config_keys.keys()
+                }
+
+                cls._instance._configs[DEFAULT_CONFIG_KEY] = DjangoViteConfig(**config)
+
+                # Manifest is only used in production.
+                if not cls._instance._configs[DEFAULT_CONFIG_KEY].dev_mode:
+                    cls._instance._parse_manifest(DEFAULT_CONFIG_KEY)
+
+        return cls._instance
+
+    def generate_vite_ws_client(
+        self, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+    ) -> str:
+        """
+        Generates the script tag for the Vite WS client for HMR.
+        Only used in development, in production this method returns
+        an empty string.
+
+        Arguments:
+            config_key {str} -- Key of the configuration to use.
+
+        Returns:
+            str -- The script tag or an empty string.
+
+        Keyword Arguments:
+            **kwargs {Dict[str, str]} -- Adds new attributes to generated
+                script tags.
+        """
+
+        config = self._get_config(config_key)
+
+        if not config.dev_mode:
+            return ""
+
+        return self._generate_script_tag(
+            self._generate_vite_server_url(config.ws_client_url, config),
+            {"type": "module", **kwargs},
+        )
+
+    @staticmethod
+    def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
+        """
+        Generates an HTML script tag.
+
+        Arguments:
+            src {str} -- Source of the script.
+
+        Keyword Arguments:
+            attrs {Dict[str, str]} -- List of custom attributes
+                for the tag.
+
+        Returns:
+            str -- The script tag.
+        """
+
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
+
+        return f'<script {attrs_str} src="{src}"></script>'
+
+    @staticmethod
+    def _generate_stylesheet_tag(href: str) -> str:
+        """
+        Generates an HTML <link> stylesheet tag for CSS.
+
+        Arguments:
+            href {str} -- CSS file URL.
+
+        Returns:
+            str -- CSS link tag.
+        """
+
+        return f'<link rel="stylesheet" href="{href}" />'
+
+    def _generate_stylesheet_preload_tag(href: str) -> str:
+        """
+        Generates an HTML <link> preload tag for CSS.
+
+        Arguments:
+            href {str} -- CSS file URL.
+
+        Returns:
+            str -- CSS link tag.
+        """
+
+        return f'<link rel="preload" href="{href}" as="style" />'
+
+    @staticmethod
+    def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
+        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
+
+        return f'<link href="{href}" {attrs_str} />'
+
+    @staticmethod
+    def _generate_vite_server_url(
+        path: str,
+        config: Type[DjangoViteConfig],
+    ) -> str:
+        """
+        Generates an URL to and asset served by the Vite development server.
+
+        Keyword Arguments:
+            path {str} -- Path to the asset.
+            config {Type[DjangoViteConfig]} -- Config object to use.
+
+        Returns:
+            str -- Full URL to the asset.
+        """
+
+        return urljoin(
+            f"{config.dev_server_protocol}://"
+            f"{config.dev_server_host}:{config.dev_server_port}",
+            urljoin(config.static_url, path),
+        )
+
+    def generate_vite_react_refresh_url(
+        self, config_key: str = DEFAULT_CONFIG_KEY
+    ) -> str:
+        """
+        Generates the script for the Vite React Refresh for HMR.
+        Only used in development, in production this method returns
+        an empty string.
+
+        Returns:
+            str -- The script or an empty string.
+            config_key {str} -- Key of the configuration to use.
+        """
+        config = self._get_config(config_key)
+
+        if not config.dev_mode:
+            return ""
+
+        return f"""<script type="module">
+            import RefreshRuntime from \
+            '{self._generate_vite_server_url(config.react_refresh_url, config)}'
+            RefreshRuntime.injectIntoGlobalHook(window)
+            window.$RefreshReg$ = () => {{}}
+            window.$RefreshSig$ = () => (type) => type
+            window.__vite_plugin_react_preamble_installed__ = true
+        </script>"""
+
+    @staticmethod
+    def _generate_production_server_url(path: str, static_url_prefix="") -> str:
+        """
+        Generates an URL to an asset served during production.
+
+        Keyword Arguments:
+            path {str} -- Path to the asset.
+
+        Returns:
+            str -- Full URL to the asset.
+        """
+
+        production_server_url = path
+        if prefix := static_url_prefix:
+            if not static_url_prefix.endswith("/"):
+                prefix += "/"
+            production_server_url = urljoin(prefix, path)
+
+        if apps.is_installed("django.contrib.staticfiles"):
+            from django.contrib.staticfiles.storage import staticfiles_storage
+
+            return staticfiles_storage.url(production_server_url)
+
+        return production_server_url
+
+
+@register.simple_tag
+@mark_safe
+def vite_hmr_client(
+    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+) -> str:
+    """
+    Generates the script tag for the Vite WS client for HMR.
+    Only used in development, in production this method returns
+    an empty string.
+
+    Arguments:
+        config {str} -- Configuration to use.
+
+    Returns:
+        str -- The script tag or an empty string.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+    """
+
+    return DjangoViteAssetLoader.instance().generate_vite_ws_client(
+        config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_asset(
+    path: str,
+    config_key: str = DEFAULT_CONFIG_KEY,
+    **kwargs: Dict[str, str],
+) -> str:
+    """
+    Generates a <script> tag for this JS/TS asset, a <link> tag for
+    all of its CSS dependencies, and a <link rel="modulepreload">
+    for all js dependencies, as listed in the manifest file
+    In development Vite loads all by itself.
+
+    Arguments:
+        path {str} -- Path to a Vite JS/TS asset to include.
+        config {str} -- Configuration to use.
+
+    Returns:
+        str -- All tags to import this file in your HTML page.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If cannot find the file path in the
+            manifest (only in production).
+
+    Returns:
+        str -- The <script> tag and all <link> tags to import this
+            asset in your page.
+    """
+
+    assert path is not None
+    assert config_key is not None
+
+    return DjangoViteAssetLoader.instance().generate_vite_asset(
+        path, config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_preload_asset(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
+    """
+    Generates preloadmodule tag for this JS/TS asset and preloads
+    all of its CSS and JS dependencies by reading the manifest
+    file (for production only).
+    In development does nothing.
+
+    Arguments:
+        path {str} -- Path to a Vite JS/TS asset to include.
+
+    Returns:
+        str -- All tags to import this file in your HTML page.
+
+    Raises:
+        RuntimeError: If cannot find the file path in the
+            manifest (only in production).
+
+    """
+    assert path is not None
+
+    return DjangoViteAssetLoader.instance().preload_vite_asset(path, config_key)
+
+
+@register.simple_tag
+def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
+    """
+    Generates only the URL of an asset managed by ViteJS.
+    Warning, this function does not generate URLs for dependant assets.
+
+    Arguments:
+        path {str} -- Path to a Vite asset.
+        config {str} -- Configuration to use.
+
+    Raises:
+        RuntimeError: If cannot find the asset path in the
+            manifest (only in production).
+
+    Returns:
+        str -- The URL of this asset.
+    """
+    assert path is not None
+    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, config_key)
+
+
+@register.simple_tag
+@mark_safe
+def vite_legacy_polyfills(
+    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
+) -> str:
+    """
+    Generates a <script> tag to the polyfills generated
+    by '@vitejs/plugin-legacy' if used.
+    This tag must be included at end of the <body> before including
+    other legacy scripts.
+
+    Arguments:
+        config_key {str} -- Configuration to use.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If polyfills path not found inside
+            the 'manifest.json' (only in production).
+
+    Returns:
+        str -- The script tag to the polyfills.
+    """
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
+        config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_legacy_asset(
+    path: str,
+    config_key: str = DEFAULT_CONFIG_KEY,
+    **kwargs: Dict[str, str],
+) -> str:
+    """
+    Generates a <script> tag for legacy assets JS/TS
+    generated by '@vitejs/plugin-legacy'
+    (in production only, in development do nothing).
+
+    Arguments:
+        path {str} -- Path to a Vite asset to include
+            (must contains '-legacy' in its name).
+        config_key {str} -- Configuration to use.
+
+    Keyword Arguments:
+        **kwargs {Dict[str, str]} -- Adds new attributes to generated
+            script tags.
+
+    Raises:
+        RuntimeError: If cannot find the asset path in
+            the manifest (only in production).
+
+    Returns:
+        str -- The script tag of this legacy asset.
+    """
+
+    assert path is not None
+
+    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
+        path, config_key, **kwargs
+    )
+
+
+@register.simple_tag
+@mark_safe
+def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
+    """
+    Generates the script for the Vite React Refresh for HMR.
+    Only used in development, in production this method returns
+    an empty string.
+
+    Returns:
+        str -- The script or an empty string.
+    """
+    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(config_key)

From 1ecbed86ace75c69c791d5371d0301c280384290 Mon Sep 17 00:00:00 2001
From: Niicck <niicck@users.noreply.github.com>
Date: Mon, 16 Oct 2023 15:15:46 -0700
Subject: [PATCH 29/29] enable multi-app configs with DjangoViteAppClient

---
 django_vite/__init__.py                 |    4 +
 django_vite/exceptions.py               |    6 +
 django_vite/loader.py                   | 1032 +++++++++--------------
 django_vite/tag_generator.py            |  915 +-------------------
 django_vite/templatetags/django_vite.py |  804 +-----------------
 pyproject.toml                          |    1 +
 tests/conftest.py                       |   16 +-
 tests/tests/test_asset_loader.py        |   15 +-
 8 files changed, 469 insertions(+), 2324 deletions(-)

diff --git a/django_vite/__init__.py b/django_vite/__init__.py
index e69de29..09f8c32 100644
--- a/django_vite/__init__.py
+++ b/django_vite/__init__.py
@@ -0,0 +1,4 @@
+from .loader import DjangoViteConfig
+
+
+__all__ = ["DjangoViteConfig"]
diff --git a/django_vite/exceptions.py b/django_vite/exceptions.py
index 1030ca8..ec13a28 100644
--- a/django_vite/exceptions.py
+++ b/django_vite/exceptions.py
@@ -8,3 +8,9 @@ class DjangoViteAssetNotFoundError(RuntimeError):
     """Vite Asset could not be found."""
 
     pass
+
+
+class DjangoViteConfigNotFoundError(RuntimeError):
+    """DjangoViteConfig not found in DJANGO_VITE settings."""
+
+    pass
diff --git a/django_vite/loader.py b/django_vite/loader.py
index c1e9956..75bfeb1 100644
--- a/django_vite/loader.py
+++ b/django_vite/loader.py
@@ -1,40 +1,27 @@
 import json
 from pathlib import Path
-from typing import Dict, List, Callable, NamedTuple, Optional, Type, Union
+from typing import Dict, List, Callable, NamedTuple, Optional, Union, Tuple
 from urllib.parse import urljoin
+import warnings
 
-from django import template
 from django.apps import apps
 from django.conf import settings
-from django.utils.safestring import mark_safe
 
-from django_vite.exceptions import DjangoViteAssetNotFoundError, DjangoViteManifestError
+from django_vite.exceptions import (
+    DjangoViteManifestError,
+    DjangoViteAssetNotFoundError,
+    DjangoViteConfigNotFoundError,
+)
+from django_vite.tag_generator import Tag, TagGenerator
 
-DEFAULT_CONFIG_KEY = "default"
-
-register = template.Library()
-
-
-class DjangoViteManifest(NamedTuple):
-    """
-    Represent an entry for a file inside the "manifest.json".
-    """
-
-    file: str
-    src: Optional[str] = None
-    isEntry: Optional[bool] = False
-    css: Optional[List[str]] = []
-    imports: Optional[List[str]] = []
+DEFAULT_APP_NAME = "default"
 
 
 class DjangoViteConfig(NamedTuple):
     """
-    Represent the Django Vite configuration structure.
+    The arguments for a DjangoViteAppClient.
     """
 
-    # Location of Vite compiled assets (only used in Vite production mode).
-    assets_path: Union[Path, str]
-
     # If using in development or production mode.
     dev_mode: bool = False
 
@@ -47,75 +34,201 @@ class DjangoViteConfig(NamedTuple):
     # Default Vite server port.
     dev_server_port: int = 3000
 
-    # Default Vite server path to HMR script.
-    ws_client_url: str = "@vite/client"
-
     # Prefix for STATIC_URL.
     static_url_prefix: str = ""
 
+    # Path to your manifest file generated by Vite.
+    manifest_path: Optional[Union[Path, str]] = None
+
     # Motif in the "manifest.json" to find the polyfills generated by Vite.
     legacy_polyfills_motif: str = "legacy-polyfills"
 
-    # Path to your manifest file generated by Vite.
-    manifest_path: Union[Path, str] = ""
+    # Default Vite server path to HMR script.
+    ws_client_url: str = "@vite/client"
 
     # Default Vite server path to React RefreshRuntime for @vitejs/plugin-react.
     react_refresh_url: str = "@react-refresh"
 
-    @property
-    def static_url(self) -> str:
-        url = urljoin(settings.STATIC_URL, self.static_url_prefix)
-        if not url.endswith("/"):
-            url += "/"
-        return url
 
-    @property
-    def static_root(self) -> Path:
+class ManifestEntry(NamedTuple):
+    """
+    Represent an entry for a file inside the "manifest.json".
+    """
+
+    file: str
+    src: Optional[str] = None
+    isEntry: Optional[bool] = False
+    css: Optional[List[str]] = []
+    imports: Optional[List[str]] = []
+
+
+class ManifestClient:
+    """
+    A client for accessing entries in the compiled vite config's "manifest.json".
+    Only parses manifest.json if dev_mode=False.
+
+    Public Methods:
+        get(path: str) -- return the ManifestEntry for the given path.
+    """
+
+    def __init__(self, config: DjangoViteConfig):
+        self._config = config
+
+        self.dev_mode = config.dev_mode
+        self.manifest_path = self._clean_manifest_path()
+        self.legacy_polyfills_motif = config.legacy_polyfills_motif
+
+        self._entries: Dict[str, ManifestEntry] = {}
+        self.legacy_polyfills_entry: Optional[ManifestEntry] = None
+
+        if not self.dev_mode:
+            self._entries, self.legacy_polyfills_entry = self._parse_manifest()
+
+    def _clean_manifest_path(self) -> Path:
         """
-        Compute the static root URL of assets.
+        Get the manifest_path from the config.
+        If it wasn't provided, set the default location to
+        STATIC_ROOT / static_url_prefix / "manifest.json".
 
         Returns:
-            Union[Path, str] -- Static root URL.
+            Path -- the path to the vite config's manifest.json
+        """
+        initial_manifest_path = self._config.manifest_path
+        if not initial_manifest_path:
+            return (
+                Path(settings.STATIC_ROOT)
+                / self._config.static_url_prefix
+                / "manifest.json"
+            )
+        elif not isinstance(initial_manifest_path, Path):
+            return Path(initial_manifest_path)
+        else:
+            return initial_manifest_path
+
+    def _parse_manifest(self) -> Tuple[Dict[str, ManifestEntry], ManifestEntry]:
         """
+        Read and parse the Vite manifest file.
 
-        return (
-            Path(self.assets_path)
-            if self.dev_mode
-            else Path(settings.STATIC_ROOT) / self.static_url_prefix
-        )
+        Returns:
+            entries {Dict[str, ManifestEntry]} -- All entries within the manifest
 
-    def get_computed_manifest_path(self) -> Union[Path, str]:
+            legacy_polyfills_entry {ManifestEntry} -- The manifest entry for legacy
+                polyfills, if it exists within the manifest.json
+
+        Raises:
+            DjangoViteManifestError: if cannot load the file or JSON in file is
+                malformed.
         """
-        Compute the path to the "manifest.json".
+        entries: Dict[str, ManifestEntry] = {}
+        legacy_polyfills_entry: Optional[ManifestEntry] = None
+
+        try:
+            with open(self.manifest_path, "r") as manifest_file:
+                manifest_content = manifest_file.read()
+                manifest_json = json.loads(manifest_content)
+
+                for path, manifest_entry_data in manifest_json.items():
+                    manifest_entry = ManifestEntry(**manifest_entry_data)
+                    entries[path] = manifest_entry
+                    if self.legacy_polyfills_motif in path:
+                        legacy_polyfills_entry = manifest_entry
+
+                return (entries, legacy_polyfills_entry)
+
+        except Exception as error:
+            raise DjangoViteManifestError(
+                f"Cannot read Vite manifest file at "
+                f"{self.manifest_path} : {str(error)}"
+            ) from error
+
+    def get(self, path: str) -> ManifestEntry:
+        """
+        Gets the manifest_entry for given path.
 
         Returns:
-            Union[Path, str] -- Path to the "manifest.json".
+            ManifestEntry -- the ManifestEntry for your path
+
+        Raises:
+            DjangoViteAssetNotFoundError: if cannot find the file path in the manifest
+                or if manifest was never parsed due to dev_mode=True.
         """
+        if path not in self._entries:
+            raise DjangoViteAssetNotFoundError(
+                f"Cannot find {path} in Vite manifest " f"at {self.manifest_path}"
+            )
 
-        return (
-            self.manifest_path
-            if self.manifest_path
-            else self.static_root / "manifest.json"
-        )
+        return self._entries[path]
 
 
-class DjangoViteAssetLoader:
+class DjangoViteAppClient:
     """
-    Class handling Vite asset loading.
+    An interface for generating assets and urls from one vite app.
+    DjangoViteConfig provides the arguments for the client.
     """
 
-    _instance = None
+    def __init__(self, config: DjangoViteConfig) -> None:
+        self._config = config
 
-    _configs = Dict[str, Type[DjangoViteConfig]]
-    _manifests: Dict[str, Type[DjangoViteManifest]]
+        self.dev_mode = config.dev_mode
+        self.dev_server_protocol = config.dev_server_protocol
+        self.dev_server_host = config.dev_server_host
+        self.dev_server_port = config.dev_server_port
+        self.static_url_prefix = config.static_url_prefix
+        self.ws_client_url = config.ws_client_url
+        self.react_refresh_url = config.react_refresh_url
 
-    def __init__(self) -> None:
-        raise RuntimeError("Use the instance() method instead.")
+        self.manifest = ManifestClient(config)
+
+    def _get_dev_server_url(
+        self,
+        path: str,
+    ) -> str:
+        """
+        Generates an URL to an asset served by the Vite development server.
+
+        Keyword Arguments:
+            path {str} -- Path to the asset.
+
+        Returns:
+            str -- Full URL to the asset.
+        """
+        static_url_base = urljoin(settings.STATIC_URL, self.static_url_prefix)
+        if not static_url_base.endswith("/"):
+            static_url_base += "/"
+
+        return urljoin(
+            f"{self.dev_server_protocol}://"
+            f"{self.dev_server_host}:{self.dev_server_port}",
+            urljoin(static_url_base, path),
+        )
+
+    def _get_production_server_url(self, path: str) -> str:
+        """
+        Generates an URL to an asset served during production.
+
+        Keyword Arguments:
+            path {str} -- Path to the asset.
+
+        Returns:
+            str -- Full URL to the asset.
+        """
+
+        production_server_url = path
+        if prefix := self.static_url_prefix:
+            if not prefix.endswith("/"):
+                prefix += "/"
+            production_server_url = urljoin(prefix, path)
+
+        if apps.is_installed("django.contrib.staticfiles"):
+            from django.contrib.staticfiles.storage import staticfiles_storage
+
+            return staticfiles_storage.url(production_server_url)
+
+        return production_server_url
 
     def generate_vite_asset(
         self,
         path: str,
-        config_key: str,
         **kwargs: Dict[str, str],
     ) -> str:
         """
@@ -127,7 +240,6 @@ def generate_vite_asset(
 
         Arguments:
             path {str} -- Path to a Vite JS/TS asset to include.
-            config_key {str} -- Key of the configuration to use.
 
         Returns:
             str -- All tags to import this file in your HTML page.
@@ -136,44 +248,30 @@ def generate_vite_asset(
             **kwargs {Dict[str, str]} -- Adds new attributes to generated
                 script tags.
 
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the file path in the
-                manifest (only in production).
-
         Returns:
             str -- The <script> tag and all <link> tags to import
                 this asset in your page.
         """
 
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_vite_server_url(path, config),
-                {"type": "module", **kwargs},
-            )
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
+        if self.dev_mode:
+            url = self._get_dev_server_url(path)
+            return TagGenerator.script(
+                url,
+                attrs={"type": "module", **kwargs},
             )
 
-        manifest_entry = manifest[path]
-        tags = []
+        tags: List[Tag] = []
+        manifest_entry = self.manifest.get(path)
         scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
 
         # Add dependent CSS
-        tags.extend(self._load_css_files_of_asset(path, config_key))
+        tags.extend(self._load_css_files_of_asset(path))
 
         # Add the script by itself
+        url = self._get_production_server_url(manifest_entry.file)
         tags.append(
-            DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_production_server_url(
-                    manifest_entry.file, config.static_url_prefix
-                ),
+            TagGenerator.script(
+                url,
                 attrs=scripts_attrs,
             )
         )
@@ -187,13 +285,11 @@ def generate_vite_asset(
         }
 
         for dep in manifest_entry.imports:
-            dep_manifest_entry = manifest[dep]
+            dep_manifest_entry = self.manifest.get(dep)
             dep_file = dep_manifest_entry.file
-            url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file, config.static_url_prefix
-            )
+            url = self._get_production_server_url(dep_file)
             tags.append(
-                DjangoViteAssetLoader._generate_preload_tag(
+                TagGenerator.preload(
                     url,
                     attrs=preload_attrs,
                 )
@@ -204,45 +300,29 @@ def generate_vite_asset(
     def preload_vite_asset(
         self,
         path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
     ) -> str:
         """
         Generates a <link modulepreload> tag for this JS/TS asset, a
         <link preload> tag for all of its CSS dependencies,
         and a <link modulepreload> for the js dependencies.
         In development this template tag renders nothing,
-        since files aren't compiled yet"
+        since files aren't compiled yet.
 
         Arguments:
             path {str} -- Path to a Vite JS/TS asset to preload.
-            config_key {str} -- Key of the configuration to use.
 
         Returns:
             str -- All tags to preload this file in your HTML page.
 
-        Raises:
-            DjangoViteAssetNotFoundError: if cannot find the file path in the
-                manifest.
-
         Returns:
             str -- all <link> tags to preload
                 this asset.
         """
-        tags = []
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
+        if self.dev_mode:
             return ""
 
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        manifest_entry = manifest[path]
+        tags: List[Tag] = []
+        manifest_entry = self.manifest.get(path)
 
         # Add the script by itself
         script_attrs = {
@@ -253,28 +333,24 @@ def preload_vite_asset(
         }
 
         manifest_file = manifest_entry.file
-        url = DjangoViteAssetLoader._generate_production_server_url(
-            manifest_file, config.static_url_prefix
-        )
+        url = self._get_production_server_url(manifest_file)
         tags.append(
-            DjangoViteAssetLoader._generate_preload_tag(
+            TagGenerator.preload(
                 url,
                 attrs=script_attrs,
             )
         )
 
         # Add dependent CSS
-        tags.extend(self._preload_css_files_of_asset(path, config_key))
+        tags.extend(self._preload_css_files_of_asset(path))
 
         # Preload imports
         for dep in manifest_entry.imports:
-            dep_manifest_entry = manifest[dep]
+            dep_manifest_entry = self.manifest.get(dep)
             dep_file = dep_manifest_entry.file
-            url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file, config.static_url_prefix
-            )
+            url = self._get_production_server_url(dep_file)
             tags.append(
-                DjangoViteAssetLoader._generate_preload_tag(
+                TagGenerator.preload(
                     url,
                     attrs=script_attrs,
                 )
@@ -285,34 +361,27 @@ def preload_vite_asset(
     def _preload_css_files_of_asset(
         self,
         path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
+    ) -> List[Tag]:
         return self._generate_css_files_of_asset(
             path,
-            [],
-            DjangoViteAssetLoader._generate_stylesheet_preload_tag,
-            config_key,
-        )
+            tag_generator=TagGenerator.stylesheet_preload,
+        )[0]
 
     def _load_css_files_of_asset(
         self,
         path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
+    ) -> List[Tag]:
         return self._generate_css_files_of_asset(
             path,
-            [],
-            DjangoViteAssetLoader._generate_stylesheet_tag,
-            config_key,
-        )
+            tag_generator=TagGenerator.stylesheet,
+        )[0]
 
     def _generate_css_files_of_asset(
         self,
         path: str,
-        already_processed: List[str],
-        tag_generator: Callable,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
+        already_processed: Optional[List[str]] = None,
+        tag_generator: Callable[[str], Tag] = TagGenerator.stylesheet,
+    ) -> Tuple[List[Tag], List[str]]:
         """
         Generates all CSS tags for dependencies of an asset.
 
@@ -322,71 +391,50 @@ def _generate_css_files_of_asset(
             already_processed {list} -- List of already processed CSS file.
 
         Returns:
-            list -- List of CSS tags.
+            tags -- List of CSS tags.
+            already_processed -- List of already processed css paths
         """
-
-        tags = []
-        config = self._get_config(config_key)
-        manifest = self._get_manifest(config_key)
-        manifest_entry = manifest[path]
+        already_processed = already_processed or []
+        tags: List[Tag] = []
+        manifest_entry = self.manifest.get(path)
 
         for import_path in manifest_entry.imports:
-            tags.extend(
-                self._generate_css_files_of_asset(
-                    import_path, already_processed, tag_generator, config_key
-                )
+            new_tags, new_already_processed = self._generate_css_files_of_asset(
+                import_path, already_processed, tag_generator
             )
+            tags.extend(new_tags)
+            already_processed.extend(new_already_processed)
 
         for css_path in manifest_entry.css:
             if css_path not in already_processed:
-                url = DjangoViteAssetLoader._generate_production_server_url(
-                    css_path, config.static_url_prefix
-                )
+                url = self._get_production_server_url(css_path)
                 tags.append(tag_generator(url))
 
             already_processed.append(css_path)
 
-        return tags
+        return (tags, already_processed)
 
-    def generate_vite_asset_url(self, path: str, config_key: str) -> str:
+    def generate_vite_asset_url(self, path: str) -> str:
         """
         Generates only the URL of an asset managed by ViteJS.
         Warning, this function does not generate URLs for dependant assets.
 
         Arguments:
             path {str} -- Path to a Vite asset.
-            config_key {str} -- Key of the configuration to use.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the asset path in the
-                manifest (only in production).
 
         Returns:
             str -- The URL of this asset.
         """
 
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return DjangoViteAssetLoader._generate_vite_server_url(path, config)
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
+        if self.dev_mode:
+            return self._get_dev_server_url(path)
 
-        manifest_entry = manifest[path]
+        manifest_entry = self.manifest.get(path)
 
-        return DjangoViteAssetLoader._generate_production_server_url(
-            manifest_entry.file, config.static_url_prefix
-        )
+        return self._get_production_server_url(manifest_entry.file)
 
     def generate_vite_legacy_polyfills(
         self,
-        config_key: str,
         **kwargs: Dict[str, str],
     ) -> str:
         """
@@ -395,47 +443,40 @@ def generate_vite_legacy_polyfills(
         This tag must be included at end of the <body> before
         including other legacy scripts.
 
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
         Keyword Arguments:
             **kwargs {Dict[str, str]} -- Adds new attributes to generated
                 script tags.
 
         Raises:
             DjangoViteAssetNotFoundError: If polyfills path not found inside
-                the 'manifest.json' (only in production).
+                the 'manifest.json'.
 
         Returns:
             str -- The script tag to the polyfills.
         """
 
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
+        if self.dev_mode:
             return ""
 
-        manifest = self._get_manifest(config_key)
-        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
+        polyfills_manifest_entry = self.manifest.legacy_polyfills_entry
 
-        for path, manifest_entry in manifest.items():
-            if config.legacy_polyfills_motif in path:
-                return DjangoViteAssetLoader._generate_script_tag(
-                    DjangoViteAssetLoader._generate_production_server_url(
-                        manifest_entry.file, config.static_url_prefix
-                    ),
-                    attrs=scripts_attrs,
-                )
+        if not polyfills_manifest_entry:
+            raise DjangoViteAssetNotFoundError(
+                f"Vite legacy polyfills not found in manifest "
+                f"at {self.manifest.manifest_path}"
+            )
+
+        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
+        url = self._get_production_server_url(polyfills_manifest_entry.file)
 
-        raise DjangoViteAssetNotFoundError(
-            f"Vite legacy polyfills not found in manifest "
-            f"at {config.get_computed_manifest_path()}"
+        return TagGenerator.script(
+            url,
+            attrs=scripts_attrs,
         )
 
     def generate_vite_legacy_asset(
         self,
         path: str,
-        config_key: str,
         **kwargs: Dict[str, str],
     ) -> str:
         """
@@ -446,7 +487,6 @@ def generate_vite_legacy_asset(
         Arguments:
             path {str} -- Path to a Vite asset to include
                 (must contains '-legacy' in its name).
-            config_key {str} -- Key of the configuration to use.
 
         Keyword Arguments:
             **kwargs {Dict[str, str]} -- Adds new attributes to generated
@@ -460,166 +500,24 @@ def generate_vite_legacy_asset(
             str -- The script tag of this legacy asset .
         """
 
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
+        if self.dev_mode:
             return ""
 
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
+        manifest_entry = self.manifest.get(path)
         scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
+        url = self._get_production_server_url(manifest_entry.file)
 
-        url = DjangoViteAssetLoader._generate_production_server_url(
-            manifest[path].file, config.static_url_prefix
-        )
-        return DjangoViteAssetLoader._generate_script_tag(
+        return TagGenerator.script(
             url,
             attrs=scripts_attrs,
         )
 
-    def _get_config(self, config_key: str) -> Type[DjangoViteConfig]:
-        """
-        Get configuration object registered with the key passed in
-        parameter.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to retrieve.
-
-        Raises:
-            RuntimeError: If no configuration exists for this key.
-
-        Returns:
-            Type[DjangoViteConfig] -- The configuration.
-        """
-
-        if config_key not in self._configs:
-            raise RuntimeError(f'Cannot find "{config_key}" configuration')
-
-        return self._configs[config_key]
-
-    def _parse_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
-        """
-        Read and parse the Vite manifest file.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Raises:
-            DjangoViteManifestError: if cannot load the file or JSON in file is
-            malformed.
-        """
-
-        config = self._get_config(config_key)
-
-        try:
-            with open(config.get_computed_manifest_path(), "r") as manifest_file:
-                manifest_content = manifest_file.read()
-                manifest_json = json.loads(manifest_content)
-
-                return {k: DjangoViteManifest(**v) for k, v in manifest_json.items()}
-
-        except Exception as error:
-            raise DjangoViteManifestError(
-                f"Cannot read Vite manifest file at "
-                f"{config.get_computed_manifest_path()} : {str(error)}"
-            ) from error
-
-    def _get_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
-        """
-        Load if needed and parse the "manifest.json" of the specified
-        configuration.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            Dict[str, Type[DjangoViteManifest]] -- Parsed content of
-                the "manifest.json"
-        """
-
-        if config_key not in self._manifests:
-            self._manifests[config_key] = self._parse_manifest(config_key)
-
-        return self._manifests[config_key]
-
-    @classmethod
-    def instance(cls):
-        """
-        Singleton.
-        Uses singleton to keep parsed manifests in memory after
-        the first time they are loaded.
-
-        Returns:
-            DjangoViteAssetLoader -- only instance of the class.
-        """
-
-        if cls._instance is None:
-            cls._instance = cls.__new__(cls)
-            cls._instance._configs = {}
-            cls._instance._manifests = {}
-
-            if hasattr(settings, "DJANGO_VITE"):
-                config = getattr(settings, "DJANGO_VITE")
-
-                for config_key, config_values in config.items():
-                    if isinstance(config_values, DjangoViteConfig):
-                        cls._instance._configs[config_key] = config_values
-                    elif isinstance(config_values, dict):
-                        cls._instance._configs[config_key] = DjangoViteConfig(
-                            **config_values
-                        )
-                    else:
-                        raise RuntimeError(
-                            f"Cannot read configuration for key '{config_key}'"
-                        )
-            else:
-                # Warning : This branch will be remove in further
-                # releases. Please use new way of handling configuration.
-
-                _config_keys = {
-                    "DJANGO_VITE_DEV_MODE": "dev_mode",
-                    "DJANGO_VITE_DEV_SERVER_PROTOCOL": "dev_server_protocol",
-                    "DJANGO_VITE_DEV_SERVER_HOST": "dev_server_host",
-                    "DJANGO_VITE_DEV_SERVER_PORT": "dev_server_port",
-                    "DJANGO_VITE_WS_CLIENT_URL": "ws_client_url",
-                    "DJANGO_VITE_ASSETS_PATH": "assets_path",
-                    "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
-                    "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
-                    "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
-                    "DJANGO_VITE_REACT_REFRESH_URL": "react_refresh_url",
-                }
-
-                config = {
-                    _config_keys[setting_key]: getattr(settings, setting_key)
-                    for setting_key in dir(settings)
-                    if setting_key in _config_keys.keys()
-                }
-
-                cls._instance._configs[DEFAULT_CONFIG_KEY] = DjangoViteConfig(**config)
-
-                # Manifest is only used in production.
-                if not cls._instance._configs[DEFAULT_CONFIG_KEY].dev_mode:
-                    cls._instance._parse_manifest(DEFAULT_CONFIG_KEY)
-
-        return cls._instance
-
-    def generate_vite_ws_client(
-        self, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-    ) -> str:
+    def generate_vite_ws_client(self, **kwargs: Dict[str, str]) -> str:
         """
         Generates the script tag for the Vite WS client for HMR.
         Only used in development, in production this method returns
         an empty string.
 
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
         Returns:
             str -- The script tag or an empty string.
 
@@ -628,94 +526,17 @@ def generate_vite_ws_client(
                 script tags.
         """
 
-        config = self._get_config(config_key)
-
-        if not config.dev_mode:
+        if not self.dev_mode:
             return ""
 
-        return self._generate_script_tag(
-            self._generate_vite_server_url(config.ws_client_url, config),
-            {"type": "module", **kwargs},
-        )
-
-    @staticmethod
-    def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
-        """
-        Generates an HTML script tag.
-
-        Arguments:
-            src {str} -- Source of the script.
-
-        Keyword Arguments:
-            attrs {Dict[str, str]} -- List of custom attributes
-                for the tag.
-
-        Returns:
-            str -- The script tag.
-        """
-
-        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
-
-        return f'<script {attrs_str} src="{src}"></script>'
-
-    @staticmethod
-    def _generate_stylesheet_tag(href: str) -> str:
-        """
-        Generates an HTML <link> stylesheet tag for CSS.
-
-        Arguments:
-            href {str} -- CSS file URL.
-
-        Returns:
-            str -- CSS link tag.
-        """
-
-        return f'<link rel="stylesheet" href="{href}" />'
+        url = self._get_dev_server_url(self.ws_client_url)
 
-    def _generate_stylesheet_preload_tag(href: str) -> str:
-        """
-        Generates an HTML <link> preload tag for CSS.
-
-        Arguments:
-            href {str} -- CSS file URL.
-
-        Returns:
-            str -- CSS link tag.
-        """
-
-        return f'<link rel="preload" href="{href}" as="style" />'
-
-    @staticmethod
-    def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
-        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
-
-        return f'<link href="{href}" {attrs_str} />'
-
-    @staticmethod
-    def _generate_vite_server_url(
-        path: str,
-        config: Type[DjangoViteConfig],
-    ) -> str:
-        """
-        Generates an URL to and asset served by the Vite development server.
-
-        Keyword Arguments:
-            path {str} -- Path to the asset.
-            config {Type[DjangoViteConfig]} -- Config object to use.
-
-        Returns:
-            str -- Full URL to the asset.
-        """
-
-        return urljoin(
-            f"{config.dev_server_protocol}://"
-            f"{config.dev_server_host}:{config.dev_server_port}",
-            urljoin(config.static_url, path),
+        return TagGenerator.script(
+            url,
+            attrs={"type": "module", **kwargs},
         )
 
-    def generate_vite_react_refresh_url(
-        self, config_key: str = DEFAULT_CONFIG_KEY
-    ) -> str:
+    def generate_vite_react_refresh_url(self) -> str:
         """
         Generates the script for the Vite React Refresh for HMR.
         Only used in development, in production this method returns
@@ -725,234 +546,217 @@ def generate_vite_react_refresh_url(
             str -- The script or an empty string.
             config_key {str} -- Key of the configuration to use.
         """
-        config = self._get_config(config_key)
 
-        if not config.dev_mode:
+        if not self.dev_mode:
             return ""
 
+        url = self._get_dev_server_url(self.react_refresh_url)
+
         return f"""<script type="module">
-            import RefreshRuntime from \
-            '{self._generate_vite_server_url(config.react_refresh_url, config)}'
+            import RefreshRuntime from '{url}'
             RefreshRuntime.injectIntoGlobalHook(window)
             window.$RefreshReg$ = () => {{}}
             window.$RefreshSig$ = () => (type) => type
             window.__vite_plugin_react_preamble_installed__ = true
         </script>"""
 
-    @staticmethod
-    def _generate_production_server_url(path: str, static_url_prefix="") -> str:
-        """
-        Generates an URL to an asset served during production.
-
-        Keyword Arguments:
-            path {str} -- Path to the asset.
-
-        Returns:
-            str -- Full URL to the asset.
-        """
-
-        production_server_url = path
-        if prefix := static_url_prefix:
-            if not static_url_prefix.endswith("/"):
-                prefix += "/"
-            production_server_url = urljoin(prefix, path)
-
-        if apps.is_installed("django.contrib.staticfiles"):
-            from django.contrib.staticfiles.storage import staticfiles_storage
-
-            return staticfiles_storage.url(production_server_url)
-
-        return production_server_url
-
 
-@register.simple_tag
-@mark_safe
-def vite_hmr_client(
-    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-) -> str:
-    """
-    Generates the script tag for the Vite WS client for HMR.
-    Only used in development, in production this method returns
-    an empty string.
-
-    Arguments:
-        config {str} -- Configuration to use.
-
-    Returns:
-        str -- The script tag or an empty string.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-    """
-
-    return DjangoViteAssetLoader.instance().generate_vite_ws_client(
-        config_key, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_asset(
-    path: str,
-    config_key: str = DEFAULT_CONFIG_KEY,
-    **kwargs: Dict[str, str],
-) -> str:
+class DjangoViteAssetLoader:
     """
-    Generates a <script> tag for this JS/TS asset, a <link> tag for
-    all of its CSS dependencies, and a <link rel="modulepreload">
-    for all js dependencies, as listed in the manifest file
-    In development Vite loads all by itself.
-
-    Arguments:
-        path {str} -- Path to a Vite JS/TS asset to include.
-        config {str} -- Configuration to use.
-
-    Returns:
-        str -- All tags to import this file in your HTML page.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Raises:
-        RuntimeError: If cannot find the file path in the
-            manifest (only in production).
+    Class handling Vite asset loading.
 
-    Returns:
-        str -- The <script> tag and all <link> tags to import this
-            asset in your page.
+    Routes asset and url generation to the proper DjangoViteAppClient.
     """
 
-    assert path is not None
-    assert config_key is not None
-
-    return DjangoViteAssetLoader.instance().generate_vite_asset(
-        path, config_key, **kwargs
-    )
+    _instance = None
+    _apps: Dict[str, DjangoViteAppClient]
+
+    DJANGO_VITE = "DJANGO_VITE"
+
+    LEGACY_DJANGO_VITE_SETTINGS: Dict[str, Optional[str]] = {
+        "DJANGO_VITE_DEV_MODE": "dev_mode",
+        "DJANGO_VITE_DEV_SERVER_PROTOCOL": "dev_server_protocol",
+        "DJANGO_VITE_DEV_SERVER_HOST": "dev_server_host",
+        "DJANGO_VITE_DEV_SERVER_PORT": "dev_server_port",
+        "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
+        "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
+        "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
+        "DJANGO_VITE_WS_CLIENT_URL": "ws_client_url",
+        "DJANGO_VITE_REACT_REFRESH_URL": "react_refresh_url",
+        "DJANGO_VITE_ASSETS_PATH": None,
+    }
 
+    def __init__(self) -> None:
+        raise RuntimeError("Use the instance() method instead.")
 
-@register.simple_tag
-@mark_safe
-def vite_preload_asset(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
-    """
-    Generates preloadmodule tag for this JS/TS asset and preloads
-    all of its CSS and JS dependencies by reading the manifest
-    file (for production only).
-    In development does nothing.
+    @classmethod
+    def instance(cls):
+        """
+        Singleton.
+        Uses singleton to keep parsed manifests in memory after
+        the first time they are loaded.
 
-    Arguments:
-        path {str} -- Path to a Vite JS/TS asset to include.
+        Returns:
+            DjangoViteAssetLoader -- only instance of the class.
+        """
 
-    Returns:
-        str -- All tags to import this file in your HTML page.
+        if cls._instance is None:
+            cls._instance = cls.__new__(cls)
+            cls._instance._apps = {}
 
-    Raises:
-        RuntimeError: If cannot find the file path in the
-            manifest (only in production).
+            cls._apply_django_vite_settings()
+            cls._apply_legacy_django_vite_settings()
+            cls._apply_default_fallback()
 
-    """
-    assert path is not None
+        return cls._instance
 
-    return DjangoViteAssetLoader.instance().preload_vite_asset(path, config_key)
+    @classmethod
+    def _apply_django_vite_settings(cls):
+        """
+        Takes DjangoViteConfigs from the DJANGO_VITE setting, and plugs them into
+        DjangoViteAppClients.
+        """
 
+        django_vite_settings = getattr(settings, cls.DJANGO_VITE, None)
 
-@register.simple_tag
-def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
-    """
-    Generates only the URL of an asset managed by ViteJS.
-    Warning, this function does not generate URLs for dependant assets.
+        if not django_vite_settings:
+            return
 
-    Arguments:
-        path {str} -- Path to a Vite asset.
-        config {str} -- Configuration to use.
+        for app_name, config in django_vite_settings.items():
+            if not isinstance(config, DjangoViteConfig):
+                config = DjangoViteConfig(**config)
+            cls._instance._apps[app_name] = DjangoViteAppClient(config)
 
-    Raises:
-        RuntimeError: If cannot find the asset path in the
-            manifest (only in production).
+    @classmethod
+    def _apply_legacy_django_vite_settings(cls):
+        """
+        If the project hasn't yet migrated to the new way of configuring django-vite,
+        then plug values from the legacy settings into an app named "default".
+        """
+
+        applied_settings = dir(settings)
+        legacy_settings_keys = cls.LEGACY_DJANGO_VITE_SETTINGS.keys()
+        applied_legacy_settings = [
+            key for key in legacy_settings_keys if key in applied_settings
+        ]
+
+        if not applied_legacy_settings:
+            return
+
+        # If there are both new DJANGO_VITE settings as well as legacy settings, then
+        # allow _apply_django_vite_settings to apply only the DJANGO_VITE configs and
+        # ignore the legacy settings.
+        if cls.DJANGO_VITE in applied_settings:
+            warnings.warn(
+                f"You're mixing the new {cls.DJANGO_VITE} setting with these "
+                f"legacy settings: [{applied_legacy_settings.join(',')}]. Those legacy "
+                f"settings will be ignored since you have a {cls.DJANGO_VITE}"
+                " setting configured. Please remove those legacy django-vite settings.",
+                DeprecationWarning,
+            )
+            return
+
+        # Apply legacy settings, with a warning.
+        warnings.warn(
+            f"The settings [{', '.join(applied_legacy_settings)}] will be removed "
+            "in future releases of django-vite. Please switch to defining your "
+            f'settings as {cls.DJANGO_VITE} = {{"default": {{...}},}}.',
+            DeprecationWarning,
+        )
 
-    Returns:
-        str -- The URL of this asset.
-    """
-    assert path is not None
-    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, config_key)
+        legacy_config = {}
+        for legacy_setting in applied_legacy_settings:
+            new_config_name = cls.LEGACY_DJANGO_VITE_SETTINGS[legacy_setting]
+            if new_config_name:
+                legacy_config[new_config_name] = getattr(settings, legacy_setting)
 
+        legacy_config = DjangoViteConfig(**legacy_config)
+        cls._instance._apps[DEFAULT_APP_NAME] = DjangoViteAppClient(legacy_config)
 
-@register.simple_tag
-@mark_safe
-def vite_legacy_polyfills(
-    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-) -> str:
-    """
-    Generates a <script> tag to the polyfills generated
-    by '@vitejs/plugin-legacy' if used.
-    This tag must be included at end of the <body> before including
-    other legacy scripts.
+    @classmethod
+    def _apply_default_fallback(cls):
+        """
+        If no settings at all were provided (and no DjangoViteAppClient were
+        instantiated), we can create a "default" DjangoViteAppClient using the default
+        values of DjangoViteConfig.
+        """
 
-    Arguments:
-        config_key {str} -- Configuration to use.
+        if not cls._instance._apps:
+            default_config = DjangoViteConfig()
+            cls._instance._apps[DEFAULT_APP_NAME] = DjangoViteAppClient(default_config)
 
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
+    def _get_app_client(self, app: str) -> DjangoViteAppClient:
+        """
+        Gets the DjangoViteAppClient for given app.
 
-    Raises:
-        RuntimeError: If polyfills path not found inside
-            the 'manifest.json' (only in production).
+        Returns:
+            DjangoViteAppClient -- the client for your app
 
-    Returns:
-        str -- The script tag to the polyfills.
-    """
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
-        config_key, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_legacy_asset(
-    path: str,
-    config_key: str = DEFAULT_CONFIG_KEY,
-    **kwargs: Dict[str, str],
-) -> str:
-    """
-    Generates a <script> tag for legacy assets JS/TS
-    generated by '@vitejs/plugin-legacy'
-    (in production only, in development do nothing).
+        Raises:
+            DjangoViteConfigNotFoundError: If app was not found in DJANGO_VITE
+                settings.
+        """
 
-    Arguments:
-        path {str} -- Path to a Vite asset to include
-            (must contains '-legacy' in its name).
-        config_key {str} -- Configuration to use.
+        if app not in self._apps:
+            raise DjangoViteConfigNotFoundError(
+                f"Cannot find {app} in {self.DJANGO_VITE} settings."
+            )
 
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
+        return self._apps[app]
 
-    Raises:
-        RuntimeError: If cannot find the asset path in
-            the manifest (only in production).
+    def generate_vite_asset(
+        self,
+        path: str,
+        app: str = DEFAULT_APP_NAME,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        app_client = self._get_app_client(app)
+        return app_client.generate_vite_asset(path, **kwargs)
 
-    Returns:
-        str -- The script tag of this legacy asset.
-    """
+    def preload_vite_asset(
+        self,
+        path: str,
+        app: str = DEFAULT_APP_NAME,
+    ) -> str:
+        app_client = self._get_app_client(app)
+        return app_client.preload_vite_asset(path)
 
-    assert path is not None
+    def generate_vite_asset_url(
+        self,
+        path: str,
+        app: str = DEFAULT_APP_NAME,
+    ) -> str:
+        app_client = self._get_app_client(app)
+        return app_client.generate_vite_asset_url(path)
 
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
-        path, config_key, **kwargs
-    )
+    def generate_vite_legacy_polyfills(
+        self,
+        app: str = DEFAULT_APP_NAME,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        app_client = self._get_app_client(app)
+        return app_client.generate_vite_legacy_polyfills(**kwargs)
 
+    def generate_vite_legacy_asset(
+        self,
+        path: str,
+        app: str = DEFAULT_APP_NAME,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        app_client = self._get_app_client(app)
+        return app_client.generate_vite_legacy_asset(path, **kwargs)
 
-@register.simple_tag
-@mark_safe
-def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
-    """
-    Generates the script for the Vite React Refresh for HMR.
-    Only used in development, in production this method returns
-    an empty string.
+    def generate_vite_ws_client(
+        self,
+        app: str = DEFAULT_APP_NAME,
+        **kwargs: Dict[str, str],
+    ) -> str:
+        app_client = self._get_app_client(app)
+        return app_client.generate_vite_ws_client(**kwargs)
 
-    Returns:
-        str -- The script or an empty string.
-    """
-    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(config_key)
+    def generate_vite_react_refresh_url(
+        self,
+        app: str = DEFAULT_APP_NAME,
+    ) -> str:
+        app_client = self._get_app_client(app)
+        return app_client.generate_vite_react_refresh_url()
diff --git a/django_vite/tag_generator.py b/django_vite/tag_generator.py
index c1e9956..f9f3c8f 100644
--- a/django_vite/tag_generator.py
+++ b/django_vite/tag_generator.py
@@ -1,645 +1,11 @@
-import json
-from pathlib import Path
-from typing import Dict, List, Callable, NamedTuple, Optional, Type, Union
-from urllib.parse import urljoin
+from typing import Dict
 
-from django import template
-from django.apps import apps
-from django.conf import settings
-from django.utils.safestring import mark_safe
+Tag = str
 
-from django_vite.exceptions import DjangoViteAssetNotFoundError, DjangoViteManifestError
-
-DEFAULT_CONFIG_KEY = "default"
-
-register = template.Library()
-
-
-class DjangoViteManifest(NamedTuple):
-    """
-    Represent an entry for a file inside the "manifest.json".
-    """
-
-    file: str
-    src: Optional[str] = None
-    isEntry: Optional[bool] = False
-    css: Optional[List[str]] = []
-    imports: Optional[List[str]] = []
-
-
-class DjangoViteConfig(NamedTuple):
-    """
-    Represent the Django Vite configuration structure.
-    """
-
-    # Location of Vite compiled assets (only used in Vite production mode).
-    assets_path: Union[Path, str]
-
-    # If using in development or production mode.
-    dev_mode: bool = False
-
-    # Default Vite server protocol (http or https)
-    dev_server_protocol: str = "http"
-
-    # Default vite server hostname.
-    dev_server_host: str = "localhost"
-
-    # Default Vite server port.
-    dev_server_port: int = 3000
-
-    # Default Vite server path to HMR script.
-    ws_client_url: str = "@vite/client"
-
-    # Prefix for STATIC_URL.
-    static_url_prefix: str = ""
-
-    # Motif in the "manifest.json" to find the polyfills generated by Vite.
-    legacy_polyfills_motif: str = "legacy-polyfills"
-
-    # Path to your manifest file generated by Vite.
-    manifest_path: Union[Path, str] = ""
-
-    # Default Vite server path to React RefreshRuntime for @vitejs/plugin-react.
-    react_refresh_url: str = "@react-refresh"
-
-    @property
-    def static_url(self) -> str:
-        url = urljoin(settings.STATIC_URL, self.static_url_prefix)
-        if not url.endswith("/"):
-            url += "/"
-        return url
-
-    @property
-    def static_root(self) -> Path:
-        """
-        Compute the static root URL of assets.
-
-        Returns:
-            Union[Path, str] -- Static root URL.
-        """
-
-        return (
-            Path(self.assets_path)
-            if self.dev_mode
-            else Path(settings.STATIC_ROOT) / self.static_url_prefix
-        )
-
-    def get_computed_manifest_path(self) -> Union[Path, str]:
-        """
-        Compute the path to the "manifest.json".
-
-        Returns:
-            Union[Path, str] -- Path to the "manifest.json".
-        """
-
-        return (
-            self.manifest_path
-            if self.manifest_path
-            else self.static_root / "manifest.json"
-        )
-
-
-class DjangoViteAssetLoader:
-    """
-    Class handling Vite asset loading.
-    """
-
-    _instance = None
-
-    _configs = Dict[str, Type[DjangoViteConfig]]
-    _manifests: Dict[str, Type[DjangoViteManifest]]
-
-    def __init__(self) -> None:
-        raise RuntimeError("Use the instance() method instead.")
-
-    def generate_vite_asset(
-        self,
-        path: str,
-        config_key: str,
-        **kwargs: Dict[str, str],
-    ) -> str:
-        """
-        Generates a <script> tag for this JS/TS asset, a <link> tag for
-        all of its CSS dependencies, and a <link modulepreload>
-        for the js dependencies, as listed in the manifest file
-        (for production only).
-        In development Vite loads all by itself.
-
-        Arguments:
-            path {str} -- Path to a Vite JS/TS asset to include.
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            str -- All tags to import this file in your HTML page.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the file path in the
-                manifest (only in production).
-
-        Returns:
-            str -- The <script> tag and all <link> tags to import
-                this asset in your page.
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_vite_server_url(path, config),
-                {"type": "module", **kwargs},
-            )
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        manifest_entry = manifest[path]
-        tags = []
-        scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
-
-        # Add dependent CSS
-        tags.extend(self._load_css_files_of_asset(path, config_key))
-
-        # Add the script by itself
-        tags.append(
-            DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_production_server_url(
-                    manifest_entry.file, config.static_url_prefix
-                ),
-                attrs=scripts_attrs,
-            )
-        )
-
-        # Preload imports
-        preload_attrs = {
-            "type": "text/javascript",
-            "crossorigin": "anonymous",
-            "rel": "modulepreload",
-            "as": "script",
-        }
-
-        for dep in manifest_entry.imports:
-            dep_manifest_entry = manifest[dep]
-            dep_file = dep_manifest_entry.file
-            url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file, config.static_url_prefix
-            )
-            tags.append(
-                DjangoViteAssetLoader._generate_preload_tag(
-                    url,
-                    attrs=preload_attrs,
-                )
-            )
-
-        return "\n".join(tags)
-
-    def preload_vite_asset(
-        self,
-        path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> str:
-        """
-        Generates a <link modulepreload> tag for this JS/TS asset, a
-        <link preload> tag for all of its CSS dependencies,
-        and a <link modulepreload> for the js dependencies.
-        In development this template tag renders nothing,
-        since files aren't compiled yet"
-
-        Arguments:
-            path {str} -- Path to a Vite JS/TS asset to preload.
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            str -- All tags to preload this file in your HTML page.
-
-        Raises:
-            DjangoViteAssetNotFoundError: if cannot find the file path in the
-                manifest.
-
-        Returns:
-            str -- all <link> tags to preload
-                this asset.
-        """
-        tags = []
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return ""
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        manifest_entry = manifest[path]
-
-        # Add the script by itself
-        script_attrs = {
-            "type": "text/javascript",
-            "crossorigin": "anonymous",
-            "rel": "modulepreload",
-            "as": "script",
-        }
-
-        manifest_file = manifest_entry.file
-        url = DjangoViteAssetLoader._generate_production_server_url(
-            manifest_file, config.static_url_prefix
-        )
-        tags.append(
-            DjangoViteAssetLoader._generate_preload_tag(
-                url,
-                attrs=script_attrs,
-            )
-        )
-
-        # Add dependent CSS
-        tags.extend(self._preload_css_files_of_asset(path, config_key))
-
-        # Preload imports
-        for dep in manifest_entry.imports:
-            dep_manifest_entry = manifest[dep]
-            dep_file = dep_manifest_entry.file
-            url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file, config.static_url_prefix
-            )
-            tags.append(
-                DjangoViteAssetLoader._generate_preload_tag(
-                    url,
-                    attrs=script_attrs,
-                )
-            )
-
-        return "\n".join(tags)
-
-    def _preload_css_files_of_asset(
-        self,
-        path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
-        return self._generate_css_files_of_asset(
-            path,
-            [],
-            DjangoViteAssetLoader._generate_stylesheet_preload_tag,
-            config_key,
-        )
-
-    def _load_css_files_of_asset(
-        self,
-        path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
-        return self._generate_css_files_of_asset(
-            path,
-            [],
-            DjangoViteAssetLoader._generate_stylesheet_tag,
-            config_key,
-        )
-
-    def _generate_css_files_of_asset(
-        self,
-        path: str,
-        already_processed: List[str],
-        tag_generator: Callable,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
-        """
-        Generates all CSS tags for dependencies of an asset.
-
-        Arguments:
-            path {str} -- Path to an asset in the 'manifest.json'.
-            config_key {str} -- Key of the configuration to use.
-            already_processed {list} -- List of already processed CSS file.
-
-        Returns:
-            list -- List of CSS tags.
-        """
-
-        tags = []
-        config = self._get_config(config_key)
-        manifest = self._get_manifest(config_key)
-        manifest_entry = manifest[path]
-
-        for import_path in manifest_entry.imports:
-            tags.extend(
-                self._generate_css_files_of_asset(
-                    import_path, already_processed, tag_generator, config_key
-                )
-            )
-
-        for css_path in manifest_entry.css:
-            if css_path not in already_processed:
-                url = DjangoViteAssetLoader._generate_production_server_url(
-                    css_path, config.static_url_prefix
-                )
-                tags.append(tag_generator(url))
-
-            already_processed.append(css_path)
-
-        return tags
-
-    def generate_vite_asset_url(self, path: str, config_key: str) -> str:
-        """
-        Generates only the URL of an asset managed by ViteJS.
-        Warning, this function does not generate URLs for dependant assets.
-
-        Arguments:
-            path {str} -- Path to a Vite asset.
-            config_key {str} -- Key of the configuration to use.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the asset path in the
-                manifest (only in production).
-
-        Returns:
-            str -- The URL of this asset.
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return DjangoViteAssetLoader._generate_vite_server_url(path, config)
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        manifest_entry = manifest[path]
-
-        return DjangoViteAssetLoader._generate_production_server_url(
-            manifest_entry.file, config.static_url_prefix
-        )
-
-    def generate_vite_legacy_polyfills(
-        self,
-        config_key: str,
-        **kwargs: Dict[str, str],
-    ) -> str:
-        """
-        Generates a <script> tag to the polyfills
-        generated by '@vitejs/plugin-legacy' if used.
-        This tag must be included at end of the <body> before
-        including other legacy scripts.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If polyfills path not found inside
-                the 'manifest.json' (only in production).
-
-        Returns:
-            str -- The script tag to the polyfills.
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return ""
-
-        manifest = self._get_manifest(config_key)
-        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
-
-        for path, manifest_entry in manifest.items():
-            if config.legacy_polyfills_motif in path:
-                return DjangoViteAssetLoader._generate_script_tag(
-                    DjangoViteAssetLoader._generate_production_server_url(
-                        manifest_entry.file, config.static_url_prefix
-                    ),
-                    attrs=scripts_attrs,
-                )
-
-        raise DjangoViteAssetNotFoundError(
-            f"Vite legacy polyfills not found in manifest "
-            f"at {config.get_computed_manifest_path()}"
-        )
-
-    def generate_vite_legacy_asset(
-        self,
-        path: str,
-        config_key: str,
-        **kwargs: Dict[str, str],
-    ) -> str:
-        """
-        Generates a <script> tag for legacy assets JS/TS
-        generated by '@vitejs/plugin-legacy'
-        (in production only, in development do nothing).
-
-        Arguments:
-            path {str} -- Path to a Vite asset to include
-                (must contains '-legacy' in its name).
-            config_key {str} -- Key of the configuration to use.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the asset path in the
-                manifest (only in production).
-
-        Returns:
-            str -- The script tag of this legacy asset .
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return ""
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
-
-        url = DjangoViteAssetLoader._generate_production_server_url(
-            manifest[path].file, config.static_url_prefix
-        )
-        return DjangoViteAssetLoader._generate_script_tag(
-            url,
-            attrs=scripts_attrs,
-        )
-
-    def _get_config(self, config_key: str) -> Type[DjangoViteConfig]:
-        """
-        Get configuration object registered with the key passed in
-        parameter.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to retrieve.
-
-        Raises:
-            RuntimeError: If no configuration exists for this key.
-
-        Returns:
-            Type[DjangoViteConfig] -- The configuration.
-        """
-
-        if config_key not in self._configs:
-            raise RuntimeError(f'Cannot find "{config_key}" configuration')
-
-        return self._configs[config_key]
-
-    def _parse_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
-        """
-        Read and parse the Vite manifest file.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Raises:
-            DjangoViteManifestError: if cannot load the file or JSON in file is
-            malformed.
-        """
-
-        config = self._get_config(config_key)
-
-        try:
-            with open(config.get_computed_manifest_path(), "r") as manifest_file:
-                manifest_content = manifest_file.read()
-                manifest_json = json.loads(manifest_content)
-
-                return {k: DjangoViteManifest(**v) for k, v in manifest_json.items()}
-
-        except Exception as error:
-            raise DjangoViteManifestError(
-                f"Cannot read Vite manifest file at "
-                f"{config.get_computed_manifest_path()} : {str(error)}"
-            ) from error
-
-    def _get_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
-        """
-        Load if needed and parse the "manifest.json" of the specified
-        configuration.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            Dict[str, Type[DjangoViteManifest]] -- Parsed content of
-                the "manifest.json"
-        """
-
-        if config_key not in self._manifests:
-            self._manifests[config_key] = self._parse_manifest(config_key)
-
-        return self._manifests[config_key]
-
-    @classmethod
-    def instance(cls):
-        """
-        Singleton.
-        Uses singleton to keep parsed manifests in memory after
-        the first time they are loaded.
-
-        Returns:
-            DjangoViteAssetLoader -- only instance of the class.
-        """
-
-        if cls._instance is None:
-            cls._instance = cls.__new__(cls)
-            cls._instance._configs = {}
-            cls._instance._manifests = {}
-
-            if hasattr(settings, "DJANGO_VITE"):
-                config = getattr(settings, "DJANGO_VITE")
-
-                for config_key, config_values in config.items():
-                    if isinstance(config_values, DjangoViteConfig):
-                        cls._instance._configs[config_key] = config_values
-                    elif isinstance(config_values, dict):
-                        cls._instance._configs[config_key] = DjangoViteConfig(
-                            **config_values
-                        )
-                    else:
-                        raise RuntimeError(
-                            f"Cannot read configuration for key '{config_key}'"
-                        )
-            else:
-                # Warning : This branch will be remove in further
-                # releases. Please use new way of handling configuration.
-
-                _config_keys = {
-                    "DJANGO_VITE_DEV_MODE": "dev_mode",
-                    "DJANGO_VITE_DEV_SERVER_PROTOCOL": "dev_server_protocol",
-                    "DJANGO_VITE_DEV_SERVER_HOST": "dev_server_host",
-                    "DJANGO_VITE_DEV_SERVER_PORT": "dev_server_port",
-                    "DJANGO_VITE_WS_CLIENT_URL": "ws_client_url",
-                    "DJANGO_VITE_ASSETS_PATH": "assets_path",
-                    "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
-                    "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
-                    "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
-                    "DJANGO_VITE_REACT_REFRESH_URL": "react_refresh_url",
-                }
-
-                config = {
-                    _config_keys[setting_key]: getattr(settings, setting_key)
-                    for setting_key in dir(settings)
-                    if setting_key in _config_keys.keys()
-                }
-
-                cls._instance._configs[DEFAULT_CONFIG_KEY] = DjangoViteConfig(**config)
-
-                # Manifest is only used in production.
-                if not cls._instance._configs[DEFAULT_CONFIG_KEY].dev_mode:
-                    cls._instance._parse_manifest(DEFAULT_CONFIG_KEY)
-
-        return cls._instance
-
-    def generate_vite_ws_client(
-        self, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-    ) -> str:
-        """
-        Generates the script tag for the Vite WS client for HMR.
-        Only used in development, in production this method returns
-        an empty string.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            str -- The script tag or an empty string.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-        """
-
-        config = self._get_config(config_key)
-
-        if not config.dev_mode:
-            return ""
-
-        return self._generate_script_tag(
-            self._generate_vite_server_url(config.ws_client_url, config),
-            {"type": "module", **kwargs},
-        )
 
+class TagGenerator:
     @staticmethod
-    def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
+    def script(src: str, attrs: Dict[str, str]) -> Tag:
         """
         Generates an HTML script tag.
 
@@ -659,7 +25,7 @@ def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
         return f'<script {attrs_str} src="{src}"></script>'
 
     @staticmethod
-    def _generate_stylesheet_tag(href: str) -> str:
+    def stylesheet(href: str) -> Tag:
         """
         Generates an HTML <link> stylesheet tag for CSS.
 
@@ -672,7 +38,8 @@ def _generate_stylesheet_tag(href: str) -> str:
 
         return f'<link rel="stylesheet" href="{href}" />'
 
-    def _generate_stylesheet_preload_tag(href: str) -> str:
+    @staticmethod
+    def stylesheet_preload(href: str) -> Tag:
         """
         Generates an HTML <link> preload tag for CSS.
 
@@ -686,273 +53,7 @@ def _generate_stylesheet_preload_tag(href: str) -> str:
         return f'<link rel="preload" href="{href}" as="style" />'
 
     @staticmethod
-    def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
+    def preload(href: str, attrs: Dict[str, str]) -> Tag:
         attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
 
         return f'<link href="{href}" {attrs_str} />'
-
-    @staticmethod
-    def _generate_vite_server_url(
-        path: str,
-        config: Type[DjangoViteConfig],
-    ) -> str:
-        """
-        Generates an URL to and asset served by the Vite development server.
-
-        Keyword Arguments:
-            path {str} -- Path to the asset.
-            config {Type[DjangoViteConfig]} -- Config object to use.
-
-        Returns:
-            str -- Full URL to the asset.
-        """
-
-        return urljoin(
-            f"{config.dev_server_protocol}://"
-            f"{config.dev_server_host}:{config.dev_server_port}",
-            urljoin(config.static_url, path),
-        )
-
-    def generate_vite_react_refresh_url(
-        self, config_key: str = DEFAULT_CONFIG_KEY
-    ) -> str:
-        """
-        Generates the script for the Vite React Refresh for HMR.
-        Only used in development, in production this method returns
-        an empty string.
-
-        Returns:
-            str -- The script or an empty string.
-            config_key {str} -- Key of the configuration to use.
-        """
-        config = self._get_config(config_key)
-
-        if not config.dev_mode:
-            return ""
-
-        return f"""<script type="module">
-            import RefreshRuntime from \
-            '{self._generate_vite_server_url(config.react_refresh_url, config)}'
-            RefreshRuntime.injectIntoGlobalHook(window)
-            window.$RefreshReg$ = () => {{}}
-            window.$RefreshSig$ = () => (type) => type
-            window.__vite_plugin_react_preamble_installed__ = true
-        </script>"""
-
-    @staticmethod
-    def _generate_production_server_url(path: str, static_url_prefix="") -> str:
-        """
-        Generates an URL to an asset served during production.
-
-        Keyword Arguments:
-            path {str} -- Path to the asset.
-
-        Returns:
-            str -- Full URL to the asset.
-        """
-
-        production_server_url = path
-        if prefix := static_url_prefix:
-            if not static_url_prefix.endswith("/"):
-                prefix += "/"
-            production_server_url = urljoin(prefix, path)
-
-        if apps.is_installed("django.contrib.staticfiles"):
-            from django.contrib.staticfiles.storage import staticfiles_storage
-
-            return staticfiles_storage.url(production_server_url)
-
-        return production_server_url
-
-
-@register.simple_tag
-@mark_safe
-def vite_hmr_client(
-    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-) -> str:
-    """
-    Generates the script tag for the Vite WS client for HMR.
-    Only used in development, in production this method returns
-    an empty string.
-
-    Arguments:
-        config {str} -- Configuration to use.
-
-    Returns:
-        str -- The script tag or an empty string.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-    """
-
-    return DjangoViteAssetLoader.instance().generate_vite_ws_client(
-        config_key, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_asset(
-    path: str,
-    config_key: str = DEFAULT_CONFIG_KEY,
-    **kwargs: Dict[str, str],
-) -> str:
-    """
-    Generates a <script> tag for this JS/TS asset, a <link> tag for
-    all of its CSS dependencies, and a <link rel="modulepreload">
-    for all js dependencies, as listed in the manifest file
-    In development Vite loads all by itself.
-
-    Arguments:
-        path {str} -- Path to a Vite JS/TS asset to include.
-        config {str} -- Configuration to use.
-
-    Returns:
-        str -- All tags to import this file in your HTML page.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Raises:
-        RuntimeError: If cannot find the file path in the
-            manifest (only in production).
-
-    Returns:
-        str -- The <script> tag and all <link> tags to import this
-            asset in your page.
-    """
-
-    assert path is not None
-    assert config_key is not None
-
-    return DjangoViteAssetLoader.instance().generate_vite_asset(
-        path, config_key, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_preload_asset(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
-    """
-    Generates preloadmodule tag for this JS/TS asset and preloads
-    all of its CSS and JS dependencies by reading the manifest
-    file (for production only).
-    In development does nothing.
-
-    Arguments:
-        path {str} -- Path to a Vite JS/TS asset to include.
-
-    Returns:
-        str -- All tags to import this file in your HTML page.
-
-    Raises:
-        RuntimeError: If cannot find the file path in the
-            manifest (only in production).
-
-    """
-    assert path is not None
-
-    return DjangoViteAssetLoader.instance().preload_vite_asset(path, config_key)
-
-
-@register.simple_tag
-def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
-    """
-    Generates only the URL of an asset managed by ViteJS.
-    Warning, this function does not generate URLs for dependant assets.
-
-    Arguments:
-        path {str} -- Path to a Vite asset.
-        config {str} -- Configuration to use.
-
-    Raises:
-        RuntimeError: If cannot find the asset path in the
-            manifest (only in production).
-
-    Returns:
-        str -- The URL of this asset.
-    """
-    assert path is not None
-    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, config_key)
-
-
-@register.simple_tag
-@mark_safe
-def vite_legacy_polyfills(
-    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-) -> str:
-    """
-    Generates a <script> tag to the polyfills generated
-    by '@vitejs/plugin-legacy' if used.
-    This tag must be included at end of the <body> before including
-    other legacy scripts.
-
-    Arguments:
-        config_key {str} -- Configuration to use.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Raises:
-        RuntimeError: If polyfills path not found inside
-            the 'manifest.json' (only in production).
-
-    Returns:
-        str -- The script tag to the polyfills.
-    """
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
-        config_key, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_legacy_asset(
-    path: str,
-    config_key: str = DEFAULT_CONFIG_KEY,
-    **kwargs: Dict[str, str],
-) -> str:
-    """
-    Generates a <script> tag for legacy assets JS/TS
-    generated by '@vitejs/plugin-legacy'
-    (in production only, in development do nothing).
-
-    Arguments:
-        path {str} -- Path to a Vite asset to include
-            (must contains '-legacy' in its name).
-        config_key {str} -- Configuration to use.
-
-    Keyword Arguments:
-        **kwargs {Dict[str, str]} -- Adds new attributes to generated
-            script tags.
-
-    Raises:
-        RuntimeError: If cannot find the asset path in
-            the manifest (only in production).
-
-    Returns:
-        str -- The script tag of this legacy asset.
-    """
-
-    assert path is not None
-
-    return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
-        path, config_key, **kwargs
-    )
-
-
-@register.simple_tag
-@mark_safe
-def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
-    """
-    Generates the script for the Vite React Refresh for HMR.
-    Only used in development, in production this method returns
-    an empty string.
-
-    Returns:
-        str -- The script or an empty string.
-    """
-    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(config_key)
diff --git a/django_vite/templatetags/django_vite.py b/django_vite/templatetags/django_vite.py
index c1e9956..1ffdf62 100644
--- a/django_vite/templatetags/django_vite.py
+++ b/django_vite/templatetags/django_vite.py
@@ -1,775 +1,17 @@
-import json
-from pathlib import Path
-from typing import Dict, List, Callable, NamedTuple, Optional, Type, Union
-from urllib.parse import urljoin
+from typing import Dict
 
 from django import template
-from django.apps import apps
-from django.conf import settings
 from django.utils.safestring import mark_safe
 
-from django_vite.exceptions import DjangoViteAssetNotFoundError, DjangoViteManifestError
+from django_vite.loader import DjangoViteAssetLoader, DEFAULT_APP_NAME
 
-DEFAULT_CONFIG_KEY = "default"
 
 register = template.Library()
 
 
-class DjangoViteManifest(NamedTuple):
-    """
-    Represent an entry for a file inside the "manifest.json".
-    """
-
-    file: str
-    src: Optional[str] = None
-    isEntry: Optional[bool] = False
-    css: Optional[List[str]] = []
-    imports: Optional[List[str]] = []
-
-
-class DjangoViteConfig(NamedTuple):
-    """
-    Represent the Django Vite configuration structure.
-    """
-
-    # Location of Vite compiled assets (only used in Vite production mode).
-    assets_path: Union[Path, str]
-
-    # If using in development or production mode.
-    dev_mode: bool = False
-
-    # Default Vite server protocol (http or https)
-    dev_server_protocol: str = "http"
-
-    # Default vite server hostname.
-    dev_server_host: str = "localhost"
-
-    # Default Vite server port.
-    dev_server_port: int = 3000
-
-    # Default Vite server path to HMR script.
-    ws_client_url: str = "@vite/client"
-
-    # Prefix for STATIC_URL.
-    static_url_prefix: str = ""
-
-    # Motif in the "manifest.json" to find the polyfills generated by Vite.
-    legacy_polyfills_motif: str = "legacy-polyfills"
-
-    # Path to your manifest file generated by Vite.
-    manifest_path: Union[Path, str] = ""
-
-    # Default Vite server path to React RefreshRuntime for @vitejs/plugin-react.
-    react_refresh_url: str = "@react-refresh"
-
-    @property
-    def static_url(self) -> str:
-        url = urljoin(settings.STATIC_URL, self.static_url_prefix)
-        if not url.endswith("/"):
-            url += "/"
-        return url
-
-    @property
-    def static_root(self) -> Path:
-        """
-        Compute the static root URL of assets.
-
-        Returns:
-            Union[Path, str] -- Static root URL.
-        """
-
-        return (
-            Path(self.assets_path)
-            if self.dev_mode
-            else Path(settings.STATIC_ROOT) / self.static_url_prefix
-        )
-
-    def get_computed_manifest_path(self) -> Union[Path, str]:
-        """
-        Compute the path to the "manifest.json".
-
-        Returns:
-            Union[Path, str] -- Path to the "manifest.json".
-        """
-
-        return (
-            self.manifest_path
-            if self.manifest_path
-            else self.static_root / "manifest.json"
-        )
-
-
-class DjangoViteAssetLoader:
-    """
-    Class handling Vite asset loading.
-    """
-
-    _instance = None
-
-    _configs = Dict[str, Type[DjangoViteConfig]]
-    _manifests: Dict[str, Type[DjangoViteManifest]]
-
-    def __init__(self) -> None:
-        raise RuntimeError("Use the instance() method instead.")
-
-    def generate_vite_asset(
-        self,
-        path: str,
-        config_key: str,
-        **kwargs: Dict[str, str],
-    ) -> str:
-        """
-        Generates a <script> tag for this JS/TS asset, a <link> tag for
-        all of its CSS dependencies, and a <link modulepreload>
-        for the js dependencies, as listed in the manifest file
-        (for production only).
-        In development Vite loads all by itself.
-
-        Arguments:
-            path {str} -- Path to a Vite JS/TS asset to include.
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            str -- All tags to import this file in your HTML page.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the file path in the
-                manifest (only in production).
-
-        Returns:
-            str -- The <script> tag and all <link> tags to import
-                this asset in your page.
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_vite_server_url(path, config),
-                {"type": "module", **kwargs},
-            )
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        manifest_entry = manifest[path]
-        tags = []
-        scripts_attrs = {"type": "module", "crossorigin": "", **kwargs}
-
-        # Add dependent CSS
-        tags.extend(self._load_css_files_of_asset(path, config_key))
-
-        # Add the script by itself
-        tags.append(
-            DjangoViteAssetLoader._generate_script_tag(
-                DjangoViteAssetLoader._generate_production_server_url(
-                    manifest_entry.file, config.static_url_prefix
-                ),
-                attrs=scripts_attrs,
-            )
-        )
-
-        # Preload imports
-        preload_attrs = {
-            "type": "text/javascript",
-            "crossorigin": "anonymous",
-            "rel": "modulepreload",
-            "as": "script",
-        }
-
-        for dep in manifest_entry.imports:
-            dep_manifest_entry = manifest[dep]
-            dep_file = dep_manifest_entry.file
-            url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file, config.static_url_prefix
-            )
-            tags.append(
-                DjangoViteAssetLoader._generate_preload_tag(
-                    url,
-                    attrs=preload_attrs,
-                )
-            )
-
-        return "\n".join(tags)
-
-    def preload_vite_asset(
-        self,
-        path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> str:
-        """
-        Generates a <link modulepreload> tag for this JS/TS asset, a
-        <link preload> tag for all of its CSS dependencies,
-        and a <link modulepreload> for the js dependencies.
-        In development this template tag renders nothing,
-        since files aren't compiled yet"
-
-        Arguments:
-            path {str} -- Path to a Vite JS/TS asset to preload.
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            str -- All tags to preload this file in your HTML page.
-
-        Raises:
-            DjangoViteAssetNotFoundError: if cannot find the file path in the
-                manifest.
-
-        Returns:
-            str -- all <link> tags to preload
-                this asset.
-        """
-        tags = []
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return ""
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        manifest_entry = manifest[path]
-
-        # Add the script by itself
-        script_attrs = {
-            "type": "text/javascript",
-            "crossorigin": "anonymous",
-            "rel": "modulepreload",
-            "as": "script",
-        }
-
-        manifest_file = manifest_entry.file
-        url = DjangoViteAssetLoader._generate_production_server_url(
-            manifest_file, config.static_url_prefix
-        )
-        tags.append(
-            DjangoViteAssetLoader._generate_preload_tag(
-                url,
-                attrs=script_attrs,
-            )
-        )
-
-        # Add dependent CSS
-        tags.extend(self._preload_css_files_of_asset(path, config_key))
-
-        # Preload imports
-        for dep in manifest_entry.imports:
-            dep_manifest_entry = manifest[dep]
-            dep_file = dep_manifest_entry.file
-            url = DjangoViteAssetLoader._generate_production_server_url(
-                dep_file, config.static_url_prefix
-            )
-            tags.append(
-                DjangoViteAssetLoader._generate_preload_tag(
-                    url,
-                    attrs=script_attrs,
-                )
-            )
-
-        return "\n".join(tags)
-
-    def _preload_css_files_of_asset(
-        self,
-        path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
-        return self._generate_css_files_of_asset(
-            path,
-            [],
-            DjangoViteAssetLoader._generate_stylesheet_preload_tag,
-            config_key,
-        )
-
-    def _load_css_files_of_asset(
-        self,
-        path: str,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
-        return self._generate_css_files_of_asset(
-            path,
-            [],
-            DjangoViteAssetLoader._generate_stylesheet_tag,
-            config_key,
-        )
-
-    def _generate_css_files_of_asset(
-        self,
-        path: str,
-        already_processed: List[str],
-        tag_generator: Callable,
-        config_key: str = DEFAULT_CONFIG_KEY,
-    ) -> List[str]:
-        """
-        Generates all CSS tags for dependencies of an asset.
-
-        Arguments:
-            path {str} -- Path to an asset in the 'manifest.json'.
-            config_key {str} -- Key of the configuration to use.
-            already_processed {list} -- List of already processed CSS file.
-
-        Returns:
-            list -- List of CSS tags.
-        """
-
-        tags = []
-        config = self._get_config(config_key)
-        manifest = self._get_manifest(config_key)
-        manifest_entry = manifest[path]
-
-        for import_path in manifest_entry.imports:
-            tags.extend(
-                self._generate_css_files_of_asset(
-                    import_path, already_processed, tag_generator, config_key
-                )
-            )
-
-        for css_path in manifest_entry.css:
-            if css_path not in already_processed:
-                url = DjangoViteAssetLoader._generate_production_server_url(
-                    css_path, config.static_url_prefix
-                )
-                tags.append(tag_generator(url))
-
-            already_processed.append(css_path)
-
-        return tags
-
-    def generate_vite_asset_url(self, path: str, config_key: str) -> str:
-        """
-        Generates only the URL of an asset managed by ViteJS.
-        Warning, this function does not generate URLs for dependant assets.
-
-        Arguments:
-            path {str} -- Path to a Vite asset.
-            config_key {str} -- Key of the configuration to use.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the asset path in the
-                manifest (only in production).
-
-        Returns:
-            str -- The URL of this asset.
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return DjangoViteAssetLoader._generate_vite_server_url(path, config)
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        manifest_entry = manifest[path]
-
-        return DjangoViteAssetLoader._generate_production_server_url(
-            manifest_entry.file, config.static_url_prefix
-        )
-
-    def generate_vite_legacy_polyfills(
-        self,
-        config_key: str,
-        **kwargs: Dict[str, str],
-    ) -> str:
-        """
-        Generates a <script> tag to the polyfills
-        generated by '@vitejs/plugin-legacy' if used.
-        This tag must be included at end of the <body> before
-        including other legacy scripts.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If polyfills path not found inside
-                the 'manifest.json' (only in production).
-
-        Returns:
-            str -- The script tag to the polyfills.
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return ""
-
-        manifest = self._get_manifest(config_key)
-        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
-
-        for path, manifest_entry in manifest.items():
-            if config.legacy_polyfills_motif in path:
-                return DjangoViteAssetLoader._generate_script_tag(
-                    DjangoViteAssetLoader._generate_production_server_url(
-                        manifest_entry.file, config.static_url_prefix
-                    ),
-                    attrs=scripts_attrs,
-                )
-
-        raise DjangoViteAssetNotFoundError(
-            f"Vite legacy polyfills not found in manifest "
-            f"at {config.get_computed_manifest_path()}"
-        )
-
-    def generate_vite_legacy_asset(
-        self,
-        path: str,
-        config_key: str,
-        **kwargs: Dict[str, str],
-    ) -> str:
-        """
-        Generates a <script> tag for legacy assets JS/TS
-        generated by '@vitejs/plugin-legacy'
-        (in production only, in development do nothing).
-
-        Arguments:
-            path {str} -- Path to a Vite asset to include
-                (must contains '-legacy' in its name).
-            config_key {str} -- Key of the configuration to use.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-
-        Raises:
-            DjangoViteAssetNotFoundError: If cannot find the asset path in the
-                manifest (only in production).
-
-        Returns:
-            str -- The script tag of this legacy asset .
-        """
-
-        config = self._get_config(config_key)
-
-        if config.dev_mode:
-            return ""
-
-        manifest = self._get_manifest(config_key)
-
-        if path not in manifest:
-            raise DjangoViteAssetNotFoundError(
-                f"Cannot find {path} in Vite manifest "
-                f"at {config.get_computed_manifest_path()}"
-            )
-
-        scripts_attrs = {"nomodule": "", "crossorigin": "", **kwargs}
-
-        url = DjangoViteAssetLoader._generate_production_server_url(
-            manifest[path].file, config.static_url_prefix
-        )
-        return DjangoViteAssetLoader._generate_script_tag(
-            url,
-            attrs=scripts_attrs,
-        )
-
-    def _get_config(self, config_key: str) -> Type[DjangoViteConfig]:
-        """
-        Get configuration object registered with the key passed in
-        parameter.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to retrieve.
-
-        Raises:
-            RuntimeError: If no configuration exists for this key.
-
-        Returns:
-            Type[DjangoViteConfig] -- The configuration.
-        """
-
-        if config_key not in self._configs:
-            raise RuntimeError(f'Cannot find "{config_key}" configuration')
-
-        return self._configs[config_key]
-
-    def _parse_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
-        """
-        Read and parse the Vite manifest file.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Raises:
-            DjangoViteManifestError: if cannot load the file or JSON in file is
-            malformed.
-        """
-
-        config = self._get_config(config_key)
-
-        try:
-            with open(config.get_computed_manifest_path(), "r") as manifest_file:
-                manifest_content = manifest_file.read()
-                manifest_json = json.loads(manifest_content)
-
-                return {k: DjangoViteManifest(**v) for k, v in manifest_json.items()}
-
-        except Exception as error:
-            raise DjangoViteManifestError(
-                f"Cannot read Vite manifest file at "
-                f"{config.get_computed_manifest_path()} : {str(error)}"
-            ) from error
-
-    def _get_manifest(self, config_key: str) -> Dict[str, Type[DjangoViteManifest]]:
-        """
-        Load if needed and parse the "manifest.json" of the specified
-        configuration.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            Dict[str, Type[DjangoViteManifest]] -- Parsed content of
-                the "manifest.json"
-        """
-
-        if config_key not in self._manifests:
-            self._manifests[config_key] = self._parse_manifest(config_key)
-
-        return self._manifests[config_key]
-
-    @classmethod
-    def instance(cls):
-        """
-        Singleton.
-        Uses singleton to keep parsed manifests in memory after
-        the first time they are loaded.
-
-        Returns:
-            DjangoViteAssetLoader -- only instance of the class.
-        """
-
-        if cls._instance is None:
-            cls._instance = cls.__new__(cls)
-            cls._instance._configs = {}
-            cls._instance._manifests = {}
-
-            if hasattr(settings, "DJANGO_VITE"):
-                config = getattr(settings, "DJANGO_VITE")
-
-                for config_key, config_values in config.items():
-                    if isinstance(config_values, DjangoViteConfig):
-                        cls._instance._configs[config_key] = config_values
-                    elif isinstance(config_values, dict):
-                        cls._instance._configs[config_key] = DjangoViteConfig(
-                            **config_values
-                        )
-                    else:
-                        raise RuntimeError(
-                            f"Cannot read configuration for key '{config_key}'"
-                        )
-            else:
-                # Warning : This branch will be remove in further
-                # releases. Please use new way of handling configuration.
-
-                _config_keys = {
-                    "DJANGO_VITE_DEV_MODE": "dev_mode",
-                    "DJANGO_VITE_DEV_SERVER_PROTOCOL": "dev_server_protocol",
-                    "DJANGO_VITE_DEV_SERVER_HOST": "dev_server_host",
-                    "DJANGO_VITE_DEV_SERVER_PORT": "dev_server_port",
-                    "DJANGO_VITE_WS_CLIENT_URL": "ws_client_url",
-                    "DJANGO_VITE_ASSETS_PATH": "assets_path",
-                    "DJANGO_VITE_STATIC_URL_PREFIX": "static_url_prefix",
-                    "DJANGO_VITE_MANIFEST_PATH": "manifest_path",
-                    "DJANGO_VITE_LEGACY_POLYFILLS_MOTIF": "legacy_polyfills_motif",
-                    "DJANGO_VITE_REACT_REFRESH_URL": "react_refresh_url",
-                }
-
-                config = {
-                    _config_keys[setting_key]: getattr(settings, setting_key)
-                    for setting_key in dir(settings)
-                    if setting_key in _config_keys.keys()
-                }
-
-                cls._instance._configs[DEFAULT_CONFIG_KEY] = DjangoViteConfig(**config)
-
-                # Manifest is only used in production.
-                if not cls._instance._configs[DEFAULT_CONFIG_KEY].dev_mode:
-                    cls._instance._parse_manifest(DEFAULT_CONFIG_KEY)
-
-        return cls._instance
-
-    def generate_vite_ws_client(
-        self, config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-    ) -> str:
-        """
-        Generates the script tag for the Vite WS client for HMR.
-        Only used in development, in production this method returns
-        an empty string.
-
-        Arguments:
-            config_key {str} -- Key of the configuration to use.
-
-        Returns:
-            str -- The script tag or an empty string.
-
-        Keyword Arguments:
-            **kwargs {Dict[str, str]} -- Adds new attributes to generated
-                script tags.
-        """
-
-        config = self._get_config(config_key)
-
-        if not config.dev_mode:
-            return ""
-
-        return self._generate_script_tag(
-            self._generate_vite_server_url(config.ws_client_url, config),
-            {"type": "module", **kwargs},
-        )
-
-    @staticmethod
-    def _generate_script_tag(src: str, attrs: Dict[str, str]) -> str:
-        """
-        Generates an HTML script tag.
-
-        Arguments:
-            src {str} -- Source of the script.
-
-        Keyword Arguments:
-            attrs {Dict[str, str]} -- List of custom attributes
-                for the tag.
-
-        Returns:
-            str -- The script tag.
-        """
-
-        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
-
-        return f'<script {attrs_str} src="{src}"></script>'
-
-    @staticmethod
-    def _generate_stylesheet_tag(href: str) -> str:
-        """
-        Generates an HTML <link> stylesheet tag for CSS.
-
-        Arguments:
-            href {str} -- CSS file URL.
-
-        Returns:
-            str -- CSS link tag.
-        """
-
-        return f'<link rel="stylesheet" href="{href}" />'
-
-    def _generate_stylesheet_preload_tag(href: str) -> str:
-        """
-        Generates an HTML <link> preload tag for CSS.
-
-        Arguments:
-            href {str} -- CSS file URL.
-
-        Returns:
-            str -- CSS link tag.
-        """
-
-        return f'<link rel="preload" href="{href}" as="style" />'
-
-    @staticmethod
-    def _generate_preload_tag(href: str, attrs: Dict[str, str]) -> str:
-        attrs_str = " ".join([f'{key}="{value}"' for key, value in attrs.items()])
-
-        return f'<link href="{href}" {attrs_str} />'
-
-    @staticmethod
-    def _generate_vite_server_url(
-        path: str,
-        config: Type[DjangoViteConfig],
-    ) -> str:
-        """
-        Generates an URL to and asset served by the Vite development server.
-
-        Keyword Arguments:
-            path {str} -- Path to the asset.
-            config {Type[DjangoViteConfig]} -- Config object to use.
-
-        Returns:
-            str -- Full URL to the asset.
-        """
-
-        return urljoin(
-            f"{config.dev_server_protocol}://"
-            f"{config.dev_server_host}:{config.dev_server_port}",
-            urljoin(config.static_url, path),
-        )
-
-    def generate_vite_react_refresh_url(
-        self, config_key: str = DEFAULT_CONFIG_KEY
-    ) -> str:
-        """
-        Generates the script for the Vite React Refresh for HMR.
-        Only used in development, in production this method returns
-        an empty string.
-
-        Returns:
-            str -- The script or an empty string.
-            config_key {str} -- Key of the configuration to use.
-        """
-        config = self._get_config(config_key)
-
-        if not config.dev_mode:
-            return ""
-
-        return f"""<script type="module">
-            import RefreshRuntime from \
-            '{self._generate_vite_server_url(config.react_refresh_url, config)}'
-            RefreshRuntime.injectIntoGlobalHook(window)
-            window.$RefreshReg$ = () => {{}}
-            window.$RefreshSig$ = () => (type) => type
-            window.__vite_plugin_react_preamble_installed__ = true
-        </script>"""
-
-    @staticmethod
-    def _generate_production_server_url(path: str, static_url_prefix="") -> str:
-        """
-        Generates an URL to an asset served during production.
-
-        Keyword Arguments:
-            path {str} -- Path to the asset.
-
-        Returns:
-            str -- Full URL to the asset.
-        """
-
-        production_server_url = path
-        if prefix := static_url_prefix:
-            if not static_url_prefix.endswith("/"):
-                prefix += "/"
-            production_server_url = urljoin(prefix, path)
-
-        if apps.is_installed("django.contrib.staticfiles"):
-            from django.contrib.staticfiles.storage import staticfiles_storage
-
-            return staticfiles_storage.url(production_server_url)
-
-        return production_server_url
-
-
 @register.simple_tag
 @mark_safe
-def vite_hmr_client(
-    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-) -> str:
+def vite_hmr_client(app: str = DEFAULT_APP_NAME, **kwargs: Dict[str, str]) -> str:
     """
     Generates the script tag for the Vite WS client for HMR.
     Only used in development, in production this method returns
@@ -786,16 +28,14 @@ def vite_hmr_client(
             script tags.
     """
 
-    return DjangoViteAssetLoader.instance().generate_vite_ws_client(
-        config_key, **kwargs
-    )
+    return DjangoViteAssetLoader.instance().generate_vite_ws_client(app, **kwargs)
 
 
 @register.simple_tag
 @mark_safe
 def vite_asset(
     path: str,
-    config_key: str = DEFAULT_CONFIG_KEY,
+    app: str = DEFAULT_APP_NAME,
     **kwargs: Dict[str, str],
 ) -> str:
     """
@@ -823,18 +63,13 @@ def vite_asset(
         str -- The <script> tag and all <link> tags to import this
             asset in your page.
     """
-
     assert path is not None
-    assert config_key is not None
-
-    return DjangoViteAssetLoader.instance().generate_vite_asset(
-        path, config_key, **kwargs
-    )
+    return DjangoViteAssetLoader.instance().generate_vite_asset(path, app, **kwargs)
 
 
 @register.simple_tag
 @mark_safe
-def vite_preload_asset(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
+def vite_preload_asset(path: str, app: str = DEFAULT_APP_NAME) -> str:
     """
     Generates preloadmodule tag for this JS/TS asset and preloads
     all of its CSS and JS dependencies by reading the manifest
@@ -853,12 +88,11 @@ def vite_preload_asset(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
 
     """
     assert path is not None
-
-    return DjangoViteAssetLoader.instance().preload_vite_asset(path, config_key)
+    return DjangoViteAssetLoader.instance().preload_vite_asset(path, app)
 
 
 @register.simple_tag
-def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
+def vite_asset_url(path: str, app: str = DEFAULT_APP_NAME) -> str:
     """
     Generates only the URL of an asset managed by ViteJS.
     Warning, this function does not generate URLs for dependant assets.
@@ -875,14 +109,12 @@ def vite_asset_url(path: str, config_key: str = DEFAULT_CONFIG_KEY) -> str:
         str -- The URL of this asset.
     """
     assert path is not None
-    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, config_key)
+    return DjangoViteAssetLoader.instance().generate_vite_asset_url(path, app)
 
 
 @register.simple_tag
 @mark_safe
-def vite_legacy_polyfills(
-    config_key: str = DEFAULT_CONFIG_KEY, **kwargs: Dict[str, str]
-) -> str:
+def vite_legacy_polyfills(app: str = DEFAULT_APP_NAME, **kwargs: Dict[str, str]) -> str:
     """
     Generates a <script> tag to the polyfills generated
     by '@vitejs/plugin-legacy' if used.
@@ -890,7 +122,7 @@ def vite_legacy_polyfills(
     other legacy scripts.
 
     Arguments:
-        config_key {str} -- Configuration to use.
+        app {str} -- Configuration to use.
 
     Keyword Arguments:
         **kwargs {Dict[str, str]} -- Adds new attributes to generated
@@ -904,7 +136,7 @@ def vite_legacy_polyfills(
         str -- The script tag to the polyfills.
     """
     return DjangoViteAssetLoader.instance().generate_vite_legacy_polyfills(
-        config_key, **kwargs
+        app, **kwargs
     )
 
 
@@ -912,7 +144,7 @@ def vite_legacy_polyfills(
 @mark_safe
 def vite_legacy_asset(
     path: str,
-    config_key: str = DEFAULT_CONFIG_KEY,
+    app: str = DEFAULT_APP_NAME,
     **kwargs: Dict[str, str],
 ) -> str:
     """
@@ -923,7 +155,7 @@ def vite_legacy_asset(
     Arguments:
         path {str} -- Path to a Vite asset to include
             (must contains '-legacy' in its name).
-        config_key {str} -- Configuration to use.
+        app {str} -- Configuration to use.
 
     Keyword Arguments:
         **kwargs {Dict[str, str]} -- Adds new attributes to generated
@@ -940,13 +172,13 @@ def vite_legacy_asset(
     assert path is not None
 
     return DjangoViteAssetLoader.instance().generate_vite_legacy_asset(
-        path, config_key, **kwargs
+        path, app, **kwargs
     )
 
 
 @register.simple_tag
 @mark_safe
-def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
+def vite_react_refresh(app: str = DEFAULT_APP_NAME) -> str:
     """
     Generates the script for the Vite React Refresh for HMR.
     Only used in development, in production this method returns
@@ -955,4 +187,4 @@ def vite_react_refresh(config_key: str = DEFAULT_CONFIG_KEY) -> str:
     Returns:
         str -- The script or an empty string.
     """
-    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(config_key)
+    return DjangoViteAssetLoader.instance().generate_vite_react_refresh_url(app)
diff --git a/pyproject.toml b/pyproject.toml
index 02d02fa..3fc9f6c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,6 +8,7 @@ addopts = '''
   --cov-report term-missing
   --cov-branch
 '''
+filterwarnings = "ignore::DeprecationWarning:django_vite.*"
 
 [tool.black]
 line-length = 88
diff --git a/tests/conftest.py b/tests/conftest.py
index 2eedec1..ff92a35 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,26 +1,20 @@
 from typing import Dict, Any
 import pytest
-from importlib import reload
-from django.apps import apps
-from django_vite.templatetags import django_vite
+
+from django_vite.loader import DjangoViteAssetLoader
 
 
 def reload_django_vite():
-    reload(django_vite)
-    django_vite_app_config = apps.get_app_config("django_vite")
-    django_vite_app_config.ready()
+    DjangoViteAssetLoader._instance = None
+    DjangoViteAssetLoader.instance()
 
 
 @pytest.fixture()
 def patch_settings(settings):
     """
     1. Patch new settings into django.conf.settings.
-    2. Reload django_vite module so that variables on the module level that use settings
-    get recalculated.
+    2. Recreate DjangoViteAssetLoader._instance with new settings applied.
     3. Restore the original settings once the test is over.
-
-    TODO: refactor django_vite so that we don't set variables on the module level using
-    settings.
     """
     __PYTEST_EMPTY__ = "__PYTEST_EMPTY__"
     original_settings_cache = {}
diff --git a/tests/tests/test_asset_loader.py b/tests/tests/test_asset_loader.py
index 470867c..fe56a32 100644
--- a/tests/tests/test_asset_loader.py
+++ b/tests/tests/test_asset_loader.py
@@ -1,5 +1,7 @@
 import pytest
+from contextlib import suppress
 
+from django_vite.exceptions import DjangoViteManifestError
 from django_vite.templatetags.django_vite import DjangoViteAssetLoader
 from django_vite.apps import check_loader_instance
 
@@ -21,12 +23,13 @@ def test_check_loader_instance_happy(patch_settings):
 
 
 def test_check_loader_instance_warnings(patch_settings):
-    patch_settings(
-        {
-            "DJANGO_VITE_DEV_MODE": False,
-            "DJANGO_VITE_MANIFEST_PATH": "fake.json",
-        }
-    )
+    with suppress(DjangoViteManifestError):
+        patch_settings(
+            {
+                "DJANGO_VITE_DEV_MODE": False,
+                "DJANGO_VITE_MANIFEST_PATH": "fake.json",
+            }
+        )
     DjangoViteAssetLoader._instance = None
     warnings = check_loader_instance()
     assert len(warnings) == 1