Add additional contexts

10up · Sep 30, 2024 · 8af17da · 8af17da
1 parent fd69010
commit 8af17da
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/safe-svg.php b/safe-svg.php
@@ -128,8 +128,14 @@ public function __construct() {
 			$this->sanitizer = new Sanitizer();
 			$this->sanitizer->minify( true );
 
-			add_action( 'init', array( $this, 'setup_blocks' ) );
+			// Allow SVG uploads from specific contexts.
 			add_action( 'load-upload.php', array( $this, 'allow_svg_from_upload' ) );
+			add_action( 'load-post-new.php', array( $this, 'allow_svg_from_upload' ) );
+			add_action( 'load-post.php', array( $this, 'allow_svg_from_upload' ) );
+			add_action( 'load-site-editor.php', array( $this, 'allow_svg_from_upload' ) );
+
+			// Init all the things.
+			add_action( 'init', array( $this, 'setup_blocks' ) );
 			add_filter( 'wp_handle_sideload_prefilter', array( $this, 'check_for_svg' ) );
 			add_filter( 'wp_handle_upload_prefilter', array( $this, 'check_for_svg' ) );
 			add_filter( 'wp_prepare_attachment_for_js', array( $this, 'fix_admin_preview' ), 10, 3 );