feat: add getStat method.

jaywcjlove · Apr 25, 2022 · 82f60a1 · 82f60a1
1 parent 1d04d65
commit 82f60a1
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -150,6 +150,7 @@ export interface IFileDirStat extends Partial<fs.Stats> {
 declare type Callback = (filepath: string, stat: IFileDirStat) => void;
 export default function recursiveReaddirFiles(rootPath: string, options?: RecursiveReaddirFilesOptions, callback?: Callback): Promise<IFileDirStat[]>;
 export { recursiveReaddirFiles };
+export declare const getStat: (filepath: string) => Promise<IFileDirStat>;
 /**
  * Get ext
  * @param {String} filePath `/a/b.jpg` => `jpg`

diff --git a/src/index.test.ts b/src/index.test.ts
@@ -1,7 +1,7 @@
 /// <reference types="jest" />
 
 import path from 'path';
-import recursiveReaddirFiles from './';
+import recursiveReaddirFiles, { getStat } from './';
 
 it('ignored test case', async () => {
   const files = await recursiveReaddirFiles(process.cwd(), {
@@ -144,3 +144,10 @@ it('filter options test case', async () => {
   expect(files.length).toBe(1);
   expect(files[0].name).toEqual('README.md');
 });
+
+it('getStat test case', async () => {
+  const stat = await getStat(path.resolve(process.cwd(), 'package.json'));
+  expect(stat.ext).toEqual('json');
+  expect(stat.name).toEqual('package.json');
+  expect(stat.path.endsWith('package.json')).toBeTruthy();
+});
diff --git a/src/index.ts b/src/index.ts
@@ -75,30 +75,20 @@ async function getFiles(
     });
   if (callback) {
     fileDir.map(async (item: IFileDirStat) => {
-      const stat = (await fs.promises.stat(item.path)) as IFileDirStat;
-      stat.ext = '';
+      const stat = await getStat(item.path);
       if (stat.isDirectory()) {
         getFiles(item.path, options, [], callback);
-      } else if (stat.isFile()) {
-        stat.ext = getExt(item.path);
-        stat.name = item.name;
-        stat.path = item.path;
       }
       callback(item.path, stat);
     });
   } else {
     await Promise.all(
       fileDir.map(async (item: IFileDirStat) => {
-        const stat = (await fs.promises.stat(item.path)) as IFileDirStat;
-        // item.size = stat.size;
-        stat.ext = '';
+        const stat = await getStat(item.path);
         if (stat.isDirectory()) {
           const arr = await getFiles(item.path, options, []);
           files = files.concat(arr);
         } else if (stat.isFile()) {
-          stat.ext = getExt(item.path);
-          stat.name = item.name;
-          stat.path = item.path;
           files.push(stat);
         }
       }),
@@ -112,6 +102,17 @@ async function getFiles(
   });
 }
 
+export const getStat = async (filepath: string): Promise<IFileDirStat> => {
+  const stat = (await fs.promises.stat(filepath)) as IFileDirStat;
+  stat.ext = '';
+  if (stat.isFile()) {
+    stat.ext = getExt(filepath);
+    stat.name = path.basename(filepath);
+    stat.path = path.resolve(filepath);
+  }
+  return stat;
+};
+
 /**
  * Get ext
  * @param {String} filePath `/a/b.jpg` => `jpg`