feat: improve rewrite rules

Author: Justineo

rewrite/config.ts

@@ -1,11 +1,30 @@
-export const SRC_PATTERN = /\/figma_app/
-
-export const REWRITE_PATTERN = /\.appModel\.isReadOnly/g
-
-export const REWRITE_REPLACER = '.appModel.__isReadOnly__'
-
-const MARKERS = ['.appModel.isReadOnly']
+export type Replacement = {
+  pattern: string | RegExp
+  replacer: string | ((...args: any[]) => string)
+}
 
-export function matchFile(src: string, content: string) {
-  return SRC_PATTERN.test(src) && MARKERS.every((marker) => content.includes(marker))
+export type Group = {
+  markers?: string[]
+  replacements: Replacement[]
 }
+
+export const GROUPS: Group[] = [
+  {
+    markers: ['.appModel.isReadOnly'],
+    replacements: [
+      {
+        pattern: /\.appModel\.isReadOnly/g,
+        replacer: '.appModel.__isReadOnly__'
+      }
+    ]
+  },
+  {
+    markers: ['dispnf.fyufotjpo;00', 'np{.fyufotjpo;00'],
+    replacements: [
+      {
+        pattern: /dispnf\.fyufotjpo;00|np{\.fyufotjpo;00/g,
+        replacer: 'FIGMA_PLEASE_STOP'
+      }
+    ]
+  }
+]

rewrite/figma.ts

@@ -1,4 +1,5 @@
-import { matchFile, REWRITE_PATTERN, REWRITE_REPLACER } from './config'
+import type { Group } from './config'
+import { GROUPS } from './config'
 
 async function rewriteScript() {
   const current = document.currentScript as HTMLScriptElement
@@ -13,15 +14,38 @@ async function rewriteScript() {
     current.replaceWith(script)
   }
 
+  function applyGroup(content: string, group: Group) {
+    const markers = group.markers || []
+
+    if (!markers.every((marker) => content.includes(marker))) {
+      return content
+    }
+
+    let out = content
+    for (const { pattern, replacer } of group.replacements) {
+      if (typeof pattern === 'string') {
+        // @ts-ignore
+        out = out.replaceAll(pattern, replacer)
+      } else {
+        // @ts-ignore
+        out = out.replace(pattern, replacer)
+      }
+    }
+    return out
+  }
+
   try {
-    let content = await (await fetch(src)).text()
+    const original = await (await fetch(src)).text()
+    let content = original
+
+    for (const group of GROUPS) {
+      content = applyGroup(content, group)
+    }
 
-    if (matchFile(src, content)) {
-      content = content.replace(REWRITE_PATTERN, REWRITE_REPLACER)
+    if (content !== original) {
       console.log(`Rewrote script: ${src}`)
     }
 
-    // delete window.figma may throw Error in strict mode
     content = content.replaceAll('delete window.figma', 'window.figma = undefined')
 
     Object.defineProperty(document, 'currentScript', {