From f8157123478712014a4f7cfeeca1a6c656e9ec38 Mon Sep 17 00:00:00 2001
From: KoloMl <thekolomoets@hotmail.com>
Date: Sat, 30 Mar 2024 21:29:31 +0400
Subject: [PATCH] Support building content scripts defined in manifest.json

---
 manifest.json            |  16 ++++++
 vite.config.extension.js | 109 +++++++++++++++++++++++++++++----------
 2 files changed, 98 insertions(+), 27 deletions(-)

diff --git a/manifest.json b/manifest.json
index d6cdca5..55ae042 100644
--- a/manifest.json
+++ b/manifest.json
@@ -5,6 +5,22 @@
     "host_permissions": [
         "*://*.furbooru.org/"
     ],
+    "content_scripts": [
+        {
+            "matches": [
+                "*://*.furbooru.org/",
+                "*://*.furbooru.org/images?*",
+                "*://*.furbooru.org/search?*",
+                "*://*.furbooru.org/tags/*"
+            ],
+            "js": [
+                "src/content/listing.js"
+            ],
+            "css": [
+                "src/styles/content/listing.scss"
+            ]
+        }
+    ],
     "action": {
         "default_popup": "index.html"
     },
diff --git a/vite.config.extension.js b/vite.config.extension.js
index 761c447..a7e7f9d 100644
--- a/vite.config.extension.js
+++ b/vite.config.extension.js
@@ -1,4 +1,4 @@
-import {defineConfig} from 'vite';
+import {defineConfig, normalizePath} from 'vite';
 import path from "path";
 import fs from "fs";
 import {load} from "cheerio";
@@ -7,19 +7,89 @@ import crypto from "crypto";
 const packageJsonPath = path.resolve(__dirname, 'package.json');
 const manifestJsonPath = path.resolve(__dirname, 'manifest.json');
 
+const buildDirectoryPath = path.resolve(__dirname, 'build');
+const contentScriptsDirectoryPath = path.resolve(buildDirectoryPath, 'assets', 'content');
+
+if (!fs.existsSync(manifestJsonPath)) {
+  throw new Error(
+    `The manifest.json file is missing from the root of the project.`
+  );
+}
+
+if (!fs.existsSync(packageJsonPath)) {
+  throw new Error(
+    `The package.json file is missing from the root of the project.`
+  );
+}
+
+const packageInformation = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
+const manifestInformation = JSON.parse(fs.readFileSync(manifestJsonPath, 'utf8'));
+
+/** @type {import('vite').RollupOptions} */
+const rollupOptions = {
+  input: {},
+  output: {
+    dir: contentScriptsDirectoryPath,
+    entryFileNames: '[name].js',
+    assetFileNames: '[name].[ext]',
+  }
+};
+
+function hashFilePath(filePath) {
+  return crypto
+    .createHash('sha256')
+    .update(filePath)
+    .digest('base64url')
+    .slice(0, 8);
+}
+
+// This is somewhat hacky, but it works for now. This code goes over the list of content scripts, adding them to the
+// rollupOptions.input object, and then modifying the content_scripts array to point to the built file names.
+// TODO: This fragment should probably somehow be moved into a plugin, together with the code that copies the source
+//  mainifest.json file to the build directory.
+if (manifestInformation?.['content_scripts']) {
+  manifestInformation['content_scripts'] = manifestInformation['content_scripts'].map(entry => {
+    if (entry.js) {
+      entry.js = entry.js.map(filePath => {
+        const fileName = path.basename(filePath);
+        const baseName = fileName.split('.').slice(0, -1).join('.');
+        const outputBaseName = `${baseName}-${hashFilePath(filePath)}`;
+
+        rollupOptions.input[outputBaseName] = filePath;
+
+        return normalizePath(
+          path.relative(
+            buildDirectoryPath,
+            path.resolve(contentScriptsDirectoryPath, outputBaseName + '.js')
+          )
+        );
+      });
+    }
+
+    if (entry.css) {
+      entry.css = entry.css.map(filePath => {
+        const fileName = path.basename(filePath);
+        const baseName = fileName.split('.').slice(0, -1).join('.');
+        const outputBaseName = `${baseName}-${hashFilePath(filePath)}`;
+
+        rollupOptions.input[outputBaseName] = filePath;
+
+        return normalizePath(
+          path.relative(
+            buildDirectoryPath,
+            path.resolve(contentScriptsDirectoryPath, outputBaseName + '.css')
+          )
+        );
+      })
+    }
+
+    return entry;
+  });
+}
+
 export default defineConfig({
   build: {
-    rollupOptions: {
-      input: {
-        listing: 'src/content/listing.js',
-      },
-      output: {
-        dir: path.resolve(__dirname, 'build', 'assets', 'content'),
-        inlineDynamicImports: false,
-        entryFileNames: '[name].js',
-        format: "iife"
-      },
-    },
+    rollupOptions,
     emptyOutDir: false,
   },
   resolve: {
@@ -63,21 +133,6 @@ export default defineConfig({
     {
       name: "bypass-manifest-extension",
       async buildEnd() {
-        if (!fs.existsSync(manifestJsonPath)) {
-          throw new Error(
-            `The manifest.json file is missing from the root of the project.`
-          );
-        }
-
-        if (!fs.existsSync(packageJsonPath)) {
-          throw new Error(
-            `The package.json file is missing from the root of the project.`
-          );
-        }
-
-        const packageInformation = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
-        const manifestInformation = JSON.parse(fs.readFileSync(manifestJsonPath, 'utf8'));
-
         manifestInformation.version = packageInformation.version;
 
         fs.writeFileSync(