gfwleak/appsketch-works-asw-controller
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: ASW-192 playbook支持python 脚本类型

Browse Source
This commit is contained in:
zhangshuai
2024-11-26 10:13:20 +08:00
parent afcbbe30b6
commit b189c5d9d0
3 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/net/geedge/asw/module/runner/job/JobPlaybookExecutor.java
View File

@@ -212,6 +212,7 @@ public class JobPlaybookExecutor extends QuartzJobBean {
request.form("file", zipFile);
request.form("id", job.getId());
request.form("packageName", packageName);
request.form("type", playbook.getType());
for (Map.Entry<String, Object> param : params.entrySet()) {
request.form(param.getKey(), param.getValue());
}

12
src/main/java/net/geedge/asw/module/runner/service/impl/PlaybookServiceImpl.java
View File

@@ -70,6 +70,7 @@ public class PlaybookServiceImpl extends ServiceImpl<PlaybookDao, PlaybookEntity
FileUtils.copyInputStreamToFile(file.getInputStream(), destination);
playbook.setPath(destination.getPath());
if (T.StrUtil.equalsIgnoreCase(type, "airtest_script")){
playbook.setType("airtest_script");
unzip = T.ZipUtil.unzip(destination);
List<File> fileList = Arrays.stream(unzip.listFiles()).toList();
if (T.CollUtil.isEmpty(fileList) || fileList.size() != 1) {
@@ -87,6 +88,17 @@ public class PlaybookServiceImpl extends ServiceImpl<PlaybookDao, PlaybookEntity
throw new ASWException(RCode.PLAYBOOK_INVALID_FILE);
}
}
if (T.StrUtil.equalsIgnoreCase(type, "python")){
playbook.setType("python");
unzip = T.ZipUtil.unzip(destination);
List<File> fileList = Arrays.stream(unzip.listFiles()).toList();
fileList = fileList.stream().filter(x -> {
return T.StrUtil.equals("main.py", x.getName());
}).toList();
if (T.CollUtil.isEmpty(fileList)) {
throw new ASWException(RCode.PLAYBOOK_INVALID_FILE);
}
}
this.save(playbook);
} catch (Exception e) {
log.error(e, "[savePlaybook] [error] [file: {}]", file.getName());