headlamp-plugin: Allow to copy extra dist files into dist folder

Signed-off-by: Joaquim Rocha <[email protected]>

Author: joaquimrocha

plugins/headlamp-plugin/bin/headlamp-plugin.js

@@ -229,6 +229,63 @@ async function calculateChecksum(filePath) {
   }
 }
 
+/**
+ * Copy extra files specified in package.json to the dist folder
+ *
+ * @param {string} [packagePath='.'] - Path to the package root containing package.json
+ * @returns {Promise<void>}
+ */
+async function copyExtraDistFiles(packagePath = '.') {
+  try {
+    const packageJsonPath = path.join(packagePath, 'package.json');
+    if (!fs.existsSync(packageJsonPath)) {
+      return; // No package.json, nothing to do
+    }
+
+    const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
+    if (!packageJson.headlamp || !packageJson.headlamp.extraDist) {
+      return; // No extra files to copy
+    }
+
+    const extraDist = packageJson.headlamp.extraDist;
+    const distFolder = path.resolve(packagePath, 'dist');
+
+    // Create dist folder if it doesn't exist (although it should by this point)
+    if (!fs.existsSync(distFolder)) {
+      fs.mkdirSync(distFolder, { recursive: true });
+    }
+
+    // Process all entries in extraDist
+    for (const [target, source] of Object.entries(extraDist)) {
+      const targetPath = path.join(distFolder, target);
+      const sourcePath = path.resolve(packagePath, source);
+
+      // Skip if source doesn't exist
+      if (!fs.existsSync(sourcePath)) {
+        console.warn(`Warning: extraDist source "${sourcePath}" does not exist, skipping.`);
+        continue;
+      }
+
+      // Create target directory if needed
+      fs.mkdirSync(path.dirname(targetPath), { recursive: true });
+
+      // Copy based on whether it's a directory or file
+      const sourceStats = fs.statSync(sourcePath);
+      if (sourceStats.isDirectory()) {
+        console.log(`Copying extra directory "${sourcePath}" to "${targetPath}"`);
+        fs.copySync(sourcePath, targetPath);
+      } else {
+        console.log(`Copying extra file "${sourcePath}" to "${targetPath}"`);
+        fs.copyFileSync(sourcePath, targetPath);
+      }
+    }
+
+    console.log('Successfully copied extra dist files');
+  } catch (error) {
+    console.error('Error copying extra dist files:', error);
+  }
+}
+
 /**
  * Creates a tarball of the plugin package. The tarball is placed in the outputFolderPath.
  * It moves files from:
@@ -274,6 +331,10 @@ async function createArchive(pluginDir, outputDir) {
 
   // Create temporary folder
   const tempFolder = fs.mkdtempSync(path.join(os.tmpdir(), 'headlamp-plugin-'));
+
+  // Make sure any extraDist files are in the dist folder before extraction
+  await copyExtraDistFiles(pluginPath);
+
   if (extract(pluginPath, tempFolder, false) !== 0) {
     console.error(
       `Error: Failed to extract plugin package to "${tempFolder}". Not creating archive.`
@@ -381,6 +442,16 @@ async function start() {
     config.build.sourcemap = 'inline';
   }
 
+  // Add file copy hook to be executed after each build
+  if (config.plugins) {
+    config.plugins.push({
+      name: 'headlamp-copy-extra-dist',
+      closeBundle: async () => {
+        await copyExtraDistFiles();
+      },
+    });
+  }
+
   try {
     await vite.build(config);
   } catch (e) {
@@ -579,6 +650,10 @@ async function build(packageFolder) {
     const vite = await vitePromise;
     try {
       await vite.build(config.default);
+
+      // Copy extra dist files after successful build
+      await copyExtraDistFiles('.');
+
       console.log(`Finished building "${folder}" for production.`);
     } catch (e) {
       console.error(e);