From c5832e3980dd1715a048cb01ed428404122b9e9a Mon Sep 17 00:00:00 2001
From: Andrei Vladykin <vladikin@gmail.com>
Date: Sat, 1 Apr 2023 04:14:45 +0400
Subject: [PATCH] ISSUE-181: Support for Figma. Refactor Canva support. (#184)

* ISSUE-181: Support for Figma. Refactor Canva support.

* ISSUE-181: Missed project details for Figma.

* ISSUE-181: code review. missed conditional break for non project page

* ISSUE-181: bump manifest's version up to 3.0.10

* ISSUE-181: fix for sendPostRequestToApi
---
 src/manifests/chrome.json  |  2 +-
 src/manifests/firefox.json |  2 +-
 src/wakatimeScript.ts      | 62 +++++++++++++++++++++++++++++++++-----
 3 files changed, 56 insertions(+), 10 deletions(-)

diff --git a/src/manifests/chrome.json b/src/manifests/chrome.json
index 019accc..f5b6ed4 100644
--- a/src/manifests/chrome.json
+++ b/src/manifests/chrome.json
@@ -33,5 +33,5 @@
     "page": "options.html"
   },
   "permissions": ["alarms", "tabs", "storage", "idle"],
-  "version": "3.0.9"
+  "version": "3.0.10"
 }
diff --git a/src/manifests/firefox.json b/src/manifests/firefox.json
index 4cddeb3..aa12d7e 100644
--- a/src/manifests/firefox.json
+++ b/src/manifests/firefox.json
@@ -46,5 +46,5 @@
     "storage",
     "idle"
   ],
-  "version": "3.0.9"
+  "version": "3.0.10"
 }
diff --git a/src/wakatimeScript.ts b/src/wakatimeScript.ts
index 0f1a0f3..2978eb1 100644
--- a/src/wakatimeScript.ts
+++ b/src/wakatimeScript.ts
@@ -92,6 +92,11 @@ const sendPostRequestToApi = async (
   hostname = '',
 ): Promise<void> => {
   try {
+    const items = await browser.storage.sync.get({
+      apiUrl: config.apiUrl,
+      heartbeatApiEndPoint: config.heartbeatApiEndPoint,
+    });
+
     const request: RequestInit = {
       body: JSON.stringify(payload),
       credentials: 'omit',
@@ -102,7 +107,10 @@ const sendPostRequestToApi = async (
         'X-Machine-Name': hostname,
       };
     }
-    const response = await fetch(`${config.heartbeatApiUrl}?api_key=${apiKey}`, request);
+    const response = await fetch(
+      `${items.apiUrl}${items.heartbeatApiEndPoint}?api_key=${apiKey}`,
+      request,
+    );
     await response.json();
   } catch (err: unknown) {
     console.log('Error', err);
@@ -243,21 +251,59 @@ const recordHeartbeat = async (apiKey: string, payload: Record<string, unknown>)
   }
 };
 
+interface DesignProject {
+  editor: string;
+  language: string;
+  project: string;
+}
+
+const parseCanva = (): DesignProject | undefined => {
+  const canvaProject = document.getElementsByClassName('rF765A');
+  if (canvaProject.length === 0) return;
+
+  const projectName = (document.head.querySelector('meta[property="og:title"]') as HTMLMetaElement)
+    .content;
+  return {
+    editor: 'Canva',
+    language: 'Canva Design',
+    project: projectName,
+  };
+};
+
+const parseFigma = (): DesignProject | undefined => {
+  const figmaProject = document.getElementsByClassName('gpu-view-content');
+  if (figmaProject.length === 0) return;
+
+  const projectName = (document.querySelector('span[data-testid="filename"]') as HTMLElement)
+    .innerText;
+  return {
+    editor: 'Figma',
+    language: 'Figma Design',
+    project: projectName,
+  };
+};
+
+const getParser: {
+  [key: string]:
+    | (() => { editor: string; language: string; project: string } | undefined)
+    | undefined;
+} = {
+  'www.canva.com': parseCanva,
+  'www.figma.com': parseFigma,
+};
+
 const init = async () => {
   const apiKey = await getApiKey();
   if (!apiKey) return;
 
   const { hostname } = document.location;
-  const canvaProject = document.getElementsByClassName('rF765A');
 
-  if (hostname === 'www.canva.com' && canvaProject.length > 0) {
-    const ogTitle = (document.head.querySelector('meta[property="og:title"]') as HTMLMetaElement)
-      .content;
+  const projectDetails = getParser[hostname]?.();
+
+  if (projectDetails) {
     await recordHeartbeat(apiKey, {
       category: 'Designing',
-      editor: 'Canva',
-      language: 'Canva Design',
-      project: ogTitle,
+      ...projectDetails,
     });
   }
 };