Fix passing of configuration into CrystalCollector

Author: Blacksmoke16

mkdocstrings_handlers/crystal/__init__.py

@@ -1,15 +1,17 @@
 from __future__ import annotations
 
-from collections.abc import Mapping, Sequence
+from collections.abc import Mapping, MutableMapping, Sequence
 from typing import Any
 
+from markdown import Extension
+from mkdocs.config.defaults import MkDocsConfig
 from mkdocstrings import BaseHandler, HandlerOptions
 
 from . import inventory
 from .collector import CrystalCollector
 from .renderer import CrystalRenderer
 
-__version__ = "0.3.8"
+__version__ = "0.3.9"
 
 
 class CrystalHandler(CrystalCollector, CrystalRenderer, BaseHandler):
@@ -22,15 +24,17 @@ def __init__(
         *,
         theme: str,
         custom_templates: str | None = None,
-        config: Mapping[str, Any] | None = None,
-        **kwargs: Any,
+        mdx: Sequence[str | Extension],
+        mdx_config: Mapping[str, Any],
+        handler_config: MutableMapping[str, Any],
+        tool_config: MkDocsConfig,
     ) -> None:
-        config = config or {}
-        BaseHandler.__init__(self, theme=theme, custom_templates=custom_templates, **kwargs)
+        BaseHandler.__init__(
+            self, theme=theme, custom_templates=custom_templates, mdx=mdx, mdx_config=mdx_config
+        )
         CrystalCollector.__init__(
             self,
-            crystal_docs_flags=config.get("crystal_docs_flags", ()),
-            source_locations=config.get("source_locations", {}),
+            **handler_config,
         )
 
     def get_options(self, local_options: Mapping[str, Any]) -> HandlerOptions: