feima-shortcuts 0.3.1 → 0.3.3-beta.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "feima-shortcuts",
-  "version": "0.3.1",
+  "version": "0.3.3-beta.1",
   "description": "快捷指令",
   "main": "index.js",
   "directories": {

package/src/scripts/restful-api/old.js

@@ -1,9 +1,10 @@
 const fs = require("fs");
 const { makeDir } = require("../../utils/makeDir");
 
-const fileName = "index.ts";
+const fileNameTs = "index.ts";
+const fileNameJs = "index.js";
 
-const postTemplat = (apiPath, functionName) => {
+const postTemplatTs = (apiPath, functionName) => {
   return `export const ${functionName} = (options: any) => {
   const isFormData = options?.data instanceof FormData;
 
@@ -27,6 +28,30 @@ const postTemplat = (apiPath, functionName) => {
 `
 };
 
+const postTemplatJs = (apiPath, functionName) => {
+  return `export const ${functionName} = (options) => {
+  const isFormData = options?.data instanceof FormData;
+
+  const data = isFormData
+    ? (() => {
+        const formData = options.data;
+        formData.append("api", "${apiPath}");
+        return formData;
+      })()
+    : {
+        ...(options?.data || {}),
+        api: "${apiPath}",
+      };
+
+  return request({
+    method: "post",
+    ...(options || {}),
+    data,
+  });
+};
+`
+};
+
 function splitString(input) {
   const parts = input.split('.');
   const path = parts.slice(0, -1).join('/'); // 取前面的部分并用 '/' 连接
@@ -34,10 +59,10 @@ function splitString(input) {
   return { path, functionName };
 }
 
-const fileData = (answers, functionName) => { 
+const fileData = (fileName, functionName) => { 
   if (fs.existsSync(`./${fileName}`)) {
     const data = fs.readFileSync(`./${fileName}`, "utf8");
-    if (data.includes(`export const ${functionName} `)) {
+    if (data.includes(`export const ${functionName} `) || data.includes(`export const ${functionName}: any`)) {
       return "";
     }
     return data;
@@ -45,18 +70,35 @@ const fileData = (answers, functionName) => {
   return `import request from '@/utils/request'`
 }
 
-const apiAllTemplate = (answers,functionName) => {
-  const fileTemplate = fileData(answers, functionName);
+const apiAllTemplate = (answers, functionName, fileName, templateFn) => {
+  const fileTemplate = fileData(fileName, functionName);
 
   if (!fileTemplate) return console.error(`无需重复添加 ${answers.path}`);
 
   const template = `${fileTemplate}
 
-${postTemplat(answers.path, functionName)}`;
+${templateFn(answers.path, functionName)}`;
 
   fs.writeFileSync(`./${fileName}`, template);
 };
 
+const resolveFileAndTemplate = () => {
+  const tsExists = fs.existsSync(`./${fileNameTs}`);
+  const jsExists = fs.existsSync(`./${fileNameJs}`);
+
+  if (tsExists && jsExists) {
+    console.error("index.ts 和 index.js 同时存在，请手动处理后再运行脚本。");
+    return null;
+  }
+
+  if (jsExists) {
+    return { fileName: fileNameJs, templateFn: postTemplatJs };
+  }
+
+  // 只存在 ts 或两个都不存在时，统一使用 ts
+  return { fileName: fileNameTs, templateFn: postTemplatTs };
+};
+
 exports.run = function (answers) {
   const { path, functionName } = splitString(answers.path);
   const pagePath = `./src/api/${path}`;
@@ -64,6 +106,11 @@ exports.run = function (answers) {
   makeDir(pagePath);
   process.chdir(pagePath);
   console.log(`import { ${functionName} } from '@/api/${path}'`)
+  
+  const resolved = resolveFileAndTemplate();
+  if (!resolved) return;
+
+  const { fileName, templateFn } = resolved;
 
-  apiAllTemplate(answers, functionName);
+  apiAllTemplate(answers, functionName, fileName, templateFn);
 };