org.dromara.jpom.build.BuildUtil Maven / Gradle / Ivy
/*
* Copyright (c) 2019 Of Him Code Technology Studio
* Jpom is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
* http://license.coscl.org.cn/MulanPSL2
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
*/
package org.dromara.jpom.build;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.URLUtil;
import cn.hutool.core.util.ZipUtil;
import cn.hutool.crypto.SecureUtil;
import cn.hutool.extra.compress.CompressUtil;
import cn.hutool.extra.compress.archiver.Archiver;
import org.dromara.jpom.JpomApplication;
import org.dromara.jpom.common.ServerConst;
import org.dromara.jpom.common.i18n.I18nMessageUtil;
import org.dromara.jpom.model.data.BuildInfoModel;
import org.dromara.jpom.model.data.RepositoryModel;
import org.springframework.util.Assert;
import java.io.File;
import java.nio.charset.Charset;
import java.util.function.BiFunction;
/**
* 构建工具类
*
* @author bwcx_jzy
* @since 2019/7/19
*/
public class BuildUtil {
public static Long buildCacheSize = 0L;
public static final String USE_TAR_GZ = "USE_TAR_GZ";
/**
* 刷新存储文件大小
*/
public static void reloadCacheSize() {
File buildDataDir = BuildUtil.getBuildDataDir();
BuildUtil.buildCacheSize = FileUtil.size(buildDataDir);
//
}
public static File getBuildDataFile(String id) {
return FileUtil.file(getBuildDataDir(), id);
}
// /**
// * 获取代码路径
// *
// * @param buildModel 实体
// * @return file
// * @see BuildUtil#getSourceById
// */
// @Deprecated
// public static File getSource(BuildModel buildModel) {
// return FileUtil.file(BuildUtil.getBuildDataFile(buildModel.getId()), "source");
// }
/**
* @param id 构建ID
* @return file
* @author Hotstrip
* 新版本获取代码路径
* @since 2021-08-22
*/
public static File getSourceById(String id) {
return FileUtil.file(BuildUtil.getBuildDataFile(id), "source");
}
public static File getBuildDataDir() {
return FileUtil.file(JpomApplication.getInstance().getDataPath(), "build");
}
/**
* 获取构建产物存放路径
*
* @param buildModelId 构建实体
* @param buildId id
* @param resultFile 结果目录
* @return file
*/
public static File getHistoryPackageFile(String buildModelId, int buildId, String resultFile) {
if (buildId <= 0) {
// 没有 0 号构建id，避免生成 #0 文件夹
return null;
}
if (StrUtil.isEmpty(buildModelId) || StrUtil.isEmpty(resultFile)) {
return null;
}
ResultDirFileAction resultDirFileAction = ResultDirFileAction.parse(resultFile);
ResultDirFileAction.Type type = resultDirFileAction.getType();
if (type == ResultDirFileAction.Type.ANT_PATH) {
// ANT 模式 不能直接获取，避免提前创建文件夹
return null;
}
File result = FileUtil.file(getBuildDataFile(buildModelId), "history", BuildInfoModel.getBuildIdStr(buildId), "result");
return FileUtil.file(result, resultFile);
}
/**
* 插件构建产物存放路径
*
* @param buildModelId 构建实体
* @param buildId id
*/
public static void mkdirHistoryPackageFile(String buildModelId, int buildId) {
File result = FileUtil.file(getBuildDataFile(buildModelId), "history", BuildInfoModel.getBuildIdStr(buildId), "result");
FileUtil.mkdir(result);
}
/**
* 获取构建产物存放路径
*
* @param buildModelId 构建实体
* @param buildId id
* @return file
*/
public static File getHistoryPackageZipFile(String buildModelId, int buildId) {
return FileUtil.file(getBuildDataFile(buildModelId),
"history",
BuildInfoModel.getBuildIdStr(buildId),
"zip");
}
/**
* 获取日志记录文件
*
* @param buildModelId buildModelId
* @param buildId 构建编号
* @return file
*/
public static File getLogFile(String buildModelId, int buildId) {
if (StrUtil.isEmpty(buildModelId)) {
return null;
}
return FileUtil.file(getBuildDataFile(buildModelId),
"history",
BuildInfoModel.getBuildIdStr(buildId),
"info.log");
}
/**
* 如果为文件夹自动打包为zip ,反之返回null
*
* @param file file
* @return 压缩包文件
*/
private static File isDirPackage(String id, int buildNumberId, File file, boolean tarGz) {
Assert.state(file != null && file.exists(), I18nMessageUtil.get("i18n.product_file_does_not_exist.ee13"));
if (file.isFile()) {
return null;
}
Assert.state(!FileUtil.isDirEmpty(file), I18nMessageUtil.get("i18n.empty_folder_cannot_be_packed.5a75") + buildNumberId);
String name = FileUtil.getName(file);
// 如果产物配置 / 时无法获取文件名，采用 result
name = StrUtil.emptyToDefault(name, "result");
// 保存目录存放值 history 路径
File packageFile = BuildUtil.getHistoryPackageZipFile(id, buildNumberId);
File zipFile = tarGz ? FileUtil.file(packageFile, name + ".tar.gz") : FileUtil.file(packageFile, name + ".zip");
// 不存在则打包
if (tarGz) {
try (Archiver archiver = CompressUtil.createArchiver(Charset.defaultCharset(), "tar.gz", zipFile)) {
archiver.add(file);
}
} else {
ZipUtil.zip(file.getAbsolutePath(), zipFile.getAbsolutePath());
}
return zipFile;
}
/**
* 如果为文件夹自动打包为zip ,反之返回null
*
* @param file file
* @param id 构建Id
* @param buildNumberId 构建序号
* @param tarGz 是否打包 为 tar
* @param consumer 文件回调
* @return 执行结果
*/
public static T loadDirPackage(String id, int buildNumberId, File file, boolean tarGz, BiFunction consumer) {
File dirPackage = isDirPackage(id, buildNumberId, file, tarGz);
if (dirPackage == null) {
return consumer.apply(false, file);
} else {
return consumer.apply(true, dirPackage);
}
}
/**
* get rsa file
*
* @param path 文件名
* @return file
*/
public static File getRepositoryRsaFile(String path) {
File sshDir = FileUtil.file(JpomApplication.getInstance().getDataPath(), ServerConst.SSH_KEY);
return FileUtil.file(sshDir, path);
}
/**
* get rsa file
*
* @param repositoryModel 仓库
* @return 文件
*/
public static File getRepositoryRsaFile(RepositoryModel repositoryModel) {
if (StrUtil.isEmpty(repositoryModel.getRsaPrv())) {
return null;
}
// ssh
File rsaFile;
if (StrUtil.startWith(repositoryModel.getRsaPrv(), URLUtil.FILE_URL_PREFIX)) {
String rsaPath = StrUtil.removePrefix(repositoryModel.getRsaPrv(), URLUtil.FILE_URL_PREFIX);
rsaFile = FileUtil.file(rsaPath);
} else {
if (StrUtil.isEmpty(repositoryModel.getId())) {
rsaFile = FileUtil.file(JpomApplication.getInstance().getTempPath(), ServerConst.SSH_KEY, SecureUtil.sha1(repositoryModel.getGitUrl()) + ServerConst.ID_RSA);
} else {
rsaFile = BuildUtil.getRepositoryRsaFile(repositoryModel.getId() + ServerConst.ID_RSA);
}
// 写入
FileUtil.writeUtf8String(repositoryModel.getRsaPrv(), rsaFile);
}
Assert.state(FileUtil.isFile(rsaFile), I18nMessageUtil.get("i18n.repository_key_file_does_not_exist_or_is_abnormal.1d78"));
return rsaFile;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy