Track Teams as meeting category

Author: alanhamlett

src/background.ts

@@ -60,12 +60,14 @@ browser.tabs.onUpdated.addListener(async (tabId, changeInfo) => {
   }
 });
 
-browser.runtime.onMessage.addListener(async (request: { task: string }, sender) => {
-  if (request.task === 'handleActivity') {
-    if (!sender.tab?.id) return;
-    await WakaTimeCore.handleActivity(sender.tab.id);
-  }
-});
+browser.runtime.onMessage.addListener(
+  async (request: { isPassiveActivity?: boolean; task: string }, sender) => {
+    if (request.task === 'handleActivity') {
+      if (!sender.tab?.id) return;
+      await WakaTimeCore.handleActivity(sender.tab.id, request.isPassiveActivity);
+    }
+  },
+);
 
 /**
  * "Persistent" service worker via bug exploit

src/core/WakaTimeCore.ts

@@ -1,7 +1,7 @@
-import browser, { Tabs } from 'webextension-polyfill';
 import { openDB } from 'idb';
 import moment from 'moment';
 import { v4 as uuid4 } from 'uuid';
+import browser, { Tabs } from 'webextension-polyfill';
 import config, { ExtensionStatus } from '../config/config';
 import { EntityType, Heartbeat, HeartbeatsBulkResponse } from '../types/heartbeats';
 import getDomainFromUrl, { getDomain } from '../utils/getDomainFromUrl';
@@ -88,7 +88,7 @@ class WakaTimeCore {
     return site?.projectName;
   }
 
-  async handleActivity(tabId: number) {
+  async handleActivity(tabId: number, isPassiveActivity: boolean = false) {
     const settings = await getSettings();
     if (!settings.loggingEnabled) {
       await changeExtensionStatus('trackingDisabled');
@@ -109,7 +109,10 @@ class WakaTimeCore {
       await changeExtensionStatus('allGood');
     }
 
-    const heartbeat = await this.buildHeartbeat(url, settings, activeTab);
+    const heartbeat = await this.buildHeartbeat(url, settings, activeTab, isPassiveActivity);
+    if (!heartbeat) {
+      return;
+    }
 
     if (!this.shouldSendHeartbeat(heartbeat)) return;
 
@@ -130,7 +133,12 @@ class WakaTimeCore {
     return activeTab;
   }
 
-  async buildHeartbeat(url: string, settings: Settings, tab: browser.Tabs.Tab): Promise<Heartbeat> {
+  async buildHeartbeat(
+    url: string,
+    settings: Settings,
+    tab: browser.Tabs.Tab,
+    isPassiveActivity: boolean,
+  ): Promise<Heartbeat | undefined> {
     if (!tab.id) {
       throw Error('Missing tab id.');
     }
@@ -141,6 +149,10 @@ class WakaTimeCore {
       }
     ).heartbeat;
 
+    if (isPassiveActivity && !heartbeat) {
+      return;
+    }
+
     const entity = settings.loggingType === 'domain' ? getDomainFromUrl(url) : url;
 
     const projectNameFromList = this.getProjectNameFromList(url, settings);

src/types/sites.ts

@@ -8,6 +8,7 @@ export enum KnownSite {
   github = 'github',
   gitlab = 'gitlab',
   googlemeet = 'googlemeet',
+  msteams = 'msteams',
   slack = 'slack',
   stackoverflow = 'stackoverflow',
   travisci = 'travisci',
@@ -29,6 +30,7 @@ export type HeartbeatParser = (url: string) => OptionalHeartbeat | undefined;
 
 export interface SiteParser {
   parser: HeartbeatParser;
+  trackWithoutMouseMoving?: boolean;
   urls: RegExp[] | string[];
 }
 

src/utils/sites.ts

@@ -281,6 +281,23 @@ const GoogleMeet: HeartbeatParser = (_url: string): OptionalHeartbeat | undefine
   };
 };
 
+const MicrosoftTeams: HeartbeatParser = (_url: string): OptionalHeartbeat | undefined => {
+  const leaveButton = document.querySelector('#hangup-button');
+  if (!leaveButton) return;
+
+  const title = document.querySelector('title')?.innerText;
+  if (!title) return;
+
+  const meetingTitle = title.split(' | ')[1];
+  if (!meetingTitle.trim()) return;
+
+  return {
+    category: Category.meeting,
+    plugin: 'Microsoft Teams',
+    project: meetingTitle.trim(),
+  };
+};
+
 const Slack: HeartbeatParser = (_url: string): OptionalHeartbeat | undefined => {
   const title = document.querySelector('title')?.textContent?.split(' - ');
   if (!title || title.length < 3 || title[-1] !== 'Slack') {
@@ -374,8 +391,14 @@ const SITES: Record<KnownSite, SiteParser> = {
   },
   googlemeet: {
     parser: GoogleMeet,
+    trackWithoutMouseMoving: true,
     urls: [/^https?:\/\/meet.google.com\//],
   },
+  msteams: {
+    parser: MicrosoftTeams,
+    trackWithoutMouseMoving: true,
+    urls: [/^https:\/\/teams.live.com\/v2\//, /^https:\/\/teams.microsoft.com\/v1\//],
+  },
   slack: {
     parser: Slack,
     urls: [/^https:\/\/app.slack.com\/client\//],
@@ -394,6 +417,7 @@ const SITES: Record<KnownSite, SiteParser> = {
   },
   zoom: {
     parser: Zoom,
+    trackWithoutMouseMoving: true,
     urls: [/^https:\/\/(.+\.)?zoom.us\/[^?]+\/join/],
   },
 };

src/wakatimeScript.ts

@@ -30,6 +30,10 @@ const sendHeartbeat = debounce(async () => {
   void chrome.runtime.sendMessage({ task: 'handleActivity' });
 });
 
+const sendHeartbeatMeeting = async () => {
+  await chrome.runtime.sendMessage({ isPassiveActivity: true, task: 'handleActivity' });
+};
+
 chrome.runtime.onMessage.addListener(
   (request: { task: string; url: string }, sender, sendResponse) => {
     if (request.task === 'getHeartbeatFromPage') {
@@ -49,17 +53,14 @@ document.body.addEventListener('click', sendHeartbeat, true);
 document.body.addEventListener('keypress', sendHeartbeat, true);
 
 const checkIfInAMeeting = () => {
-  if (!window.location.href.startsWith('https://meet.google.com/')) {
+  const site = getSite(window.location.href);
+  if (!site?.trackWithoutMouseMoving) {
     return;
   }
 
-  const isActiveMeeting = !!document.querySelector('[data-meeting-title]');
-  if (isActiveMeeting) {
-    sendHeartbeat();
-  }
+  void sendHeartbeatMeeting();
 
   setTimeout(checkIfInAMeeting, oneMinute);
 };
 
-// Google Meet
 checkIfInAMeeting();