From c37f680e9f421da78c3771e61f42f54bceef18d2 Mon Sep 17 00:00:00 2001
From: KoloMl <thekolomoets@hotmail.com>
Date: Fri, 8 Aug 2025 19:38:52 +0400
Subject: [PATCH] Provide current site name & identifier for build-time
 modifications

---
 .vite/lib/content-scripts.js | 17 ++++++++++++-
 package.json                 |  1 +
 vite.config.ts               | 49 ++++++++++++++++++++++++------------
 3 files changed, 50 insertions(+), 17 deletions(-)

diff --git a/.vite/lib/content-scripts.js b/.vite/lib/content-scripts.js
index f9385ca..5a84761 100644
--- a/.vite/lib/content-scripts.js
+++ b/.vite/lib/content-scripts.js
@@ -2,6 +2,8 @@ import { build } from "vite";
 import { createHash } from "crypto";
 import path from "path";
 import fs from "fs";
+import { SwapDefinedVariablesPlugin } from "../plugins/swap-defined-variables.js";
+import { ScssViteReadEnvVariableFunctionPlugin } from "../plugins/scss-read-env-variable-function.js";
 
 /**
  * Create the result base file name for the file.
@@ -192,6 +194,15 @@ export async function buildScriptsAndStyles(buildOptions) {
           .get(fileName)
           ?.push(...dependencies);
       }),
+      ScssViteReadEnvVariableFunctionPlugin(),
+      SwapDefinedVariablesPlugin({
+        envVariable: 'SITE',
+        expectedValue: 'derpibooru',
+        define: {
+          __CURRENT_SITE__: JSON.stringify('derpibooru'),
+          __CURRENT_SITE_NAME__: JSON.stringify('Derpibooru'),
+        }
+      }),
     ]
   });
 
@@ -215,7 +226,11 @@ export async function buildScriptsAndStyles(buildOptions) {
     },
     plugins: [
       wrapScriptIntoIIFE(),
-    ]
+    ],
+    define: {
+      __CURRENT_SITE__: JSON.stringify('furbooru'),
+      __CURRENT_SITE_NAME__: JSON.stringify('Furbooru'),
+    }
   });
 
   return pathsReplacement;
diff --git a/package.json b/package.json
index c2a991f..b7a9a94 100644
--- a/package.json
+++ b/package.json
@@ -4,6 +4,7 @@
     "private": true,
     "scripts": {
         "build": "npm run build:popup && npm run build:extension",
+        "build:derpibooru": "cross-env SITE=derpibooru npm run build",
         "build:popup": "vite build",
         "build:extension": "node build-extension.js",
         "check": "svelte-kit sync && svelte-check --tsconfig ./tsconfig.json",
diff --git a/vite.config.ts b/vite.config.ts
index f8d70c5..b6d5b66 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -1,21 +1,38 @@
 import { sveltekit } from '@sveltejs/kit/vite';
 import { defineConfig } from 'vitest/config';
+import { ScssViteReadEnvVariableFunctionPlugin } from "./.vite/plugins/scss-read-env-variable-function";
+import { SwapDefinedVariablesPlugin } from "./.vite/plugins/swap-defined-variables";
 
-export default defineConfig({
-  build: {
-    // SVGs imported from the FA6 don't need to be inlined!
-    assetsInlineLimit: 0
-  },
-  plugins: [
-    sveltekit(),
-  ],
-  test: {
-    globals: true,
-    environment: 'jsdom',
-    exclude: ['**/node_modules/**', '.*\\.d\\.ts$', '.*\\.spec\\.ts$'],
-    coverage: {
-      reporter: ['text', 'html'],
-      include: ['src/lib/**/*.{js,ts}'],
+export default defineConfig(() => {
+  return {
+    build: {
+      // SVGs imported from the FA6 don't need to be inlined!
+      assetsInlineLimit: 0
+    },
+    plugins: [
+      sveltekit(),
+      ScssViteReadEnvVariableFunctionPlugin(),
+      SwapDefinedVariablesPlugin({
+        envVariable: 'SITE',
+        expectedValue: 'derpibooru',
+        define: {
+          __CURRENT_SITE__: JSON.stringify('derpibooru'),
+          __CURRENT_SITE_NAME__: JSON.stringify('Derpibooru'),
+        }
+      }),
+    ],
+    test: {
+      globals: true,
+      environment: 'jsdom',
+      exclude: ['**/node_modules/**', '.*\\.d\\.ts$', '.*\\.spec\\.ts$'],
+      coverage: {
+        reporter: ['text', 'html'],
+        include: ['src/lib/**/*.{js,ts}'],
+      }
+    },
+    define: {
+      __CURRENT_SITE__: JSON.stringify('furbooru'),
+      __CURRENT_SITE_NAME__: JSON.stringify('Furbooru'),
     }
-  }
+  };
 });