Merge pull request #44 from relative-ci/fix-types

fix types
relative-ci · Jan 11, 2025 · 011c9bd · 011c9bd
2 parents 4c78577 + 0f923cc
commit 011c9bd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 11 deletions.
diff --git a/src/extract.ts b/src/extract.ts
@@ -15,7 +15,7 @@ export type ChunkStats = Omit<OutputChunk, 'code' | 'modules'> & {
   modules: Record<string, ModuleStats>;
 };
 
-export type Stats = OutputBundle;
+export type Stats = Record<string, AssetStats | ChunkStats>;
 
 export type StatsOptions = {
   /**
@@ -49,7 +49,7 @@ export default function extractRollupStats(bundle: OutputBundle, options: StatsO
 
       // Skip asset source if options source is false
       if (!source) {
-        assetStats = omit(assetStats, 'source');
+        assetStats = omit(assetStats, ['source']); 
       }
 
       output[bundleEntryFilepath] = assetStats;
@@ -62,7 +62,7 @@ export default function extractRollupStats(bundle: OutputBundle, options: StatsO
 
       // Skip chunk source if options source is false
       if (!source) {
-        chunkStats = omit(chunkStats, 'code');
+        chunkStats = omit(chunkStats, ['code']);
       }
 
       // Extract chunk modules stats
@@ -78,7 +78,7 @@ export default function extractRollupStats(bundle: OutputBundle, options: StatsO
 
         // Skip module source if options source is false
         if (!source) {
-          moduleStats = omit(moduleStats, 'code');
+          moduleStats = omit(moduleStats, ['code']);
         }
 
         chunkModulesStats[bundleModuleFilepath] = moduleStats;

diff --git a/src/utils/omit.ts b/src/utils/omit.ts
@@ -1,12 +1,13 @@
-export function omit(
-  data: { [key: string]: unknown },
-  keys: Array<string>
-): { [key: string]: unknown } {
-  const result = {};
+export function omit<D extends object, K extends keyof D>(
+  data: D,
+  keys: K[],
+): Omit<D, K> {
+  const result = {} as D;
+  const objectKeys = Object.keys(data) as Array<K>;
 
-  Object.entries(data).forEach(([key, value]) => {
+  objectKeys.forEach((key) => {
     if (!keys.includes(key)) {
-      result[key] = value;
+      result[key] = data[key];
     }
   });