✨ Add allowed option for inline/block attributes

mdit_py_plugins/attrs/index.py

@@ -11,12 +11,18 @@
 from .parse import ParseError, parse
 
 
+class NotAllowedAttributesError(ValueError):
+    pass
+
+
 def attrs_plugin(
     md: MarkdownIt,
     *,
     after: Sequence[str] = ("image", "code_inline", "link_close", "span_close"),
     spans: bool = False,
     span_after: str = "link",
+    allowed_attributes: Sequence[str] = [],
+    strict: bool = False,
 ) -> None:
     """Parse inline attributes that immediately follow certain inline elements::
 
@@ -58,6 +64,17 @@ def _attr_inline_rule(state: StateInline, silent: bool) -> bool:
             return False
         try:
             new_pos, attrs = parse(state.src[state.pos :])
+            if allowed_attributes:
+                if strict:
+                    attrs = {
+                        k: v for k, v in attrs.items() if k not in allowed_attributes
+                    }
+                    if attrs:
+                        raise NotAllowedAttributesError(
+                            f"These attributes are not allowed {attrs}"
+                        )
+                else:
+                    attrs = {k: v for k, v in attrs.items() if k in allowed_attributes}
         except ParseError:
             return False
         token_index = _find_opening(state.tokens, len(state.tokens) - 1)