Support vite publicDir config

Author: garronej

src/bin/copy-dsfr-to-public.ts

@@ -1,41 +1,147 @@
 #!/usr/bin/env node
 
-import { join as pathJoin } from "path";
+import { join as pathJoin, relative as pathRelative } from "path";
 import * as fs from "fs";
 import { getProjectRoot } from "./tools/getProjectRoot";
+import { assert } from "tsafe/assert";
+import type { Equals } from "tsafe";
 
-const projectDirPath = process.cwd();
+(async () => {
+    const projectDirPath = process.cwd();
 
-const publicDirPath = pathJoin(projectDirPath, "public");
+    const viteConfigFilePath = (() => {
+        for (const ext of [".js", ".ts"]) {
+            const candidateFilePath = pathJoin(projectDirPath, `vite.config${ext}`);
 
-if (!fs.existsSync(publicDirPath)) {
+            if (!fs.existsSync(candidateFilePath)) {
+                continue;
+            }
+
+            return candidateFilePath;
+        }
+
+        return undefined;
+    })();
+
+    const publicDirPath =
+        viteConfigFilePath !== undefined
+            ? await (async function getVitePublicDirPath() {
+                  const viteConfig = fs.readFileSync(viteConfigFilePath).toString("utf8");
+
+                  if (!viteConfig.includes("publicDir")) {
+                      return pathJoin(projectDirPath, "public");
+                  }
+
+                  const [, afterPublicDir] = viteConfig.split(/\s["']?publicDir["']?\s*:/);
+
+                  for (let indexEnd = 0; indexEnd < afterPublicDir.length; indexEnd++) {
+                      // eslint-disable-next-line @typescript-eslint/no-unused-vars
+                      const {
+                          default: path,
+                          basename,
+                          dirname,
+                          delimiter,
+                          extname,
+                          format,
+                          isAbsolute,
+                          join,
+                          normalize,
+                          parse,
+                          posix,
+                          relative,
+                          resolve,
+                          sep,
+                          toNamespacedPath,
+                          win32,
+                          ...rest
+                      } = await import("path");
+                      assert<Equals<keyof typeof rest, never>>();
+
+                      const part = afterPublicDir
+                          .substring(0, indexEnd)
+                          .replace(/__dirname/g, `"${projectDirPath}"`);
+
+                      let candidate: string;
+
+                      try {
+                          candidate = eval(part);
+                      } catch {
+                          continue;
+                      }
+
+                      if (typeof candidate !== "string") {
+                          continue;
+                      }
+
+                      return candidate;
+                  }
+
+                  console.error(
+                      `Can't parse the vite configuration please open an issue about it ${getRepoIssueUrl()}`
+                  );
+
+                  process.exit(-1);
+              })()
+            : pathJoin(projectDirPath, "public");
+
+    edit_gitignore: {
+        const gitignoreFilePath = pathJoin(projectDirPath, ".gitignore");
+
+        if (!fs.existsSync(gitignoreFilePath)) {
+            fs.writeFileSync(gitignoreFilePath, Buffer.from("", "utf8"));
+        }
+
+        const gitignoreRaw = fs.readFileSync(gitignoreFilePath).toString("utf8");
+
+        const pathToIgnore = `/${pathJoin(pathRelative(projectDirPath, publicDirPath), "dsfr")}/`;
+
+        if (gitignoreRaw.split("\n").find(line => line.startsWith(pathToIgnore)) !== undefined) {
+            break edit_gitignore;
+        }
+
+        fs.writeFileSync(
+            gitignoreFilePath,
+            Buffer.from(`${gitignoreRaw}\n${pathToIgnore}\n`, "utf8")
+        );
+    }
+
+    if (!fs.existsSync(publicDirPath)) {
+        if (viteConfigFilePath === undefined) {
+            console.error(
+                [
+                    "There is no public/ directory in the current working directory, we don't know your framework",
+                    "you are not calling this script at the right location or we don't know your React framework",
+                    `please submit an issue about it here ${getRepoIssueUrl()}`
+                ].join(" ")
+            );
+
+            process.exit(-1);
+        }
+
+        fs.mkdirSync(publicDirPath, { "recursive": true });
+    }
+
+    const dsfrDirPath = pathJoin(publicDirPath, "dsfr");
+
+    if (fs.existsSync(dsfrDirPath)) {
+        fs.rmSync(dsfrDirPath, { "recursive": true, "force": true });
+    }
+
+    fs.cpSync(
+        pathJoin(projectDirPath, "node_modules", "@codegouvfr", "react-dsfr", "dsfr"),
+        dsfrDirPath,
+        {
+            "recursive": true
+        }
+    );
+})();
+
+function getRepoIssueUrl() {
     const reactDsfrRepoUrl = JSON.parse(
         fs.readFileSync(pathJoin(getProjectRoot(), "package.json")).toString("utf8")
     )
         ["repository"]["url"].replace(/^git/, "https:")
         .replace(/\.git$/, "");
 
-    console.error(
-        [
-            "There is no public/ directory in the current working directory, we don't know your framework",
-            "you are not calling this script at the right location or we don't know your React framework",
-            `please submit an issue about it here ${reactDsfrRepoUrl}/issues`
-        ].join(" ")
-    );
-
-    process.exit(-1);
+    return `${reactDsfrRepoUrl}/issues`;
 }
-
-const dsfrDirPath = pathJoin(publicDirPath, "dsfr");
-
-if (fs.existsSync(dsfrDirPath)) {
-    fs.rmSync(dsfrDirPath, { "recursive": true, "force": true });
-}
-
-fs.cpSync(
-    pathJoin(projectDirPath, "node_modules", "@codegouvfr", "react-dsfr", "dsfr"),
-    dsfrDirPath,
-    {
-        "recursive": true
-    }
-);

test/integration/vite/.gitignore

@@ -1,11 +1,3 @@
-# Logs
-logs
-*.log
-npm-debug.log*
-yarn-debug.log*
-yarn-error.log*
-pnpm-debug.log*
-lerna-debug.log*
 
 node_modules
 dist
@@ -23,4 +15,4 @@ dist-ssr
 *.sln
 *.sw?
 
-/public/dsfr
+/public/dsfr/

test/integration/vite/vite.config.ts

@@ -4,6 +4,7 @@ import react from "@vitejs/plugin-react";
 // https://vitejs.dev/config/
 export default defineConfig({
     plugins: [react()],
+    publicDir: "public",
     server: {
         fs: {
           allow: ['../../..']