116 lines
3.1 KiB
JavaScript
116 lines
3.1 KiB
JavaScript
/* global browser */
|
|
|
|
// Core
|
|
var WakaTimeCore = require("./core/WakaTimeCore").default;
|
|
|
|
// initialize class
|
|
var wakatime = new WakaTimeCore();
|
|
|
|
// Holds currently open connections (ports) with devtools
|
|
// Uses tabId as index key.
|
|
var connections = {};
|
|
|
|
// Add a listener to resolve alarms
|
|
browser.alarms.onAlarm.addListener(function (alarm) {
|
|
// |alarm| can be undefined because onAlarm also gets called from
|
|
// window.setTimeout on old chrome versions.
|
|
if (alarm && alarm.name == 'heartbeatAlarm') {
|
|
|
|
console.log('recording a heartbeat - alarm triggered');
|
|
|
|
wakatime.recordHeartbeat();
|
|
}
|
|
});
|
|
|
|
// Create a new alarm for heartbeats.
|
|
browser.alarms.create('heartbeatAlarm', {periodInMinutes: 2});
|
|
|
|
/**
|
|
* Whenever a active tab is changed it records a heartbeat with that tab url.
|
|
*/
|
|
browser.tabs.onActivated.addListener(function (activeInfo) {
|
|
|
|
browser.tabs.get(activeInfo.tabId).then(function (tab) {
|
|
|
|
console.log('recording a heartbeat - active tab changed');
|
|
|
|
wakatime.recordHeartbeat();
|
|
});
|
|
|
|
});
|
|
|
|
/**
|
|
* Whenever a active window is changed it records a heartbeat with the active tab url.
|
|
*/
|
|
browser.windows.onFocusChanged.addListener(function (windowId) {
|
|
|
|
if (windowId != browser.windows.WINDOW_ID_NONE) {
|
|
console.log('recording a heartbeat - active window changed');
|
|
|
|
wakatime.recordHeartbeat();
|
|
} else {
|
|
console.log('lost focus');
|
|
}
|
|
|
|
});
|
|
|
|
/**
|
|
* Whenever any tab is updated it checks if the updated tab is the tab that is
|
|
* currently active and if it is, then it records a heartbeat.
|
|
*/
|
|
browser.tabs.onUpdated.addListener(function (tabId, changeInfo, tab) {
|
|
|
|
if (changeInfo.status === 'complete') {
|
|
// Get current tab URL.
|
|
browser.tabs.query({currentWindow: true, active: true}).then(function(tabs) {
|
|
// If tab updated is the same as active tab
|
|
if (tabId == tabs[0].id) {
|
|
console.log('recording a heartbeat - tab updated');
|
|
|
|
wakatime.recordHeartbeat();
|
|
}
|
|
});
|
|
}
|
|
|
|
});
|
|
|
|
|
|
/**
|
|
* This is in charge of detecting if devtools are opened or closed
|
|
* and sending a heartbeat depending on that.
|
|
*/
|
|
browser.runtime.onConnect.addListener(function (port) {
|
|
|
|
if (port.name == "devtools-page") {
|
|
|
|
// Listen to messages sent from the DevTools page
|
|
port.onMessage.addListener(function (message, sender, sendResponse) {
|
|
if (message.name == "init") {
|
|
|
|
connections[message.tabId] = port;
|
|
|
|
wakatime.setTabsWithDevtoolsOpen(Object.keys(connections));
|
|
|
|
wakatime.recordHeartbeat();
|
|
}
|
|
});
|
|
|
|
port.onDisconnect.addListener(function (port) {
|
|
|
|
var tabs = Object.keys(connections);
|
|
|
|
for (var i = 0, len = tabs.length; i < len; i ++) {
|
|
if (connections[tabs[i]] == port) {
|
|
delete connections[tabs[i]];
|
|
break;
|
|
}
|
|
}
|
|
|
|
wakatime.setTabsWithDevtoolsOpen(Object.keys(connections));
|
|
|
|
wakatime.recordHeartbeat();
|
|
});
|
|
|
|
}
|
|
});
|