@wordpress/edit-post/src/components/browser-url/use-classic-revision-redirect.js

Edit Post module for WordPress.

import { useRegistry, useSelect } from '@wordpress/data';
import { store as coreStore } from '@wordpress/core-data';
import { store as editorStore } from '@wordpress/editor';
import { useEffect } from '@wordpress/element';
import { addQueryArgs } from '@wordpress/url';
import { unlock } from '../../lock-unlock';

/**
 * Redirects an editor URL that already contains a revision ID to the classic
 * screen when visual revisions are disabled. Revision buttons already link
 * there, but the URL can load before the editor finishes checking for meta boxes.
 */
export default function useClassicRevisionRedirect() {
	const registry = useRegistry();
	const { disableVisualRevisions, currentRevisionId, postType, postId } =
		useSelect( ( select ) => {
			const editor = select( editorStore );
			return {
				disableVisualRevisions:
					!! editor.getEditorSettings().disableVisualRevisions,
				currentRevisionId: unlock( editor ).getCurrentRevisionId(),
				postType: editor.getCurrentPostType(),
				postId: editor.getCurrentPostId(),
			};
		}, [] );
	const hasClassicRevisionScreen = ! [
		'wp_template',
		'wp_template_part',
	].includes( postType );

	useEffect( () => {
		if (
			! disableVisualRevisions ||
			! hasClassicRevisionScreen ||
			! currentRevisionId ||
			! postType ||
			! postId
		) {
			return;
		}

		let cancelled = false;
		// Invalid IDs trigger `wp_die()` before JavaScript loads. Check that the
		// revision belongs to this post before redirecting.
		registry
			.resolveSelect( coreStore )
			.getRevision( 'postType', postType, postId, currentRevisionId, {
				context: 'edit',
				_fields: 'id',
			} )
			.then( ( revision ) => {
				if ( cancelled || ! revision ) {
					return;
				}

				const editor = registry.select( editorStore );
				const isStillSelected =
					!! editor.getEditorSettings().disableVisualRevisions &&
					editor.getCurrentPostType() === postType &&
					String( editor.getCurrentPostId() ) === String( postId ) &&
					unlock( editor ).getCurrentRevisionId() ===
						currentRevisionId;
				if ( ! isStillSelected ) {
					return;
				}

				// Replace the editor URL so Back does not reopen it and trigger
				// the redirect again.
				window.location.replace(
					addQueryArgs( 'revision.php', {
						revision: currentRevisionId,
					} )
				);
			} );

		return () => {
			cancelled = true;
		};
	}, [
		disableVisualRevisions,
		hasClassicRevisionScreen,
		currentRevisionId,
		postType,
		postId,
		registry,
	] );
}