purovps/jshERP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

给plugin模块增加权限校验

Browse Source
This commit is contained in:
季圣华
2022-06-18 11:55:14 +08:00
parent 4a96bb3c33
commit 5e7e6fd998
1 changed files with 78 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

106
jshERP-boot/src/main/java/com/jsh/erp/controller/PluginController.java
View File

@@ -3,6 +3,9 @@ package com.jsh.erp.controller;
import com.gitee.starblues.integration.application.PluginApplication; import com.gitee.starblues.integration.application.PluginApplication;
import com.gitee.starblues.integration.operator.PluginOperator; import com.gitee.starblues.integration.operator.PluginOperator;
import com.gitee.starblues.integration.operator.module.PluginInfo; import com.gitee.starblues.integration.operator.module.PluginInfo;
import com.jsh.erp.constants.BusinessConstants;
import com.jsh.erp.datasource.entities.User;
import com.jsh.erp.service.user.UserService;
import com.jsh.erp.utils.BaseResponseInfo; import com.jsh.erp.utils.BaseResponseInfo;
import com.jsh.erp.utils.ComputerInfo; import com.jsh.erp.utils.ComputerInfo;
import com.jsh.erp.utils.StringUtil; import com.jsh.erp.utils.StringUtil;
@@ -13,6 +16,7 @@ import org.springframework.util.DigestUtils;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponse;
import java.nio.file.Paths; import java.nio.file.Paths;
@@ -28,6 +32,8 @@ import java.util.*;
@Api(tags = {"插件管理"}) @Api(tags = {"插件管理"})
public class PluginController { public class PluginController {
@Resource
private UserService userService;
private final PluginOperator pluginOperator; private final PluginOperator pluginOperator;
@@ -49,14 +55,17 @@ public class PluginController {
Map<String, Object> map = new HashMap<String, Object>(); Map<String, Object> map = new HashMap<String, Object>();
try { try {
List<PluginInfo> resList = new ArrayList<>(); List<PluginInfo> resList = new ArrayList<>();
List<PluginInfo> list = pluginOperator.getPluginInfo(); User userInfo = userService.getCurrentUser();
if(StringUtil.isEmpty(name)) { if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
resList = list; List<PluginInfo> list = pluginOperator.getPluginInfo();
} else { if (StringUtil.isEmpty(name)) {
for(PluginInfo pi : list) { resList = list;
String desc = pi.getPluginDescriptor().getPluginDescription(); } else {
if(desc.contains(name)) { for (PluginInfo pi : list) {
resList.add(pi); String desc = pi.getPluginDescriptor().getPluginDescription();
if (desc.contains(name)) {
resList.add(pi);
}
} }
} }
} }
@@ -80,7 +89,12 @@ public class PluginController {
@ApiOperation(value = "获取插件jar文件名") @ApiOperation(value = "获取插件jar文件名")
public Set<String> getPluginFilePaths(){ public Set<String> getPluginFilePaths(){
try { try {
return pluginOperator.getPluginFilePaths(); User userInfo = userService.getCurrentUser();
if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
return pluginOperator.getPluginFilePaths();
} else {
return null;
}
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
return null; return null;
@@ -100,10 +114,15 @@ public class PluginController {
Map<String, Object> map = new HashMap<String, Object>(); Map<String, Object> map = new HashMap<String, Object>();
String message = ""; String message = "";
try { try {
if(pluginOperator.stop(id)){ User userInfo = userService.getCurrentUser();
message = "plugin '" + id +"' stop success"; if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
if (pluginOperator.stop(id)) {
message = "plugin '" + id + "' stop success";
} else {
message = "plugin '" + id + "' stop failure";
}
} else { } else {
message = "plugin '" + id +"' stop failure"; message = "power is limit";
} }
map.put("message", message); map.put("message", message);
res.code = 200; res.code = 200;
@@ -129,10 +148,15 @@ public class PluginController {
Map<String, Object> map = new HashMap<String, Object>(); Map<String, Object> map = new HashMap<String, Object>();
String message = ""; String message = "";
try { try {
if(pluginOperator.start(id)){ User userInfo = userService.getCurrentUser();
message = "plugin '" + id +"' start success"; if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
if (pluginOperator.start(id)) {
message = "plugin '" + id + "' start success";
} else {
message = "plugin '" + id + "' start failure";
}
} else { } else {
message = "plugin '" + id +"' start failure"; message = "power is limit";
} }
map.put("message", message); map.put("message", message);
res.code = 200; res.code = 200;
@@ -159,10 +183,15 @@ public class PluginController {
Map<String, Object> map = new HashMap<String, Object>(); Map<String, Object> map = new HashMap<String, Object>();
String message = ""; String message = "";
try { try {
if(pluginOperator.uninstall(id, true)){ User userInfo = userService.getCurrentUser();
message = "plugin '" + id +"' uninstall success"; if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
if (pluginOperator.uninstall(id, true)) {
message = "plugin '" + id + "' uninstall success";
} else {
message = "plugin '" + id + "' uninstall failure";
}
} else { } else {
message = "plugin '" + id +"' uninstall failure"; message = "power is limit";
} }
map.put("message", message); map.put("message", message);
res.code = 200; res.code = 200;
@@ -186,8 +215,13 @@ public class PluginController {
@ApiOperation(value = "根据插件路径安装插件") @ApiOperation(value = "根据插件路径安装插件")
public String install(@RequestParam("path") String path){ public String install(@RequestParam("path") String path){
try { try {
if(pluginOperator.install(Paths.get(path))){ User userInfo = userService.getCurrentUser();
return "installByPath success"; if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
if (pluginOperator.install(Paths.get(path))) {
return "installByPath success";
} else {
return "installByPath failure";
}
} else { } else {
return "installByPath failure"; return "installByPath failure";
} }
@@ -208,9 +242,15 @@ public class PluginController {
public BaseResponseInfo install(MultipartFile file, HttpServletRequest request, HttpServletResponse response){ public BaseResponseInfo install(MultipartFile file, HttpServletRequest request, HttpServletResponse response){
BaseResponseInfo res = new BaseResponseInfo(); BaseResponseInfo res = new BaseResponseInfo();
try { try {
pluginOperator.uploadPluginAndStart(file); User userInfo = userService.getCurrentUser();
res.code = 200; if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
res.data = "导入成功"; pluginOperator.uploadPluginAndStart(file);
res.code = 200;
res.data = "导入成功";
} else {
res.code = 500;
res.data = "抱歉，无操作权限！";
}
} catch(Exception e){ } catch(Exception e){
e.printStackTrace(); e.printStackTrace();
res.code = 500; res.code = 500;
@@ -228,10 +268,15 @@ public class PluginController {
@ApiOperation(value = "上传插件的配置文件") @ApiOperation(value = "上传插件的配置文件")
public String uploadConfig(@RequestParam("configFile") MultipartFile multipartFile){ public String uploadConfig(@RequestParam("configFile") MultipartFile multipartFile){
try { try {
if(pluginOperator.uploadConfigFile(multipartFile)){ User userInfo = userService.getCurrentUser();
return "uploadConfig success"; if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
if (pluginOperator.uploadConfigFile(multipartFile)) {
return "uploadConfig success";
} else {
return "uploadConfig failure";
}
} else { } else {
return "uploadConfig failure"; return "installByPath failure";
} }
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
@@ -249,8 +294,13 @@ public class PluginController {
@ApiOperation(value = "备份插件") @ApiOperation(value = "备份插件")
public String backupPlugin(@PathVariable("pluginId") String pluginId){ public String backupPlugin(@PathVariable("pluginId") String pluginId){
try { try {
if(pluginOperator.backupPlugin(pluginId, "testBack")){ User userInfo = userService.getCurrentUser();
return "backupPlugin success"; if(BusinessConstants.DEFAULT_MANAGER.equals(userInfo.getLoginName())) {
if (pluginOperator.backupPlugin(pluginId, "testBack")) {
return "backupPlugin success";
} else {
return "backupPlugin failure";
}
} else { } else {
return "backupPlugin failure"; return "backupPlugin failure";
} }