You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.3 KiB
46 lines
1.3 KiB
import type { PropsWithChildren } from 'react'; |
|
import React from 'react'; |
|
|
|
import { createBrowserHistory } from 'history'; |
|
import { Router as OriginalRouter } from 'react-router'; |
|
|
|
import { layoutFromWindow } from 'mastodon/is_mobile'; |
|
|
|
interface MastodonLocationState { |
|
fromMastodon?: boolean; |
|
mastodonModalKey?: string; |
|
} |
|
|
|
const browserHistory = createBrowserHistory< |
|
MastodonLocationState | undefined |
|
>(); |
|
const originalPush = browserHistory.push.bind(browserHistory); |
|
const originalReplace = browserHistory.replace.bind(browserHistory); |
|
|
|
browserHistory.push = (path: string, state?: MastodonLocationState) => { |
|
state = state ?? {}; |
|
state.fromMastodon = true; |
|
|
|
if (layoutFromWindow() === 'multi-column' && !path.startsWith('/deck')) { |
|
originalPush(`/deck${path}`, state); |
|
} else { |
|
originalPush(path, state); |
|
} |
|
}; |
|
|
|
browserHistory.replace = (path: string, state?: MastodonLocationState) => { |
|
if (browserHistory.location.state?.fromMastodon) { |
|
state = state ?? {}; |
|
state.fromMastodon = true; |
|
} |
|
|
|
if (layoutFromWindow() === 'multi-column' && !path.startsWith('/deck')) { |
|
originalReplace(`/deck${path}`, state); |
|
} else { |
|
originalReplace(path, state); |
|
} |
|
}; |
|
|
|
export const Router: React.FC<PropsWithChildren> = ({ children }) => { |
|
return <OriginalRouter history={browserHistory}>{children}</OriginalRouter>; |
|
};
|
|
|