From 7a50bd2cf62d7fde49cc6b714ce7aa2fe13439a8 Mon Sep 17 00:00:00 2001
From: Miorel-Lucian Palii <mlpalii@gmail.com>
Date: Sun, 13 Oct 2024 23:13:48 -0700
Subject: [PATCH] Process `__NEXT_DATA__` in Chrome extension

The LeetCode study plans (e.g. https://leetcode.com/studyplan/top-100-liked/) get their data in this way, so we inject middleware here for this use case as well.
---
 .../extension/injectJsonScriptMiddleware.ts   | 53 +++++++++++++++++++
 .../leetcode-zen-mode/src/extension/main.ts   |  6 +++
 workspaces/util/src/jsonParseSafe.ts          | 11 ++++
 3 files changed, 70 insertions(+)
 create mode 100644 workspaces/leetcode-zen-mode/src/extension/injectJsonScriptMiddleware.ts
 create mode 100644 workspaces/util/src/jsonParseSafe.ts

diff --git a/workspaces/leetcode-zen-mode/src/extension/injectJsonScriptMiddleware.ts b/workspaces/leetcode-zen-mode/src/extension/injectJsonScriptMiddleware.ts
new file mode 100644
index 00000000..2d0f34e8
--- /dev/null
+++ b/workspaces/leetcode-zen-mode/src/extension/injectJsonScriptMiddleware.ts
@@ -0,0 +1,53 @@
+import invariant from "invariant";
+import type { JsonValue } from "type-fest";
+
+import { jsonParseSafe } from "@code-chronicles/util/jsonParseSafe";
+
+type ProxiedPropertyKey = "innerHTML" | "innerText" | "textContent";
+
+type MiddlewareFn = (
+  data: JsonValue,
+  script: HTMLScriptElement,
+  property: ProxiedPropertyKey,
+) => JsonValue;
+
+function inject<TProto extends { constructor: Function }>(
+  proto: TProto,
+  property: ProxiedPropertyKey,
+  middlewareFn: MiddlewareFn,
+): void {
+  const prevDescriptor = Object.getOwnPropertyDescriptor(proto, property);
+
+  invariant(
+    prevDescriptor && prevDescriptor.get,
+    `\`${proto.constructor.name}.prototype.${property}\` property descriptor didn't have the expected form!`,
+  );
+  const prevDescriptorGet = prevDescriptor.get;
+
+  Object.defineProperty(HTMLScriptElement.prototype, property, {
+    ...prevDescriptor,
+    get(this: HTMLScriptElement) {
+      const data = prevDescriptorGet.call(this);
+
+      // If the data doesn't parse as JSON, we pass it through unchanged.
+      // If it does parse as JSON, we'll run the middleware.
+      const parsedData = jsonParseSafe(data);
+      if (parsedData) {
+        return JSON.stringify(middlewareFn(parsedData.data, this, property));
+      }
+
+      return data;
+    },
+  });
+}
+
+/**
+ * Injects a function to process and possibly replace JSON data stored within
+ * <script> tags. The middleware will run when properties such as `innerHTML`
+ * get accessed, and it must return the possibly updated JSON data.
+ */
+export function injectJsonScriptMiddleware(middlewareFn: MiddlewareFn): void {
+  inject(Element.prototype, "innerHTML", middlewareFn);
+  inject(HTMLElement.prototype, "innerText", middlewareFn);
+  inject(Node.prototype, "textContent", middlewareFn);
+}
diff --git a/workspaces/leetcode-zen-mode/src/extension/main.ts b/workspaces/leetcode-zen-mode/src/extension/main.ts
index 8979efc8..ac2a5b84 100644
--- a/workspaces/leetcode-zen-mode/src/extension/main.ts
+++ b/workspaces/leetcode-zen-mode/src/extension/main.ts
@@ -1,5 +1,6 @@
 import { mapJsonBlobData } from "@code-chronicles/util/mapJsonBlobData";
 
+import { injectJsonScriptMiddleware } from "./injectJsonScriptMiddleware.ts";
 import { injectXhrBlobResponseMiddleware } from "./injectXhrBlobResponseMiddleware.ts";
 import { patchWebpackChunkLoading } from "./patchWebpackChunkLoading.ts";
 import { rewriteLeetCodeGraphQLData } from "./rewriteLeetCodeGraphQLData.ts";
@@ -23,6 +24,11 @@ function main() {
     return blob;
   });
 
+  // LeetCode also reads data stored as JSON within some <script> tags within
+  // the page, for example the __NEXT_DATA__ used by Next.js. We will inject
+  // the middleware there as well.
+  injectJsonScriptMiddleware(rewriteLeetCodeGraphQLData);
+
   // Additionally, we will patch some of the actual page code! We will do so
   // by trying to intercept `webpack` chunk loading, so that we can patch the
   // modules used by the page.
diff --git a/workspaces/util/src/jsonParseSafe.ts b/workspaces/util/src/jsonParseSafe.ts
new file mode 100644
index 00000000..d5997dc9
--- /dev/null
+++ b/workspaces/util/src/jsonParseSafe.ts
@@ -0,0 +1,11 @@
+import type { JsonValue } from "type-fest";
+
+export function jsonParseSafe(
+  ...args: Parameters<typeof JSON.parse>
+): { data: JsonValue } | undefined {
+  try {
+    return { data: JSON.parse(...args) };
+  } catch {
+    return undefined;
+  }
+}