react-use
Version:
Collection of React Hooks
59 lines (58 loc) • 2.18 kB
JavaScript
import { useEffect, useState } from 'react';
import { isBrowser, off, on } from './misc/util';
var patchHistoryMethod = function (method) {
var history = window.history;
var original = history[method];
history[method] = function (state) {
var result = original.apply(this, arguments);
var event = new Event(method.toLowerCase());
event.state = state;
window.dispatchEvent(event);
return result;
};
};
if (isBrowser) {
patchHistoryMethod('pushState');
patchHistoryMethod('replaceState');
}
var useLocationServer = function () { return ({
trigger: 'load',
length: 1,
}); };
var buildState = function (trigger) {
var _a = window.history, state = _a.state, length = _a.length;
var _b = window.location, hash = _b.hash, host = _b.host, hostname = _b.hostname, href = _b.href, origin = _b.origin, pathname = _b.pathname, port = _b.port, protocol = _b.protocol, search = _b.search;
return {
trigger: trigger,
state: state,
length: length,
hash: hash,
host: host,
hostname: hostname,
href: href,
origin: origin,
pathname: pathname,
port: port,
protocol: protocol,
search: search,
};
};
var useLocationBrowser = function () {
var _a = useState(buildState('load')), state = _a[0], setState = _a[1];
useEffect(function () {
var onPopstate = function () { return setState(buildState('popstate')); };
var onPushstate = function () { return setState(buildState('pushstate')); };
var onReplacestate = function () { return setState(buildState('replacestate')); };
on(window, 'popstate', onPopstate);
on(window, 'pushstate', onPushstate);
on(window, 'replacestate', onReplacestate);
return function () {
off(window, 'popstate', onPopstate);
off(window, 'pushstate', onPushstate);
off(window, 'replacestate', onReplacestate);
};
}, []);
return state;
};
var hasEventConstructor = typeof Event === 'function';
export default isBrowser && hasEventConstructor ? useLocationBrowser : useLocationServer;