Last "external" route tracking

useHistoryMemorizer.ts

import { useEffect } from 'react';
import { useLastLocation } from 'react-router-last-location';
import create from 'zustand';

type LocationMap = {
	[key: string]: string | undefined;
};

type HistoryStore = {
	locations: LocationMap;
	setLocation: (location: string, lastLocation: string | undefined) => void;
};

const useHistoryStore = create<HistoryStore>((set) => ({
	locations: {},
	setLocation: (location, lastLocation) => {
		set(() => {
			const newState: { locations: LocationMap } = { locations: {} };
			newState.locations[location] = lastLocation;
			return newState;
		});
	}
}));

export const useHistoryMemorizer = (location: string) => {
	const lastLocation = useLastLocation();
	const { backUrl, setUrl } = useHistoryStore((state) => ({
		backUrl: state.locations[location],
		setUrl: state.setLocation
	}));

	useEffect(() => {
		if (!lastLocation?.pathname.includes(location)) {
			setUrl(location, lastLocation?.pathname);
		}
	});

	return backUrl;
};