Fix broken snippet code verification functionality

php/admin-menus/class-edit-menu.php

@@ -21,11 +21,6 @@ public function __construct() {
 	public function run() {
 		parent::run();
 		$this->remove_debug_bar_codemirror();
-
-		if ( isset( $_POST['save_snippet'] ) && $_POST['save_snippet'] ) {
-			add_action( 'code_snippets/allow_execute_snippet', array( $this, 'prevent_exec_on_save' ), 10, 2 );
-		}
-
 	}
 
 	/**
@@ -46,30 +41,6 @@ public function register() {
 		);
 	}
 
-	/**
-	 * Prevent the snippet currently being saved from being executed
-	 * so it is not run twice (once normally, once
-	 *
-	 * @param bool $exec    Whether the snippet will be executed
-	 * @param int  $exec_id The ID of the snippet being executed
-	 *
-	 * @return bool Whether the snippet will be executed
-	 */
-	function prevent_exec_on_save( $exec, $exec_id ) {
-
-		if ( ! isset( $_POST['save_snippet'], $_POST['snippet_id'] ) ) {
-			return $exec;
-		}
-
-		$id = intval( $_POST['snippet_id'] );
-
-		if ( $id == $exec_id ) {
-			return false;
-		}
-
-		return $exec;
-	}
-
 	/**
 	 * Executed when the menu is loaded
 	 */

php/class-admin.php

@@ -38,6 +38,10 @@ public function run() {
 		add_filter( 'plugin_action_links_' . plugin_basename( CODE_SNIPPETS_FILE ), array( $this, 'plugin_settings_link' ) );
 		add_filter( 'plugin_row_meta', array( $this, 'plugin_meta_links' ), 10, 2 );
 		add_action( 'code_snippets/admin/manage', array( $this, 'survey_message' ) );
+
+		if ( isset( $_POST['save_snippet'] ) && $_POST['save_snippet'] ) {
+			add_action( 'code_snippets/allow_execute_snippet', array( $this, 'prevent_exec_on_save' ), 10, 2 );
+		}
 	}
 
 	/**
@@ -99,6 +103,30 @@ function enqueue_admin_stylesheet( $hook ) {
 		);
 	}
 
+	/**
+	 * Prevent the snippet currently being saved from being executed
+	 * so it is not run twice (once normally, once
+	 *
+	 * @param bool $exec    Whether the snippet will be executed
+	 * @param int  $exec_id The ID of the snippet being executed
+	 *
+	 * @return bool Whether the snippet will be executed
+	 */
+	function prevent_exec_on_save( $exec, $exec_id ) {
+
+		if ( ! isset( $_POST['save_snippet'], $_POST['snippet_id'] ) ) {
+			return $exec;
+		}
+
+		$id = intval( $_POST['snippet_id'] );
+
+		if ( $id == $exec_id ) {
+			return false;
+		}
+
+		return $exec;
+	}
+
 	/**
 	 * Adds a link pointing to the Manage Snippets page
 	 *