+
+/**
+ * Check lock status for posts displayed on the Posts screen
+ *
+ * @since 3.6.0
+ */
+function wp_check_locked_posts( $response, $data, $screen_id ) {
+ $checked = array();
+
+ if ( array_key_exists( 'wp-check-locked-posts', $data ) && is_array( $data['wp-check-locked-posts'] ) ) {
+ foreach ( $data['wp-check-locked-posts'] as $key ) {
+ if ( ! $post_id = absint( substr( $key, 5 ) ) )
+ continue;
+
+ if ( ( $user_id = wp_check_post_lock( $post_id ) ) && ( $user = get_userdata( $user_id ) ) && current_user_can( 'edit_post', $post_id ) ) {
+ $send = array( 'text' => sprintf( __( '%s is currently editing' ), $user->display_name ) );
+
+ if ( ( $avatar = get_avatar( $user->ID, 18 ) ) && preg_match( "|src='([^']+)'|", $avatar, $matches ) )
+ $send['avatar_src'] = $matches[1];
+
+ $checked[$key] = $send;
+ }
+ }
+ }
+
+ if ( ! empty( $checked ) )
+ $response['wp-check-locked-posts'] = $checked;
+
+ return $response;
+}
+add_filter( 'heartbeat_received', 'wp_check_locked_posts', 10, 3 );
+
+/**
+ * Check lock status on the New/Edit Post screen and refresh the lock
+ *
+ * @since 3.6.0
+ */
+function wp_refresh_post_lock( $response, $data, $screen_id ) {
+ if ( array_key_exists( 'wp-refresh-post-lock', $data ) ) {
+ $received = $data['wp-refresh-post-lock'];
+ $send = array();
+
+ if ( ! $post_id = absint( $received['post_id'] ) )
+ return $response;
+
+ if ( ! current_user_can('edit_post', $post_id) )
+ return $response;
+
+ if ( ( $user_id = wp_check_post_lock( $post_id ) ) && ( $user = get_userdata( $user_id ) ) ) {
+ $error = array(
+ 'text' => sprintf( __( '%s has taken over and is currently editing.' ), $user->display_name )
+ );
+
+ if ( $avatar = get_avatar( $user->ID, 64 ) ) {
+ if ( preg_match( "|src='([^']+)'|", $avatar, $matches ) )
+ $error['avatar_src'] = $matches[1];
+ }
+
+ $send['lock_error'] = $error;
+ } else {
+ if ( $new_lock = wp_set_post_lock( $post_id ) )
+ $send['new_lock'] = implode( ':', $new_lock );
+ }
+
+ $response['wp-refresh-post-lock'] = $send;
+ }
+
+ return $response;
+}
+add_filter( 'heartbeat_received', 'wp_refresh_post_lock', 10, 3 );
+
+/**
+ * Check nonce expiration on the New/Edit Post screen and refresh if needed
+ *
+ * @since 3.6.0
+ */
+function wp_refresh_post_nonces( $response, $data, $screen_id ) {
+ if ( array_key_exists( 'wp-refresh-post-nonces', $data ) ) {
+ $received = $data['wp-refresh-post-nonces'];
+ $response['wp-refresh-post-nonces'] = array( 'check' => 1 );
+
+ if ( ! $post_id = absint( $received['post_id'] ) )
+ return $response;
+
+ if ( ! current_user_can( 'edit_post', $post_id ) || empty( $received['post_nonce'] ) )
+ return $response;
+
+ if ( 2 === wp_verify_nonce( $received['post_nonce'], 'update-post_' . $post_id ) ) {
+ $response['wp-refresh-post-nonces'] = array(
+ 'replace' => array(
+ 'getpermalinknonce' => wp_create_nonce('getpermalink'),
+ 'samplepermalinknonce' => wp_create_nonce('samplepermalink'),
+ 'closedpostboxesnonce' => wp_create_nonce('closedpostboxes'),
+ '_ajax_linking_nonce' => wp_create_nonce( 'internal-linking' ),
+ '_wpnonce' => wp_create_nonce( 'update-post_' . $post_id ),
+ ),
+ 'heartbeatNonce' => wp_create_nonce( 'heartbeat-nonce' ),
+ );
+ }
+ }
+
+ return $response;
+}
+add_filter( 'heartbeat_received', 'wp_refresh_post_nonces', 10, 3 );
+
+/**
+ * Disable suspension of Heartbeat on the Add/Edit Post screens.
+ *
+ * @since 3.8.0
+ *
+ * @param array $settings An array of Heartbeat settings.
+ * @return array Filtered Heartbeat settings.
+ */
+function wp_heartbeat_set_suspension( $settings ) {
+ global $pagenow;
+
+ if ( 'post.php' === $pagenow || 'post-new.php' === $pagenow ) {
+ $settings['suspension'] = 'disable';
+ }
+
+ return $settings;
+}
+add_filter( 'heartbeat_settings', 'wp_heartbeat_set_suspension' );
+
+/**
+ * Autosave with heartbeat
+ *
+ * @since 3.9.0
+ */
+function heartbeat_autosave( $response, $data ) {
+ if ( ! empty( $data['wp_autosave'] ) ) {
+ $saved = wp_autosave( $data['wp_autosave'] );
+
+ if ( is_wp_error( $saved ) ) {
+ $response['wp_autosave'] = array( 'success' => false, 'message' => $saved->get_error_message() );
+ } elseif ( empty( $saved ) ) {
+ $response['wp_autosave'] = array( 'success' => false, 'message' => __( 'Error while saving.' ) );
+ } else {
+ /* translators: draft saved date format, see http://php.net/date */
+ $draft_saved_date_format = __( 'g:i:s a' );
+ /* translators: %s: date and time */
+ $response['wp_autosave'] = array( 'success' => true, 'message' => sprintf( __( 'Draft saved at %s.' ), date_i18n( $draft_saved_date_format ) ) );
+ }
+ }
+
+ return $response;
+}
+// Run later as we have to set DOING_AUTOSAVE for back-compat
+add_filter( 'heartbeat_received', 'heartbeat_autosave', 500, 2 );
+
+/**
+ * Disables autocomplete on the 'post' form (Add/Edit Post screens) for WebKit browsers,
+ * as they disregard the autocomplete setting on the editor textarea. That can break the editor
+ * when the user navigates to it with the browser's Back button. See #28037
+ *
+ * @since 4.0
+ */
+function post_form_autocomplete_off() {
+ global $is_safari, $is_chrome;
+
+ if ( $is_safari || $is_chrome ) {
+ echo ' autocomplete="off"';
+ }
+}
+
+add_action( 'post_edit_form_tag', 'post_form_autocomplete_off' );