flexysnap 0.1.3-alpha.2 → 0.1.4-alpha.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "flexysnap",
-  "version": "0.1.3-alpha.2",
+  "version": "0.1.4-alpha.1",
   "description": "Flexible visual snapshot testing for Playwright, built for e-commerce.",
   "keywords": [
     "playwright",

package/src/index.js

@@ -23,4 +23,10 @@ export {
 
 export { areWireframesStable } from './wireframeStability.js';
 
-export { expectWireframe, getRGBHistogramFromBuffer } from './wireframeUtils.js';
+export { expectWireframe, getRGBHistogramFromBuffer } from './wireframeUtils.js';
+
+export {
+    setWireframeOutputRoot,
+    getWireframeOutputRoot,
+    resolveWireframeOutputDir
+} from './wireframeOutput.js';

package/src/wireframeOutput.js

@@ -0,0 +1,29 @@
+import path from 'path';
+
+let outputRoot = process.cwd();
+
+function setWireframeOutputRoot(rootPath) {
+    outputRoot = rootPath || process.cwd();
+}
+
+function getWireframeOutputRoot() {
+    return outputRoot;
+}
+
+function resolveWireframeOutputDir(outputDir) {
+    if (!outputDir || typeof outputDir !== 'string') {
+        throw new Error('resolveWireframeOutputDir requires a non-empty string outputDir.');
+    }
+
+    if (path.isAbsolute(outputDir)) {
+        return outputDir;
+    }
+
+    return path.join(outputRoot, outputDir);
+}
+
+export {
+    setWireframeOutputRoot,
+    getWireframeOutputRoot,
+    resolveWireframeOutputDir
+};

package/src/wireframeUtils.js

@@ -3,6 +3,7 @@ import path from 'path';
 import { logTimestamp } from './testUtils.js';
 import { loadImage, createCanvas } from 'canvas';
 import { areWireframesStable } from './wireframeStability.js';
+import { resolveWireframeOutputDir } from './wireframeOutput.js';
 
 const FLEXYSNAP_ELEMENT_ID_ATTRIBUTE = 'data-flexysnap-id';
 
@@ -270,13 +271,11 @@ async function extractStableWireframe(page, elementGroups, retryDelay, maxRetryC
     return {wireframeData: currentWireframeData, scrollPosition: currentScrollPosition};
 }
 
-async function expectWireframe(page, elementGroups, configName, outputFile, outputName, retryDelay = 1000, maxRetryCount = 10) {
-        logTimestamp(`Starting wireframe capture for: ${outputFile}`);
-    const { wireframeData, scrollPosition } = await extractStableWireframe(page, elementGroups, retryDelay, maxRetryCount);
+async function expectWireframe(page, elementGroups, outputDir, outputFile, outputName, options = {}) {
+    const { retryDelay = 1000, maxRetryCount = 10, metadata = {} } = options;
 
-    const userType = process.env.USER_TYPE;
-    const deviceType = process.env.DEVICE_TYPE;
-    const testType = process.env.TEST_TYPE;
+    logTimestamp(`Starting wireframe capture for: ${outputFile}`);
+    const { wireframeData, scrollPosition } = await extractStableWireframe(page, elementGroups, retryDelay, maxRetryCount);
 
     const screenshotScrollPosition = await page.evaluate(() => ({
         x: window.scrollX,
@@ -286,25 +285,25 @@ async function expectWireframe(page, elementGroups, configName, outputFile, outp
     const wireframeOutput = {
         name: outputName,
         timestamp: new Date().toISOString(),
-        deviceType: process.env.DEVICE_TYPE,
-        userType: process.env.USER_TYPE,
         scrollPosition,
         screenshotScrollPosition,
-        elementGroups: wireframeData
+        elementGroups: wireframeData,
+        ...metadata
     };
 
+    const resolvedDir = resolveWireframeOutputDir(outputDir);
+    fs.mkdirSync(resolvedDir, {recursive: true});
+
     const fileName = `${outputFile}.json`;
-    const filePath = path.join(process.cwd(), 'wireframes', 'test', testType, configName, deviceType, userType, fileName);
-    const fileDir = path.dirname(filePath);
-    fs.mkdirSync(fileDir, {recursive: true});
+    const filePath = path.join(resolvedDir, fileName);
     fs.writeFileSync(filePath, JSON.stringify(wireframeOutput, null, 2));
     logTimestamp(`Wireframe captured and saved to: ${fileName}`);
 
-    const screenshotPath = path.join(fileDir, `${outputFile}.png`);
+    const screenshotPath = path.join(resolvedDir, `${outputFile}.png`);
     await page.screenshot({ path: screenshotPath });
     logTimestamp(`Wireframe screenshot saved to: ${screenshotPath}`);
 
-    return {wireframeOutput, screenshotPath};
+    return {wireframeOutput, screenshotPath, outputDir: resolvedDir};
 }
 
 export { expectWireframe, getRGBHistogramFromBuffer };