WordPressの投稿画面で「このサイトを離れますか？」を出すようにする方法

WordPressで記事書いていて、間違えて閉じてしまった経験がある人は、
いるんじゃないでしょうか。。

私も、一回だけやってしまって、んぐぐ、となったので、
今回は、閉じようとしたたときに、「このサイトを離れますか？」が表示されるように変更しようと思います。

プログラムができなくても、子テーマをすでに使っていれば簡単に実装できます。

functions.phpに以下を追加すれば表示されるようになります。

// 「このサイトを離れますか？」を表示する
function confirm_beforeunload()
{
  wp_enqueue_script('jquery');
?>
  <script>
    jQuery(function($) {
      $(window).on('beforeunload', function(e) {
        e.preventDefault();
        return '';
      });
      $("input[type='submit']").on('click', function(e) {
        $(window).off('beforeunload');
      });
    });
  </script>
<?php
}
add_action('admin_head-post-new.php', 'confirm_beforeunload');
add_action('admin_head-post.php', 'confirm_beforeunload');