move internal plugin back to staticfiles module (#187)

* move internal plugin back to `staticfiles` module

* change import

Author: joshuadavidthomas

src/django_bird/plugins/file_assets.py

@@ -10,7 +10,7 @@
 pm.add_hookspecs(hookspecs)
 
 DEFAULT_PLUGINS: list[str] = [
-    "django_bird.plugins.file_assets",
+    "django_bird.staticfiles",
 ]
 
 

src/django_bird/plugins/manager.py

@@ -16,6 +16,8 @@
 from django.core.checks import CheckMessage
 from django.core.files.storage import FileSystemStorage
 
+from django_bird import hookimpl
+
 from ._typing import override
 from .apps import DjangoBirdAppConfig
 from .conf import app_settings
@@ -107,6 +109,16 @@ def url(self) -> str | None:
         return self.storage.url(str(static_relative_path))
 
 
+@hookimpl
+def collect_component_assets(template_path: Path) -> Iterable[Asset]:
+    assets: list[Asset] = []
+    for asset_type in AssetType:
+        asset_path = template_path.with_suffix(asset_type.ext)
+        if asset_path.exists():
+            assets.append(Asset(path=asset_path, type=asset_type))
+    return assets
+
+
 @final
 class BirdAssetStorage(StaticFilesStorage):
     def __init__(self, *args: Any, prefix: str, **kwargs: Any):

src/django_bird/staticfiles.py

@@ -14,6 +14,7 @@
 from django_bird.staticfiles import Asset
 from django_bird.staticfiles import AssetType
 from django_bird.staticfiles import BirdAssetFinder
+from django_bird.staticfiles import collect_component_assets
 
 from .utils import TestAsset
 from .utils import TestComponent
@@ -199,6 +200,40 @@ def test_url_nonexistent(self, templates_dir):
         assert asset.url is None
 
 
+def test_asset_collection(templates_dir):
+    button = TestComponent(name="button", content="<button>Click me</button>").create(
+        templates_dir
+    )
+    button_css = TestAsset(
+        component=button,
+        content=".button { color: blue; }",
+        asset_type=AssetType.CSS,
+    ).create()
+    button_js = TestAsset(
+        component=button, content="console.log('button');", asset_type=AssetType.JS
+    ).create()
+
+    assets = collect_component_assets(button.file)
+
+    assert Asset(button_css.file, button_css.asset_type) in assets
+    assert Asset(button_js.file, button_js.asset_type) in assets
+
+
+def test_asset_collection_nested(templates_dir):
+    button = TestComponent(
+        name="button", content="<button>Click me</button>", sub_dir="nested"
+    ).create(templates_dir)
+    button_css = TestAsset(
+        component=button,
+        content=".button { color: blue; }",
+        asset_type=AssetType.CSS,
+    ).create()
+
+    assets = collect_component_assets(button.file)
+
+    assert Asset(button_css.file, button_css.asset_type) in assets
+
+
 class TestBirdAssetFinder:
     def test_check(self):
         finder = BirdAssetFinder()