feat(feature): lint 指定后缀文件

Author: OSpoon

README.md

@@ -25,7 +25,7 @@ npm i -g code-genius
 | cup    | --ignore \<path\>                              | 清理运行时生成的文件                           |
 | ihooks | --                                             | 使用且有修改 simple-git-hooks 后需要重新初始化 |
 | run    | --                                             | 运行列出的脚本                                 |
-| lint   | --eslintrc \<file\>, --path \<path\>, --staged | 检查代码并尝试修复                             |
+| lint   | --eslintrc \<file\>, --path \<path\>, --staged, --suffix \<suffix\> | 检查代码并尝试修复                             |
 
 ## API
 

src/command/eslint-fix.ts

@@ -2,7 +2,7 @@ import { ESLint } from "eslint";
 import path from "node:path";
 
 import {
-  execCommand,
+  getLintFiles,
   loggerError,
   loggerInfo,
   loggerSuccess,
@@ -15,12 +15,11 @@ export const eslintFix = async (
   options: EsLintOptions
 ) => {
   try {
-    const { eslintrc, staged, paths } = options;
+    const { eslintrc, staged, paths, suffix } = options;
 
     if (ACTIVATION) {
       loggerInfo("eslintFix 参数信息: \n");
       console.table(cwd);
-      console.table(staged);
       console.table(options);
     }
 
@@ -31,28 +30,7 @@ export const eslintFix = async (
       overrideConfigFile: path.join(cwd, eslintrc),
     });
 
-    let files: string[] = [];
-    if (staged) {
-      const result = await execCommand("git", [
-        "diff",
-        "--name-only",
-        "--cached",
-      ]);
-      files =
-        result
-          ?.split("\n")
-          .map((path) => `${cwd}/${path}`)
-          .filter(
-            (path) =>
-              path.endsWith(".js") ||
-              path.endsWith(".jsx") ||
-              path.endsWith(".ts") ||
-              path.endsWith(".tsx")
-          ) || [];
-    } else {
-      files = paths.map((path) => `${cwd}/${path}`);
-    }
-
+    const files = await getLintFiles(cwd, staged, paths, suffix);
     const results = await eslint.lintFiles(files);
     await ESLint.outputFixes(results);
     const formatter = await eslint.loadFormatter("stylish");

src/setup.ts

@@ -78,12 +78,17 @@ export const setupSet: SetupSet = {
       .option("--staged", "Inspecte staged files", {
         default: false,
       })
+      .option("--suffix <suffix>", "Inspecte files with specified suffixes", {
+        default: ".js,.jsx,.ts,.tsx",
+      })
       .action(async (options) => {
-        const { eslintrc, staged, path } = options;
+        const { eslintrc, staged, path, suffix } = options;
+        console.log(suffix)
         await eslintFix(cwd, {
           eslintrc,
           staged,
           paths: typeof path === "string" ? [path] : path,
+          suffix: suffix.split(",")
         });
       });
   },

src/shared/config.ts

@@ -98,4 +98,5 @@ export const esLintOptions: EsLintOptions = {
   eslintrc: ".eslintrc.json",
   paths: ["src"],
   staged: true,
+  suffix: [".js", "jsx", ".ts", ".tsx"],
 };

src/shared/index.ts

@@ -1,4 +1,6 @@
 import { execa } from "execa";
+import fs from "fs";
+import path from "path";
 import type { Options } from "execa";
 import { consola } from "consola";
 import process from "node:process";
@@ -58,3 +60,71 @@ export const loggerSuccess = (content: string) => {
 export const loggerError = (content: string | unknown) => {
   console.log(lightRed(`[ERROR]： ${content}`));
 };
+
+export function filterFiles(
+  fileList: string[],
+  allowedExtensions: string[]
+): string[] {
+  const filteredFiles: string[] = [];
+
+  for (let i = 0; i < fileList.length; i++) {
+    const file = fileList[i];
+
+    for (let j = 0; j < allowedExtensions.length; j++) {
+      const extension = allowedExtensions[j];
+
+      if (file.endsWith(extension)) {
+        filteredFiles.push(file);
+        break;
+      }
+    }
+  }
+  return filteredFiles;
+}
+
+export function getAllFiles(dirPaths: string[]): string[] {
+  const fileList: string[] = [];
+  function traverseDirectory(dirPath: string) {
+    const files = fs.readdirSync(dirPath);
+
+    files.forEach((file) => {
+      const filePath = path.join(dirPath, file);
+      const stat = fs.statSync(filePath);
+
+      if (stat.isDirectory()) {
+        // 如果是文件夹，则递归遍历
+        traverseDirectory(filePath);
+      } else {
+        // 如果是文件，则将文件路径添加到列表中
+        fileList.push(filePath);
+      }
+    });
+  }
+
+  dirPaths.forEach((dirPath) => {
+    traverseDirectory(dirPath);
+  });
+
+  const uniqueFiles = Array.from(new Set(fileList));
+  return uniqueFiles;
+}
+
+export const getLintFiles = async (
+  cwd: string,
+  staged: boolean,
+  paths: string[],
+  suffix: string[]
+) => {
+  let files: string[] = [];
+  if (staged) {
+    const result = await execCommand("git", [
+      "diff",
+      "--name-only",
+      "--cached",
+    ]);
+    files = result?.split("\n").map((path) => `${cwd}/${path}`) || [];
+  } else {
+    files = getAllFiles(paths.map((path) => `${cwd}/${path}`));
+  }
+  return filterFiles(files, suffix);
+};

src/shared/types.ts

@@ -18,5 +18,6 @@ export interface CommitScope {
 export interface EsLintOptions {
   eslintrc: string;
   paths: string[];
+  suffix: string[];
   staged: boolean;
 }