feat: support diff image ext (#65)

reg-viz · Aug 18, 2024 · c50b0f9 · c50b0f9
1 parent 9198450
commit c50b0f9
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/src/states/entity.ts b/src/states/entity.ts
@@ -46,11 +46,13 @@ export const initializeEntityState = (
     actual: data.actualDir,
   };
 
+  const diffExtension = data.diffImageExtension ?? 'png';
+
   store.set(defaultEntityAtom, {
-    new: toEntities('new', dirs, data.newItems),
-    passed: toEntities('passed', dirs, data.passedItems),
-    failed: toEntities('changed', dirs, data.failedItems),
-    deleted: toEntities('deleted', dirs, data.deletedItems),
+    new: toEntities('new', dirs, data.newItems, diffExtension),
+    passed: toEntities('passed', dirs, data.passedItems, diffExtension),
+    failed: toEntities('changed', dirs, data.failedItems, diffExtension),
+    deleted: toEntities('deleted', dirs, data.deletedItems, diffExtension),
   });
 
   const defaultEntity = store.get(defaultEntityAtom);

diff --git a/src/types/reg.ts b/src/types/reg.ts
@@ -30,6 +30,7 @@ export type RegData = {
   deletedItems: RegItem[];
   ximgdiffConfig?: XIMGDiffConfig;
   links?: RegLink[];
+  diffImageExtension?: 'webp' | 'png';
 };
 
 export type RegEntity = {

diff --git a/src/utils/transformer.ts b/src/utils/transformer.ts
@@ -15,6 +15,7 @@ export const toEntities = (
   variant: RegVariant,
   dirs: Dirs,
   items: RegItem[],
+  diffExtension: 'png' | 'webp' = 'png',
 ): RegEntity[] => {
   const join = (key: keyof Dirs, to: string) =>
     dirs[key].replace(/\/$/, '') + '/' + to.replace(/^\//, '');
@@ -26,7 +27,7 @@ export const toEntities = (
       id,
       variant,
       name: item.raw,
-      diff: join('diff', item.encoded).replace(/\.[^.]+$/, '.png'),
+      diff: join('diff', item.encoded).replace(/\.[^.]+$/, `.${diffExtension}`),
       before: join('expected', item.encoded),
       after: join('actual', item.encoded),
     };