feat: support write file close #1

Author: a145789

package.json

@@ -2,15 +2,18 @@
   "name": "vue3-script-to-setup",
   "type": "module",
   "version": "0.0.0",
+  "author": "clencat <2091927351@qq.com>",
   "packageManager": "pnpm@7.12.0",
-  "description": "",
+  "description": "transform your vue3 script to setup mode",
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/antfu/vue3-script-to-setup.git"
+    "url": "git+https://github.com/a145789/vue3-script-to-setup"
   },
-  "bugs": "https://github.com/antfu/vue3-script-to-setup/issues",
-  "keywords": [],
+  "bugs": "https://github.com/a145789/vue3-script-to-setup/issues",
+  "keywords": [
+    "vue3"
+  ],
   "sideEffects": false,
   "exports": {
     ".": {
@@ -36,6 +39,9 @@
   "files": [
     "dist"
   ],
+  "engines": {
+    "node": ">=14"
+  },
   "scripts": {
     "build": "unbuild",
     "dev": "unbuild --stub",

rome.json

@@ -4,7 +4,7 @@
     "rules": {
       "recommended": true,
       "correctness": {
-        "noUnusedVariables": "warn"
+        "noUnusedVariables": "error"
       },
       "suspicious": {
         "noExplicitAny": "warn"
@@ -14,8 +14,5 @@
   "formatter": {
     "indentStyle": "space",
     "indentSize": 2
-  },
-  "javascript": {
-    "formatter": {}
   }
 }

src/index.ts

@@ -1,60 +1,7 @@
-import { findUpSync } from "find-up";
-import { getTheFileAbsolutePath, output, useConfigPath } from "./utils";
-import { parseSfc } from "./transform";
-import { CommandsOption } from "./constants";
+import { DefaultOption } from "./constants";
+import setup from "./setup";
 
-const CONFIG_FILE_NAME = "tosetup.config" as const;
-
-async function setup() {
-  const argv = process.argv.slice(2).filter(Boolean);
-
-  let { pathNames, commands } = argv.reduce<{
-    pathNames: string[];
-    commands: CommandsOption;
-  }>(
-    (p, c) => {
-      if (c.startsWith("--")) {
-        switch (c.split("--")[1] as keyof typeof p.commands) {
-          case "propsNotOnlyTs":
-            p.commands.propsNotOnlyTs = true;
-            break;
-          case "notUseNewFile":
-            p.commands.notUseNewFile = true;
-            break;
-
-          default:
-            break;
-        }
-        return p;
-      }
-
-      const absolutePath = getTheFileAbsolutePath(c);
-      if (absolutePath) {
-        p.pathNames.push(absolutePath);
-      }
-
-      return p;
-    },
-    { pathNames: [], commands: {} },
-  );
-
-  if (!pathNames.length) {
-    const configPath = findUpSync(CONFIG_FILE_NAME);
-    if (!configPath) {
-      output.error(
-        `Please enter a file path or use a ${CONFIG_FILE_NAME} file.`,
-      );
-      process.exit(1);
-    }
-
-    const config = await useConfigPath(CONFIG_FILE_NAME);
-
-    pathNames = config.pathNames;
-
-    commands = { ...commands, ...config.option };
-  }
-
-  parseSfc(pathNames, commands);
+export function defineConfig(option: DefaultOption) {
+  return option;
 }
-
 setup();

src/setup.ts

@@ -0,0 +1,76 @@
+import { findUpSync } from "find-up";
+import { getTheFileAbsolutePath, output, useConfigPath } from "./utils";
+import { transformSfc } from "./transform";
+import { CommandsOption } from "./constants";
+import writeFile from "./writeFile";
+
+const CONFIG_FILE_NAME = "tosetup.config" as const;
+
+async function setup() {
+  output.log("tosetup begin...\n");
+  const argv = process.argv.slice(2).filter(Boolean);
+
+  let { pathNames, commands } = argv.reduce<{
+    pathNames: string[];
+    commands: CommandsOption;
+  }>(
+    (p, c) => {
+      if (c.startsWith("--")) {
+        switch (c.split("--")[1] as keyof typeof p.commands) {
+          case "propsNotOnlyTs":
+            p.commands.propsNotOnlyTs = true;
+            break;
+          case "notUseNewFile":
+            p.commands.notUseNewFile = true;
+            break;
+
+          default:
+            break;
+        }
+        return p;
+      }
+
+      const absolutePath = getTheFileAbsolutePath(c);
+      if (absolutePath) {
+        p.pathNames.push(absolutePath);
+      }
+
+      return p;
+    },
+    { pathNames: [], commands: {} },
+  );
+
+  if (!pathNames.length) {
+    const configPath = findUpSync(CONFIG_FILE_NAME);
+    if (!configPath) {
+      output.error(
+        `Please enter a file path or use a ${CONFIG_FILE_NAME} file.`,
+      );
+      process.exit(1);
+    }
+
+    const config = await useConfigPath(CONFIG_FILE_NAME);
+
+    pathNames = config.pathNames;
+
+    commands = { ...commands, ...config.option };
+  }
+  for (const path of pathNames) {
+    output.log(`File ${path} start of transform...`);
+    const code = transformSfc(path, commands);
+    if (code) {
+      try {
+        const file = writeFile(code, path, commands);
+        output.success(`File ${file} transform success.\n`);
+      } catch (error) {
+        output.error(`write ${path} failure.\n`);
+        console.log(error);
+      }
+    } else {
+      output.error(`File ${path} transform failure.\n`);
+    }
+  }
+  output.log("\ntosetup end.");
+}
+
+export default setup;

src/transform/attrsAndSlots.ts

@@ -6,7 +6,7 @@ import type {
   TsType,
 } from "@swc/core";
 import { Config, SetupAst, USE_ATTRS, USE_SLOTS } from "../constants";
-import Visitor from "@swc/core/Visitor";
+import { Visitor } from "@swc/core/Visitor";
 import { getSetupSecondParams } from "../utils";
 import MagicString from "magic-string";
 

src/transform/index.ts

@@ -1,36 +1 @@
-import { parse } from "vue/compiler-sfc";
-import { readFileSync } from "fs";
-import { CommandsOption, FileType } from "../constants";
-import transformScript from "./transformScript";
-import { output } from "../utils";
-
-export function parseSfc(pathNames: string[], option: CommandsOption) {
-  for (const path of pathNames) {
-    const sfc = readFileSync(path).toString();
-    const {
-      descriptor: { script, scriptSetup },
-    } = parse(sfc);
-
-    if (scriptSetup || !script) {
-      continue;
-    }
-
-    let code: string | null = null;
-    try {
-      code = transformScript({
-        ...option,
-        fileType: script.lang === "ts" ? FileType.ts : FileType.js,
-        script: script.content,
-        offset: 0,
-        fileAbsolutePath: path,
-        setupScript: "",
-      });
-    } catch (error) {
-      output.error(`transform script failed in the ${path}`);
-      console.log(error);
-    } finally {
-      if (!code) {
-      }
-    }
-  }
-}
+export { transformSfc } from "./sfc";

src/transform/transformScript.ts renamed to src/transform/script.ts

@@ -15,7 +15,7 @@ import type {
 import { transformSync } from "@swc/core";
 
 import { parseSync } from "@swc/core";
-import Visitor from "@swc/core/Visitor";
+import { Visitor } from "@swc/core/Visitor";
 
 import {
   Config,

src/transform/sfc.ts

@@ -0,0 +1,47 @@
+import { parse } from "vue/compiler-sfc";
+import { readFileSync } from "fs";
+import { CommandsOption, FileType } from "../constants";
+import transformScript from "./script";
+import { output } from "../utils";
+import MagicString from "magic-string";
+
+export function transformSfc(path: string, option: CommandsOption) {
+  const sfc = readFileSync(path).toString();
+  const {
+    descriptor: { script, scriptSetup },
+  } = parse(sfc);
+
+  if (scriptSetup || !script) {
+    return null;
+  }
+
+  let code: string | null = null;
+  try {
+    code = transformScript({
+      ...option,
+      fileType: script.lang === "ts" ? FileType.ts : FileType.js,
+      script: script.content,
+      offset: 0,
+      fileAbsolutePath: path,
+      setupScript: "",
+    });
+  } catch (error) {
+    output.error(`transform script failed in the ${path}`);
+    console.log(error);
+  } finally {
+    if (code) {
+      const ms = new MagicString(sfc);
+      ms.update(script.loc.start.offset, script.loc.end.offset, `\n${code}`);
+      ms.replaceAll(/\<\bscript\b.*\>/g, (str) => {
+        const lastIdx = str.length - 1;
+        return `${str.slice(0, lastIdx)} setup${str[lastIdx]}`;
+      });
+
+      return ms.toString();
+    } else {
+      return null;
+    }
+  }
+}
+
+export default transformSfc;

src/utils.ts

@@ -12,7 +12,7 @@ import type {
   Options,
   TsTypeReference,
 } from "@swc/core";
-import Visitor from "@swc/core/Visitor";
+import { Visitor } from "@swc/core/Visitor";
 import { blue, green, red, yellow } from "colorette";
 import slash from "slash";
 import { DefaultOption, SetupAst, VisitorCb } from "./constants";
@@ -76,10 +76,6 @@ function getFgVueFile(paths: string[]) {
   return fg.sync(paths).filter((p) => p.endsWith(".vue"));
 }
 
-export function defineConfig(option: DefaultOption) {
-  return option;
-}
-
 export function getPropsValueIdentifier(
   ast: Expression,
   keyValue: string,

src/writeFile.ts

@@ -0,0 +1,17 @@
+import { writeFileSync } from "fs";
+import { CommandsOption } from "./constants";
+
+function writeFile(
+  code: string,
+  path: string,
+  { notUseNewFile }: CommandsOption,
+) {
+  const index = path.indexOf(".vue");
+  const file = notUseNewFile ? path : `${path.slice(0, index)}.new.vue`;
+
+  writeFileSync(file, code);
+
+  return file;
+}
+
+export default writeFile;