Retrieve the commit SHA from the details-menu element for pull requests, as refs/pull/{number} does not yield results from the qlty.sh API

Author: dejan-lukic

src/api.ts

@@ -1,25 +1,9 @@
 export async function readCoverageData(
   path: string,
+  reference: string,
 ): Promise<GetFileCoverageResponse | null> {
   const pathParts = window.location.pathname.split("/");
   const [workspace, project] = pathParts.slice(1, 3);
-
-  let commitShaIndex = pathParts.indexOf("pull");
-  let reference: string | null = null;
-  if (commitShaIndex >= 0) {
-    reference = `refs/pull/${pathParts[commitShaIndex + 1]}`;
-  } else {
-    commitShaIndex = pathParts.indexOf("commit");
-    if (commitShaIndex >= 0) {
-      reference = pathParts[commitShaIndex + 1];
-    }
-  }
-
-  if (!reference) {
-    console.error("[qlty] No commit SHA found in the URL.");
-    return null;
-  }
-
   return new Promise((resolve, reject) => {
     chrome.runtime.sendMessage(
       {

src/github.ts

@@ -10,8 +10,66 @@ export function tryInjectDiffUI(): boolean {
   }
 }
 
+function getRef() {
+  const commitSha = getPullRequestCommitSha();
+  if (commitSha) {
+    return commitSha;
+  }
+
+  const ref = getRefFromUrl();
+  if (ref) {
+    return ref;
+  }
+
+  return null;
+}
+
+function getPullRequestCommitSha() {
+  const getRepoElement = document.querySelector("get-repo");
+  if (!getRepoElement) {
+    return null;
+  }
+
+  const detailsMenu = getRepoElement.querySelector('details-menu');
+  if (!detailsMenu) {
+    return null;
+  }
+
+  // src="/{org}/{repo}/pull/{number}/open_with_menu?ref={sha}"
+  const src = detailsMenu.getAttribute('src');
+  const match = src?.match(/ref=([a-z0-9]+)/);
+  if (!match || match.length < 2) {
+    return null;
+  }
+
+  return match[1];
+}
+
+function getRefFromUrl() {
+  const pathParts = window.location.pathname.split("/");
+
+  let commitShaIndex = pathParts.indexOf("pull");
+  let reference: string | null = null;
+  if (commitShaIndex >= 0) {
+    reference = `refs/pull/${pathParts[commitShaIndex + 1]}`;
+  } else {
+    commitShaIndex = pathParts.indexOf("commit");
+    if (commitShaIndex >= 0) {
+      reference = pathParts[commitShaIndex + 1];
+    }
+  }
+
+  return reference;
+}
+
 async function loadCoverageForPath(path: string): Promise<FileCoverage | null> {
-  const data = await readCoverageData(path);
+  const ref = getRef();
+  if (!ref) {
+    console.warn("[qlty] No reference found for coverage data");
+    return null;
+  }
+
+  const data = await readCoverageData(path, ref);
   const coverage = data?.files.find(
     (file) => normalizePath(file.path) === normalizePath(path),
   );