diff --git a/yudao-server/src/test/java/cn/iocoder/yudao/ProjectReactor.java b/yudao-server/src/test/java/cn/iocoder/yudao/ProjectReactor.java index f961b066f..2cb2f9640 100644 --- a/yudao-server/src/test/java/cn/iocoder/yudao/ProjectReactor.java +++ b/yudao-server/src/test/java/cn/iocoder/yudao/ProjectReactor.java @@ -1,17 +1,24 @@ package cn.iocoder.yudao; +import cn.hutool.core.io.FileTypeUtil; import cn.hutool.core.io.FileUtil; import cn.hutool.core.util.StrUtil; +import cn.iocoder.yudao.framework.common.util.collection.SetUtils; import lombok.extern.slf4j.Slf4j; import org.apache.commons.io.FileUtils; import java.io.File; import java.nio.charset.StandardCharsets; import java.util.Collection; +import java.util.Set; +import java.util.regex.Matcher; +import java.util.stream.Collectors; + +import static java.io.File.separator; /** * 项目修改器，一键替换 Maven 的 groupId、artifactId，项目的 package 等 - * + *

* 通过修改 groupIdNew、artifactIdNew、projectBaseDirNew 三个变量 * * @author 芋道源码 @@ -24,8 +31,16 @@ public class ProjectReactor { private static final String PACKAGE_NAME = "cn.iocoder.yudao"; private static final String TITLE = "芋道管理系统"; + /** + * 白名单文件，不进行重写，避免出问题 + */ + private static final Set WHITE_FILE_TYPES = SetUtils.asSet("gif", "jpg", "svg", "png", // 图片 + "eot", "woff2", "ttf", "woff"); // 字体 + public static void main(String[] args) { + long start = System.currentTimeMillis(); String projectBaseDir = getProjectBaseDir(); + log.info("[main][原项目路劲改地址 ({})]", projectBaseDir); // ========== 配置，需要你手动修改 ========== String groupIdNew = "cn.star.gg"; @@ -33,41 +48,61 @@ public class ProjectReactor { String packageNameNew = "cn.start.pp"; String titleNew = "土豆管理系统"; String projectBaseDirNew = projectBaseDir + "-new"; // 一键改名后，“新”项目所在的目录 - // ========== ========== + log.info("[main][新项目路径地址 ({})]", projectBaseDirNew); // 获得需要复制的文件 - log.info("[main][开始获得需要重写的文件]"); + log.info("[main][开始获得需要重写的文件，预计需要 10-20 秒]"); Collection files = listFiles(projectBaseDir); - log.info("[main][需要重写的文件数量：{}，预计需要 5-10 秒]", files.size()); + log.info("[main][需要重写的文件数量：{}，预计需要 15-30 秒]", files.size()); // 写入文件 files.forEach(file -> { + // 如果是白名单的文件类型，不进行重写，直接拷贝 + String fileType = FileTypeUtil.getType(file); + if (WHITE_FILE_TYPES.contains(fileType)) { + copyFile(file, projectBaseDir, projectBaseDirNew, packageNameNew, artifactIdNew); + return; + } + // 如果非白名单的文件类型，重写内容，在生成文件 String content = replaceFileContent(file, groupIdNew, artifactIdNew, packageNameNew, titleNew); writeFile(file, content, projectBaseDir, projectBaseDirNew, packageNameNew, artifactIdNew); }); - log.info("[main][重写完成]"); + log.info("[main][重写完成]共耗时：{} 秒", (System.currentTimeMillis() - start) / 1000); } private static String getProjectBaseDir() { - // noinspection ConstantConditions - return StrUtil.subBefore(ProjectReactor.class.getClassLoader().getResource("").getFile(), - "/yudao-server", false); + String baseDir = System.getProperty("user.dir"); + if (StrUtil.isEmpty(baseDir)) { + throw new NullPointerException("项目基础路径不存在"); + } + return baseDir; } private static Collection listFiles(String projectBaseDir) { Collection files = FileUtils.listFiles(new File(projectBaseDir), null, true); - files.removeIf(file -> file.getPath().contains("/target/")); - files.removeIf(file -> file.getPath().contains("/node_modules/")); - files.removeIf(file -> file.getPath().contains("/.idea/")); // 移除 IDEA 自身的文件 - files.removeIf(file -> file.getPath().contains("/.git/")); // 移除 Git 自身的文件 - files.removeIf(file -> file.getPath().contains("/dist/")); // 移除 Node 编译出来的 + // 移除 IDEA、Git 自身的文件、Node 编译出来的文件 + files = files.stream() + .filter(file -> !file.getPath().contains(separator + "target" + separator) + && !file.getPath().contains(separator + "node_modules" + separator) + && !file.getPath().contains(separator + ".idea" + separator) + && !file.getPath().contains(separator + ".git" + separator) + && !file.getPath().contains(separator + "dist" + separator) + && !file.getPath().contains(".iml") + && !file.getPath().contains(".html.gz")) + .collect(Collectors.toList()); return files; } private static String replaceFileContent(File file, String groupIdNew, String artifactIdNew, String packageNameNew, String titleNew) { - return FileUtil.readString(file, StandardCharsets.UTF_8) - .replaceAll(GROUP_ID, groupIdNew) + String content = FileUtil.readString(file, StandardCharsets.UTF_8); + // 如果是白名单的文件类型，不进行重写 + String fileType = FileTypeUtil.getType(file); + if (WHITE_FILE_TYPES.contains(fileType)) { + return content; + } + // 执行文件内容都重写 + return content.replaceAll(GROUP_ID, groupIdNew) .replaceAll(PACKAGE_NAME, packageNameNew) .replaceAll(ARTIFACT_ID, artifactIdNew) // 必须放在最后替换，因为 ARTIFACT_ID 太短！ .replaceAll(StrUtil.upperFirst(ARTIFACT_ID), StrUtil.upperFirst(artifactIdNew)) @@ -76,12 +111,23 @@ public class ProjectReactor { private static void writeFile(File file, String fileContent, String projectBaseDir, String projectBaseDirNew, String packageNameNew, String artifactIdNew) { - String newPath = file.getPath().replace(projectBaseDir, projectBaseDirNew) // 新目录 - .replace(PACKAGE_NAME.replaceAll("\\.", File.separator), - packageNameNew.replaceAll("\\.", File.separator)) - .replace(ARTIFACT_ID, artifactIdNew) // - .replaceAll(StrUtil.upperFirst(ARTIFACT_ID), StrUtil.upperFirst(artifactIdNew)); + String newPath = buildNewFilePath(file, projectBaseDir, projectBaseDirNew, packageNameNew, artifactIdNew); FileUtil.writeUtf8String(fileContent, newPath); } + private static void copyFile(File file, String projectBaseDir, + String projectBaseDirNew, String packageNameNew, String artifactIdNew) { + String newPath = buildNewFilePath(file, projectBaseDir, projectBaseDirNew, packageNameNew, artifactIdNew); + FileUtil.copyFile(file, new File(newPath)); + } + + private static String buildNewFilePath(File file, String projectBaseDir, + String projectBaseDirNew, String packageNameNew, String artifactIdNew) { + return file.getPath().replace(projectBaseDir, projectBaseDirNew) // 新目录 + .replace(PACKAGE_NAME.replaceAll("\\.", Matcher.quoteReplacement(separator)), + packageNameNew.replaceAll("\\.", Matcher.quoteReplacement(separator))) + .replace(ARTIFACT_ID, artifactIdNew) // + .replaceAll(StrUtil.upperFirst(ARTIFACT_ID), StrUtil.upperFirst(artifactIdNew)); + } + }