watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-04 04:37:11 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Capture pushState events.

Browse source
This commit is contained in:
Mike Cao 2020-07-18 22:51:17 -07:00
parent 58a1c63407
commit eebf145efc
6 changed files with 81 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

110
scripts/umami/index.js
View file

@ -1,63 +1,89 @@
import 'promise-polyfill/src/polyfill';
import 'unfetch/polyfill';
const HOST_URL = process.env.UMAMI_URL;
const SESSION_VAR = 'umami.session';
const {
screen: { width, height },
navigator: { language },
location: { hostname, pathname, search },
localStorage: store,
document,
history,
} = window;
const post = (url, params) =>
fetch(url, {
method: 'post',
cache: 'no-cache',
headers: {
Accept: 'application/json',
'Content-Type': 'application/json',
},
body: JSON.stringify(params),
}).then(res => res.json());
const createSession = data =>
post(`${HOST_URL}/api/session`, data).then(({ success, ...session }) => {
if (success) {
store.setItem(SESSION_VAR, JSON.stringify(session));
return success;
}
});
const getSession = () => JSON.parse(store.getItem(SESSION_VAR));
const pageView = (url, referrer) =>
post(`${HOST_URL}/api/collect`, {
type: 'pageview',
payload: { url, referrer, session: getSession() },
}).then(({ success }) => {
if (!success) {
store.removeItem(SESSION_VAR);
}
return success;
});
const script = document.querySelector('script[data-website-id]');
if (script) {
const website_id = script.getAttribute('data-website-id');
if (website_id) {
const referrer = document.referrer;
const sessionKey = 'umami.session';
const hostUrl = new URL(script.src).origin;
const screen = `${width}x${height}`;
const url = `${pathname}${search}`;
const data = { website_id, hostname, url, screen, language };
let currentUrl = `${pathname}${search}`;
let currenrRef = document.referrer;
if (!store.getItem(SESSION_VAR)) {
createSession(data).then(success => success && pageView(url, referrer));
} else {
pageView(url, referrer).then(success => !success && createSession(data));
}
const post = (url, params) =>
fetch(url, {
method: 'post',
cache: 'no-cache',
headers: {
Accept: 'application/json',
'Content-Type': 'application/json',
},
body: JSON.stringify(params),
}).then(res => res.json());
const createSession = data =>
post(`${hostUrl}/api/session`, data).then(({ success, ...session }) => {
if (success) {
store.setItem(sessionKey, JSON.stringify(session));
return success;
}
});
const getSession = () => JSON.parse(store.getItem(sessionKey));
const getSessionData = url => ({ website_id, hostname, url, screen, language });
const pageView = (url, referrer) =>
post(`${hostUrl}/api/collect`, {
type: 'pageview',
payload: { url, referrer, session: getSession() },
}).then(({ success }) => {
if (!success) {
store.removeItem(sessionKey);
}
return success;
});
const execute = (url, referrer) => {
const data = getSessionData(url);
if (!store.getItem(sessionKey)) {
createSession(data).then(success => success && pageView(url, referrer));
} else {
pageView(url, referrer).then(success => !success && createSession(data));
}
};
const handlePush = (state, title, url) => {
currenrRef = currentUrl;
currentUrl = url;
execute(currentUrl, currenrRef);
};
const hook = (type, cb) => {
const orig = history[type];
return (state, title, url) => {
const args = [state, title, url];
cb.apply(null, args);
return orig.apply(history, args);
};
};
history.pushState = hook('pushState', handlePush);
history.replaceState = hook('replaceState', handlePush);
execute(currentUrl, currenrRef);
}
}